CombinedTrajectoryFactory.cc

Go to the documentation of this file.

#include <memory>
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "Alignment/ReferenceTrajectories/interface/TrajectoryFactoryPlugin.h"
 
#include <TString.h>
#include <TObjArray.h>
 
#include "Alignment/ReferenceTrajectories/interface/TrajectoryFactoryBase.h"
 
 
class CombinedTrajectoryFactory : public TrajectoryFactoryBase {
public:
  CombinedTrajectoryFactory(const edm::ParameterSet &config);
  ~CombinedTrajectoryFactory() override;
 
  const ReferenceTrajectoryCollection trajectories(const edm::EventSetup &setup,
                                                   const ConstTrajTrackPairCollection &tracks,
                                                   const reco::BeamSpot &beamSpot) const override;
 
  const ReferenceTrajectoryCollection trajectories(const edm::EventSetup &setup,
                                                   const ConstTrajTrackPairCollection &tracks,
                                                   const ExternalPredictionCollection &external,
                                                   const reco::BeamSpot &beamSpot) const override;
 
  CombinedTrajectoryFactory *clone() const override { return new CombinedTrajectoryFactory(*this); }
 
private:
  CombinedTrajectoryFactory(const CombinedTrajectoryFactory &other)
      : TrajectoryFactoryBase(other), theUseAllFactories{other.theUseAllFactories} {
    theFactories.reserve(other.theFactories.size());
    for (const auto &f : other.theFactories) {
      theFactories.emplace_back(f->clone());
    }
  }
 
  std::vector<std::unique_ptr<TrajectoryFactoryBase>> theFactories;
  bool theUseAllFactories;  
};
 
 
using namespace std;
 
CombinedTrajectoryFactory::CombinedTrajectoryFactory(const edm::ParameterSet &config)
    : TrajectoryFactoryBase(config), theUseAllFactories(config.getParameter<bool>("useAllFactories")) {
  vector<string> factoryNames = config.getParameter<vector<string>>("TrajectoryFactoryNames");
  for (auto const &factoryName : factoryNames) {
    // auto_ptr to avoid missing a delete due to throw...
    std::unique_ptr<TObjArray> namePset(TString(factoryName.c_str()).Tokenize(","));
    if (namePset->GetEntriesFast() != 2) {
      throw cms::Exception("BadConfig") << "@SUB=CombinedTrajectoryFactory"
                                        << "TrajectoryFactoryNames must contain 2 comma "
                                        << "separated strings, but is '" << factoryName << "'";
    }
    const edm::ParameterSet factoryCfg = config.getParameter<edm::ParameterSet>(namePset->At(1)->GetName());
    theFactories.emplace_back(TrajectoryFactoryPlugin::get()->create(namePset->At(0)->GetName(), factoryCfg));
  }
}
 
CombinedTrajectoryFactory::~CombinedTrajectoryFactory(void) {}
 
const CombinedTrajectoryFactory::ReferenceTrajectoryCollection CombinedTrajectoryFactory::trajectories(
    const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const reco::BeamSpot &beamSpot) const {
  ReferenceTrajectoryCollection trajectories;
  ReferenceTrajectoryCollection tmpTrajectories;  // outside loop for efficiency
 
  for (auto const &factory : theFactories) {
    tmpTrajectories = factory->trajectories(setup, tracks, beamSpot);
    trajectories.insert(trajectories.end(), tmpTrajectories.begin(), tmpTrajectories.end());
 
    if (!theUseAllFactories && !trajectories.empty())
      break;
  }
 
  return trajectories;
}
 
const CombinedTrajectoryFactory::ReferenceTrajectoryCollection CombinedTrajectoryFactory::trajectories(
    const edm::EventSetup &setup,
    const ConstTrajTrackPairCollection &tracks,
    const ExternalPredictionCollection &external,
    const reco::BeamSpot &beamSpot) const {
  ReferenceTrajectoryCollection trajectories;
  ReferenceTrajectoryCollection tmpTrajectories;  // outside loop for efficiency
 
  for (auto const &factory : theFactories) {
    tmpTrajectories = factory->trajectories(setup, tracks, external, beamSpot);
    trajectories.insert(trajectories.end(), tmpTrajectories.begin(), tmpTrajectories.end());
 
    if (!theUseAllFactories && !trajectories.empty())
      break;
  }
 
  return trajectories;
}
 
DEFINE_EDM_PLUGIN(TrajectoryFactoryPlugin, CombinedTrajectoryFactory, "CombinedTrajectoryFactory");