Logo ROOT   6.10/09
Reference Guide
TGDMLWrite.h
Go to the documentation of this file.
1 // @(#)root/gdml:$Id$
2 // Author: Anton Pytel 15/9/2011
3 
4 /*************************************************************************
5  * Copyright (C) 1995-2011, Rene Brun and Fons Rademakers. *
6  * All rights reserved. *
7  * *
8  * For the licensing terms see $ROOTSYS/LICENSE. *
9  * For the list of contributors see $ROOTSYS/README/CREDITS. *
10  *************************************************************************/
11 
12 #ifndef ROOT_TGDMLWRITE
13 #define ROOT_TGDMLWRITE
14 
15 #include "TGeoMatrix.h"
16 
17 #include "TXMLEngine.h"
18 
19 #include "TGeoVolume.h"
20 
21 #include "TGeoParaboloid.h"
22 
23 #include "TGeoSphere.h"
24 
25 #include "TGeoArb8.h"
26 
27 #include "TGeoCone.h"
28 
29 #include "TGeoPara.h"
30 
31 #include "TGeoTrd1.h"
32 
33 #include "TGeoTrd2.h"
34 
35 #include "TGeoTube.h"
36 
37 #include "TGeoPcon.h"
38 
39 #include "TGeoTorus.h"
40 
41 #include "TGeoPgon.h"
42 
43 #include "TGeoXtru.h"
44 
45 #include "TGeoPgon.h"
46 
47 #include "TGeoEltu.h"
48 
49 #include "TGeoHype.h"
50 
51 #include "TGeoBoolNode.h"
52 
53 #include "TGeoCompositeShape.h"
54 
55 #include "TGeoScaledShape.h"
56 
57 #include <map>
58 #include <vector>
59 #include <iostream>
60 
61 ////////////////////////////////////////////////////////////////////////////
62 // //
63 // TGDMLWrite - Class for exporting geometries From ROOT's gGeoManager //
64 // (instance of TGeoManager class) To GDML file. More about GDML //
65 // see http://gdml.web.cern.ch. //
66 // //
67 ////////////////////////////////////////////////////////////////////////////
68 
69 class TGDMLWrite : public TObject {
70 public:
71  TGDMLWrite();
72  virtual ~TGDMLWrite();
73 
74  static void StartGDMLWriting(TGeoManager * geomanager, const char* filename, TString option) {
75  //static function -
76  //options:
77  // g - set by default - geant4 compatibility
78  // f,n - if none of this two is set then naming convention is
79  // with incremental suffix, if "f" then suffix is pointer
80  // if "n" then there is no suffix, but uniqness of names
81  // is not secured.
83  writer->WriteGDMLfile(geomanager, filename, option);
84  delete writer;
85  }
86  //wrapper of all main methods for extraction
87  void WriteGDMLfile(TGeoManager * geomanager, const char* filename = "test.gdml", TString option = "");
88  enum ENamingType {
92  };
93  void SetNamingSpeed(ENamingType naming);
94  void SetG4Compatibility(Bool_t G4Compatible) {
95  fgG4Compatibility = G4Compatible;
96  };
97 
98 private:
99  struct Xyz {
103  };
104 
105  typedef std::map<TString, Bool_t> NameList;
106  typedef std::map<TString, TString> NameListS;
107  typedef std::map<TString, Int_t> NameListI;
108  typedef std::map<TString, Float_t> NameListF;
109  struct StructLst {
110  NameList fLst;
111  }; //to store pointers
112  struct NameLst {
113  NameListS fLst; //to map pointers with names
114  NameListI fLstIter; //to store all the iterators for repeating names
115  };
116 
117  //General lists
118  StructLst *fIsotopeList; //list of isotopes
119  StructLst *fElementList; //list of elements
120  StructLst *fAccPatt; //list of accepted patterns for division
121  StructLst *fRejShape; //list of rejected shapes
122 
123  NameLst *fNameList; //list of names (pointer mapped)
124 
125  //Data members
126  static TGDMLWrite *fgGDMLWrite; //pointer to gdml writer
127  Int_t fgNamingSpeed; //input option for volume and solid naming
128  Bool_t fgG4Compatibility; //input option for Geant4 compatibility
129  XMLDocPointer_t fGdmlFile; //pointer storing xml file
130  TString fTopVolumeName; //name of top volume
131  TXMLEngine *fGdmlE; //xml engine pointer
132 
133  XMLNodePointer_t fDefineNode; //main <define> node...
134  XMLNodePointer_t fMaterialsNode; //main <materials> node...
135  XMLNodePointer_t fSolidsNode; //main <solids> node...
136  XMLNodePointer_t fStructureNode; //main <structure> node...
137  Int_t fVolCnt; //count of volumes
138  Int_t fPhysVolCnt; //count of physical volumes
139  UInt_t fActNameErr; //count of name errors
140  UInt_t fSolCnt; //count of name solids
141 
142  static const UInt_t fgkProcBit = BIT(14); //14th bit is set when solid is processed
143  static const UInt_t fgkProcBitVol = BIT(19); //19th bit is set when volume is processed
144  static const UInt_t fgkMaxNameErr = 5; //maximum number of errors for naming
145 
146  //I. Methods processing the gGeoManager geometry object structure
147  //1. Main methods to extract everything from ROOT gGeoManager
148  XMLNodePointer_t ExtractMaterials(TList* materialsLst); //result <materials>...
149  TString ExtractSolid(TGeoShape* volShape); //adds <shape> to <solids>
150  void ExtractVolumes(TGeoVolume* volume); //result <volume> node... + corresp. shape
151 
152 
153  //1.1 Materials sub methods - creating Nodes
154  XMLNodePointer_t CreateAtomN(Double_t atom, const char * unit = "g/mole");
155  XMLNodePointer_t CreateDN(Double_t density, const char * unit = "g/cm3");
156  XMLNodePointer_t CreateFractionN(Double_t percentage, const char * refName);
157 
158  XMLNodePointer_t CreateIsotopN(TGeoIsotope * isotope, const char * name);
159  XMLNodePointer_t CreateElementN(TGeoElement * element, XMLNodePointer_t materials, const char * name);
162 
163 
164  //1.2 Solids sub methods
167 
190 
192 
193  //1.3 Volume sub methods
194  XMLNodePointer_t CreatePhysVolN(const char * name, Int_t copyno, const char * volref, const char * posref, const char * rotref, XMLNodePointer_t scaleN);
195  XMLNodePointer_t CreateDivisionN(Double_t offset, Double_t width, Int_t number, const char * axis, const char * unit, const char * volref);
196 
197  XMLNodePointer_t CreateSetupN(const char * topVolName , const char * name = "default", const char * version = "1.0");
198  XMLNodePointer_t StartVolumeN(const char * name, const char * solid, const char * material);
199  XMLNodePointer_t StartAssemblyN(const char * name);
200 
201 
202  //II. Utility methods
203  Xyz GetXYZangles(const Double_t * rotationMatrix);
204  //nodes to create position, rotation and similar types first-position/rotation...
205  XMLNodePointer_t CreatePositionN(const char * name, Xyz position, const char * type = "position", const char * unit = "cm");
206  XMLNodePointer_t CreateRotationN(const char * name, Xyz rotation, const char * type = "rotation", const char * unit = "deg");
207  TGeoCompositeShape* CreateFakeCtub(TGeoCtub * geoShape); //create fake cut tube as intersection
208 
209  //check name (2nd parameter) whether it is in the list (1st parameter)
210  Bool_t IsInList(NameList list, TString name2check);
211  TString GenName(TString oldname);
212  TString GenName(TString oldname, TString objPointer);
213  Bool_t CanProcess(TObject *pointer);
214  TString GetPattAxis(Int_t divAxis, const char * pattName, TString& unit);
215  Bool_t IsNullParam(Double_t parValue, TString parName, TString objName);
216  void UnsetTemporaryBits(TGeoManager * geoMng);
217 
218  ClassDef(TGDMLWrite, 0) //imports GDML using DOM and binds it to ROOT
219 };
220 
221 #endif /* ROOT_TGDMLWRITE */
XMLNodePointer_t CreateMixtureN(TGeoMixture *mixture, XMLNodePointer_t materials, TString mname)
Creates "material" node for GDML with references to other sub elements.
Definition: TGDMLWrite.cxx:651
Spherical shell class.
Definition: TGeoSphere.h:17
Cylindrical tube class.
Definition: TGeoTube.h:17
XMLNodePointer_t CreateHypeN(TGeoHype *geoShape)
Creates "hype" node for GDML.
The manager class for any TGeo geometry.
Definition: TGeoManager.h:37
Box class.
Definition: TGeoBBox.h:17
XMLNodePointer_t CreateFractionN(Double_t percentage, const char *refName)
Creates "fraction" node for GDML.
Definition: TGDMLWrite.cxx:563
XMLNodePointer_t CreateParaN(TGeoPara *geoShape)
Creates "para" node for GDML.
Definition: TGDMLWrite.cxx:883
XMLNodePointer_t fStructureNode
Definition: TGDMLWrite.h:136
static TGDMLWrite * fgGDMLWrite
Definition: TGDMLWrite.h:126
Gtra is a twisted trapezoid.
Definition: TGeoArb8.h:143
XMLNodePointer_t StartVolumeN(const char *name, const char *solid, const char *material)
Creates "volume" node for GDML.
StructLst * fRejShape
Definition: TGDMLWrite.h:121
A polycone.
Definition: TGeoPcon.h:17
A polygone.
Definition: TGeoPgon.h:19
#define BIT(n)
Definition: Rtypes.h:75
TGeoVolume, TGeoVolumeMulti, TGeoVolumeAssembly are the volume classes.
Definition: TGeoVolume.h:48
TString ExtractSolid(TGeoShape *volShape)
Method creating solid to xml file and returning its name.
Definition: TGDMLWrite.cxx:363
Torus segment class.
Definition: TGeoTorus.h:17
void UnsetTemporaryBits(TGeoManager *geoMng)
Unsetting bits that were changed in gGeoManager during export so that export can be run more times wi...
XMLNodePointer_t CreateMaterialN(TGeoMaterial *material, TString mname)
Creates "material" node for GDML.
Definition: TGDMLWrite.cxx:704
Bool_t CanProcess(TObject *pointer)
Method which tests whether solids can be processed.
Basic string class.
Definition: TString.h:129
Base class describing materials.
Definition: TGeoMaterial.h:29
int Int_t
Definition: RtypesCore.h:41
bool Bool_t
Definition: RtypesCore.h:59
XMLNodePointer_t CreatePositionN(const char *name, Xyz position, const char *type="position", const char *unit="cm")
Creates "position" kind of node for GDML.
void WriteGDMLfile(TGeoManager *geomanager, const char *filename="test.gdml", TString option="")
Wrapper of all exporting methods Creates blank GDML file and fills it with gGeoManager structure conv...
Definition: TGDMLWrite.cxx:209
std::map< TString, Float_t > NameListF
Definition: TGDMLWrite.h:108
NameListI fLstIter
Definition: TGDMLWrite.h:114
XMLNodePointer_t CreateXtrusionN(TGeoXtru *geoShape)
Creates "xtru" node for GDML.
A shape scaled by a TGeoScale transformation.
Definition: writer.py:1
XMLNodePointer_t CreateEllipsoidN(TGeoCompositeShape *geoShape, TString elName)
Creates "ellipsoid" node for GDML this is a special case, because ellipsoid is not defined in ROOT so...
XMLNodePointer_t StartAssemblyN(const char *name)
Creates "assembly" node for GDML.
A trapezoid with only x length varying with z.
Definition: TGeoTrd1.h:17
Paraboloid class.
XMLNodePointer_t CreateTwistedTrapN(TGeoGtra *geoShape)
Creates "twistedtrap" node for GDML.
Definition: TGDMLWrite.cxx:948
XMLDocPointer_t fGdmlFile
Definition: TGDMLWrite.h:129
XMLNodePointer_t CreateElConeN(TGeoScaledShape *geoShape)
Creates "elcone" (elliptical cone) node for GDML this is a special case, because elliptical cone is n...
XMLNodePointer_t CreateArb8N(TGeoArb8 *geoShape)
Creates "arb8" node for GDML.
Definition: TGDMLWrite.cxx:799
XMLNodePointer_t CreateZplaneN(Double_t z, Double_t rmin, Double_t rmax)
Creates "zplane" node for GDML.
XMLNodePointer_t CreateConeN(TGeoConeSeg *geoShape)
Creates "cone" node for GDML from TGeoConeSeg object.
Definition: TGDMLWrite.cxx:833
#define ClassDef(name, id)
Definition: Rtypes.h:297
A phi segment of a conical tube.
Definition: TGeoCone.h:98
XMLNodePointer_t CreateTrapN(TGeoTrap *geoShape)
Creates "trap" node for GDML.
Definition: TGDMLWrite.cxx:903
Bool_t IsInList(NameList list, TString name2check)
Checks whether name2check is in (NameList) list.
XMLNodePointer_t CreateIsotopN(TGeoIsotope *isotope, const char *name)
Creates "isotope" node for GDML.
Definition: TGDMLWrite.cxx:574
void * XMLDocPointer_t
Definition: TXMLEngine.h:20
Int_t fPhysVolCnt
Definition: TGDMLWrite.h:138
XMLNodePointer_t ExtractMaterials(TList *materialsLst)
Method exporting materials.
Definition: TGDMLWrite.cxx:331
TXMLEngine * fGdmlE
Definition: TGDMLWrite.h:131
UInt_t fActNameErr
Definition: TGDMLWrite.h:139
virtual ~TGDMLWrite()
Destructor.
Definition: TGDMLWrite.cxx:185
Base class for chemical elements.
Definition: TGeoElement.h:36
TGDMLWrite()
Default constructor.
Definition: TGDMLWrite.cxx:157
TRAP is a general trapezoid, i.e.
Definition: TGeoArb8.h:89
void SetG4Compatibility(Bool_t G4Compatible)
Definition: TGDMLWrite.h:94
XMLNodePointer_t CreateEltubeN(TGeoEltu *geoShape)
Creates "eltube" node for GDML.
Bool_t IsNullParam(Double_t parValue, TString parName, TString objName)
Check for null parameter to skip the NULL objects.
A doubly linked list.
Definition: TList.h:43
XMLNodePointer_t fDefineNode
Definition: TGDMLWrite.h:133
XMLNodePointer_t fSolidsNode
Definition: TGDMLWrite.h:135
Class handling Boolean composition of shapes.
A trapezoid with both x and y lengths varying with z.
Definition: TGeoTrd2.h:17
Parallelepiped class.
Definition: TGeoPara.h:17
Xyz GetXYZangles(const Double_t *rotationMatrix)
Retrieves X Y Z angles from rotation matrix.
XMLNodePointer_t CreatePhysVolN(const char *name, Int_t copyno, const char *volref, const char *posref, const char *rotref, XMLNodePointer_t scaleN)
Creates "physvol" node for GDML.
Base abstract class for all shapes.
Definition: TGeoShape.h:25
void SetNamingSpeed(ENamingType naming)
Set convetion of naming solids and volumes.
Definition: TGDMLWrite.cxx:199
XMLNodePointer_t CreateBoxN(TGeoBBox *geoShape)
Creates "box" node for GDML.
Definition: TGDMLWrite.cxx:734
std::map< TString, Bool_t > NameList
Definition: TGDMLWrite.h:105
XMLNodePointer_t CreatePolyconeN(TGeoPcon *geoShape)
Creates "polycone" node for GDML.
XMLNodePointer_t CreateParaboloidN(TGeoParaboloid *geoShape)
Creates "paraboloid" node for GDML.
Definition: TGDMLWrite.cxx:755
XMLNodePointer_t CreateTorusN(TGeoTorus *geoShape)
Creates "torus" node for GDML.
StructLst * fIsotopeList
Definition: TGDMLWrite.h:118
unsigned int UInt_t
Definition: RtypesCore.h:42
Hyperboloid class defined by 5 parameters.
Definition: TGeoHype.h:17
TGeoCompositeShape * CreateFakeCtub(TGeoCtub *geoShape)
Method creating cutTube as an intersection of tube and two boxes.
TString fTopVolumeName
Definition: TGDMLWrite.h:130
void * XMLNodePointer_t
Definition: TXMLEngine.h:17
Bool_t fgG4Compatibility
Definition: TGDMLWrite.h:128
TString GenName(TString oldname)
NCNAME basic restrictions Replace "$" character with empty character etc.
A tube segment cut with 2 planes.
Definition: TGeoTube.h:168
XMLNodePointer_t CreateDivisionN(Double_t offset, Double_t width, Int_t number, const char *axis, const char *unit, const char *volref)
Creates "divisionvol" node for GDML.
An extrusion with fixed outline shape in x-y and a sequence of z extents (segments).
Definition: TGeoXtru.h:21
double Double_t
Definition: RtypesCore.h:55
Int_t fVolCnt
Definition: TGDMLWrite.h:137
int type
Definition: TGX11.cxx:120
Conical tube class.
Definition: TGeoCone.h:17
An arbitrary trapezoid with less than 8 vertices standing on.
Definition: TGeoArb8.h:17
XMLNodePointer_t CreateTrdN(TGeoTrd1 *geoShape)
Creates "trd" node for GDML from object TGeoTrd1.
static const UInt_t fgkProcBitVol
Definition: TGDMLWrite.h:143
XMLNodePointer_t CreatePolyhedraN(TGeoPgon *geoShape)
Creates "polyhedra" node for GDML.
XMLNodePointer_t CreateAtomN(Double_t atom, const char *unit="g/mole")
Creates "atom" node for GDML.
Definition: TGDMLWrite.cxx:541
Mother of all ROOT objects.
Definition: TObject.h:37
XMLNodePointer_t ChooseObject(TGeoShape *geoShape)
Chooses the object and method that should be used for processing object.
XMLNodePointer_t CreateSphereN(TGeoSphere *geoShape)
Creates "sphere" node for GDML.
Definition: TGDMLWrite.cxx:775
void ExtractVolumes(TGeoVolume *volume)
Method extracting geometry structure recursively.
Definition: TGDMLWrite.cxx:381
XMLNodePointer_t fMaterialsNode
Definition: TGDMLWrite.h:134
Elliptical tube class.
Definition: TGeoEltu.h:17
NameLst * fNameList
Definition: TGDMLWrite.h:123
XMLNodePointer_t CreateTubeN(TGeoTubeSeg *geoShape)
Creates "tube" node for GDML from object TGeoTubeSeg.
Int_t fgNamingSpeed
Definition: TGDMLWrite.h:127
StructLst * fElementList
Definition: TGDMLWrite.h:119
XMLNodePointer_t CreateDN(Double_t density, const char *unit="g/cm3")
Creates "D" density node for GDML.
Definition: TGDMLWrite.cxx:552
static const UInt_t fgkMaxNameErr
Definition: TGDMLWrite.h:144
XMLNodePointer_t CreateElementN(TGeoElement *element, XMLNodePointer_t materials, const char *name)
Creates "element" node for GDML element node and attribute.
Definition: TGDMLWrite.cxx:588
std::map< TString, TString > NameListS
Definition: TGDMLWrite.h:106
XMLNodePointer_t CreateCommonBoolN(TGeoCompositeShape *geoShape)
Creates common part of union intersection and subtraction nodes.
UInt_t fSolCnt
Definition: TGDMLWrite.h:140
std::map< TString, Int_t > NameListI
Definition: TGDMLWrite.h:107
XMLNodePointer_t CreateSetupN(const char *topVolName, const char *name="default", const char *version="1.0")
Creates "setup" node for GDML.
TString GetPattAxis(Int_t divAxis, const char *pattName, TString &unit)
Method that retrieves axis and unit along which object is divided.
static void StartGDMLWriting(TGeoManager *geomanager, const char *filename, TString option)
Definition: TGDMLWrite.h:74
XMLNodePointer_t CreateRotationN(const char *name, Xyz rotation, const char *type="rotation", const char *unit="deg")
Creates "rotation" kind of node for GDML.
StructLst * fAccPatt
Definition: TGDMLWrite.h:120
static const UInt_t fgkProcBit
Definition: TGDMLWrite.h:142
XMLNodePointer_t CreateCutTubeN(TGeoCtub *geoShape)
Creates "cutTube" node for GDML.
A phi segment of a tube.
Definition: TGeoTube.h:88