g4simulation/g4micromegas/PHG4MicromegasDisplayAction.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef G4MICROMEGAS_PHG4MICROMEGASDISPLAYACTION_H
0004 #define G4MICROMEGAS_PHG4MICROMEGASDISPLAYACTION_H
0005
0006 #include <g4main/PHG4DisplayAction.h>
0007
0008 #include <Geant4/G4Color.hh>
0009
0010 #include <map>
0011 #include <string>  // for string
0012 #include <vector>
0013
0014 class G4LogicalVolume;
0015 class G4VisAttributes;
0016 class G4VPhysicalVolume;
0017
0018 class PHG4MicromegasDisplayAction : public PHG4DisplayAction
0019 {
0020  public:
0021   PHG4MicromegasDisplayAction(const std::string &name);
0022
0023   ~PHG4MicromegasDisplayAction() override;
0024
0025   void ApplyDisplayAction(G4VPhysicalVolume *physvol) override;
0026   void AddVolume(G4LogicalVolume *logvol, const G4Colour &col) { m_LogicalVolumeMap[logvol] = col; }
0027
0028  private:
0029   std::map<G4LogicalVolume *, G4Color> m_LogicalVolumeMap;
0030   std::vector<G4VisAttributes *> m_VisAttVec;
0031 };
0032
0033 #endif  // G4MICROMEGAS_PHG4MICROMEGASDISPLAYACTION_H