// @(#)root/geom:$Name: $:$Id: TGeoManager.cxx,v 1.24 2002/12/03 16:01:39 brun Exp $
// Author: Andrei Gheata 25/10/01
/*************************************************************************
* Copyright (C) 1995-2000, Rene Brun and Fons Rademakers. *
* All rights reserved. *
* *
* For the licensing terms see $ROOTSYS/LICENSE. *
* For the list of contributors see $ROOTSYS/README/CREDITS. *
*************************************************************************/
////////////////////////////////////////////////////////////////////////////////
// General architecture
// --------------------
//
// The new ROOT geometry package is a tool designed for building, browsing,
// tracking and visualizing a detector geometry. The code is independent from
// other external MC for simulation, therefore it does not contain any
// constraints related to physics. However, the package defines a number of
// hooks for tracking, such as materials, magnetic field or track state flags,
// in order to allow interfacing to tracking MC's. The final goal is to be
// able to use the same geometry for several purposes, such as tracking,
// reconstruction or visualization, taking advantage of the ROOT features
// related to bookkeeping, I/O, histograming, browsing and GUI's.
//
// The geometrical modeler is the most important component of the package and
// it provides answers to the basic questions like "Where am I ?" or "How far
// from the next boundary ?", but also to more complex ones like "How far from
// the closest surface ?" or "Which is the next crossing along a helix ?".
//
// The architecture of the modeler is a combination between a GEANT-like
// containment scheme and a normal CSG binary tree at the level of shapes. An
// important common feature of all detector geometry descriptions is the
// mother-daughter concept. This is the most natural approach when tracking
// is concerned and imposes a set of constraints to the way geometry is defined.
// Constructive solid geometry composition is used only in order to create more
// complex shapes from an existing set of primitives through boolean operations.
// This feature is not implemented yet but in future full definition of boolean
// expressions will be supported.
//
// Practically every geometry defined in GEANT style can be mapped by the modeler.
// The basic components used for building the logical hierarchy of the geometry
// are called "volumes" and "nodes". Volumes (sometimes called "solids") are fully
// defined geometrical objects having a given shape and material and possibly
// containing a list of nodes. Nodes represent just positioned instances of volumes
// inside a container volume and they are not directly defined by user. They are
// automatically created as a result of adding one volume inside other or dividing
// a volume. The geometrical transformation hold by nodes is always defined with
// respect to their mother (relative positioning). Reflection matrices are allowed.
// All volumes have to be fully aware of their containees when the geometry is
// closed. They will build aditional structures (voxels) in order to fasten-up
// the search algorithms. Finally, nodes can be regarded as bidirectional links
// between containers and containees objects.
//
// The structure defined in this way is a graph structure since volumes are
// replicable (same volume can become daughter node of several other volumes),
// every volume becoming a branch in this graph. Any volume in the logical graph
// can become the actual top volume at run time (see TGeoManager::SetTopVolume()).
// All functionalities of the modeler will behave in this case as if only the
// corresponding branch starting from this volume is the registered geometry.
//
//
/*
*/
//
//
// A given volume can be positioned several times in the geometry. A volume
// can be divided according default or user-defined patterns, creating automatically
// the list of division nodes inside. The elementary volumes created during the
// dividing process follow the same scheme as usual volumes, therefore it is possible
// to position further geometrical structures inside or to divide them further more
// (see TGeoVolume::Divide()).
//
// The primitive shapes supported by the package are basically the GEANT3
// shapes (see class TGeoShape), arbitrary wedges with eight vertices on two parallel
// planes. All basic primitives inherits from class TGeoBBox since the bounding box
// of a solid is essential for the tracking algorithms. They also implement the
// virtual methods defined in the virtual class TGeoShape (point and segment
// classification). User-defined primitives can be direcly plugged into the modeler
// provided that they override these methods. Composite shapes will be soon supported
// by the modeler. In order to build a TGeoCompositeShape, one will have to define
// first the primitive components. The object that handle boolean
// operations among components is called TGeoBoolCombinator and it has to be
// constructed providing a string boolean expression between the components names.
//
//
// Example for building a simple geometry :
//-----------------------------------------
//
//______________________________________________________________________________
//void rootgeom()
//{
////--- Definition of a simple geometry
// gSystem->Load("libGeom");
// TGeoManager *geom = new TGeoManager("simple1", "Simple geometry");
//
// //--- define some materials
// TGeoMaterial *mat;
// mat = new TGeoMaterial("mat1", "Vacuum", 0,0,0);
// mat = new TGeoMaterial("mat2", "Al", 26.98,13,2.7);
//
// //--- define the transformations
// TGeoTranslation *tr1 = new TGeoTranslation(20., 0, 0.);
// TGeoTranslation *tr2 = new TGeoTranslation(10., 0., 0.);
// TGeoTranslation *tr3 = new TGeoTranslation(10., 20., 0.);
// TGeoTranslation *tr4 = new TGeoTranslation(5., 10., 0.);
// TGeoTranslation *tr5 = new TGeoTranslation(20., 0., 0.);
// TGeoTranslation *tr6 = new TGeoTranslation(-5., 0., 0.);
// TGeoTranslation *tr7 = new TGeoTranslation(7.5, 7.5, 0.);
// TGeoRotation *rot1 = new TGeoRotation("rot1", 90., 0., 90., 270., 0., 0.);
// TGeoCombiTrans *combi1 = new TGeoCombiTrans(7.5, -7.5, 0., rot1);
// TGeoTranslation *tr8 = new TGeoTranslation(7.5, -5., 0.);
// TGeoTranslation *tr9 = new TGeoTranslation(7.5, 20., 0.);
// TGeoTranslation *tr10 = new TGeoTranslation(85., 0., 0.);
// TGeoTranslation *tr11 = new TGeoTranslation(35., 0., 0.);
// TGeoTranslation *tr12 = new TGeoTranslation(-15., 0., 0.);
// TGeoTranslation *tr13 = new TGeoTranslation(-65., 0., 0.);
//
// TGeoTranslation *tr14 = new TGeoTranslation(0,0,-100);
// TGeoCombiTrans *combi2 = new TGeoCombiTrans(0,0,100,
// new TGeoRotation("rot2",90,180,90,90,180,0));
// TGeoCombiTrans *combi3 = new TGeoCombiTrans(100,0,0,
// new TGeoRotation("rot3",90,270,0,0,90,180));
// TGeoCombiTrans *combi4 = new TGeoCombiTrans(-100,0,0,
// new TGeoRotation("rot4",90,90,0,0,90,0));
// TGeoCombiTrans *combi5 = new TGeoCombiTrans(0,100,0,
// new TGeoRotation("rot5",0,0,90,180,90,270));
// TGeoCombiTrans *combi6 = new TGeoCombiTrans(0,-100,0,
// new TGeoRotation("rot6",180,0,90,180,90,90));
//
// //--- make the top container volume
// Double_t worldx = 110.;
// Double_t worldy = 50.;
// Double_t worldz = 5.;
// TGeoVolume *top = geom->MakeBox("TOP", "mat1", 270., 270., 120.);
// geom->SetTopVolume(top); // mandatory !
// //--- build other container volumes
// TGeoVolume *replica = geom->MakeBox("REPLICA", "mat1",120,120,120);
// replica->SetVisibility(kFALSE);
// TGeoVolume *rootbox = geom->MakeBox("ROOT", "mat1", 110., 50., 5.);
// rootbox->SetVisibility(kFALSE); // this will hold word 'ROOT'
//
// //--- make letter 'R'
// TGeoVolume *R = geom->MakeBox("R", "mat1", 25., 25., 5.);
// R->SetVisibility(kFALSE);
// TGeoVolume *bar1 = geom->MakeBox("bar1", "mat2", 5., 25, 5.);
// bar1->SetLineColor(kRed);
// R->AddNode(bar1, 1, tr1);
// TGeoVolume *bar2 = geom->MakeBox("bar2", "mat2", 5., 5., 5.);
// bar2->SetLineColor(kRed);
// R->AddNode(bar2, 1, tr2);
// R->AddNode(bar2, 2, tr3);
// TGeoVolume *tub1 = geom->MakeTubs("tub1", "mat2", 5., 15., 5., 90., 270.);
// tub1->SetLineColor(kRed);
// R->AddNode(tub1, 1, tr4);
// TGeoVolume *bar3 = geom->MakeArb8("bar3", "mat2", 5.);
// bar3->SetLineColor(kRed);
// TGeoArb8 *arb = (TGeoArb8*)bar3->GetShape();
// arb->SetVertex(0, 15., -5.);
// arb->SetVertex(1, 5., -5.);
// arb->SetVertex(2, -10., -25.);
// arb->SetVertex(3, 0., -25.);
// arb->SetVertex(4, 15., -5.);
// arb->SetVertex(5, 5., -5.);
// arb->SetVertex(6, -10., -25.);
// arb->SetVertex(7, 0., -25.);
// R->AddNode(bar3, 1, gGeoIdentity);
//
// //--- make letter 'O'
// TGeoVolume *O = geom->MakeBox("O", "mat1", 25., 25., 5.);
// O->SetVisibility(kFALSE);
// TGeoVolume *bar4 = geom->MakeBox("bar4", "mat2", 5., 7.5, 5.);
// bar4->SetLineColor(kYellow);
// O->AddNode(bar4, 1, tr5);
// O->AddNode(bar4, 2, tr6);
// TGeoVolume *tub2 = geom->MakeTubs("tub1", "mat2", 7.5, 17.5, 5., 0., 180.);
// tub2->SetLineColor(kYellow);
// O->AddNode(tub2, 1, tr7);
// O->AddNode(tub2, 2, combi1);
//
// //--- make letter 'T'
// TGeoVolume *T = geom->MakeBox("T", "mat1", 25., 25., 5.);
// T->SetVisibility(kFALSE);
// TGeoVolume *bar5 = geom->MakeBox("bar5", "mat2", 5., 20., 5.);
// bar5->SetLineColor(kBlue);
// T->AddNode(bar5, 1, tr8);
// TGeoVolume *bar6 = geom->MakeBox("bar6", "mat2", 17.5, 5., 5.);
// bar6->SetLineColor(kBlue);
// T->AddNode(bar6, 1, tr9);
//
// //--- add letters to 'ROOT' container
// rootbox->AddNode(R, 1, tr10);
// rootbox->AddNode(O, 1, tr11);
// rootbox->AddNode(O, 2, tr12);
// rootbox->AddNode(T, 1, tr13);
//
// //--- add word 'ROOT' on each face of a cube
// replica->AddNode(rootbox, 1, tr14);
// replica->AddNode(rootbox, 2, combi2);
// replica->AddNode(rootbox, 3, combi3);
// replica->AddNode(rootbox, 4, combi4);
// replica->AddNode(rootbox, 5, combi5);
// replica->AddNode(rootbox, 6, combi6);
//
// //--- add four replicas of this cube to top volume
// top->AddNode(replica, 1, new TGeoTranslation(-150, -150, 0));
// top->AddNode(replica, 2, new TGeoTranslation(150, -150, 0));
// top->AddNode(replica, 3, new TGeoTranslation(150, 150, 0));
// top->AddNode(replica, 4, new TGeoTranslation(-150, 150, 0));
//
// //--- close the geometry
// geom->CloseGeometry();
//
// //--- draw the ROOT box
// geom->SetVisLevel(4);
// top->Draw();
// if (gPad) gPad->x3d();
//}
//______________________________________________________________________________
//
//
//
/*
*/
//
//
//
// TGeoManager - the manager class for the geometry package.
// ---------------------------------------------------------
//
// TGeoManager class is embedding all the API needed for building and tracking
// a geometry. It defines a global pointer (gGeoManager) in order to be fully
// accessible from external code. The mechanism of handling multiple geometries
// at the same time will be soon implemented.
//
// TGeoManager is the owner of all geometry objects defined in a session,
// therefore users must not try to control their deletion. It contains lists of
// materials, transformations, shapes and volumes. Logical nodes (positioned
// volumes) are created and destroyed by the TGeoVolume class. Physical
// nodes and their global transformations are subjected to a caching mechanism
// due to the sometimes very large memory requirements of logical graph expansion.
// The caching mechanism is triggered by the total number of physical instances
// of volumes and the cache manager is a client of TGeoManager. The manager class
// also controls the painter client. This is linked with ROOT graphical libraries
// loaded on demand in order to control visualization actions.
//
// Rules for building a valid geometry
// -----------------------------------
//
// A given geometry can be built in various ways, but there are mandatory steps
// that have to be followed in order to be validated by the modeler. There are
// general rules : volumes needs materials and shapes in order to be created,
// both container an containee volumes must be created before linking them together,
// and the relative transformation matrix must be provided. All branches must
// have an upper link point otherwise they will not be considered as part of the
// geometry. Visibility or tracking properties of volumes can be provided both
// at build time or after geometry is closed, but global visualization settings
// (see TGeoPainter class) should not be provided at build time, otherwise the
// drawing package will be loaded. There is also a list of specific rules :
// positioned daughters should not extrude their mother or intersect with sisters
// unless this is specified (see TGeoVolume::AddNodeOverlap()), the top volume
// (containing all geometry tree) must be specified before closing the geometry
// and must not be positioned - it represents the global reference frame. After
// building the full geometry tree, the geometry must be closed
// (see TGeoManager::CloseGeometry()). Voxelization can be redone per volume after
// this process.
//
//
// Below is the general scheme of the manager class.
//
//
/*
*/
//
//
// An interactive session
// ------------------------
//
// Provided that a geometry was successfully built and closed (for instance the
// previous example $ROOTSYS/tutorials/rootgeom.C ), the manager class will register
// itself to ROOT and the logical/physical structures will become immediately browsable.
// The ROOT browser will display starting from the geometry folder : the list of
// transformations and materials, the top volume and the top logical node. These last
// two can be fully expanded, any intermediate volume/node in the browser being subject
// of direct access context menu operations (right mouse button click). All user
// utilities of classes TGeoManager, TGeoVolume and TGeoNode can be called via the
// context menu.
//
//
/*
*/
//
//
// --- Drawing the geometry
//
// Any logical volume can be drawn via TGeoVolume::Draw() member function.
// This can be direcly accessed from the context menu of the volume object
// directly from the browser.
// There are several drawing options that can be set with
// TGeoManager::SetVisOption(Int_t opt) method :
// opt=0 - only the content of the volume is drawn, N levels down (default N=3).
// This is the default behavior. The number of levels to be drawn can be changed
// via TGeoManager::SetVisLevel(Int_t level) method.
//
//
/*
*/
//
//
// opt=1 - the final leaves (e.g. daughters with no containment) of the branch
// starting from volume are drawn down to the current number of levels.
// WARNING : This mode is memory consuming
// depending of the size of geometry, so drawing from top level within this mode
// should be handled with care for expensive geometries. In future there will be
// a limitation on the maximum number of nodes to be visualized.
//
//
/*
*/
//
//
// opt=2 - only the clicked volume is visualized. This is automatically set by
// TGeoVolume::DrawOnly() method
// opt=3 - only a given path is visualized. This is automatically set by
// TGeoVolume::DrawPath(const char *path) method
//
// The current view can be exploded in cartesian, cylindrical or spherical
// coordinates :
// TGeoManager::SetExplodedView(Int_t opt). Options may be :
// - 0 - default (no bombing)
// - 1 - cartesian coordinates. The bomb factor on each axis can be set with
// TGeoManager::SetBombX(Double_t bomb) and corresponding Y and Z.
// - 2 - bomb in cylindrical coordinates. Only the bomb factors on Z and R
// are considered
//
//
/*
*/
//
//
// - 3 - bomb in radial spherical coordinate : TGeoManager::SetBombR()
//
// Volumes themselves support different visualization settings :
// - TGeoVolume::SetVisibility() : set volume visibility.
// - TGeoVolume::VisibleDaughters() : set daughters visibility.
// All these actions automatically updates the current view if any.
//
// --- Checking the geometry
//
// Several checking methods are accessible from the volume context menu. They
// generally apply only to the visible parts of the drawn geometry in order to
// ease geometry checking, and their implementation is in the TGeoChecker class
// from the painting package.
//
// 1. Checking a given point.
// Can be called from TGeoManager::CheckPoint(Double_t x, Double_t y, Double_t z).
// This method is drawing the daughters of the volume containing the point one
// level down, printing the path to the deepest physical node holding this point.
// It also computes the closest distance to any boundary. The point will be drawn
// in red.
//
//
/*
*/
//
//
// 2. Shooting random points.
// Can be called from TGeoVolume::RandomPoints() (context menu function) and
// it will draw this volume with current visualization settings. Random points
// are generated in the bounding box of the top drawn volume. The points are
// classified and drawn with the color of their deepest container. Only points
// in visible nodes will be drawn.
//
//
/*
*/
//
//
//
// 3. Raytracing.
// Can be called from TGeoVolume::RandomRays() (context menu of volumes) and
// will shoot rays from a given point in the local reference frame with random
// directions. The intersections with displayed nodes will appear as segments
// having the color of the touched node. Drawn geometry will be then made invisible
// in order to enhance rays.
//
//
/*
*/
//
#include "Riostream.h"
#include "TROOT.h"
#include "TSystem.h"
#include "TStyle.h"
#include "TVirtualPad.h"
#include "TBrowser.h"
#include "TFile.h"
#include "TKey.h"
#include "TGeoMaterial.h"
#include "TGeoMatrix.h"
#include "TGeoManager.h"
#include "TGeoPara.h"
#include "TGeoTube.h"
#include "TGeoEltu.h"
#include "TGeoCone.h"
#include "TGeoSphere.h"
#include "TGeoArb8.h"
#include "TGeoPgon.h"
#include "TGeoTrd1.h"
#include "TGeoTrd2.h"
#include "TGeoCompositeShape.h"
#include "TVirtualGeoPainter.h"
#include "TGeoManager.h"
// statics and globals
TGeoManager *gGeoManager = 0;
const char *kGeoOutsidePath = " ";
ClassImp(TGeoManager)
//-----------------------------------------------------------------------------
TGeoManager::TGeoManager()
{
// Default constructor.
if (TClass::IsCallingNew() == TClass::kDummyNew)
fBits = (UChar_t*) -1;
else {
Init();
gGeoIdentity = 0;
}
}
//-----------------------------------------------------------------------------
TGeoManager::TGeoManager(const char *name, const char *title)
:TNamed(name, title)
{
// Constructor.
Init();
gGeoIdentity = new TGeoIdentity("Identity");
BuildDefaultMaterials();
}
//-----------------------------------------------------------------------------
void TGeoManager::Init()
{
// Initialize manager class.
if (gGeoManager) {
Warning("Init","Deleting previous geometry: %s/%s",gGeoManager->GetName(),gGeoManager->GetTitle());
delete gGeoManager;
}
gGeoManager = this;
fPhiCut = kFALSE;
fPhimin = 0;
fPhimax = 360;
fStreamVoxels = kFALSE;
fIsGeomReading = kFALSE;
fSearchOverlaps = kFALSE;
fLoopVolumes = kFALSE;
fStartSafe = kTRUE;
fSafety = 0;
fStep = 0;
fBits = new UChar_t[50000]; // max 25000 nodes per volume
fMaterials = new TList();
fMatrices = new TList();
fNodes = new TObjArray(30);
fNNodes = 0;
fLevel = 0;
fPoint = new Double_t[3];
fDirection = new Double_t[3];
fNormalChecked = new Double_t[3];
fCldirChecked = new Double_t[3];
fNormal = new Double_t[3];
fCldir = new Double_t[3];
fVolumes = new TList();
fShapes = new TList();
fGVolumes = new TList();
fGShapes = new TList();
fTopVolume = 0;
fTopNode = 0;
fCurrentVolume = 0;
fMasterVolume = 0;
fCurrentNode = 0;
fLastNode = 0;
fCurrentOverlapping = kFALSE;
fPath = "";
fCache = 0;
fPainter = 0;
fIsEntering = kFALSE;
fIsExiting = kFALSE;
fIsStepEntering = kFALSE;
fIsStepExiting = kFALSE;
fIsOutside = kFALSE;
fIsOnBoundary = kFALSE;
fIsNullStep = kFALSE;
fVisLevel = 3;
fVisOption = 0;
fExplodedView = 0;
fNsegments = 20;
fCurrentMatrix = 0;
}
//-----------------------------------------------------------------------------
TGeoManager::~TGeoManager()
{
// Destructor
if (fBits == (UChar_t*) -1)
return;
Warning("dtor", "deleting previous geometry: %s/%s",GetName(),GetTitle());
// gROOT->GetListOfGeometries()->Remove(this);
gROOT->GetListOfBrowsables()->Remove(this);
TSeqCollection *brlist = gROOT->GetListOfBrowsers();
TIter next(brlist);
TBrowser *browser = 0;
while ((browser=(TBrowser*)next())) {
browser->RecursiveRemove(this);
// browser->Refresh();
printf("browser refreshedn");
}
delete [] fBits;
if (fCache) delete fCache;
if (fMatrices) {fMatrices->Delete(); delete fMatrices;}
if (fNodes) delete fNodes;
if (fMaterials) {fMaterials->Delete(); delete fMaterials;}
if (fShapes) {fShapes->Delete(); delete fShapes;}
if (fVolumes) {fVolumes->Delete(); delete fVolumes;}
CleanGarbage();
if (fPainter) delete fPainter;
delete [] fPoint;
delete [] fDirection;
delete [] fNormalChecked;
delete [] fCldirChecked;
delete [] fNormal;
delete [] fCldir;
delete fGVolumes;
delete fGShapes;
gGeoIdentity = 0;
gGeoManager = 0;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::AddMaterial(TGeoMaterial *material)
{
// Add a material to the list. Returns index of the material in list.
if (!material) {
Error("AddMaterial", "invalid material");
return -1;
}
Int_t index = GetMaterialIndex(material->GetName());
if (index >= 0) return index;
index = fMaterials->GetSize();
fMaterials->Add(material);
return index;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::AddTransformation(TGeoMatrix *matrix)
{
// Add a matrix to the list. Returns index of the matrix in list.
if (!matrix) {
Error("AddMatrix", "invalid matrix");
return -1;
}
Int_t index = fMatrices->GetSize();
fMatrices->Add(matrix);
return index;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::AddShape(TGeoShape *shape)
{
// Add a shape to the list. Returns index of the shape in list.
if (!shape) {
Error("AddShape", "invalid shape");
return -1;
}
TList *list = fShapes;
if (shape->IsRunTimeShape()) list = fGShapes;;
Int_t index = list->GetSize();
list->Add(shape);
return index;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::AddVolume(TGeoVolume *volume)
{
// Add a volume to the list. Returns index of the volume in list.
if (!volume) {
Error("AddVolume", "invalid volume");
return -1;
}
TList *list = fVolumes;
if (volume->IsRunTime()) list = fGVolumes;
Int_t index = list->GetSize();
list->Add(volume);
return index;
}
//-----------------------------------------------------------------------------
void TGeoManager::Browse(TBrowser *b)
{
// Describe how to browse this object.
if (!b) return;
if (fMaterials) b->Add(fMaterials, "Materials");
if (fMatrices) b->Add(fMatrices, "Local transformations");
if (fTopVolume) b->Add(fTopVolume);
if (fTopNode) b->Add(fTopNode);
}
//-----------------------------------------------------------------------------
void TGeoManager::BombTranslation(const Double_t *tr, Double_t *bombtr)
{
// Get the new 'bombed' translation vector according current exploded view mode.
if (fPainter) fPainter->BombTranslation(tr, bombtr);
return;
}
//-----------------------------------------------------------------------------
void TGeoManager::UnbombTranslation(const Double_t *tr, Double_t *bombtr)
{
// Get the new 'unbombed' translation vector according current exploded view mode.
if (fPainter) fPainter->UnbombTranslation(tr, bombtr);
return;
}
//-----------------------------------------------------------------------------
void TGeoManager::BuildCache()
{
// Builds the cache for physical nodes and global matrices.
if (!fCache) {
if (fNNodes>5000000) // temporary - works without
// build dummy cache
fCache = new TGeoCacheDummy(fTopNode);
else
// build real cache
fCache = new TGeoNodeCache(0);
}
}
//-----------------------------------------------------------------------------
void TGeoManager::ClearAttributes()
{
// Reset all attributes to default ones. Default attributes for visualization
// are those defined before closing the geometry.
if (gPad) delete gPad;
gPad = 0;
SetVisOption(0);
SetVisLevel(3);
SetExplodedView(0);
SetBombFactors();
if (!gStyle) return;
TIter next(fVolumes);
TGeoVolume *vol = 0;
while ((vol=(TGeoVolume*)next())) {
if (!vol->IsVisTouched()) continue;
vol->SetVisibility(kTRUE);
vol->SetVisDaughters(kTRUE);
vol->SetLineStyle(gStyle->GetLineStyle());
vol->SetLineWidth(gStyle->GetLineWidth());
vol->SetVisTouched(kFALSE);
}
}
//-----------------------------------------------------------------------------
void TGeoManager::CloseGeometry()
{
// Closing geometry implies checking the geometry validity, fixing shapes
// with negative parameters (run-time shapes)building the cache manager,
// voxelizing all volumes, counting the total number of physical nodes and
// registring the manager class to the browser.
gROOT->GetListOfBrowsables()->Add(this);
TSeqCollection *brlist = gROOT->GetListOfBrowsers();
TIter next(brlist);
TBrowser *browser = 0;
while ((browser=(TBrowser*)next())) {
browser->Refresh();
printf("%s added to browsern", GetName());
}
if (fIsGeomReading) {
printf("### Geometry loaded from file...n");
gGeoIdentity=(TGeoIdentity *)fMatrices->At(0);
if (!fTopNode) {
if (!fMasterVolume) {
Error("CloseGeometry", "Master volume not streamed");
return;
}
SetTopVolume(fMasterVolume);
if (fStreamVoxels) printf("### Voxelization retrieved from filen");
Voxelize("ALL");
if (!fCache) BuildCache();
} else {
Warning("CloseGeometry", "top node was streamed!");
Voxelize("ALL");
if (!fCache) BuildCache();
}
printf("### nodes in %s : %in", gGeoManager->GetTitle(), fNNodes);
printf("----------------modeler ready----------------n");
return;
}
SelectTrackingMedia();
printf("Fixing runtime shapes...n");
CheckGeometry();
printf("Counting nodes...n");
fNNodes = gGeoManager->CountNodes();
Voxelize("ALL");
printf("Building caches for nodes and matrices...n");
BuildCache();
printf("### nodes in %s : %in", gGeoManager->GetTitle(), fNNodes);
printf("----------------modeler ready----------------n");
}
//-----------------------------------------------------------------------------
void TGeoManager::ClearShape(TGeoShape *shape)
{
// Remove a shape from the list of shapes.
if (fShapes->FindObject(shape)) fShapes->Remove(shape);
delete shape;
}
//-----------------------------------------------------------------------------
void TGeoManager::CleanGarbage()
{
// Clean temporary volumes and shapes from garbage collection.
TIter nextv(fGVolumes);
TGeoVolume *vol = 0;
while ((vol=(TGeoVolume*)nextv()))
vol->SetFinder(0);
fGVolumes->Delete();
fGShapes->Delete();
}
//-----------------------------------------------------------------------------
void TGeoManager::CdTop()
{
// Make top level node the current node. Updates the cache accordingly.
// Determine the overlapping state of current node.
fLevel = 0;
if (fCurrentOverlapping) fLastNode = fCurrentNode;
fCurrentNode = fTopNode;
fCache->CdTop();
fCurrentOverlapping = fCurrentNode->IsOverlapping();
}
//-----------------------------------------------------------------------------
void TGeoManager::CdUp()
{
// Go one level up in geometry. Updates cache accordingly.
// Determine the overlapping state of current node.
if (!fLevel) return;
fLevel--;
if (!fLevel) {
CdTop();
return;
}
fCache->CdUp();
if (fCurrentOverlapping) fLastNode = fCurrentNode;
fCurrentNode = fCache->GetNode();
if (!fCurrentNode->IsOffset()) fCurrentOverlapping = fCurrentNode->IsOverlapping();
}
//-----------------------------------------------------------------------------
void TGeoManager::CdDown(Int_t index)
{
// Make a daughter of current node current. Can be called only with a valid
// daughter index (no check). Updates cache accordingly.
TGeoNode *node = fCurrentNode->GetDaughter(index);
Bool_t is_offset = node->IsOffset();
if (is_offset)
node->GetFinder()->cd(node->GetIndex());
else
fCurrentOverlapping = node->IsOverlapping();
fCache->CdDown(index);
fCurrentNode = node;
fLevel++;
}
//-----------------------------------------------------------------------------
Bool_t TGeoManager::cd(const char *path)
{
// Browse the tree of nodes starting from fTopNode according to pathname.
// Changes the path accordingly.
if (!strlen(path)) return kFALSE;
CdTop();
TString spath = path;
TGeoVolume *vol;
Int_t length = spath.Length();
Int_t ind1 = spath.Index("/");
Int_t ind2 = 0;
Bool_t end = kFALSE;
TString name;
TGeoNode *node;
while (!end) {
ind2 = spath.Index("/", ind1+1);
if (ind2<0) {
ind2 = length;
end = kTRUE;
}
name = spath(ind1+1, ind2-ind1-1);
if (name==fTopNode->GetName()) {
ind1 = ind2;
continue;
}
vol = fCurrentNode->GetVolume();
if (vol) {
node = vol->GetNode(name.Data());
} else node = 0;
if (!node) {
Error("cd", "path not valid");
return kFALSE;
}
CdDown(fCurrentNode->GetVolume()->GetIndex(node));
ind1 = ind2;
}
return kTRUE;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::CountNodes(TGeoVolume *vol, Int_t nlevels)
{
// Count the total number of nodes starting from a volume, nlevels down.
TGeoVolume *top;
if (!vol) {
top = fTopVolume;
} else {
top = vol;
}
Int_t count = top->CountNodes(nlevels);
return count;
}
//-----------------------------------------------------------------------------
void TGeoManager::DefaultAngles()
{
// Set default angles for a given view.
if (fPainter) fPainter->DefaultAngles();
}
//-----------------------------------------------------------------------------
void TGeoManager::DrawCurrentPoint(Int_t color)
{
// Draw current point in the same view.
if (fPainter) fPainter->DrawCurrentPoint(color);
}
//-----------------------------------------------------------------------------
void TGeoManager::RandomPoints(TGeoVolume *vol, Int_t npoints, Option_t *option)
{
// Draw random points in the bounding box of a volume.
GetGeomPainter()->RandomPoints(vol, npoints, option);
}
//-----------------------------------------------------------------------------
void TGeoManager::Test(Int_t npoints, Option_t *option)
{
// Check time of finding "Where am I" for n points.
GetGeomPainter()->Test(npoints, option);
}
//-----------------------------------------------------------------------------
void TGeoManager::TestOverlaps(const char* path)
{
// Geometry overlap checker based on sampling.
GetGeomPainter()->TestOverlaps(path);
}
//-----------------------------------------------------------------------------
void TGeoManager::GetBombFactors(Double_t &bombx, Double_t &bomby, Double_t &bombz, Double_t &bombr) const
{
// Retrieve cartesian and radial bomb factors.
if (fPainter) {
fPainter->GetBombFactors(bombx, bomby, bombz, bombr);
return;
}
bombx = bomby = bombz = bombr = 1.3;
}
//-----------------------------------------------------------------------------
TGeoHMatrix *TGeoManager::GetHMatrix()
{
if (!fCurrentMatrix) fCurrentMatrix = new TGeoHMatrix();
return fCurrentMatrix;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetVisLevel() const
{
// Returns current depth to which geometry is drawn.
return fVisLevel;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetVisOption() const
{
// Returns current depth to which geometry is drawn.
return fVisOption;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetVirtualLevel()
{
// Find level of virtuality of current overlapping node (number of levels
// up having the same tracking media.
// return if the current node is ONLY
if (!fCurrentOverlapping) return 0;
Int_t new_media = 0;
Int_t imedia = fCurrentNode->GetMedia();
Int_t virtual_level = 1;
TGeoNode *mother = 0;
while ((mother=GetMother(virtual_level))) {
if (!mother->IsOverlapping() && !mother->IsOffset()) {
if (!new_media) new_media=(mother->GetMedia()==imedia)?0:virtual_level;
break;
}
if (!new_media) new_media=(mother->GetMedia()==imedia)?0:virtual_level;
virtual_level++;
}
return (new_media==0)?virtual_level:(new_media-1);
}
//-----------------------------------------------------------------------------
Bool_t TGeoManager::GotoSafeLevel()
{
// Go upwards the tree until a non-overlaping node
while (fCurrentOverlapping && fLevel) CdUp();
return kTRUE;
}
//-----------------------------------------------------------------------------
TGeoNode *TGeoManager::FindInCluster(Int_t *cluster, Int_t nc)
{
// Find a node inside a cluster of overlapping nodes. Current node must
// be on top of all the nodes in cluster. Always nc>1.
TGeoNode *clnode = 0;
TGeoNode *priority = fLastNode;
// save current node
TGeoNode *current = fCurrentNode;
TGeoNode *found = 0;
// save path
Int_t ipop = PushPath();
// mark this search
fSearchOverlaps = kTRUE;
Int_t deepest = fLevel;
Int_t deepest_virtual = fLevel-GetVirtualLevel();
Int_t found_virtual = 0;
Bool_t replace = kFALSE;
Bool_t added = kFALSE;
Int_t i;
for (i=0; i<nc; i++) {
clnode = current->GetDaughter(cluster[i]);
CdDown(cluster[i]);
found = SearchNode(kTRUE, clnode);
if (!fSearchOverlaps) {
// an only was found during the search -> exiting
PopDummy(ipop);
return found;
}
found_virtual = fLevel-GetVirtualLevel();
if (added) {
// we have put something in stack -> check it
if (found_virtual>deepest_virtual) {
replace = kTRUE;
} else {
if (found_virtual==deepest_virtual) {
if (fLevel>deepest) {
replace = kTRUE;
} else {
if ((fLevel==deepest) && (clnode==priority)) replace=kTRUE;
else replace = kFALSE;
}
} else replace = kFALSE;
}
// if this was the last checked node
if (i==(nc-1)) {
if (replace) {
PopDummy(ipop);
return found;
} else {
fCurrentOverlapping = PopPath();
PopDummy(ipop);
return fCurrentNode;
}
}
// we still have to go on
if (replace) {
// reset stack
PopDummy();
PushPath();
deepest = fLevel;
deepest_virtual = found_virtual;
}
// restore top of cluster
fCurrentOverlapping = PopPath(ipop);
} else {
// the stack was clean, push new one
PushPath();
added = kTRUE;
deepest = fLevel;
deepest_virtual = found_virtual;
// restore original path
fCurrentOverlapping = PopPath(ipop);
}
}
PopDummy(ipop);
return fCurrentNode;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetTouchedCluster(Int_t start, Double_t *point,
Int_t *check_list, Int_t ncheck, Int_t *result)
{
// Make the cluster of overlapping nodes in a voxel, containing point in reference
// of the mother. Returns number of nodes containing the point. Nodes should not be
// offsets.
// we are in the mother reference system
TGeoNode *current = fCurrentNode->GetDaughter(check_list[start]);
Int_t novlps = 0;
Int_t *ovlps = current->GetOverlaps(novlps);
if (!ovlps) return 0;
Double_t local[3];
// intersect check list with overlap list
Int_t ntotal = 0;
current->MasterToLocal(point, &local[0]);
if (current->GetVolume()->Contains(&local[0])) {
result[ntotal++]=check_list[start];
}
Int_t jst=0, i, j;
while ((ovlps[jst]<=check_list[start]) && (jst<novlps)) jst++;
if (jst==novlps) return 0;
for (i=start; i<ncheck; i++) {
for (j=jst; j<novlps; j++) {
if (check_list[i]==ovlps[j]) {
// overlapping node in voxel -> check if touched
current = fCurrentNode->GetDaughter(check_list[i]);
current->MasterToLocal(point, &local[0]);
if (current->GetVolume()->Contains(&local[0])) {
result[ntotal++]=check_list[i];
}
}
}
}
return ntotal;
}
//-----------------------------------------------------------------------------
void TGeoManager::DefaultColors()
{
// Set default volume colors according to tracking media.
if (fPainter) {
fPainter->DefaultColors();
return;
}
TIter next(fVolumes);
TGeoVolume *vol;
while ((vol=(TGeoVolume*)next()))
vol->SetLineColor(vol->GetMaterial()->GetDefaultColor());
}
//-----------------------------------------------------------------------------
void TGeoManager::SetBombFactors(Double_t bombx, Double_t bomby, Double_t bombz, Double_t bombr)
{
// Set factors that will "bomb" all translations in cartesian and cylindrical coordinates.
if (fPainter) fPainter->SetBombFactors(bombx, bomby, bombz, bombr);
}
//-----------------------------------------------------------------------------
void TGeoManager::SetVisOption(Int_t option) {
// set drawing mode :
// option=0 (default) all nodes drawn down to vislevel
// option=1 leaves and nodes at vislevel drawn
// option=2 path is drawn
GetGeomPainter();
if ((option>=0) && (option<3)) fVisOption=option;
fPainter->SetVisOption(option);
}
//-----------------------------------------------------------------------------
void TGeoManager::SetVisLevel(Int_t level) {
// set default level down to which visualization is performed
GetGeomPainter();
if (level>0) fVisLevel = level;
fPainter->SetVisLevel(level);
}
//-----------------------------------------------------------------------------
void TGeoManager::OptimizeVoxels(const char *filename)
{
// Optimize voxelization type for all volumes. Save best choice in a macro.
if (!fTopNode) {
printf("SaveAttributes - geometry must be closedn");
return;
}
ofstream out;
char *fname = new char[20];
char quote = '"';
if (!strlen(filename))
sprintf(fname, "tgeovox.C");
else
sprintf(fname, "%s", filename);
out.open(fname, ios::out);
if (!out.good()) {
Error("OptimizeVoxels", "cannot open file");
delete [] fname;
return;
}
// write header
TDatime t;
TString sname(fname);
sname.ReplaceAll(".C", "");
out << sname.Data()<<"()"<<endl;
out << "{" << endl;
out << "//=== Macro generated by ROOT version "<< gROOT->GetVersion()<<" : "<<t.AsString()<<endl;
out << "//=== Voxel optimization for " << GetTitle() << " geometry"<<endl;
out << "//===== <run this macro JUST BEFORE closing the geometry>"<<endl;
out << " TGeoVolume *vol = 0;"<<endl;
out << " // parse all voxelized volumes"<<endl;
TGeoVolume *vol = 0;
Bool_t cyltype;
TIter next(fVolumes);
while ((vol=(TGeoVolume*)next())) {
if (!vol->GetVoxels()) continue;
out<<" vol = gGeoManager->GetVolume("<<quote<<vol->GetName()<<quote<<");"<<endl;
cyltype = vol->OptimizeVoxels();
if (cyltype) {
out<<" vol->SetCylVoxels();"<<endl;
} else {
out<<" vol->SetCylVoxels(kFALSE);"<<endl;
}
}
out << "}" << endl;
out.close();
delete [] fname;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::Parse(const char *expr, TString &expr1, TString &expr2, TString &expr3)
{
// Parse a string boolean expression and do a syntax check. Find top
// level boolean operator and returns its type. Fill the two
// substrings to which this operator applies. The returned integer is :
// -1 : parse error
// 0 : no boolean operator
// 1 : union - represented as '+' in expression
// 2 : difference (subtraction) - represented as '-' in expression
// 3 : intersection - represented as '*' in expression.
// Paranthesys should be used to avoid ambiguites. For instance :
// A+B-C will be interpreted as (A+B)-C which is not the same as A+(B-C)
// eliminate not needed paranthesys
TString startstr(expr);
Int_t len = startstr.Length();
Int_t i;
TString e0 = "";
expr3 = "";
// eliminate blanks
for (i=0; i< len; i++) {
if (startstr(i)==' ') continue;
e0 += startstr(i, 1);
}
Int_t level = 0;
Int_t levmin = 999;
Int_t boolop = 0;
Int_t indop = 0;
Int_t iloop = 1;
Int_t lastop = 0;
Int_t lastdp = 0;
Int_t lastpp = 0;
Bool_t foundmat = kFALSE;
// check/eliminate paranthesys
while (iloop==1) {
iloop = 0;
lastop = 0;
lastdp = 0;
lastpp = 0;
len = e0.Length();
for (i=0; i<len; i++) {
if (e0(i)=='(') {
if (!level) iloop++;
level++;
continue;
}
if (e0(i)==')') {
level--;
if (level==0) lastpp=i;
continue;
}
if ((e0(i)=='+') || (e0(i)=='-') || (e0(i)=='*')) {
lastop = i;
if (level<levmin) {
levmin = level;
indop = i;
}
continue;
}
if ((e0(i)==':') && (level==0)) {
lastdp = i;
continue;
}
}
if (level!=0) {
printf("parse error : paranthesys does not matchn");
return -1;
}
if (iloop==1 && (e0(0)=='(') && (e0(len-1)==')')) {
// eliminate extra paranthesys
e0=e0(1, len-2);
continue;
}
if (foundmat) break;
if (((lastop==0) && (lastdp>0)) || ((lastpp>0) && (lastdp>lastpp) && (indop<lastpp))) {
expr3 = e0(lastdp+1, len-lastdp);
printf("transformation : %sn", expr3.Data());
e0=e0(0, lastdp);
foundmat = kTRUE;
iloop = 1;
continue;
} else break;
}
// loop expression and search paranthesys/operators
levmin = 999;
for (i=0; i<len; i++) {
if (e0(i)=='(') {
level++;
continue;
}
if (e0(i)==')') {
level--;
continue;
}
if (level<levmin) {
if (e0(i)=='+') {
boolop = 1; // union
levmin = level;
indop = i;
}
if (e0(i)=='-') {
boolop = 2; // difference
levmin = level;
indop = i;
}
if (e0(i)=='*') {
boolop = 3; // intersection
levmin = level;
indop = i;
}
}
}
if (indop==0) {
expr1=e0;
return indop;
}
expr1 = e0(0, indop);
expr2 = e0(indop+1, len-indop);
return boolop;
}
//-----------------------------------------------------------------------------
void TGeoManager::SaveAttributes(const char *filename)
{
// Save current attributes in a macro
if (!fTopNode) {
printf("SaveAttributes - geometry must be closedn");
return;
}
ofstream out;
char *fname = new char[20];
char quote = '"';
if (!strlen(filename))
sprintf(fname, "tgeoatt.C");
else
sprintf(fname, "%s", filename);
out.open(fname, ios::out);
if (!out.good()) {
Error("SaveAttributes", "cannot open file");
delete [] fname;
return;
}
// write header
TDatime t;
TString sname(fname);
sname.ReplaceAll(".C", "");
out << sname.Data()<<"()"<<endl;
out << "{" << endl;
out << "//=== Macro generated by ROOT version "<< gROOT->GetVersion()<<" : "<<t.AsString()<<endl;
out << "//=== Attributes for " << GetTitle() << " geometry"<<endl;
out << "//===== <run this macro AFTER loading the geometry in memory>"<<endl;
// save current top volume
out << " TGeoVolume *top = gGeoManager->GetVolume("<<quote<<gGeoManager->GetTopVolume()->GetName()<<quote<<");"<<endl;
out << " TGeoVolume *vol = 0;"<<endl;
out << " TGeoNode *node = 0;"<<endl;
out << " // clear all volume attributes and get painter"<<endl;
out << " gGeoManager->ClearAttributes();"<<endl;
out << " gGeoManager->GetGeomPainter();"<<endl;
out << " // set visualization modes and bomb factors"<<endl;
out << " gGeoManager->SetVisOption("<<gGeoManager->GetVisOption()<<");"<<endl;
out << " gGeoManager->SetVisLevel("<<gGeoManager->GetVisLevel()<<");"<<endl;
out << " gGeoManager->SetExplodedView("<<gGeoManager->GetBombMode()<<");"<<endl;
Double_t bombx, bomby, bombz, bombr;
GetBombFactors(bombx, bomby, bombz, bombr);
out << " gGeoManager->SetBombFactors("<<bombx<<","<<bomby<<","<<bombz<<","<<bombr<<");"<<endl;
out << " // iterate volumes coontainer and set new attributes"<<endl;
// out << " TIter next(gGeoManager->GetListOfVolumes());"<<endl;
TGeoVolume *vol = 0;
fTopNode->SaveAttributes(out);
TIter next(fVolumes);
while ((vol=(TGeoVolume*)next())) {
vol->SetVisStreamed(kFALSE);
}
out << " // draw top volume with new settings"<<endl;
out << " top->Draw();"<<endl;
out << " gPad->x3d();"<<endl;
out << "}" << endl;
out.close();
delete [] fname;
}
//-----------------------------------------------------------------------------
TGeoNode *TGeoManager::SearchNode(Bool_t downwards, TGeoNode *skipnode)
{
// Returns the deepest node containing fPoint, which must be set a priori.
Double_t point[3];
TGeoVolume *vol = 0;
Bool_t inside_current = kFALSE;
if (!downwards) {
// we are looking upwards until inside current node or exit
if (fStartSafe) GotoSafeLevel();
vol=fCurrentNode->GetVolume();
MasterToLocal(fPoint, &point[0]);
inside_current = vol->Contains(&point[0]);
if (!inside_current) {
TGeoNode *skip = fCurrentNode;
// check if we can go up
if (!fLevel) {
fIsOutside = kTRUE;
return 0;
}
CdUp();
return SearchNode(kFALSE, skip);
}
}
if (!inside_current) {
// we are looking downwards
vol = fCurrentNode->GetVolume();
MasterToLocal(fPoint, &point[0]);
if (fCurrentNode==skipnode) {
// in case searching down and skipping this
inside_current = kTRUE;
} else {
inside_current = vol->Contains(&point[0]);
if (!inside_current) return 0;
}
}
// point inside current (safe) node -> search downwards
TGeoNode *node;
Int_t ncheck = 0;
// if inside an non-overlapping node, reset overlap searches
if (!fCurrentOverlapping) {
fSearchOverlaps = kFALSE;
}
Int_t nd = vol->GetNdaughters();
// in case there are no daughters
if (!nd) return fCurrentNode;
TGeoPatternFinder *finder = vol->GetFinder();
// point is inside the current node
// first check if inside a division
if (finder) {
node=finder->FindNode(&point[0]);
if (node) {
// go inside the division cell and search downwards
CdDown(node->GetIndex());
return SearchNode(kTRUE, node);
}
// point is not inside the division, but might be in other nodes
// at the same level (NOT SUPPORTED YET)
return fCurrentNode;
}
// second, look if current volume is voxelized
TGeoVoxelFinder *voxels = vol->GetVoxels();
Int_t *check_list = 0;
if (voxels) {
// get the list of nodes passing thorough the current voxel
check_list = voxels->GetCheckList(&point[0], ncheck);
// if none in voxel, see if this is the last one
if (!check_list) return fCurrentNode;
// loop all nodes in voxel
for (Int_t id=0; id<ncheck; id++) {
node = vol->GetNode(check_list[id]);
if (node==skipnode) continue;
if ((id<(ncheck-1)) && node->IsOverlapping()) {
// make the cluster of overlaps
Int_t *cluster = new Int_t[ncheck-id];
Int_t nc = GetTouchedCluster(id, &point[0], check_list, ncheck, cluster);
if (nc>1) {
node = FindInCluster(cluster, nc);
delete [] cluster;
return node;
}
}
CdDown(check_list[id]);
node = SearchNode(kTRUE);
if (node) return node;
CdUp();
}
return fCurrentNode;
}
// if there are no voxels just loop all daughters
Int_t id = 0;
while ((node=fCurrentNode->GetDaughter(id++))) {
if (node==skipnode) {
if (id==nd) return fCurrentNode;
continue;
}
CdDown(id-1);
node = SearchNode(kTRUE);
if (node) return node;
CdUp();
if (id == nd) return fCurrentNode;
}
// point is not inside one of the daughters, so it is in the current vol
return fCurrentNode;
}
//-----------------------------------------------------------------------------
TGeoNode *TGeoManager::FindNextBoundary(const char *path)
{
// Find distance to target node given by path boundary on current direction. If no target
// is specified, find distance to next boundary from current point to current direction
// and store this in fStep. Returns node having this boundary. Find also
// distance to closest boundary and store it in fSafety. Set flags
// fIsStepEntering/fIsStepExiting according to the fact that current ray will enter
// or exit next node.
// convert current point and direction to local reference
// printf("-------- currently : %sn", fCurrentNode->GetName());
// printf("--- alpha=%gn", 180.*TMath::ATan2(fDirection[1], fDirection[0])/TMath::Pi());
// printf("--- point : %g, %g, %gn", fPoint[0], fPoint[1], fPoint[2]);
fStep = TGeoShape::kBig;
Double_t point[3];
Double_t dir[3];
if (strlen(path)) {
PushPath();
if (!cd(path)) {
PopPath();
return 0;
}
TGeoNode *target=fCurrentNode;
TGeoVolume *tvol=fCurrentNode->GetVolume();
MasterToLocal(fPoint, &point[0]);
MasterToLocalVect(fDirection, &dir[0]);
if (tvol->Contains(&point[0])) {
fStep=tvol->GetShape()->DistToOut(&point[0], &dir[0], 3, TGeoShape::kBig, &fSafety);
fIsStepEntering=kFALSE;
fIsStepExiting=kTRUE;
} else {
fStep=tvol->GetShape()->DistToIn(&point[0], &dir[0], 3, TGeoShape::kBig, &fSafety);
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
}
PopPath();
return target;
}
MasterToLocal(fPoint, &point[0]);
MasterToLocalVect(fDirection, &dir[0]);
// compute distance to exit point from current node and the distance to its
// closest boundary
TGeoVolume *vol = fCurrentNode->GetVolume();
// if point is outside, just check the top node
if (fIsOutside) {
fStep = fTopVolume->GetShape()->DistToIn(fPoint, fDirection, 3, TGeoShape::kBig, &fSafety);
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
return fTopNode;
}
// find distance to exiting current node
fIsStepEntering=kFALSE;
fIsStepExiting=kTRUE;
Double_t snext = TGeoShape::kBig;
fStep = vol->GetShape()->DistToOut(&point[0], &dir[0], 2, TGeoShape::kBig, &fSafety);
// printf("to exiting : %gn", fStep);
// if (fIsOnBoundary && fIsExiting) return fCurrentNode;
TGeoNode *clnode = fCurrentNode;
TGeoNode *current = 0;
TGeoVolume *mother = 0;
// if we are in an overlapping node, check also the mother
if (fCurrentOverlapping) {
// printf("overlapping node -> go to safe leveln");
Double_t mothpt[3];
Double_t vecpt[3];
Double_t dpt[3], dvec[3];
PushPath();
Int_t novlps;
while (fCurrentOverlapping) {
Int_t *ovlps = fCurrentNode->GetOverlaps(novlps);
CdUp();
mother = fCurrentNode->GetVolume();
// printf("-> up in %sn", fCurrentNode->GetName());
MasterToLocal(fPoint, &mothpt[0]);
MasterToLocalVect(fDirection, &vecpt[0]);
// check distance to out
snext = mother->GetShape()->DistToOut(&mothpt[0], &vecpt[0], 2, TGeoShape::kBig, &fSafety);
// printf("-> to out : %gn", snext);
if (snext<fStep) {
// printf(" this is closer...n");
fStep = snext;
clnode = fCurrentNode;
}
// check overlapping nodes
// printf("-> now check overlaps...n");
for (Int_t i=0; i<novlps; i++) {
current = mother->GetNode(ovlps[i]);
if (!current->IsOverlapping()) {
// printf("checking overlapping %sn", current->GetName());
current->cd();
current->GetMatrix()->MasterToLocal(&mothpt[0], &dpt[0]);
current->GetMatrix()->MasterToLocalVect(&vecpt[0], &dvec[0]);
snext = current->GetVolume()->GetShape()->DistToIn(&dpt[0], &dvec[0], 2, TGeoShape::kBig, &fSafety);
// printf("-> to in : %gn", snext);
if (snext<fStep) {
// printf(" this is closern");
fStep = snext;
clnode = current;
}
}
}
}
PopPath();
// printf("back in %sn", fCurrentNode->GetName());
}
// get number of daughters. If no daughters we are done.
Int_t nd = vol->GetNdaughters();
if (!nd) return fCurrentNode;
Double_t lpoint[3];
Double_t ldir[3];
Double_t safety = TGeoShape::kBig;
Int_t i=0;
// if only one daughter, check it and exit
if (nd<3) {
for (i=0; i<nd; i++) {
current = vol->GetNode(i);
current->cd();
current->MasterToLocal(&point[0], &lpoint[0]);
current->MasterToLocalVect(&dir[0], &ldir[0]);
snext = current->GetVolume()->GetShape()->DistToIn(&lpoint[0], &ldir[0], 2, TGeoShape::kBig, &safety);
fSafety = TMath::Min(fSafety, safety);
if (snext<fStep) {
fStep=snext;
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
clnode = current;
}
}
return clnode;
}
// if current volume is divided, we are in the non-divided region. We
// check only the first and the last cell
TGeoPatternFinder *finder = vol->GetFinder();
if (finder) {
Int_t ifirst = finder->GetDivIndex();
current = vol->GetNode(ifirst);
current->cd();
current->MasterToLocal(&point[0], &lpoint[0]);
current->MasterToLocalVect(&dir[0], &ldir[0]);
snext = current->GetVolume()->GetShape()->DistToIn(&lpoint[0], &ldir[0], 2, TGeoShape::kBig, &safety);
fSafety = TMath::Min(fSafety, safety);
if (snext<fStep) {
fStep=snext;
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
clnode = current;
}
Int_t ilast = ifirst+finder->GetNdiv()-1;
if (ilast==ifirst) return clnode;
current = vol->GetNode(ilast);
current->cd();
current->MasterToLocal(&point[0], &lpoint[0]);
current->MasterToLocalVect(&dir[0], &ldir[0]);
snext = current->GetVolume()->GetShape()->DistToIn(&lpoint[0], &ldir[0], 2, TGeoShape::kBig, &safety);
fSafety = TMath::Min(fSafety, safety);
if (snext<fStep) {
fStep=snext;
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
return current;
}
}
// if current volume is voxelized, first get current voxel
TGeoVoxelFinder *voxels = vol->GetVoxels();
// printf("---check voxelsn");
if (voxels) {
Int_t ncheck = 0;
Int_t *vlist = 0;
voxels->SortCrossedVoxels(&point[0], &dir[0]);
Bool_t first = kTRUE;
while ((vlist=voxels->GetNextVoxel(&point[0], &dir[0], ncheck))) {
// printf("---ncheck : %in", ncheck);
for (i=0; i<ncheck; i++) {
current = vol->GetNode(vlist[i]);
current->cd();
current->MasterToLocal(&point[0], &lpoint[0]);
current->MasterToLocalVect(&dir[0], &ldir[0]);
// printf("<<< CHECKING %sn", current->GetName());
if (first) {
// compute also safety if we are in the starting voxel
snext = current->GetVolume()->GetShape()->DistToIn(&lpoint[0], &ldir[0], 2, TGeoShape::kBig, &safety);
if (safety<fSafety) fSafety=safety;
} else {
snext = current->GetVolume()->GetShape()->DistToIn(&lpoint[0], &ldir[0], 3, TGeoShape::kBig, &safety);
}
// printf("<<< step : %gn", snext);
if (snext<fStep) {
// printf("%s CLOSERn", current->GetName());
fStep=snext;
fIsStepEntering=kTRUE;
fIsStepExiting=kFALSE;
clnode = current;
}
}
first=kFALSE;
}
}
return clnode;
}
//-----------------------------------------------------------------------------
Bool_t TGeoManager::IsInPhiRange() const
{
// True if current node is in phi range
if (!fPhiCut) return kTRUE;
Double_t *origin;
if (!fCurrentNode) return kFALSE;
origin = ((TGeoBBox*)fCurrentNode->GetVolume()->GetShape())->GetOrigin();
Double_t point[3];
LocalToMaster(origin, &point[0]);
Double_t phi = TMath::ATan2(point[1], point[0])*TGeoShape::kRadDeg;
if (phi<0) phi+=360.;
if ((phi>=fPhimin) && (phi<=fPhimax)) return kFALSE;
return kTRUE;
}
//-----------------------------------------------------------------------------
void TGeoManager::InitTrack(Double_t *point, Double_t *dir)
{
// Initialize current point and current direction vector (normalized)
// in MARS.
SetCurrentPoint(point);
SetCurrentDirection(dir);
FindNode();
}
//-----------------------------------------------------------------------------
void TGeoManager::InitTrack(Double_t x, Double_t y, Double_t z, Double_t nx, Double_t ny, Double_t nz)
{
// Initialize current point and current direction vector (normalized)
// in MARS.
SetCurrentPoint(x,y,z);
SetCurrentDirection(nx,ny,nz);
FindNode();
}
//-----------------------------------------------------------------------------
const char *TGeoManager::GetPath() const
{
// Get path to the current node in the form /node0/node1/...
if (fIsOutside && !fPainter) return kGeoOutsidePath;
return fCache->GetPath();
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetByteCount(Option_t * /*option*/)
{
// Get total size of geometry in bytes.
Int_t count = 0;
TIter next(fVolumes);
TGeoVolume *vol;
while ((vol=(TGeoVolume*)next())) count += vol->GetByteCount();
TIter next1(fMatrices);
TGeoMatrix *matrix;
while ((matrix=(TGeoMatrix*)next1())) count += matrix->GetByteCount();
TIter next2(fMaterials);
TGeoMaterial *mat;
while ((mat=(TGeoMaterial*)next2())) count += mat->GetByteCount();
printf("Total size of logical tree : %i bytesn", count);
return count;
}
//-----------------------------------------------------------------------------
TVirtualGeoPainter *TGeoManager::GetGeomPainter()
{
// Make a default painter if none present. Returns pointer to it.
if (!fPainter) {
fPainter=TVirtualGeoPainter::GeoPainter();
if (!fPainter) {
Error("GetGeomPainter", "could not create painter");
return 0;
}
fPainter->SetVisOption(fVisOption);
fPainter->SetVisLevel(fVisLevel);
fPainter->SetExplodedView(fExplodedView);
fPainter->SetNsegments(fNsegments);
}
return fPainter;
}
//-----------------------------------------------------------------------------
TGeoMaterial *TGeoManager::GetMaterial(const char *matname) const
{
// Search for a named material.
TGeoMaterial *mat = (TGeoMaterial*)fMaterials->FindObject(matname);
return mat;
}
//-----------------------------------------------------------------------------
TGeoMaterial *TGeoManager::GetMaterial(Int_t id) const
{
// Return material at position id.
if (id >= fMaterials->GetSize()) return 0;
TGeoMaterial *mat = (TGeoMaterial*)fMaterials->At(id);
return mat;
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetMaterialIndex(const char *matname) const
{
// Return index of named material.
TIter next(fMaterials);
TGeoMaterial *mat;
Int_t id = 0;
while ((mat = (TGeoMaterial*)next())) {
if (mat->GetName() == matname)
return id;
id++;
}
return -1; // fail
}
//-----------------------------------------------------------------------------
void TGeoManager::RandomRays(Int_t nrays, Double_t startx, Double_t starty, Double_t startz)
{
// Randomly shoot nrays and plot intersections with surfaces for current
// top node.
GetGeomPainter()->RandomRays(nrays, startx, starty, startz);
}
//-----------------------------------------------------------------------------
void TGeoManager::RemoveMaterial(Int_t index)
{
// Remove material at given index.
TObject *obj = fMaterials->At(index);
if (obj) fMaterials->Remove(obj);
}
//-----------------------------------------------------------------------------
void TGeoManager::RestoreMasterVolume()
{
// Restore the master volume of the geometry.
if (fTopVolume == fMasterVolume) return;
if (fMasterVolume) SetTopVolume(fMasterVolume);
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::GetVolume(const char *name) const
{
// Retrieves a named volume.
return ((TGeoVolume*)fVolumes->FindObject(name));
}
//-----------------------------------------------------------------------------
void TGeoManager::Voxelize(Option_t *option)
{
// Voxelize all non-divided volumes.
TGeoVolume *vol;
TGeoVoxelFinder *vox = 0;
if (!fStreamVoxels) printf("Voxelizing...n");
Int_t nentries = fVolumes->GetSize();
for (Int_t i=0; i<nentries; i++) {
vol = (TGeoVolume*)fVolumes->At(i);
if (!fIsGeomReading) vol->SortNodes();
if (!fStreamVoxels) {
vol->Voxelize(option);
} else {
vox = vol->GetVoxels();
if (vox) vox->CreateCheckList();
}
if (!fIsGeomReading) vol->FindOverlaps();
}
}
//-----------------------------------------------------------------------------
void TGeoManager::ModifiedPad() const
{
// Send "Modified" signal to painter.
if (!fPainter) return;
fPainter->ModifiedPad();
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeArb8(const char *name, const char *material,
Double_t dz, Double_t *vertices)
{
// Make an TGeoArb8 volume.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeArb8", "Material unknown");
mat = GetMaterial("default");
}
TGeoArb8 *arb = new TGeoArb8(dz, vertices);
TGeoVolume *vol = new TGeoVolume(name, arb, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeBox(const char *name, const char *material,
Double_t dx, Double_t dy, Double_t dz)
{
// Make in one step a volume pointing to a box shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeBox", "Material unknown");
mat = GetMaterial("default");
}
TGeoBBox *box = new TGeoBBox(dx, dy, dz);
TGeoVolume *vol = new TGeoVolume(name, box, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakePara(const char *name, const char *material,
Double_t dx, Double_t dy, Double_t dz,
Double_t alpha, Double_t theta, Double_t phi)
{
// Make in one step a volume pointing to a paralelipiped shape with given material.
if ((alpha==0) && (theta==0)) {
printf("Warning : para %s with alpha=0, theta=0 -> making box insteadn", name);
return MakeBox(name, material, dx, dy, dz);
}
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakePara", "Material unknown");
mat = GetMaterial("default");
}
TGeoPara *para=0;
para = new TGeoPara(dx, dy, dz, alpha, theta, phi);
TGeoVolume *vol = new TGeoVolume(name, para, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeSphere(const char *name, const char *material,
Double_t rmin, Double_t rmax, Double_t themin, Double_t themax,
Double_t phimin, Double_t phimax)
{
// Make in one step a volume pointing to a sphere shape with given material
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeSphere", " unknown");
mat = GetMaterial("default");
}
TGeoSphere *sph = new TGeoSphere(rmin, rmax, themin, themax, phimin, phimax);
TGeoVolume *vol = new TGeoVolume(name, sph, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeTube(const char *name, const char *material,
Double_t rmin, Double_t rmax, Double_t dz)
{
// Make in one step a volume pointing to a tube shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTube", "Material unknown");
mat = GetMaterial("default");
}
TGeoTube *tube = new TGeoTube(rmin, rmax, dz);
TGeoVolume *vol = new TGeoVolume(name, tube, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeTubs(const char *name, const char *material,
Double_t rmin, Double_t rmax, Double_t dz,
Double_t phi1, Double_t phi2)
{
// Make in one step a volume pointing to a tube segment shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTubs", "Material unknown");
mat = GetMaterial("default");
}
TGeoTubeSeg *tubs = new TGeoTubeSeg(rmin, rmax, dz, phi1, phi2);
TGeoVolume *vol = new TGeoVolume(name, tubs, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeEltu(const char *name, const char *material,
Double_t a, Double_t b, Double_t dz)
{
// Make in one step a volume pointing to a tube shape with given material
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTube", "Material unknown");
mat = GetMaterial("default");
}
TGeoEltu *eltu = new TGeoEltu(a, b, dz);
TGeoVolume *vol = new TGeoVolume(name, eltu, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeCtub(const char *name, const char *material,
Double_t rmin, Double_t rmax, Double_t dz, Double_t phi1, Double_t phi2,
Double_t lx, Double_t ly, Double_t lz, Double_t tx, Double_t ty, Double_t tz)
{
// Make in one step a volume pointing to a tube segment shape with given material
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTubs", "Material unknown");
mat = GetMaterial("default");
}
TGeoCtub *ctub = new TGeoCtub(rmin, rmax, dz, phi1, phi2, lx, ly, lz, tx, ty, tz);
TGeoVolume *vol = new TGeoVolume(name, ctub, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeCone(const char *name, const char *material,
Double_t dz, Double_t rmin1, Double_t rmax1,
Double_t rmin2, Double_t rmax2)
{
// Make in one step a volume pointing to a cone shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeCone", "Material unknown");
mat = GetMaterial("default");
}
TGeoCone *cone = new TGeoCone(dz, rmin1, rmax1, rmin2, rmax2);
TGeoVolume *vol = new TGeoVolume(name, cone, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeCons(const char *name, const char *material,
Double_t dz, Double_t rmin1, Double_t rmax1,
Double_t rmin2, Double_t rmax2,
Double_t phi1, Double_t phi2)
{
// Make in one step a volume pointing to a cone segment shape with given material
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeCons", "Material unknown");
mat = GetMaterial("default");
}
TGeoConeSeg *cons = new TGeoConeSeg(dz, rmin1, rmax1, rmin2, rmax2, phi1, phi2);
TGeoVolume *vol = new TGeoVolume(name, cons, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakePcon(const char *name, const char *material,
Double_t phi, Double_t dphi, Int_t nz)
{
// Make in one step a volume pointing to a polycone shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakePcon", "Material unknown");
mat = GetMaterial("default");
}
TGeoPcon *pcon = new TGeoPcon(phi, dphi, nz);
TGeoVolume *vol = new TGeoVolume(name, pcon, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakePgon(const char *name, const char *material,
Double_t phi, Double_t dphi, Int_t nedges, Int_t nz)
{
// Make in one step a volume pointing to a polygone shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakePgon", "Material unknown");
mat = GetMaterial("default");
}
TGeoPgon *pgon = new TGeoPgon(phi, dphi, nedges, nz);
TGeoVolume *vol = new TGeoVolume(name, pgon, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeTrd1(const char *name, const char *material,
Double_t dx1, Double_t dx2, Double_t dy, Double_t dz)
{
// Make in one step a volume pointing to a TGeoTrd1 shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTrd1", "Material unknown");
mat = GetMaterial("default");
}
TGeoTrd1 *trd1 = new TGeoTrd1(dx1, dx2, dy, dz);
TGeoVolume *vol = new TGeoVolume(name, trd1, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeTrd2(const char *name, const char *material,
Double_t dx1, Double_t dx2, Double_t dy1, Double_t dy2,
Double_t dz)
{
// Make in one step a volume pointing to a TGeoTrd2 shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTrd2", "Material unknown");
mat = GetMaterial("default");
}
TGeoTrd2 *trd2 = new TGeoTrd2(dx1, dx2, dy1, dy2, dz);
TGeoVolume *vol = new TGeoVolume(name, trd2, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeTrap(const char *name, const char *material,
Double_t dz, Double_t theta, Double_t phi, Double_t h1,
Double_t bl1, Double_t tl1, Double_t alpha1, Double_t h2, Double_t bl2,
Double_t tl2, Double_t alpha2)
{
// Make in one step a volume pointing to a trapezoid shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTrap", "Material unknown");
mat = GetMaterial("default");
}
TGeoTrap *trap = new TGeoTrap(dz, theta, phi, h1, bl1, tl1, alpha1, h2, bl2,
tl2, alpha2);
TGeoVolume *vol = new TGeoVolume(name, trap, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolume *TGeoManager::MakeGtra(const char *name, const char *material,
Double_t dz, Double_t theta, Double_t phi, Double_t twist, Double_t h1,
Double_t bl1, Double_t tl1, Double_t alpha1, Double_t h2, Double_t bl2,
Double_t tl2, Double_t alpha2)
{
// Make in one step a volume pointing to a twisted trapezoid shape with given material.
TGeoVolume *old = 0;
old=(TGeoVolume*)fVolumes->FindObject(name);
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeTrap", "Material unknown");
mat = GetMaterial("default");
}
TGeoGtra *gtra = new TGeoGtra(dz, theta, phi, twist, h1, bl1, tl1, alpha1, h2, bl2,
tl2, alpha2);
TGeoVolume *vol = new TGeoVolume(name, gtra, mat);
if (old) vol->MakeCopyNodes(old);
return vol;
}
//-----------------------------------------------------------------------------
TGeoVolumeMulti *TGeoManager::MakeVolumeMulti(const char *name, const char *material)
{
// Make a TGeoVolumeMulti handling a list of volumes.
TGeoMaterial *mat = GetMaterial(material);
if (!mat) {
printf("%sn", material);
Warning("MakeVolumeMulti", "Material unknown");
mat = GetMaterial("default");
}
return (new TGeoVolumeMulti(name, mat));
}
//-----------------------------------------------------------------------------
void TGeoManager::SetExplodedView(Int_t ibomb)
{
// Set type of exploding view (see TGeoPainter::SetExplodedView())
GetGeomPainter();
if ((ibomb>=0) && (ibomb<4)) fExplodedView = ibomb;
if (fPainter) fPainter->SetExplodedView(ibomb);
}
//-----------------------------------------------------------------------------
void TGeoManager::SetPhiRange(Double_t phimin, Double_t phimax)
{
// Set cut phi range
if ((phimin==0) && (phimax==360)) {
fPhiCut = kFALSE;
return;
}
fPhiCut = kTRUE;
fPhimin = phimin;
fPhimax = phimax;
}
//-----------------------------------------------------------------------------
void TGeoManager::SetNsegments(Int_t nseg)
{
// Set number of segments for approximating circles in drawing.
GetGeomPainter();
if (nseg>2) fNsegments = nseg;
if (fPainter) fPainter->SetNsegments(nseg);
}
//-----------------------------------------------------------------------------
Int_t TGeoManager::GetNsegments() const
{
// Get number of segments approximating circles
return fNsegments;
}
//-----------------------------------------------------------------------------
void TGeoManager::BuildDefaultMaterials()
{
// Build the default materials. A list of those can be found in ...
new TGeoMaterial("default", "Air", 14.61, 7.3, 0.001205);
}
//-----------------------------------------------------------------------------
TGeoNode *TGeoManager::Step(Bool_t is_geom, Bool_t cross)
{
// Make a rectiliniar step of length fStep from current point (fPoint) on current
// direction (fDirection). If the step is imposed by geometry, is_geom flag
// must be true (default). The cross flag specifies if the boundary should be
// crossed in case of a geometry step (default true). Returns new node after step.
// Set also on boundary condition.
Double_t epsil = 0;
if (fStep<1E-6) {
fIsNullStep=kTRUE;
if (fStep<0) fStep = 0.;
} else {
fIsNullStep=kFALSE;
}
if (is_geom) epsil=(cross)?1E-6:-1E-6;
TGeoNode *old = fCurrentNode;
if (fIsOutside) old = 0;
fStep += epsil;
for (Int_t i=0; i<3; i++) fPoint[i]+=fStep*fDirection[i];
TGeoNode *current = FindNode();
if (fIsOutside) current=0;
if (is_geom) {
fIsEntering = (current==old)?kFALSE:kTRUE;
fIsExiting = !fIsEntering;
if (fIsEntering && fIsNullStep) fIsNullStep = kFALSE;
fIsOnBoundary = kTRUE;
} else {
fIsEntering = fIsExiting = kFALSE;
fIsOnBoundary = kFALSE;
}
return current;
}
//-----------------------------------------------------------------------------
TGeoNode *TGeoManager::SamplePoints(Int_t npoints, Double_t &dist, Double_t epsil,
const char* g3path)
{
// shoot npoints randomly in a box of 1E-5 arround current point.
// return minimum distance to points outside
return GetGeomPainter()->SamplePoints(npoints, dist, epsil, g3path);
}
//-----------------------------------------------------------------------------
void TGeoManager::SetTopVolume(TGeoVolume *vol)
{
// Set the top volume and corresponding node as starting point of the geometry.
if (fTopVolume==vol) return;
if (fTopVolume) fTopVolume->SetTitle("");
fTopVolume = vol;
vol->SetTitle("Top volume");
if (fTopNode) delete fTopNode;
else fMasterVolume = vol;
fTopNode = new TGeoNodeMatrix(vol, gGeoIdentity);
char *name = new char[strlen(vol->GetName()+2)];
sprintf(name, "%s_1", vol->GetName());
fTopNode->SetName(name);
fTopNode->SetTitle("Top logical node");
fCurrentNode = fTopNode;
fNodes->AddAt(fTopNode, 0);
fLevel = 0;
if (fCache) {
delete fCache;
fCache = 0;
BuildCache();
}
printf("Top volume is %s. Master volume is %sn", fTopVolume->GetName(),
fMasterVolume->GetName());
}
//-----------------------------------------------------------------------------
void TGeoManager::SelectTrackingMedia()
{
// Define different tracking media.
printf("List of materials :n");
Int_t nmat = fMaterials->GetSize();
if (!nmat) {printf(" No materials !n"); return;}
Int_t *media = new Int_t[nmat];
memset(media, 0, nmat*sizeof(Int_t));
Int_t imedia = 1;
TGeoMaterial *mat, *matref;
mat = (TGeoMaterial*)fMaterials->At(0);
if (mat->GetMedia()) {
for (Int_t i=0; i<nmat; i++) {
mat = (TGeoMaterial*)fMaterials->At(i);
mat->Print();
}
return;
}
mat->SetMedia(imedia);
media[0] = imedia++;
mat->Print();
for (Int_t i=0; i<nmat; i++) {
mat = (TGeoMaterial*)fMaterials->At(i);
for (Int_t j=0; j<i; j++) {
matref = (TGeoMaterial*)fMaterials->At(j);
if (mat->IsEq(matref)) {
mat->SetMedia(media[j]);
break;
}
if (j==(i-1)) {
// different material
mat->SetMedia(imedia);
media[i] = imedia++;
mat->Print();
}
}
}
}
//-----------------------------------------------------------------------------
void TGeoManager::CheckPoint(Double_t x, Double_t y, Double_t z, Option_t *option)
{
// Classify a given point. See TGeoChecker::CheckPoint().
GetGeomPainter()->CheckPoint(x,y,z,option);
}
//-----------------------------------------------------------------------------
void TGeoManager::CheckGeometry(Option_t * /*option*/)
{
// Instanciate a TGeoChecker object and investigates the geometry according to
// option. Not implemented yet.
// check shapes first
fTopNode->CheckShapes();
}
//-----------------------------------------------------------------------------
void TGeoManager::UpdateCurrentPosition(Double_t * /*nextpoint*/)
{
// Computes and changes the current node according to the new position.
// Not implemented.
}
//-----------------------------------------------------------------------------
ULong_t TGeoManager::SizeOf(TGeoNode * /*node*/, Option_t * /*option*/)
{
// computes the total size in bytes of the branch starting with node.
// The option can specify if all the branch has to be parsed or only the node
return 0;
}
//______________________________________________________________________________
void TGeoManager::Streamer(TBuffer &R__b)
{
// Stream an object of class TGeoManager.
if (R__b.IsReading()) {
TGeoManager::Class()->ReadBuffer(R__b, this);
fIsGeomReading = kTRUE;
CloseGeometry();
fStreamVoxels = kFALSE;
fIsGeomReading = kFALSE;
} else {
TGeoManager::Class()->WriteBuffer(R__b, this);
}
}
//______________________________________________________________________________
Int_t TGeoManager::Export(const char *filename, const char *name, Option_t *option)
{
// Export this geometry on filename with a key=name
// By default the geometry is saved without the voxelisation info.
// Use option 'v" to save the voxelisation info.
TFile f(filename,"recreate");
if (f.IsZombie()) return 0;
char keyname[256];
if (name) strcpy(keyname,name);
if (strlen(keyname) == 0) strcpy(keyname,GetName());
TString opt = option;
opt.ToLower();
if (opt.Contains("v")) fStreamVoxels = kTRUE;
Int_t nbytes = Write(keyname);
fStreamVoxels = kFALSE;
return nbytes;
}
//______________________________________________________________________________
TGeoManager *TGeoManager::Import(const char *filename, const char *name, Option_t * /*option*/)
{
//static function
//Import in memory from filename the geometry with key=name.
//if name="" (default), the first TGeoManager object in the file is returned.
//Note that this function deletes the current gGeoManager (if one)
//before importing the new object.
TFile f(filename);
if (f.IsZombie()) return 0;
if (gGeoManager) delete gGeoManager;
gGeoManager = 0;
if (name && strlen(name) > 0) {
gGeoManager = (TGeoManager*)f.Get(name);
return gGeoManager;
} else {
TIter next(f.GetListOfKeys());
TKey *key;
while ((key = (TKey*)next())) {
if (strcmp(key->GetClassName(),"TGeoManager") != 0) continue;
gGeoManager = (TGeoManager*)key->ReadObj();
return gGeoManager;
}
}
return 0;
}
ROOT page - Class index - Top of the page
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.