g4simulation/g4epd/PHG4EPDDisplayAction.h

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