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

Import Geant4 5.1.0 source tree

Browse Source
This commit is contained in:
Gabriele Cosmo
2016-06-09 10:15:15 +02:00
parent 37fff30d2e
commit fbd4999cf7
4396 changed files with 56662 additions and 52446 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/visualization/HepRep/src/G4HepRepFile.cc
+1 -1
View File
@@ -44,7 +44,7 @@
//
//
// $Id: G4HepRepFile.cc,v 1.6 2002/12/13 11:18:03 gunter Exp $
// GEANT4 tag $Name: geant4-05-00 $
// GEANT4 tag $Name: geant4-05-01 $
//
//
// Joseph Perl 1 October 2001
source/visualization/HepRep/src/G4HepRepFileSceneHandler.cc
+216 -19
View File
@@ -43,8 +43,8 @@
// ********************************************************************
//
//
// $Id: G4HepRepFileSceneHandler.cc,v 1.10 2002/12/13 11:18:03 gunter Exp $
// GEANT4 tag $Name: geant4-05-00 $
// $Id: G4HepRepFileSceneHandler.cc,v 1.11 2003/01/24 21:18:48 perl Exp $
// GEANT4 tag $Name: geant4-05-01 $
//
//
// Joseph Perl 27th January 2002
@@ -68,7 +68,10 @@
#include "G4Polyhedron.hh"
#include "G4NURBS.hh"
#include "G4VTrajectory.hh"
#include "G4VTrajectoryPoint.hh"
#include "G4VHit.hh"
#include "G4AttDef.hh"
#include "G4AttValue.hh"
//HepRep
#include "HepRepXMLWriter.hh"
@@ -252,7 +255,185 @@ void G4HepRepFileSceneHandler::AddThis (const G4VTrajectory& traj) {
#ifdef G4HEPREPFILEDEBUG
G4cout << "G4HepRepFileSceneHandler::AddThis(G4VTrajectory&) " << G4endl;
#endif
G4VSceneHandler::AddThis (traj);
G4std::vector<G4AttValue>* attValues = traj.CreateAttValues();
G4std::vector<G4AttValue>::iterator iAttVal;
const G4std::map<G4String,G4AttDef>* attDefs = traj.GetAttDefs();
G4std::map<G4String,G4AttDef>::const_iterator iAttDef;
G4int i;
// Open the HepRep output file if it is not already open.
CheckFileOpen();
// Add the Event Data Type if it hasn't already been added.
// If this is the first trajectory, add the Event Data type and define attributes.
if (strcmp("Event Data",hepRepXMLWriter->prevTypeName[0])!=0) {
hepRepXMLWriter->addType("Event Data",0);
hepRepXMLWriter->addInstance();
}
// Add the Trajectories Type.
G4String previousName = hepRepXMLWriter->prevTypeName[1];
hepRepXMLWriter->addType("Trajectories",1);
// If this is the first trajectory of this event...
if (strcmp("Trajectories",previousName)!=0) {
// Specify attribute values common to all trajectories.
hepRepXMLWriter->addAttValue("DrawAs","Line");
hepRepXMLWriter->addAttValue("Layer",100);
// Specify additional attribute definitions for trajectories.
// Take all Trajectory attDefs from first trajectory.
// Would rather be able to get these attDefs without needing a reference from any
// particular trajectory, but don't know how to do that.
if (attValues && attDefs) {
for (iAttVal = attValues->begin();
iAttVal != attValues->end(); ++iAttVal) {
iAttDef = attDefs->find(iAttVal->GetName());
if (iAttDef != attDefs->end()) {
// Protect against incorrect use of Category. Anything value other than the
// standard ones will be considered to be in the physics category.
G4String category = iAttDef->second.GetCategory();
if (strcmp(category,"Draw")!=0 &&
strcmp(category,"Physics")!=0 &&
strcmp(category,"Association")!=0 &&
strcmp(category,"PickAction")!=0)
category = "Physics";
hepRepXMLWriter->addAttDef(iAttVal->GetName(), iAttDef->second.GetDesc(),
category, iAttDef->second.GetExtra());
}
}
}
// Specify additional attribute definitions for trajectory points.
// Take all TrajectoryPoint attDefs from first point of first trajectory.
// Would rather be able to get these attDefs without needing a reference from any
// particular point, but don't know how to do that.
// Note also that until we get the good separation of Types and Instances that comes
// in HepRep2, the user must be careful not to use the same AttName for two
// different Types.
if (traj.GetPointEntries()>0) {
G4VTrajectoryPoint* aTrajectoryPoint = traj.GetPoint(0);
G4std::vector<G4AttValue>* pointAttValues
= aTrajectoryPoint->CreateAttValues();
const G4std::map<G4String,G4AttDef>* pointAttDefs
= aTrajectoryPoint->GetAttDefs();
if (pointAttValues && pointAttDefs) {
for (iAttVal = pointAttValues->begin();
iAttVal != pointAttValues->end(); ++iAttVal) {
iAttDef =
pointAttDefs->find(iAttVal->GetName());
if (iAttDef != pointAttDefs->end()) {
// Protect against incorrect use of Category. Anything value other than the
// standard ones will be considered to be in the physics category.
G4String category = iAttDef->second.GetCategory();
if (strcmp(category,"Draw")!=0 &&
strcmp(category,"Physics")!=0 &&
strcmp(category,"Association")!=0 &&
strcmp(category,"PickAction")!=0)
category = "Physics";
hepRepXMLWriter->addAttDef(iAttVal->GetName(), iAttDef->second.GetDesc(),
category, iAttDef->second.GetExtra());
}
}
}
}
}
// For every trajectory, add an instance of Type Trajectory.
hepRepXMLWriter->addInstance();
// Set the LineColor attribute according to the particle charge.
float redness = 0.;
float greenness = 0.;
float blueness = 0.;
const G4double charge = traj.GetCharge();
if(charge>0.) blueness = 1.; // Blue = positive.
else if(charge<0.) redness = 1.; // Red = negative.
else greenness = 1.; // Green = neutral.
hepRepXMLWriter->addAttValue("LineColor",redness,greenness,blueness);
// Copy the current trajectory's G4AttValues to HepRepAttValues.
if (attValues && attDefs) {
for (iAttVal = attValues->begin();
iAttVal != attValues->end(); ++iAttVal) {
G4std::map<G4String,G4AttDef>::const_iterator iAttDef =
attDefs->find(iAttVal->GetName());
if (iAttDef == attDefs->end()) {
G4cout << "G4HepRepFileSceneHandler::AddThis(traj):"
"\n WARNING: no matching definition for attribute \""
<< iAttVal->GetName() << "\", value: "
<< iAttVal->GetValue();
}
else {
// Use GetDesc rather than GetName once WIRED can handle names with spaces in them.
//hepRepXMLWriter->addAttValue(iAttDef->second.GetDesc(), iAttVal->GetValue());
hepRepXMLWriter->addAttValue(iAttVal->GetName(), iAttVal->GetValue());
}
}
delete attValues; // AttValues must be deleted after use.
}
// Each trajectory is made of a single primitive, a polyline.
hepRepXMLWriter->addPrimitive();
// Specify the polyline by using the trajectory points.
for (i = 0; i < traj.GetPointEntries(); i++) {
G4VTrajectoryPoint* aTrajectoryPoint = traj.GetPoint(i);
G4Point3D vertex = aTrajectoryPoint->GetPosition();
hepRepXMLWriter->addPoint(vertex.x(), vertex.y(), vertex.z());
}
// Create Trajectory Points as a subType of Trajectories.
previousName = hepRepXMLWriter->prevTypeName[2];
hepRepXMLWriter->addType("Trajectory Points",2);
// Specify attributes common to all trajectory points.
if (strcmp("Trajectory Points",previousName)!=0) {
hepRepXMLWriter->addAttValue("DrawAs","Point");
hepRepXMLWriter->addAttValue("Layer",110);
// Change to False once related problem is fixed in WIRED.
hepRepXMLWriter->addAttValue("Visibility","True");
}
for (i = 0; i < traj.GetPointEntries(); i++) {
G4VTrajectoryPoint* aTrajectoryPoint = traj.GetPoint(i);
// Each point is a separate instance of the type Trajectory Points.
hepRepXMLWriter->addInstance();
// Copy the current trajectory point's G4AttValues to HepRepAttValues.
G4std::vector<G4AttValue>* pointAttValues
= aTrajectoryPoint->CreateAttValues();
const G4std::map<G4String,G4AttDef>* pointAttDefs
= aTrajectoryPoint->GetAttDefs();
if (pointAttValues && pointAttDefs) {
G4std::vector<G4AttValue>::iterator iAttVal;
for (iAttVal = pointAttValues->begin();
iAttVal != pointAttValues->end(); ++iAttVal) {
G4std::map<G4String,G4AttDef>::const_iterator iAttDef =
pointAttDefs->find(iAttVal->GetName());
if (iAttDef == pointAttDefs->end()) {
G4cout << "\nG4VTrajectory::ShowTrajectory:"
"\n WARNING: no matching definition for trajectory"
" point attribute \""
<< iAttVal->GetName() << "\", value: "
<< iAttVal->GetValue();
}
else {
//hepRepXMLWriter->addAttValue(iAttDef->second.GetDesc(), iAttVal->GetValue());
hepRepXMLWriter->addAttValue(iAttVal->GetName(), iAttVal->GetValue());
}
}
delete pointAttValues; // AttValues must be deleted after use.
}
// Each trajectory point is made of a single primitive, a point.
hepRepXMLWriter->addPrimitive();
G4Point3D vertex = aTrajectoryPoint->GetPosition();
hepRepXMLWriter->addPoint(vertex.x(), vertex.y(), vertex.z());
}
}
void G4HepRepFileSceneHandler::AddThis (const G4VHit& hit) {
@@ -406,22 +587,8 @@ void G4HepRepFileSceneHandler::AddHepRepInstance(const char* primName,
<< G4endl;
#endif
if (!hepRepXMLWriter->isOpen) {
char* newFileSpec;
newFileSpec = new char [100];
int length;
length = sprintf (newFileSpec, "%s%d%s","G4Data",fileCounter,".heprep");
hepRepXMLWriter->open(newFileSpec);
fileCounter++;
hepRepXMLWriter->addAttDef("LVol", "Logical Volume", "Physics","");
hepRepXMLWriter->addAttDef("Solid", "Solid Name", "Physics","");
hepRepXMLWriter->addAttDef("EType", "Entity Type", "Physics","");
hepRepXMLWriter->addAttDef("Material", "Material Name", "Physics","");
hepRepXMLWriter->addAttDef("Density", "Material Density", "Physics","");
hepRepXMLWriter->addAttDef("State", "Material State", "Physics","");
hepRepXMLWriter->addAttDef("Radlen", "Material Radiation Length", "Physics","");
}
// Open the HepRep output file if it is not already open.
CheckFileOpen();
// Handle Type declaration for Event Data.
// Should be able to just test on fReadyForTransients, but this seems to
@@ -503,3 +670,33 @@ void G4HepRepFileSceneHandler::AddHepRepInstance(const char* primName,
else
hepRepXMLWriter->addAttValue("Visibility",true);
}
void G4HepRepFileSceneHandler::CheckFileOpen() {
#ifdef G4HEPREPFILEDEBUG
G4cout <<
"G4HepRepFileSceneHandler::CheckFileOpen called."
<< G4endl;
#endif
if (!hepRepXMLWriter->isOpen) {
char* newFileSpec;
newFileSpec = new char [100];
int length;
length = sprintf (newFileSpec, "%s%d%s","G4Data",fileCounter,".heprep");
hepRepXMLWriter->open(newFileSpec);
#ifdef G4HEPREPFILEDEBUG
G4cout <<
"G4HepRepFileSceneHandler::CheckFileOpen opened file " << fileCounter
<< G4endl;
#endif
fileCounter++;
hepRepXMLWriter->addAttDef("LVol", "Logical Volume", "Physics","");
hepRepXMLWriter->addAttDef("Solid", "Solid Name", "Physics","");
hepRepXMLWriter->addAttDef("EType", "Entity Type", "Physics","");
hepRepXMLWriter->addAttDef("Material", "Material Name", "Physics","");
hepRepXMLWriter->addAttDef("Density", "Material Density", "Physics","");
hepRepXMLWriter->addAttDef("State", "Material State", "Physics","");
hepRepXMLWriter->addAttDef("Radlen", "Material Radiation Length", "Physics","");
}
}
source/visualization/HepRep/src/G4HepRepFileViewer.cc
+1 -1
View File
@@ -44,7 +44,7 @@
//
//
// $Id: G4HepRepFileViewer.cc,v 1.8 2002/12/13 11:18:03 gunter Exp $
// GEANT4 tag $Name: geant4-05-00 $
// GEANT4 tag $Name: geant4-05-01 $
#include "G4HepRepFileViewer.hh"
source/visualization/HepRep/src/HepRepXMLWriter.cc
+33 -15
View File
@@ -44,7 +44,7 @@
//
//--------------------------------------------------------------------------
// File and Version Information:
// $Id: HepRepXMLWriter.cc,v 1.9 2002/12/13 11:18:03 gunter Exp $
// $Id: HepRepXMLWriter.cc,v 1.10 2003/01/24 21:18:48 perl Exp $
//
// Description:
// Create a HepRep XML File (HepRep version 1).
@@ -129,7 +129,9 @@ void HepRepXMLWriter::addType(const char* name,int newTypeDepth)
typeDepth = newTypeDepth;
}
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addType No file is currently open." << G4endl;
#endif
}
}
@@ -144,13 +146,14 @@ void HepRepXMLWriter::addInstance()
indent();
fout << "<heprep:instance>" << G4endl;
} else {
G4cout <<
"HepRepXMLWriter:addInstance No HepRep Type is currently open"
<< G4endl;
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addInstance No HepRep Type is currently open" << G4endl;
#endif
}
} else {
G4cout << "HepRepXMLWriter:addInstance No file is currently open"
<< G4endl;
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addInstance No file is currently open" << G4endl;
#endif
}
}
@@ -165,13 +168,14 @@ void HepRepXMLWriter::addPrimitive()
indent();
fout << "<heprep:primitive>" << G4endl;
} else {
G4cout <<
"HepRepXMLWriter:addPrimitive No HepRep Instance is currently open"
<< G4endl;
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addPrimitive No HepRep Instance is currently open" << G4endl;
#endif
}
} else {
G4cout << "HepRepXMLWriter:addPrimitive No file is currently open"
<< G4endl;
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addPrimitive No file is currently open" << G4endl;
#endif
}
}
@@ -186,12 +190,14 @@ void HepRepXMLWriter::addPoint(double x, double y, double z)
indent();
fout << "<heprep:point x=\"" << x << "\" y=\"" << y << "\" z=\"" << z << "\">" << G4endl;
} else {
G4cout <<
"HepRepXMLWriter:addPoint No HepRep Primitive is currently open"
<< G4endl;
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addPoint No HepRep Primitive is currently open" << G4endl;
#endif
}
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addPoint No file is currently open" << G4endl;
#endif
}
}
@@ -203,11 +209,13 @@ void HepRepXMLWriter::addAttDef(const char* name,
if (fout.good())
{
indent();
fout << "<heprep:attdef extra=\"" << extra << "\" name=\"" << name << "\" type=\"" << type << "\"" << G4endl;
fout << " <heprep:attdef extra=\"" << extra << "\" name=\"" << name << "\" type=\"" << type << "\"" << G4endl;
indent();
fout << " desc=\"" << desc << "\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttDef No file is currently open" << G4endl;
#endif
}
}
@@ -222,7 +230,9 @@ void HepRepXMLWriter::addAttValue (const char* name,
indent();
fout << " value=\"" << value << "\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttValue No file is currently open" << G4endl;
#endif
}
}
@@ -236,7 +246,9 @@ void HepRepXMLWriter::addAttValue (const char* name,
indent();
fout << " value=\"" << value << "\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttValue No file is currently open" << G4endl;
#endif
}
}
@@ -250,7 +262,9 @@ void HepRepXMLWriter::addAttValue (const char* name,
indent();
fout << " value=\"" << value << "\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttValue No file is currently open" << G4endl;
#endif
}
}
@@ -267,7 +281,9 @@ void HepRepXMLWriter::addAttValue (const char* name,
else
fout << " value=\"False\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttValue No file is currently open" << G4endl;
#endif
}
}
@@ -286,7 +302,9 @@ void HepRepXMLWriter::addAttValue (const char* name,
indent();
fout << " value=\"" << redness << "," << greenness << "," << blueness << "\"/>" << G4endl;
} else {
#ifdef G4HEPREPFILEDEBUG
G4cout << "HepRepXMLWriter:addAttValue No file is currently open" << G4endl;
#endif
}
}