SimTracker/TrackerMaterialAnalysis/plugins/TrackingMaterialProducer.h - master - githubmirror/cmssw

SimTracker/TrackerMaterialAnalysis/plugins/TrackingMaterialProducer.h

76 строк2 KB

Ivan Razumov

Do not forward-declare G4VTouchable

12 окт 2023, 13:10

12 окт 2023, 13:10e9436e2

#ifndef TrackingMaterialProducer_h
#define TrackingMaterialProducer_h
#include <string>
#include <vector>
#include <fstream>
#include <G4VTouchable.hh>

#include "SimG4Core/Watcher/interface/SimProducer.h"
#include "SimG4Core/Notification/interface/Observer.h"

#include "G4LogicalVolume.hh"

#include "SimDataFormats/ValidationFormats/interface/MaterialAccountingTrack.h"

#include "TProfile.h"
#include "TFile.h"

class BeginOfJob;
class EndOfJob;
class BeginOfEvent;
class BeginOfTrack;
class EndOfTrack;
class G4Step;

class G4StepPoint;
class G4VPhysicalVolume;
class G4LogicalVolume;
class G4TouchableHistory;
namespace edm {
  class ParameterSet;
}

class TrackingMaterialProducer : public SimProducer,
                                 public Observer<const BeginOfJob*>,
                                 public Observer<const EndOfJob*>,
                                 public Observer<const BeginOfEvent*>,
                                 public Observer<const BeginOfTrack*>,
                                 public Observer<const G4Step*>,
                                 public Observer<const EndOfTrack*> {
public:
  TrackingMaterialProducer(const edm::ParameterSet&);
  ~TrackingMaterialProducer() override;

private:
  void update(const BeginOfJob*) override;
  void update(const BeginOfEvent*) override;
  void update(const BeginOfTrack*) override;
  void update(const G4Step*) override;
  void update(const EndOfTrack*) override;
  void update(const EndOfJob*) override;
  void produce(edm::Event&, const edm::EventSetup&) override;

  bool isSelected(const G4VTouchable* touch);
  bool isSelectedFast(const G4TouchableHistory* touch);

private:
  bool m_primaryTracks;
  std::vector<std::string> m_selectedNames;
  std::vector<const G4LogicalVolume*> m_selectedVolumes;
  std::string m_txtOutFile;
  double m_hgcalzfront;
  MaterialAccountingTrack m_track;
  const G4VPhysicalVolume* m_track_volume;
  std::vector<MaterialAccountingTrack>* m_tracks;
  TFile* output_file_;
  TProfile* radLen_vs_eta_;
  bool isHGCal;
  bool isHFNose;
  static constexpr float innerHGCalEta = 2.4;
  static constexpr float outerHGCalEta = 2.0;
  static constexpr float innerHFnoseEta = 4.;
  static constexpr float outerHFnoseEta = 3.3;
  std::ofstream outVolumeZpositionTxt;
};

#endif  // TrackingMaterialProducer_h