MomentumDependentPedeLabeler.h

Go to the documentation of this file.

#ifndef MomentumDEPENDENTPEDELABELER_H
#define MomentumDEPENDENTPEDELABELER_H

#include <vector>
#include <map>

#include <Alignment/MillePedeAlignmentAlgorithm/interface/PedeLabelerBase.h>

#include "Alignment/CommonAlignment/interface/Utilities.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

/***************************************
****************************************/
class MomentumDependentPedeLabeler : public PedeLabelerBase {
public:
  MomentumDependentPedeLabeler(const PedeLabelerBase::TopLevelAlignables &alignables, const edm::ParameterSet &config);
  ~MomentumDependentPedeLabeler() override;

  unsigned int alignableLabel(Alignable *alignable) const override;
  unsigned int alignableLabelFromParamAndInstance(Alignable *alignable,
                                                  unsigned int param,
                                                  unsigned int instance) const override;
  unsigned int lasBeamLabel(unsigned int lasBeamId) const override;
  unsigned int parameterLabel(unsigned int aliLabel, unsigned int parNum) const override;
  unsigned int parameterLabel(Alignable *alignable,
                              unsigned int parNum,
                              const AlignmentAlgorithmBase::EventInfo &eventInfo,
                              const TrajectoryStateOnSurface &tsos) const override;
  bool hasSplitParameters(Alignable *alignable) const override;
  unsigned int numberOfParameterInstances(Alignable *alignable, int param = -1) const override;
  unsigned int maxNumberOfParameterInstances() const override { return theMaxNumberOfParameterInstances; }

  unsigned int paramNumFromLabel(unsigned int paramLabel) const override;
  unsigned int alignableLabelFromLabel(unsigned int label) const override;
  Alignable *alignableFromLabel(unsigned int label) const override;
  unsigned int lasBeamIdFromLabel(unsigned int label) const override;

private:
  typedef std::map<Alignable *, unsigned int> AlignableToIdMap;
  typedef AlignableToIdMap::value_type AlignableToIdPair;
  typedef std::pair<float, float> MomentumRange;
  typedef std::vector<MomentumRange> MomentumRangeVector;
  typedef std::map<unsigned int, MomentumRangeVector> MomentumRangeParamMap;
  typedef std::map<Alignable *, MomentumRangeParamMap> AlignableToMomentumRangeMap;
  typedef AlignableToMomentumRangeMap::value_type AlignableToMomentumRangePair;
  typedef std::map<unsigned int, Alignable *> IdToAlignableMap;
  typedef std::map<unsigned int, unsigned int> UintUintMap;

  std::vector<std::string> decompose(const std::string &s, std::string::value_type delimiter) const;
  std::vector<unsigned int> convertParamSel(const std::string &selString) const;
  unsigned int buildMomentumDependencyMap(AlignableTracker *aliTracker,
                                          AlignableMuon *aliMuon,
                                          AlignableExtras *extras,
                                          const edm::ParameterSet &config);

  unsigned int buildMap(const align::Alignables &);
  unsigned int buildReverseMap();

  // data members
  AlignableToIdMap theAlignableToIdMap;  
  AlignableToMomentumRangeMap theAlignableToMomentumRangeMap;  
  IdToAlignableMap theIdToAlignableMap;                        
  UintUintMap theLasBeamToLabelMap;                            
  UintUintMap theLabelToLasBeamMap;                            
  const MomentumRange theOpenMomentumRange;
  unsigned int theMaxNumberOfParameterInstances;
};

#endif