* about the suitability of this software for any purpose. It is *
* provided "as is" without express or implied warranty. *
**************************************************************************/
-
/* $Id$ */
-
+/** @file AliFMDGeometry.cxx
+ @author Christian Holm Christensen <cholm@nbi.dk>
+ @date Mon Mar 27 12:40:37 2006
+ @brief Geometry mananger for the FMD
+*/
//____________________________________________________________________
//
// Forward Multiplicity Detector based on Silicon wafers.
#include "AliFMD2.h" // ALIFMD2_H
#include "AliFMD3.h" // ALIFMD2_H
#include "AliRecPoint.h" // ALIRECPOINT_H
-#include "AliLog.h" // ALIRECPOINT_H
+#include "AliFMDDebug.h" // ALILOG_H
#include <TVector3.h> // ROOT_TVector3
-#include <TMatrix.h> // ROOT_TMatrix
-#include <TParticle.h> // ROOT_TParticle
+// #include <TMatrix.h> // ROOT_TMatrix
+// #include <TParticle.h> // ROOT_TParticle
#include <Riostream.h>
+#include "AliFMDGeometryBuilder.h"
+// #include <TArrayI.h>
+#include <TGeoManager.h>
+#include <TGeoVolume.h>
+#include <TGeoNode.h>
+#include <TMath.h>
+static Int_t FindNodeDepth(const char* name, const char* volname);
+
//====================================================================
ClassImp(AliFMDGeometry)
AliFMDGeometry*
AliFMDGeometry::Instance()
{
+ // Return (newly created) singleton instance
if (!fgInstance) fgInstance = new AliFMDGeometry;
return fgInstance;
}
//____________________________________________________________________
AliFMDGeometry::AliFMDGeometry()
- : AliGeometry("FMD", "Forward multiplicity")
+ : AliGeometry("FMD", "Forward multiplicity"),
+ fIsInitialized(kFALSE),
+ fInner(0),
+ fOuter(0),
+ fFMD1(0),
+ fFMD2(0),
+ fFMD3(0),
+ fUseFMD1(kFALSE),
+ fUseFMD2(kFALSE),
+ fUseFMD3(kFALSE),
+ fIsInitTrans(kFALSE),
+ fBuilder(0),
+ fDetectorOff(0),
+ fModuleOff(0),
+ fRingOff(0),
+ fSectorOff(0),
+ fActive(2),
+ fDetailed(kFALSE),
+ fUseAssembly(kFALSE)
{
- fUseFMD1 = kTRUE;
- fUseFMD2 = kTRUE;
- fUseFMD3 = kTRUE;
+ // PROTECTED
+ // Default constructor
+ fUseFMD1 = kTRUE;
+ fUseFMD2 = kTRUE;
+ fUseFMD3 = kTRUE;
+ fDetailed = kTRUE;
+ fUseAssembly = kTRUE;
fInner = new AliFMDRing('I');
fOuter = new AliFMDRing('O');
fFMD1 = new AliFMD1(fInner);
fFMD2 = new AliFMD2(fInner, fOuter);
fFMD3 = new AliFMD3(fInner, fOuter);
fIsInitialized = kFALSE;
+ fActive.Reset(-1);
+}
+
+//____________________________________________________________________
+AliFMDGeometry::AliFMDGeometry(const AliFMDGeometry& other)
+ : AliGeometry(other),
+ fIsInitialized(other.fIsInitialized),
+ fInner(other.fInner),
+ fOuter(other.fOuter),
+ fFMD1(other.fFMD1),
+ fFMD2(other.fFMD2),
+ fFMD3(other.fFMD3),
+ fUseFMD1(other.fUseFMD1),
+ fUseFMD2(other.fUseFMD2),
+ fUseFMD3(other.fUseFMD3),
+ fIsInitTrans(other.fIsInitTrans),
+ fBuilder(other.fBuilder),
+ fDetectorOff(other.fDetectorOff),
+ fModuleOff(other.fModuleOff),
+ fRingOff(other.fRingOff),
+ fSectorOff(other.fSectorOff),
+ fActive(other.fActive),
+ fDetailed(other.fDetailed),
+ fUseAssembly(other.fUseAssembly)
+{
+ // PROTECTED
+ // Copy constructor
+}
+
+
+
+//____________________________________________________________________
+AliFMDGeometry&
+AliFMDGeometry::operator=(const AliFMDGeometry& other)
+{
+ // PROTECTED
+ // Assignment operator
+ fUseFMD1 = other.fUseFMD1;
+ fUseFMD2 = other.fUseFMD2;
+ fUseFMD3 = other.fUseFMD3;
+ fFMD1 = other.fFMD1;
+ fFMD2 = other.fFMD2;
+ fFMD3 = other.fFMD3;
+ fInner = other.fInner;
+ fOuter = other.fOuter;
+ fIsInitialized = other.fIsInitialized;
+ return *this;
}
//____________________________________________________________________
void
AliFMDGeometry::Init()
{
+ // Initialize the the singleton if not done so already
if (fIsInitialized) return;
fInner->Init();
fOuter->Init();
fFMD3->Init();
}
+//____________________________________________________________________
+void
+AliFMDGeometry::InitTransformations(Bool_t force)
+{
+ // Find all local <-> global transforms
+ if (force) fIsInitTrans = kFALSE;
+ if (fIsInitTrans) return;
+ if (!gGeoManager) {
+ AliError("No TGeoManager defined");
+ return;
+ }
+ AliFMDDebug(1, ("Initialising transforms for FMD geometry"));
+ if (fFMD1) fFMD1->InitTransformations();
+ if (fFMD2) fFMD2->InitTransformations();
+ if (fFMD3) fFMD3->InitTransformations();
+ fIsInitTrans = kTRUE;
+}
+
+//____________________________________________________________________
+void
+AliFMDGeometry::Build()
+{
+ // Build the geometry
+ if (!fBuilder) fBuilder = new AliFMDGeometryBuilder(fDetailed);
+ fBuilder->SetDetailed(fDetailed);
+ fBuilder->UseAssembly(fUseAssembly);
+ fBuilder->Exec();
+}
+
+//____________________________________________________________________
+void
+AliFMDGeometry::SetActive(Int_t* active, Int_t n)
+{
+ // Set active volumes
+ fActive.Set(n);
+ for (Int_t i = 0; i < n; i++) {
+ AliFMDDebug(1, ("Active vol id # %d: %d", i, active[i]));
+ fActive[i] = active[i];
+ }
+}
+
+//____________________________________________________________________
+void
+AliFMDGeometry::AddActive(Int_t active)
+{
+ // Add an active volume
+ Int_t n = fActive.fN;
+ fActive.Set(n+1);
+ fActive[n] = active;
+}
+
+//____________________________________________________________________
+Bool_t
+AliFMDGeometry::IsActive(Int_t vol) const
+{
+ // Check if a volume is active
+ for (Int_t i = 0; i < fActive.fN; i++)
+ if (fActive[i] == vol) return kTRUE;
+ return kFALSE;
+}
+
//____________________________________________________________________
AliFMDDetector*
AliFMDGeometry::GetDetector(Int_t i) const
{
+ // Get the ith detector. i should be one of 1, 2, or 3. If an
+ // invalid value is passed, 0 (NULL) is returned.
switch (i) {
case 1: return fUseFMD1 ? static_cast<AliFMDDetector*>(fFMD1) : 0;
case 2: return fUseFMD2 ? static_cast<AliFMDDetector*>(fFMD2) : 0;
AliFMDRing*
AliFMDGeometry::GetRing(Char_t i) const
{
+ // Get the ith ring. i should be one of 'I' or 'O' (case
+ // insensitive). If an invalid parameter is passed, 0 (NULL) is
+ // returned.
switch (i) {
case 'I':
case 'i': return fInner;
void
AliFMDGeometry::Enable(Int_t i)
{
+ // Enable the ith detector. i should be one of 1, 2, or 3
switch (i) {
case 1: fUseFMD1 = kTRUE; break;
case 2: fUseFMD2 = kTRUE; break;
void
AliFMDGeometry::Disable(Int_t i)
{
+ // Disable the ith detector. i should be one of 1, 2, or 3
switch (i) {
case 1: fUseFMD1 = kFALSE; break;
case 2: fUseFMD2 = kFALSE; break;
//____________________________________________________________________
void
AliFMDGeometry::Detector2XYZ(UShort_t detector,
- Char_t ring,
- UShort_t sector,
- UShort_t strip,
- Double_t& x,
- Double_t& y,
- Double_t& z) const
+ Char_t ring,
+ UShort_t sector,
+ UShort_t strip,
+ Double_t& x,
+ Double_t& y,
+ Double_t& z) const
{
+ // Translate detector coordinates (detector, ring, sector, strip) to
+ // spatial coordinates (x, y, z) in the master reference frame of
+ // ALICE.
AliFMDDetector* det = GetDetector(detector);
- if (!det) return;
+ if (!det) {
+ AliWarning(Form("Unknown detector %d", detector));
+ return;
+ }
det->Detector2XYZ(ring, sector, strip, x, y, z);
}
+//____________________________________________________________________
+Bool_t
+AliFMDGeometry::XYZ2Detector(Double_t x,
+ Double_t y,
+ Double_t z,
+ UShort_t& detector,
+ Char_t& ring,
+ UShort_t& sector,
+ UShort_t& strip) const
+{
+ // Translate spatial coordinates (x,y,z) in the master reference frame of
+ // ALICE to the detector coordinates (detector, ring, sector,
+ // strip). Note, that if this method is to be used in
+ // reconstruction or the like, then the input z-coordinate should be
+ // corrected for the events interactions points z-coordinate, like
+ // geom->XYZ2Detector(x,y,z-ipz,d,r,s,t);
+ AliFMDDetector* det = 0;
+ detector = 0;
+ for (int i = 1; i <= 3; i++) {
+ det = GetDetector(i);
+ if (!det) continue;
+ if (det->XYZ2Detector(x, y, z, ring, sector, strip)) {
+ detector = det->GetId();
+ return kTRUE;
+ }
+ }
+ return kFALSE;
+}
+
+//____________________________________________________________________
+Bool_t
+AliFMDGeometry::XYZ2REtaPhiTheta(Double_t x, Double_t y,
+ Double_t z,
+ Double_t& r, Double_t& eta,
+ Double_t& phi, Double_t& theta)
+{
+
+ // Service function to convert Cartisean XYZ to r, eta, phi, and theta.
+ //
+ // Note, that the z input should be corrected for the vertex location
+ // if needed.
+ //
+ // x Cartisean X coordinate
+ // y Cartisean Y coordinate
+ // z Cartisean Z coordinate
+ // r On return, the radius
+ // eta On return, the pseudo-rapidity
+ // phi On return, the azimuthal angle
+ // theta On return, the polar angle;
+ //
+ // Return:
+ // kFALSE in case of problems.
+
+ if (x == 0 && y == 0 && z == 0) return kFALSE;
+
+ // Correct for vertex offset.
+ phi = TMath::ATan2(y, x);
+ r = TMath::Sqrt(y * y + x * x);
+ theta = TMath::ATan2(r, z);
+ eta = -TMath::Log(TMath::Tan(theta / 2));
+
+ return kTRUE;
+}
+
//____________________________________________________________________
void
AliFMDGeometry::GetGlobal(const AliRecPoint* p,
TVector3& pos,
- TMatrix& /* mat */) const
+ TMatrixF& /* mat */) const
{
+ // Get the global coordinates cooresponding to the reconstructed
+ // point p. The coordiates is returned in the 3-vector pos passed
+ // to this member function. The matrix mat is used for rotations.
GetGlobal(p, pos);
}
void
AliFMDGeometry::GetGlobal(const AliRecPoint* p, TVector3& pos) const
{
+ // Get the global coordinates cooresponding to the reconstructed
+ // point p. The coordiates is returned in the 3-vector pos passed
+ // to this member function. Note, as AliRecPoint only has places for
+ // 3 indicies, it is assumed that the ring hit is an inner ring -
+ // which obviously needn't be the case. This makes the member
+ // function pretty darn useless.
+ // FIXME: Implement this function to work with outer rings too.
Double_t x, y, z;
TVector3 local;
p->GetLocalPosition(local);
Bool_t
AliFMDGeometry::Impact(const TParticle* /* particle */) const
{
+ // Return true, if the particle will hit the active detector
+ // elements, and false if not. Should be used for fast
+ // simulations. Note, that the function currently return false
+ // always.
+ // FIXME: Implement this function.
return kFALSE;
}
+//____________________________________________________________________
+void
+AliFMDGeometry::SetAlignableVolumes() const
+{
+ // Declare alignable volumes
+ for (Int_t d = 1; d <= 3; d++)
+ if (GetDetector(d)) GetDetector(d)->SetAlignableVolumes();
+}
+
+
+//____________________________________________________________________
+void
+AliFMDGeometry::ExtractGeomInfo()
+{
+ // Check the volume depth of some nodes, get the active volume
+ // numbers, and so forth.
+ //
+ // TODO: Here, we should actually also get the parameters of the
+ // shapes, like the verticies of the polygon shape that makes up the
+ // silicon sensor, the strip pitch, the ring radii, the z-positions,
+ // and so on - that is, all the geometric information we need for
+ // futher processing, such as simulation, digitization,
+ // reconstruction, etc.
+ Int_t detectorDepth = FindNodeDepth("F1MT_1", "ALIC");
+ Int_t ringDepth = FindNodeDepth(Form("FITV_%d", int('I')), "ALIC");
+ Int_t moduleDepth = FindNodeDepth("FIBH_0", "ALIC");
+ Int_t sectorDepth = FindNodeDepth("FISC_1", "ALIC");
+ fActive.Set(0);
+ fActive.Reset(-1);
+ AliFMDDebug(1, ("Geometry depths:\n"
+ " Sector: %d\n"
+ " Module: %d\n"
+ " Ring: %d\n"
+ " Detector: %d",
+ sectorDepth, moduleDepth, ringDepth, detectorDepth));
+ if (sectorDepth < 0 && moduleDepth < 0) {
+ fDetailed = kFALSE;
+ fSectorOff = -1;
+ fModuleOff = -1;
+ fRingOff = 0;
+ fDetectorOff = (ringDepth - detectorDepth);
+ TGeoVolume* actiVol = gGeoManager->GetVolume("FIAC");
+ TGeoVolume* actoVol = gGeoManager->GetVolume("FOAC");
+ if (actiVol) AddActive(actiVol->GetNumber());
+ if (actiVol) AddActive(actoVol->GetNumber());
+ }
+ else if (sectorDepth < 0) {
+ fDetailed = kFALSE;
+ fSectorOff = -1;
+ fModuleOff = 1;
+ fRingOff = (moduleDepth - ringDepth) + 1;
+ fDetectorOff = (moduleDepth - detectorDepth) + 1;
+ TGeoVolume* modiVol = gGeoManager->GetVolume("FIMO");
+ TGeoVolume* modoVol = gGeoManager->GetVolume("FOMO");
+ if (modiVol) AddActive(modiVol->GetNumber());
+ if (modoVol) AddActive(modoVol->GetNumber());
+ }
+ else {
+ Int_t stripDepth = FindNodeDepth("FIST_1", "ALIC");
+ fDetailed = kTRUE;
+ fSectorOff = (stripDepth - sectorDepth);
+ fModuleOff = (moduleDepth >= 0 ? (stripDepth - moduleDepth) : -1);
+ fRingOff = (stripDepth - ringDepth);
+ fDetectorOff = (stripDepth - detectorDepth );
+ TGeoVolume* striVol = gGeoManager->GetVolume("FIST");
+ TGeoVolume* stroVol = gGeoManager->GetVolume("FOST");
+ if (striVol) AddActive(striVol->GetNumber());
+ if (stroVol) AddActive(stroVol->GetNumber());
+ }
+ AliFMDDebug(1, ("Geometry offsets:\n"
+ " Sector: %d\n"
+ " Module: %d\n"
+ " Ring: %d\n"
+ " Detector: %d",
+ fSectorOff, fModuleOff, fRingOff, fDetectorOff));
+}
+
+
+//____________________________________________________________________
+static Int_t
+CheckNodes(TGeoNode* node, const char* name, Int_t& lvl)
+{
+ // If there's no node here.
+ if (!node) return -1;
+ // Check if it this one
+ TString sname(name);
+ if (sname == node->GetName()) return lvl;
+
+ // Check if the node is an immediate daugther
+ TObjArray* nodes = node->GetNodes();
+ if (!nodes) return -1;
+ // Increase the level, and search immediate sub nodes.
+ lvl++;
+ TGeoNode* found = static_cast<TGeoNode*>(nodes->FindObject(name));
+ if (found) return lvl;
+
+ // Check the sub node, if any of their sub-nodes match.
+ for (Int_t i = 0; i < nodes->GetEntries(); i++) {
+ TGeoNode* sub = static_cast<TGeoNode*>(nodes->At(i));
+ if (!sub) continue;
+ // Recurive check
+ if (CheckNodes(sub, name, lvl) >= 0) return lvl;
+ }
+ // If not found, decrease the level
+ lvl--;
+ return -1;
+}
+//____________________________________________________________________
+Int_t
+FindNodeDepth(const char* name, const char* volname)
+{
+ // Find the depth of a node
+ TGeoVolume* vol = gGeoManager->GetVolume(volname);
+ if (!vol) {
+ std::cerr << "No top volume defined" << std::endl;
+ return -1;
+ }
+
+ TGeoIterator next(vol);
+ TGeoNode* node = 0;
+ TString sName(name);
+ while ((node = next())) {
+ if (sName == node->GetName()) {
+ //std::cout << "Found node " << node->GetName() << " at level "
+ // << next.GetLevel() << std::endl;
+ return next.GetLevel();
+ }
+ }
+ return -1;
+#if 0
+ TObjArray* nodes = vol->GetNodes();
+ if (!nodes) {
+ std::cerr << "No nodes in top volume" << std::endl;
+ return -1;
+ }
+ TIter next(nodes);
+ TGeoNode* node = 0;
+ Int_t lvl = 0;
+ while ((node = static_cast<TGeoNode*>(next())))
+ if (CheckNodes(node, name, lvl) >= 0) return lvl;
+ return -1;
+#endif
+}
+
//____________________________________________________________________
//
// EOF