lars/geant4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Import Geant4 6.0.0 source tree

Browse Source
This commit is contained in:
Gabriele Cosmo
2016-06-09 10:41:53 +02:00
parent 4aea781e80
commit 96686e0c8f
6560 changed files with 153347 additions and 238155 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/visualization/HepRep/include/G4HepRepSceneHandler.hh
+108 -48
View File
@@ -31,6 +31,9 @@
#include "globals.hh"
#include <iostream>
#include <stack>
#include <map>
#include <vector>
// HepRep
#include "HEPREP/HepRep.h"
@@ -43,30 +46,29 @@
#include "G4LogicalVolume.hh"
#include "G4PhysicalVolumeModel.hh"
//class G4HepRep;
#include "G4HepRepMessenger.hh"
class G4HepRepSceneHandler: public G4VSceneHandler {
public:
G4HepRepSceneHandler (G4VGraphicsSystem& system, const G4String& name = "");
G4HepRepSceneHandler (G4VGraphicsSystem& system, G4HepRepMessenger& messenger, const G4String& name = "");
virtual ~G4HepRepSceneHandler ();
void AddThis (const G4Box&);
void AddThis (const G4Cons&);
void AddThis (const G4Tubs&);
void AddThis (const G4Trd&);
void AddThis (const G4Trap&);
void AddThis (const G4Sphere&);
void AddThis (const G4Para&);
void AddThis (const G4Torus&);
void AddThis (const G4Polycone&);
void AddThis (const G4Polyhedra&);
void AddThis (const G4VSolid&);
void AddThis (const G4Box& box) { G4VSceneHandler::AddThis (box); }
void AddThis (const G4Cons& cons) { G4VSceneHandler::AddThis (cons); }
void AddThis (const G4Tubs& tubs) { G4VSceneHandler::AddThis (tubs); }
void AddThis (const G4Trd& trd) { G4VSceneHandler::AddThis (trd); }
void AddThis (const G4Trap& trap) { G4VSceneHandler::AddThis (trap); }
void AddThis (const G4Sphere& sphere) { G4VSceneHandler::AddThis (sphere); }
void AddThis (const G4Para& para) { G4VSceneHandler::AddThis (para); }
void AddThis (const G4Torus& torus) { G4VSceneHandler::AddThis (torus); }
void AddThis (const G4Polycone& polycone) { G4VSceneHandler::AddThis (polycone); }
void AddThis (const G4Polyhedra& polyhedra) { G4VSceneHandler::AddThis (polyhedra); }
void AddThis (const G4VSolid& solid) { G4VSceneHandler::AddThis(solid); }
void AddThis (const G4VTrajectory&);
void AddThis (const G4VHit&);
void AddThis (const G4VHit& hit) { G4VSceneHandler::AddThis(hit); }
void PreAddThis (const G4Transform3D& objectTransformation,
const G4VisAttributes& visAttribs);
void PreAddThis (const G4Transform3D& objectTransformation, const G4VisAttributes& visAttribs);
void PostAddThis ();
void EstablishSpecials(G4PhysicalVolumeModel&);
@@ -79,54 +81,112 @@ class G4HepRepSceneHandler: public G4VSceneHandler {
void AddPrimitive (const G4NURBS&);
void AddPrimitive (const G4Polymarker&);
void AddPrimitive (const G4Scale& scale) {
G4VSceneHandler::AddPrimitive(scale);
}
void AddPrimitive (const G4Scale& scale) { G4VSceneHandler::AddPrimitive(scale); }
void BeginPrimitives (const G4Transform3D& objectTransformation);
void EndPrimitives ();
void BeginModeling ();
void EndModeling ();
static G4int GetSceneCount () { return sceneCount; }
static G4int getSceneCount () { return sceneCount; }
HEPREP::HepRep *GetHepRep() { return heprep; }
HEPREP::HepRepWriter *GetHepRepWriter() { return writer; }
HEPREP::HepRepFactory *GetHepRepFactory();
void open();
void close();
protected:
// void RequestPrimitives (const G4VSolid& solid);
void openHepRep();
bool closeHepRep(bool final = false);
void openFile(G4String name);
void closeFile();
private:
static G4int sceneCount; // No. of extanct scenes.
static G4int sceneIdCount; // static counter for scenes.
G4int fileNo; // file sequence number
G4int currentDepth;
G4VPhysicalVolume* currentPV;
G4LogicalVolume* currentLV;
static G4int sceneCount;
const G4ModelingParameters* originalMP; // Keeps pointer to original.
G4ModelingParameters* nonCullingMP; // For temporary non-culling.
G4Transform3D transform;
void SetColour (HEPREP::HepRepAttribute *attribute, const G4Colour& color,const G4String& key = G4String("Color"));
void SetLine (HEPREP::HepRepInstance *instance, const G4Visible& visible);
void SetMarker (HEPREP::HepRepInstance *instance, const G4VMarker& marker);
HEPREP::HepRepInstance* CreateInstance(HEPREP::HepRepInstance* p, HEPREP::HepRepType* altType);
bool IsEventData ();
std::ostream* out;
HEPREP::HepRepFactory* heprepFactory;
HEPREP::HepRepFactory* factory;
HEPREP::HepRepWriter* writer;
HEPREP::HepRepInstance *parent;
HEPREP::HepRep *heprep;
HEPREP::HepRepType *geometryType;
HEPREP::HepRepType *eventType;
HEPREP::HepRepType *trackType;
HEPREP::HepRepType *calHitType;
HEPREP::HepRepType *hitType;
// Methods
void setColor (HEPREP::HepRepAttribute *attribute, const G4Color& color,
const G4String& key = G4String("Color"));
G4Color getColor (G4double charge);
void setVisibility (HEPREP::HepRepAttribute *attribute, const G4Visible& visible);
void setLine (HEPREP::HepRepAttribute *attribute, const G4Visible& visible);
void setMarker (HEPREP::HepRepAttribute *attribute, const G4VMarker& marker);
void setAttribute (HEPREP::HepRepAttribute* attribute, G4String name, G4String value);
void setAttribute (HEPREP::HepRepAttribute* attribute, G4String name, bool value);
void setAttribute (HEPREP::HepRepAttribute* attribute, G4String name, double value);
void setAttribute (HEPREP::HepRepAttribute* attribute, G4String name, int value);
void setAttribute (HEPREP::HepRepAttribute* attribute, G4String name, double red, double green, double blue, double alpha);
bool isEventData ();
void open(G4String name);
void close();
void addAttDefs(HEPREP::HepRepDefinition* definition, const std::map<G4String,G4AttDef>* attDefs);
void addAttVals(HEPREP::HepRepAttribute* attribute, const std::map<G4String,G4AttDef>* attDefs, std::vector<G4AttValue>* attValues);
// Returns the particular instance/type or if not created, creates them and adds them to the HepRep
HEPREP::HepRep* getHepRep();
HEPREP::HepRepInstanceTree* getGeometryInstanceTree();
HEPREP::HepRepInstance* getGeometryInstance(G4LogicalVolume* volume, int depth);
HEPREP::HepRepInstance* getGeometryInstance(G4String volumeName, int depth);
HEPREP::HepRepInstance* getGeometryRootInstance();
HEPREP::HepRepTypeTree* getGeometryTypeTree();
HEPREP::HepRepType* getGeometryType(G4String volumeName, int depth);
HEPREP::HepRepType* getGeometryRootType();
HEPREP::HepRepInstanceTree* getEventInstanceTree();
HEPREP::HepRepInstance* getEventInstance();
HEPREP::HepRepTypeTree* getEventTypeTree();
HEPREP::HepRepType* getEventType();
HEPREP::HepRepType* getTrajectoryType ();
HEPREP::HepRepType* getHitType ();
HEPREP::HepRepType* getCalHitType ();
HEPREP::HepRepType* getCalHitFaceType ();
G4String getFullTypeName(G4String volumeName, int depth);
G4String getParentTypeName(int currentDepth);
// initialized Member Variables
G4HepRepMessenger& messenger;
G4String geometryLayer, eventLayer, calHitLayer;
G4String trajectoryLayer, hitLayer;
G4String rootVolumeName;
G4String baseName;
G4String eventNumberPrefix;
G4String eventNumberSuffix;
G4int eventNumber;
G4int eventNumberWidth;
G4String extension;
G4bool writeMultipleFiles;
G4int currentDepth;
G4VPhysicalVolume* currentPV;
G4LogicalVolume* currentLV;
// DO NOT USE member vars directly, use get methods.
HEPREP::HepRep* _heprep;
HEPREP::HepRepInstanceTree* _geometryInstanceTree;
std::vector<HEPREP::HepRepInstance*> _geometryInstance;
HEPREP::HepRepInstance* _geometryRootInstance;
HEPREP::HepRepTypeTree* _geometryTypeTree;
std::vector<G4String> _geometryTypeName;
std::map<G4String, HEPREP::HepRepType*> _geometryType;
HEPREP::HepRepType* _geometryRootType;
HEPREP::HepRepInstanceTree* _eventInstanceTree;
HEPREP::HepRepInstance* _eventInstance;
HEPREP::HepRepTypeTree* _eventTypeTree;
HEPREP::HepRepType* _eventType;
HEPREP::HepRepType* _trajectoryType;
HEPREP::HepRepType* _hitType;
HEPREP::HepRepType* _calHitType;
HEPREP::HepRepType* _calHitFaceType;
std::map<int, G4String> materialState;
};
#endif