HepPDTESSource.cc

Go to the documentation of this file.

#include "HepPDT/HeavyIonUnknownID.hh"
#include "SimGeneral/HepPDTESSource/interface/HepPDTESSource.h"
#include "tbb/concurrent_vector.h"

namespace {

  class CachingHeavyIonUnknownID : public HepPDT::ProcessUnknownID {
    HepPDT::ParticleData *processUnknownID(HepPDT::ParticleID id, const HepPDT::ParticleDataTable &table) final {
      // HeavyIonUnknownID constructs a new particle but does not delete it
      // we need to do that ourselves
      std::unique_ptr<HepPDT::ParticleData> p{wrapped_.processUnknownID(id, table)};
      auto *pPtr = p.get();
      if (p) {
        particles_.emplace_back(std::move(p));
      }
      return pPtr;
    }

    HepPDT::HeavyIonUnknownID wrapped_;
    tbb::concurrent_vector<std::unique_ptr<HepPDT::ParticleData>> particles_;
  };

}  // namespace

HepPDTESSource::HepPDTESSource(const edm::ParameterSet &cfg)
    : pdtFileName(cfg.getParameter<edm::FileInPath>("pdtFileName")) {
  setWhatProduced(this);
  findingRecord<PDTRecord>();
}

HepPDTESSource::~HepPDTESSource() {}

HepPDTESSource::ReturnType HepPDTESSource::produce(const PDTRecord &iRecord) {
  using namespace edm::es;
  auto pdt = std::make_unique<PDT>("PDG table", new CachingHeavyIonUnknownID);
  std::ifstream pdtFile(pdtFileName.fullPath().c_str());
  if (!pdtFile)
    throw cms::Exception("FileNotFound", "can't open pdt file") << "cannot open " << pdtFileName.fullPath();
  {  // notice: the builder has to be destroyed
    // in order to fill the table!
    HepPDT::TableBuilder builder(*pdt);
    if (!addParticleTable(pdtFile, builder)) {
      throw cms::Exception("ConfigError", "can't read pdt file") << "wrong format of " << pdtFileName.fullPath();
    }
  }
  return pdt;
}

void HepPDTESSource::setIntervalFor(const edm::eventsetup::EventSetupRecordKey &,
                                    const edm::IOVSyncValue &,
                                    edm::ValidityInterval &oInterval) {
  // the same PDT is valid for any time
  oInterval = edm::ValidityInterval(edm::IOVSyncValue::beginOfTime(), edm::IOVSyncValue::endOfTime());
}

// define this as a plug-in
// DEFINE_FWK_EVENTSETUP_SOURCE( HepPDTESSource );