EcalPyUtils.cc

Go to the documentation of this file.

/*
 * 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);
}