TrackerGeometry.h

Go to the documentation of this file.

#ifndef Geometry_TrackerGeometryBuilder_TrackerGeometry_H
#define Geometry_TrackerGeometryBuilder_TrackerGeometry_H

#include "Geometry/CommonDetUnit/interface/TrackingGeometry.h"
#include "Geometry/CommonDetUnit/interface/GeomDetEnumerators.h"
#include "Geometry/CommonDetUnit/interface/TrackerGeomDet.h"

class GeometricDet;

#include "Geometry/TrackerGeometryBuilder/interface/trackerHierarchy.h"

#include "DataFormats/Common/interface/Trie.h"

// FIXME here just to allow prototyping...
namespace trackerTrie {
  typedef GeomDet const* PDet;
  typedef edm::Trie<PDet> DetTrie;
  typedef edm::TrieNode<PDet> Node;
  typedef Node const * node_pointer; // sigh....
  typedef edm::TrieNodeIter<PDet> node_iterator;
}


class TrackerGeometry final : public TrackingGeometry {

  explicit TrackerGeometry(GeometricDet const* gd=0);  

  friend class TrackerGeomBuilderFromGeometricDet;

  void addType(GeomDetType const * p);
  void addDetUnit(GeomDetUnit const * p);
  void addDetUnitId(DetId p);
  void addDet(GeomDet const * p);
  void addDetId(DetId p);
  void finalize();

public:
  typedef GeomDetEnumerators::SubDetector SubDetector;

  enum class ModuleType {
    UNKNOWN, 
      PXB, 
      PXF, 
      IB1, 
      IB2, 
      OB1, 
      OB2, 
      W1A, 
      W2A, 
      W3A, 
      W1B, 
      W2B, 
      W3B, 
      W4, 
      W5, 
      W6, 
      W7, 
      Ph1PXB, 
      Ph1PXF, 
      Ph2PXB, 
      Ph2PXF, 
      Ph2PSP, 
      Ph2PSS, 
      Ph2SS
   };

  virtual ~TrackerGeometry() ;

  const DetTypeContainer&  detTypes()         const {return theDetTypes;}
  const DetUnitContainer&  detUnits()         const {return theDetUnits;}
  const DetContainer&      dets()             const {return theDets;}
  const DetIdContainer&    detUnitIds()       const {return theDetUnitIds;}
  const DetIdContainer&    detIds()           const { return theDetIds;}
  const TrackerGeomDet*    idToDetUnit(DetId) const;
  const TrackerGeomDet*    idToDet(DetId)     const;

  const GeomDetEnumerators::SubDetector geomDetSubDetector(int subdet) const;
  unsigned int numberOfLayers(int subdet) const;
  bool isThere(GeomDetEnumerators::SubDetector subdet) const;

  unsigned int offsetDU(SubDetector sid) const { return theOffsetDU[sid];}
  unsigned int endsetDU(SubDetector sid) const { return theEndsetDU[sid];}
  // Magic : better be called at the right moment...
  void setOffsetDU(SubDetector sid) { theOffsetDU[sid]=detUnits().size();}
  void setEndsetDU(SubDetector sid) { theEndsetDU[sid]=detUnits().size();}
  void fillTestMap(const GeometricDet* gd);

  ModuleType moduleType(const std::string& name) const;

  GeometricDet const * trackerDet() const {return  theTrackerDet;}

  const DetContainer& detsPXB() const;
  const DetContainer& detsPXF() const;
  const DetContainer& detsTIB() const;
  const DetContainer& detsTID() const;
  const DetContainer& detsTOB() const;
  const DetContainer& detsTEC() const;

  ModuleType getDetectorType(DetId) const;
  float getDetectorThickness(DetId) const;


private:

  GeometricDet const * theTrackerDet; 

  friend class GeometryAligner;

  DetTypeContainer  theDetTypes;  // owns the DetTypes
  DetUnitContainer  theDetUnits;  // they're all also into 'theDets', so we assume 'theDets' owns them
  unsigned int      theOffsetDU[6]; // offsets in the above
  unsigned int      theEndsetDU[6]; // end offsets in the above
  DetContainer      theDets;      // owns *ONLY* the GeomDet * corresponding to GluedDets.
  DetIdContainer    theDetUnitIds;
  DetIdContainer    theDetIds; 
  mapIdToDetUnit    theMapUnit; // does not own GeomDetUnit *
  mapIdToDet        theMap;     // does not own GeomDet *

  DetContainer      thePXBDets; // not owned: they're also in 'theDets'
  DetContainer      thePXFDets; // not owned: they're also in 'theDets'
  DetContainer      theTIBDets; // not owned: they're also in 'theDets'
  DetContainer      theTIDDets; // not owned: they're also in 'theDets'
  DetContainer      theTOBDets; // not owned: they're also in 'theDets'
  DetContainer      theTECDets; // not owned: they're also in 'theDets'

  GeomDetEnumerators::SubDetector theSubDetTypeMap[6];
  unsigned int theNumberOfLayers[6];
  std::vector< std::tuple< DetId, TrackerGeometry::ModuleType, float> > theDetTypetList; 
};

#endif