HcalGeometryProvider.cxx

Go to the documentation of this file.

#include <regex.h>
#include <sys/types.h>
 
#include "DetDescr/HcalGeometry.h"
#include "Framework/ConditionsObjectProvider.h"
#include "Framework/EventHeader.h"
#include "Framework/RunHeader.h"
 
namespace hcal {
 
class HcalGeometryProvider : public framework::ConditionsObjectProvider {
 public:
  HcalGeometryProvider(const std::string& name, const std::string& tagname,
                       const framework::config::Parameters& parameters,
                       framework::Process& process);
 
  virtual ~HcalGeometryProvider();
 
  virtual std::pair<const framework::ConditionsObject*,
                    framework::ConditionsIOV>
  getCondition(const ldmx::EventHeader& context);
 
  virtual void releaseConditionsObject(const framework::ConditionsObject* co) {}
 
  virtual void onNewRun(ldmx::RunHeader& rh) {
    if (detector_geometry_.empty())
      detector_geometry_ = rh.getDetectorName();
    else if (hcal_geometry_ != nullptr &&
             detector_geometry_ != rh.getDetectorName()) {
      EXCEPTION_RAISE(
          "GeometryException",
          "Attempting to run a single job with multiple geometries " +
              detector_geometry_ + " and '" + rh.getDetectorName() + "'");
    }
    // make sure detector name has been set
    if (detector_geometry_.empty())
      EXCEPTION_RAISE("GeometryException",
                      "HcalGeometryProvider unable to get the name of the "
                      "detector from the RunHeader.");
  }
 
 private:
  framework::config::Parameters params_;
  std::string detector_geometry_;
  ldmx::HcalGeometry* hcal_geometry_;
};
 
HcalGeometryProvider::HcalGeometryProvider(
    const std::string& name, const std::string& tagname,
    const framework::config::Parameters& parameters,
    framework::Process& process)
    : framework::
          ConditionsObjectProvider{ldmx::HcalGeometry::CONDITIONS_OBJECT_NAME,
                                   tagname, parameters, process},
      params_{parameters} {
  hcal_geometry_ = 0;
}
 
HcalGeometryProvider::~HcalGeometryProvider() {
  if (hcal_geometry_) delete hcal_geometry_;
  hcal_geometry_ = 0;
}
 
std::pair<const framework::ConditionsObject*, framework::ConditionsIOV>
HcalGeometryProvider::getCondition(const ldmx::EventHeader& context) {
  static const std::string keyname("detectors_valid");
 
  if (!hcal_geometry_) {
    framework::config::Parameters phex =
        (params_.exists("HcalGeometry"))
            ? (params_.get<framework::config::Parameters>("HcalGeometry"))
            : (params_);
 
    // search through the subtrees
    for (auto key : phex.keys()) {
      framework::config::Parameters pver =
          phex.get<framework::config::Parameters>(key);
 
      if (!pver.exists(keyname)) {
        ldmx_log(warn) << "No parameter " << keyname << " found in " << key;
        // log strange situation and continue
        continue;
      }
 
      std::vector<std::string> dets_valid =
          pver.get<std::vector<std::string> >(keyname);
      for (auto detregex : dets_valid) {
        std::string regex(detregex);
        if (regex.empty()) continue;  // no empty regex allowed
        if (regex[0] != '^') regex.insert(0, 1, '^');
        if (regex.back() != '$') regex += '$';
        regex_t reg;
 
        int rv =
            regcomp(&reg, regex.c_str(), REG_EXTENDED | REG_ICASE | REG_NOSUB);
        if (rv) {
          char err[1024];
          regerror(rv, &reg, err, 1024);
          EXCEPTION_RAISE(
              "GeometryException",
              "Invalid detector regular expression : '" + regex + "' " + err);
        }
        int nmatch = regexec(&reg, detector_geometry_.c_str(), 0, 0, 0);
        regfree(&reg);
        if (!nmatch) {
          hcal_geometry_ = new ldmx::HcalGeometry(pver);
          break;
        }
      }
      if (hcal_geometry_) break;
    }
    if (!hcal_geometry_) {
      EXCEPTION_RAISE("GeometryException", "Unable to create HcalGeometry");
    }
  }
 
  return std::make_pair(
      hcal_geometry_,
      framework::ConditionsIOV(context.getRun(), context.getRun(), true, true));
}
 
}  // namespace hcal
 
DECLARE_CONDITIONS_PROVIDER(hcal::HcalGeometryProvider);