|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/*
* Make some convenient Ecal function available in python
*
* \author Stefano Argiro
* \version $Id: EcalPyUtils.cc,v 1.6 2012/07/17 09:17:11 davidlt Exp $
*/
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "CondTools/Ecal/interface/EcalFloatCondObjectContainerXMLTranslator.h"
#include <pybind11/pybind11.h>
namespace py = pybind11;
namespace ecalpyutils {
std::vector<int> hashedIndexToEtaPhi(int hashedindex) {
int ieta = EBDetId::unhashIndex(hashedindex).ieta();
int iphi = EBDetId::unhashIndex(hashedindex).iphi();
std::vector<int> ret;
ret.push_back(ieta);
ret.push_back(iphi);
return ret;
}
std::vector<int> hashedIndexToXY(int hashedindex) {
int ix = EEDetId::unhashIndex(hashedindex).ix();
int iy = EEDetId::unhashIndex(hashedindex).iy();
int zside = EEDetId::unhashIndex(hashedindex).zside();
std::vector<int> ret;
ret.push_back(ix);
ret.push_back(iy);
ret.push_back(zside);
return ret;
}
int hashedIndex(int ieta, int iphi) {
EBDetId d(ieta, iphi);
return d.hashedIndex();
}
int hashedIndexEE(int ix, int iy, int iz) {
if (EEDetId::validDetId(ix, iy, iz)) {
EEDetId d(ix, iy, iz);
return d.hashedIndex();
}
return 0;
}
int ism(int ieta, int iphi) {
EBDetId d(ieta, iphi);
return d.ism();
}
std::string arraystoXML(const std::vector<float>& eb, const std::vector<float>& ee) {
EcalCondHeader h;
return EcalFloatCondObjectContainerXMLTranslator::dumpXML(h, eb, ee);
}
} // namespace ecalpyutils
PYBIND11_MODULE(pluginEcalPyUtils, m) {
// looks like these are already defined somewhere
// python access to stl integer vectors
// class_< std::vector<int> >("vectorInt")
// .def(vector_indexing_suite<std::vector<int> >())
// ;
// class_< std::vector<float> >("vectorFloat")
// .def(vector_indexing_suite<std::vector<float> >())
// ;
m.def("hashedIndexToEtaPhi", &ecalpyutils::hashedIndexToEtaPhi);
m.def("hashedIndexToXY", &ecalpyutils::hashedIndexToXY);
m.def("hashedIndex", &ecalpyutils::hashedIndex);
m.def("hashedIndexEE", &ecalpyutils::hashedIndexEE);
m.def("ism", &ecalpyutils::ism);
m.def("barrelfromXML", &EcalFloatCondObjectContainerXMLTranslator::barrelfromXML);
m.def("endcapfromXML", &EcalFloatCondObjectContainerXMLTranslator::endcapfromXML);
m.def("arraystoXML", &ecalpyutils::arraystoXML);
}
|