Alignment/CommonAlignment/interface/AlignableObjectId.h - master - githubmirror/cmssw

Alignment/CommonAlignment/interface/AlignableObjectId.h
66 строк3 KB
hyunyong
add GEM to MuonAlignment
06 мар 2021, 01:53
06 мар 2021, 01:53375f214
#ifndef Alignment_CommonAlignment_AlignableObjectId_h
#define Alignment_CommonAlignment_AlignableObjectId_h

#include <string>
#include "Alignment/CommonAlignment/interface/StructureType.h"

class TrackerGeometry;
class DTGeometry;
class CSCGeometry;
class GEMGeometry;

/// Allows conversion between type and name, and vice-versa
class AlignableObjectId {
public:
  struct entry;
  enum class Geometry { RunI, PhaseI, PhaseII, General, Unspecified };

  AlignableObjectId(Geometry);
  AlignableObjectId(const TrackerGeometry*, const DTGeometry*, const CSCGeometry*, const GEMGeometry*);
  AlignableObjectId(const AlignableObjectId&) = default;
  AlignableObjectId& operator=(const AlignableObjectId&) = default;
  AlignableObjectId(AlignableObjectId&&) = default;
  AlignableObjectId& operator=(AlignableObjectId&&) = default;
  virtual ~AlignableObjectId() = default;

  /// retrieve the geometry information
  Geometry geometry() const { return geometry_; }

  /// Convert name to type
  align::StructureType nameToType(const std::string& name) const;

  /// Convert type to name
  std::string typeToName(align::StructureType type) const;
  const char* idToString(align::StructureType type) const;
  align::StructureType stringToId(const char*) const;
  align::StructureType stringToId(const std::string& s) const { return stringToId(s.c_str()); }

  static Geometry commonGeometry(Geometry, Geometry);
  static AlignableObjectId commonObjectIdProvider(const AlignableObjectId&, const AlignableObjectId&);
  template <typename TRACKER, typename MUON>
  static AlignableObjectId commonObjectIdProvider(const TRACKER*, const MUON*);
  template <typename TRACKER>
  static AlignableObjectId commonObjectIdProvider(const TRACKER*, std::nullptr_t);

private:
  static Geometry trackerGeometry(const TrackerGeometry*);
  static Geometry muonGeometry(const DTGeometry*, const CSCGeometry*, const GEMGeometry*);

  const entry* entries_{nullptr};
  Geometry geometry_{Geometry::Unspecified};
};

template <typename TRACKER, typename MUON>
AlignableObjectId AlignableObjectId ::commonObjectIdProvider(const TRACKER* tracker, const MUON* muon) {
  auto trackerGeometry = (tracker ? tracker->objectIdProvider().geometry() : AlignableObjectId::Geometry::General);
  auto muonGeometry = (muon ? muon->objectIdProvider().geometry() : AlignableObjectId::Geometry::General);
  return AlignableObjectId::commonGeometry(trackerGeometry, muonGeometry);
}

template <typename T>
AlignableObjectId AlignableObjectId ::commonObjectIdProvider(const T* tracker, std::nullptr_t) {
  auto trackerGeometry = (tracker ? tracker->objectIdProvider().geometry() : AlignableObjectId::Geometry::General);
  return AlignableObjectId::commonGeometry(trackerGeometry, AlignableObjectId::Geometry::General);
}

#endif