FWTGeoRecoGeometryESProducer.h

Go to the documentation of this file.

#ifndef GEOMETRY_FWTGEORECO_GEOMETRY_ES_PRODUCER_H
# define GEOMETRY_FWTGEORECO_GEOMETRY_ES_PRODUCER_H

# include <memory>

# include "FWCore/Framework/interface/ESProducer.h"
# include "FWCore/Framework/interface/ESHandle.h"
# include "DataFormats/GeometryVector/interface/GlobalPoint.h"

namespace edm
{
   class ParameterSet;
}

class CaloGeometry;
class GlobalTrackingGeometry;
class TrackerGeometry;
class TrackerTopology;
class FWTGeoRecoGeometry;
class FWTGeoRecoGeometryRecord;

class TGeoManager;
class TGeoShape;
class TGeoVolume;
class TGeoMaterial;
class TGeoMedium;
class GeomDet;
class CaloCellGeometry;
class FWTGeoRecoGeometryESProducer : public edm::ESProducer
{
   enum ERecoDet  {kDummy, 
                   kSiPixel, kSiStrip,
                   kMuonDT, kMuonRPC, kMuonCSC, kMuonGEM, kMuonME0,
                   kECal, kHCal, kCaloTower,
                   kHGCE, kHGCH };
public:
   FWTGeoRecoGeometryESProducer( const edm::ParameterSet& );
   virtual ~FWTGeoRecoGeometryESProducer( void );
  
   std::shared_ptr<FWTGeoRecoGeometry> produce( const FWTGeoRecoGeometryRecord& );

private:
   FWTGeoRecoGeometryESProducer( const FWTGeoRecoGeometryESProducer& );
   const FWTGeoRecoGeometryESProducer& operator=( const FWTGeoRecoGeometryESProducer& );

   TGeoManager*      createManager( int level );
   TGeoShape*        createShape( const GeomDet *det );
   TGeoVolume*       createVolume( const std::string& name, const GeomDet *det, ERecoDet = kDummy );
   // TGeoMaterial*     createMaterial( const std::string& name );

   TGeoVolume*  GetDaughter(TGeoVolume* mother, const char* prefix, ERecoDet cidx, int id);
   TGeoVolume*  GetDaughter(TGeoVolume* mother, const char* prefix, ERecoDet cidx);
   TGeoVolume*  GetTopHolder( const char* prefix, ERecoDet cidx);

   TGeoMedium* GetMedium(ERecoDet);

   void addPixelBarrelGeometry();
   void addPixelForwardGeometry();
   void addTIBGeometry();
   void addTOBGeometry();
   void addTIDGeometry();
   void addTECGeometry();
   void addCSCGeometry();
   void addDTGeometry();
   void addRPCGeometry();
   void addGEMGeometry();
   void addME0Geometry();
   void addEcalCaloGeometry();
   void addHcalCaloGeometryBarrel();
   void addHcalCaloGeometryEndcap();
   void addHcalCaloGeometryOuter();
   void addHcalCaloGeometryForward();
   void addCaloTowerGeometry();
  
   std::map<std::string, TGeoShape*>    m_nameToShape;
   std::map<TGeoShape*, TGeoVolume*>   m_shapeToVolume;
   std::map<ERecoDet, TGeoMedium*> m_recoMedium;

   edm::ESHandle<GlobalTrackingGeometry> m_geomRecord;
   edm::ESHandle<CaloGeometry>           m_caloGeom;
   const TrackerGeometry* m_trackerGeom;
   const TrackerTopology* m_trackerTopology;
  
   std::shared_ptr<FWTGeoRecoGeometry> m_fwGeometry;

   TGeoMedium* m_dummyMedium;

   bool m_tracker;
   bool m_muon;
   bool m_calo;
};

#endif // GEOMETRY_FWTGEORECO_GEOMETRY_ES_PRODUCER_H