Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-11-06 04:14:00

0001 #ifndef DD4hep_TrackingMaterialAnalyser_h
0002 #define DD4hep_TrackingMaterialAnalyser_h
0003 #include <string>
0004 #include <vector>
0005 
0006 #include "FWCore/Framework/interface/one/EDAnalyzer.h"
0007 #include "FWCore/Framework/interface/Event.h"
0008 #include "FWCore/Framework/interface/EventSetup.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0010 #include "FWCore/Utilities/interface/InputTag.h"
0011 
0012 #include "SimDataFormats/ValidationFormats/interface/MaterialAccountingTrack.h"
0013 #include "DD4hep_MaterialAccountingGroup.h"
0014 #include "DD4hep_TrackingMaterialPlotter.h"
0015 
0016 class DD4hep_TrackingMaterialAnalyser : public edm::one::EDAnalyzer<> {
0017 public:
0018   explicit DD4hep_TrackingMaterialAnalyser(const edm::ParameterSet &);
0019   ~DD4hep_TrackingMaterialAnalyser() override;
0020 
0021 private:
0022   enum SplitMode { NEAREST_LAYER, INNER_LAYER, OUTER_LAYER, UNDEFINED };
0023 
0024   void analyze(const edm::Event &, const edm::EventSetup &) override;
0025   void beginJob() override {}
0026   void endJob() override;
0027 
0028   void split(MaterialAccountingTrack &track);
0029   int findLayer(const MaterialAccountingDetector &detector);
0030 
0031   void saveParameters(const char *name);
0032   void saveXml(const char *name);
0033   void saveLayerPlots();
0034 
0035   edm::EDGetTokenT<std::vector<MaterialAccountingTrack> > m_materialToken;
0036   edm::ESGetToken<cms::DDCompactView, IdealGeometryRecord> m_dddToken;
0037   SplitMode m_splitMode;
0038   bool m_skipAfterLastDetector;
0039   bool m_skipBeforeFirstDetector;
0040   bool m_saveSummaryPlot;
0041   bool m_saveDetailedPlots;
0042   bool m_saveParameters;
0043   bool m_saveXml;
0044   bool m_isHGCal;
0045   bool m_isHFNose;
0046   std::vector<DD4hep_MaterialAccountingGroup *> m_groups;
0047   std::vector<std::string> m_groupNames;
0048   std::unique_ptr<DD4hep_TrackingMaterialPlotter> m_plotter;
0049 };
0050 
0051 #endif  // DD4hep_TrackingMaterialAnalyser_h