The present commit corresponds to an important change in the way the
[u/mrichter/AliRoot.git] / FMD / AliFMD.cxx
index 1919114..37a2820 100644 (file)
 // These files are not in the same directory, so there's no reason to
 // ask the preprocessor to search in the current directory for these
 // files by including them with `#include "..."' 
-#include <cmath>                // __CMATH__
+#include <TBrowser.h>          // ROOT_TBrowser
 #include <TClonesArray.h>      // ROOT_TClonesArray
-#include <TGeometry.h>         // ROOT_TGeomtry
-#include <TNode.h>             // ROOT_TNode
-#include <TXTRU.h>             // ROOT_TXTRU
+#include <TGeoGlobalMagField.h> // ROOT_TGeoGlobalMagField
+#include <TGeoManager.h>        // ROOT_TGeoManager
 #include <TRotMatrix.h>                // ROOT_TRotMatrix
-#include <TTUBE.h>             // ROOT_TTUBE
 #include <TTree.h>             // ROOT_TTree
-#include <TBrowser.h>          // ROOT_TBrowser
-#include <TVirtualMC.h>                // ROOT_TVirtualMC
 #include <TVector2.h>           // ROOT_TVector2 
-#include <TGeoManager.h>        // ROOT_TGeoManager
+#include <TVirtualMC.h>                // ROOT_TVirtualMC
+#include <cmath>                // __CMATH__
 
 #include <AliRunDigitizer.h>   // ALIRUNDIGITIZER_H
 #include <AliLoader.h>         // ALILOADER_H
 #include "AliFMDDetector.h"    // ALIFMDDETECTOR_H
 #include "AliFMDRing.h"                // ALIFMDRING_H
 #include "AliFMDDigitizer.h"   // ALIFMDDIGITIZER_H
-#include "AliFMDSDigitizer.h"  // ALIFMDSDIGITIZER_H
+#include "AliFMDHitDigitizer.h"        // ALIFMDSDIGITIZER_H
+// #define USE_SSDIGITIZER 
+//#ifdef USE_SSDIGITIZER
+//# include "AliFMDSSDigitizer.h"      // ALIFMDSDIGITIZER_H
+//#endif
 // #include "AliFMDGeometryBuilder.h"
 #include "AliFMDRawWriter.h"   // ALIFMDRAWWRITER_H
-#include "AliFMDPoints.h"       // ALIFMDPOINTS_H
+#include "AliFMDRawReader.h"   // ALIFMDRAWREADER_H
+#include "AliTrackReference.h" 
+#include "AliFMDStripIndex.h"
+#include "AliFMDParameters.h"
+#include "AliFMDReconstructor.h"
 
 //____________________________________________________________________
 ClassImp(AliFMD)
@@ -260,8 +265,8 @@ void AliFMD::CreateMaterials()
   Double_t density          = 0;
   Double_t radiationLength  = 0;
   Double_t absorbtionLength = 999;
-  Int_t    fieldType        = gAlice->Field()->Integ();     // Field type 
-  Double_t maxField         = gAlice->Field()->Max();     // Field max.
+  Int_t    fieldType        = ((AliMagF*)TGeoGlobalMagField::Instance()->GetField())->Integ();     // Field type 
+  Double_t maxField         = ((AliMagF*)TGeoGlobalMagField::Instance()->GetField())->Max();     // Field max.
   Double_t maxBending       = 0;     // Max Angle
   Double_t maxStepSize      = 0.001; // Max step size 
   Double_t maxEnergyLoss    = 1;     // Max Delta E
@@ -468,227 +473,6 @@ AliFMD::FinishEvent()
 }
 
 
-//====================================================================
-//
-// Graphics and event display
-//
-//____________________________________________________________________
-void 
-AliFMD::BuildGeometry()
-{
-  //
-  // Build simple ROOT TNode geometry for event display. With the new
-  // geometry modeller, TGeoManager, this seems rather redundant. 
-  AliFMDDebug(10, ("\tCreating a simplified geometry"));
-
-  AliFMDGeometry* fmd = AliFMDGeometry::Instance();
-  
-  static TXTRU*     innerShape = 0;
-  static TXTRU*     outerShape = 0;
-  static TObjArray* innerRot   = 0;
-  static TObjArray* outerRot   = 0;
-
-  if (!innerShape || !outerShape) {
-    // Make the shapes for the modules 
-    for (Int_t i = 0; i < 2; i++) {
-      AliFMDRing* r = 0;
-      switch (i) {
-      case 0: r = fmd->GetRing('I'); break;
-      case 1: r = fmd->GetRing('O'); break;
-      }
-      if (!r) {
-       AliError(Form("no ring found for i=%d", i));
-       return;
-      }
-      Double_t    siThick  = r->GetSiThickness();
-      const Int_t knv      = r->GetNVerticies();
-      Double_t    theta    = r->GetTheta();
-      Int_t       nmod     = r->GetNModules();
-      
-      TXTRU* shape = new TXTRU(r->GetName(), r->GetTitle(), "void", knv, 2);
-      for (Int_t j = 0; j < knv; j++) {
-       TVector2* vv = r->GetVertex(knv - 1 - j);
-       shape->DefineVertex(j, vv->X(), vv->Y());
-      }
-      shape->DefineSection(0, -siThick / 2, 1, 0, 0);
-      shape->DefineSection(1, +siThick / 2, 1, 0, 0);
-      shape->SetLineColor(kYellow); //PH kYellow is the default line color in FMD
-      
-      TObjArray* rots = new TObjArray(nmod);
-      for (Int_t j = 0; j < nmod; j++) {
-       Double_t th = (j + .5) * theta * 2;
-       TString name(Form("FMD_ring_%c_rot_%02d", r->GetId(), j));
-       TString title(Form("FMD Ring %c Rotation # %d", r->GetId(), j));
-       TRotMatrix* rot = new TRotMatrix(name.Data(), title.Data(),
-                                        90, th, 90, fmod(90+th,360), 0, 0);
-       rots->AddAt(rot, j);
-      }
-      
-      switch (r->GetId()) {
-      case 'i':
-      case 'I': innerShape = shape; innerRot = rots; break;
-      case 'o':
-      case 'O': outerShape = shape; outerRot = rots; break;
-      }
-    }
-  }
-  
-  TNode* top = gAlice->GetGeometry()->GetNode("alice");
-  
-  for (Int_t i = 1; i <= 3; i++) {
-    AliFMDDetector* det = fmd->GetDetector(i);
-    if (!det) {
-      Warning("BuildGeometry", "FMD%d seems to be disabled", i);
-      continue;
-    }
-    Double_t w  = 0;
-    Double_t rh = det->GetRing('I')->GetHighR();
-    Char_t   id = 'I';
-    if (det->GetRing('O')) {
-      w  = TMath::Abs(det->GetRingZ('O') - det->GetRingZ('I'));
-      id = (TMath::Abs(det->GetRingZ('O')) 
-           > TMath::Abs(det->GetRingZ('I')) ? 'O' : 'I');
-      rh = det->GetRing('O')->GetHighR();
-    }
-    w += (det->GetRing(id)->GetModuleSpacing() +
-         det->GetRing(id)->GetSiThickness());
-    TShape* shape = new TTUBE(det->GetName(), det->GetTitle(), "void",
-                             det->GetRing('I')->GetLowR(), rh, w / 2);
-    Double_t z = (det->GetRingZ('I') - w / 2);
-    if (z > 0) z += det->GetRing(id)->GetModuleSpacing();
-    top->cd();
-    TNode* node = new TNode(det->GetName(), det->GetTitle(), shape, 
-                           0, 0, z, 0);
-    fNodes->Add(node);
-    
-    for (Int_t j = 0; j < 2; j++) {
-      AliFMDRing* r      = 0;
-      TShape*     rshape = 0;
-      TObjArray*  rots   = 0;
-      switch (j) {
-      case 0: 
-       r = det->GetRing('I'); rshape = innerShape; rots = innerRot; break;
-      case 1: 
-       r = det->GetRing('O'); rshape = outerShape; rots = outerRot; break;
-      }
-      if (!r) continue;
-      
-      Double_t    siThick  = r->GetSiThickness();
-      Int_t       nmod     = r->GetNModules();
-      Double_t    modspace = r->GetModuleSpacing();
-      Double_t    rz       = - (z - det->GetRingZ(r->GetId()));
-      
-      for (Int_t k = 0; k < nmod; k++) {
-       node->cd();
-       Double_t    offz    = (k % 2 == 1 ? modspace : 0);
-       TRotMatrix* rot     = static_cast<TRotMatrix*>(rots->At(k));
-       TString name(Form("%s%c_module_%02d", det->GetName(), r->GetId(),k));
-       TString title(Form("%s%c Module %d", det->GetName(), r->GetId(),k));
-       TNode* mnod = new TNode(name.Data(), title.Data(), rshape, 
-                               0, 0, rz - siThick / 2 
-                               + TMath::Sign(offz,z), rot);
-       mnod->SetLineColor(kYellow); //PH kYellow is the default line color in FMD
-       fNodes->Add(mnod);
-      } // for (Int_t k = 0 ; ...)
-    } // for (Int_t j = 0 ; ...)
-  } // for (Int_t i = 1 ; ...)
-}
-
-//____________________________________________________________________
-void 
-AliFMD::LoadPoints(Int_t /* track */) 
-{
-  // Store x, y, z of all hits in memory for display. 
-  // 
-  // Normally, the hits are drawn using TPolyMarker3D - however, that
-  // is not very useful for the FMD.  Therefor, this member function
-  // is overloaded to make TMarker3D, via the class AliFMDPoints.
-  // AliFMDPoints is a local class. 
-  //
-  if (!fHits) {
-    AliError(Form("fHits == 0. Name is %s",GetName()));
-    return;
-  }
-  Int_t nHits = fHits->GetEntriesFast();
-  if (nHits == 0) {
-    return;
-  }
-  Int_t tracks = gAlice->GetMCApp()->GetNtrack();
-  if (fPoints == 0) fPoints = new TObjArray(2 * tracks);
-
-  // Get geometry 
-  AliFMDGeometry* geom = AliFMDGeometry::Instance();
-  geom->Init();
-  geom->InitTransformations();
-
-  // Now make markers for each hit  
-  // AliInfo(Form("Drawing %d hits (have %d points) for track %d", 
-  //              nHits, fPoints->GetEntriesFast(), track));
-  for (Int_t ihit = 0; ihit < nHits; ihit++) {
-    AliFMDHit* hit = static_cast<AliFMDHit*>(fHits->At(ihit));
-    if (!hit) continue;
-    Double_t edep    = hit->Edep();
-    Double_t m       = hit->M();
-    Double_t poverm  = (m == 0 ? 0 : hit->P());
-    Double_t absQ    = TMath::Abs(hit->Q());
-    Bool_t   bad     = kFALSE;
-    // This `if' is to debug abnormal energy depositions.  We trigger on
-    // p/m approx larger than or equal to a MIP, and a large edep - more 
-    // than 1 keV - a MIP is 100 eV. 
-    if (edep > absQ * absQ && poverm > 1) bad = kTRUE;
-
-    AliFMDPoints* p1 = new AliFMDPoints(hit, kRed); //PH kRed is the default marker color in FMD
-    // AliPoints* p1 = new AliPoints();
-    // p1->SetMarkerColor(GetMarkerColor());
-    // p1->SetMarkerSize(GetMarkerSize());
-    // p1->SetPoint(0, hit->X(), hit->Y(), hit->Z());
-    p1->SetDetector(this);
-    p1->SetParticle(hit->GetTrack());
-    fPoints->AddAt(p1, hit->GetTrack());
-    if (bad) {
-      p1->SetMarkerColor(4);
-      // p1->SetMarkerSize(2 * GetMarkerSize());
-    }
-    
-    Double_t x, y, z;
-    geom->Detector2XYZ(hit->Detector(), hit->Ring(), hit->Sector(), 
-                      hit->Strip(), x, y, z);
-    AliFMDPoints* p = new AliFMDPoints(hit, 3);
-    // AliPoints* p = new AliPoints();
-    // p->SetMarkerColor(3);
-    // p->SetMarkerSize(GetMarkerSize());
-    // p->SetPoint(0, x, y, z);
-    p->SetDetector(this);
-    p->SetParticle(hit->GetTrack());
-    p->SetXYZ(x, y, z);
-    p->SetMarkerColor(3);
-    fPoints->AddAt(p, tracks+hit->GetTrack());
-    if (bad) {
-      p->SetMarkerColor(5);
-      // p->SetMarkerSize(2 * GetMarkerSize());
-    }
-    // AliInfo(Form("Adding point at %d", tracks+hit->GetTrack()));
-  }
-}
-
-//____________________________________________________________________
-void 
-AliFMD::DrawDetector()
-{
-  // Draw a shaded view of the Forward multiplicity detector.  This
-  // isn't really useful anymore. 
-  AliFMDDebug(10, ("\tDraw detector"));
-}
-
-//____________________________________________________________________
-Int_t 
-AliFMD::DistancetoPrimitive(Int_t, Int_t)
-{
-  // Calculate the distance from the mouse to the FMD on the screen
-  // Dummy routine.
-  //
-  return 9999;
-}
 
 //====================================================================
 //
@@ -756,6 +540,13 @@ AliFMD::SetHitsAddressBranch(TBranch *b)
   // Set the TClonesArray to read hits into. 
   b->SetAddress(&fHits);
 }
+//____________________________________________________________________
+void 
+AliFMD::SetSDigitsAddressBranch(TBranch *b)
+{
+  // Set the TClonesArray to read hits into. 
+  b->SetAddress(&fSDigits);
+}
 
 //____________________________________________________________________
 void 
@@ -865,7 +656,19 @@ AliFMD::AddHitByFields(Int_t    track,
   hit = new (a[fNhits]) AliFMDHit(fIshunt, track, detector, ring, sector, 
                                  strip, x, y, z, px, py, pz, edep, pdg, t, 
                                  l, stop);
+  // gMC->AddTrackReference(track, 12);
   fNhits++;
+  
+  //Reference track
+
+  AliMC *mcApplication = (AliMC*)gAlice->GetMCApp();
+  
+  AliTrackReference* trackRef = AddTrackReference(mcApplication->GetCurrentTrackNumber(), AliTrackReference::kFMD); 
+  UInt_t stripId = AliFMDStripIndex::Pack(detector,ring,sector,strip);
+  trackRef->SetUserId(stripId);
+  
+  
+  
   return hit;
 }
 
@@ -897,14 +700,15 @@ AliFMD::AddDigit(Int_t* digits, Int_t*)
 
 //____________________________________________________________________
 void 
-AliFMD::AddDigitByFields(UShort_t detector, 
-                        Char_t   ring, 
-                        UShort_t sector, 
-                        UShort_t strip, 
-                        UShort_t count1, 
-                        Short_t  count2,
-                        Short_t  count3, 
-                        Short_t  count4)
+AliFMD::AddDigitByFields(UShort_t       detector, 
+                        Char_t         ring, 
+                        UShort_t       sector, 
+                        UShort_t       strip, 
+                        UShort_t       count1, 
+                        Short_t        count2,
+                        Short_t        count3, 
+                        Short_t        count4,
+                        const TArrayI& refs)
 {
   // add a real digit - as coming from data
   // 
@@ -920,11 +724,13 @@ AliFMD::AddDigitByFields(UShort_t detector,
   TClonesArray& a = *(DigitsArray());
   
   new (a[fNdigits++]) 
-    AliFMDDigit(detector, ring, sector, strip, count1, count2, count3, count4);
-  AliFMDDebug(15, ("Adding digit # %5d/%5d for FMD%d%c[%2d,%3d]=(%d,%d,%d,%d)",
+    AliFMDDigit(detector, ring, sector, strip, 
+               count1, count2, count3, count4, refs);
+  AliFMDDebug(15, ("Adding digit # %5d/%5d for FMD%d%c[%2d,%3d]"
+                  "=(%d,%d,%d,%d) with %d tracks",
                   fNdigits-1, a.GetEntriesFast(),
                   detector, ring, sector, strip, 
-                  count1, count2, count3, count4));
+                  count1, count2, count3, count4, refs.fN));
   
 }
 
@@ -945,28 +751,33 @@ AliFMD::AddSDigit(Int_t* digits)
   //    digits[6]  [Short_t]  ADC Count, -1 if not used
   //    digits[7]  [Short_t]  ADC Count, -1 if not used 
   // 
-  AddSDigitByFields(UShort_t(digits[0]),  // Detector #
-                   Char_t(digits[1]),    // Ring ID
-                   UShort_t(digits[2]),  // Sector #
-                   UShort_t(digits[3]),  // Strip #
-                   Float_t(digits[4]),   // Edep
-                   UShort_t(digits[5]),  // ADC Count1 
-                   Short_t(digits[6]),   // ADC Count2 
-                   Short_t(digits[7]),   // ADC Count3 
-                   Short_t(digits[8]));
+  AddSDigitByFields(UShort_t(digits[0]),   // Detector #
+                   Char_t(digits[1]),     // Ring ID
+                   UShort_t(digits[2]),   // Sector #
+                   UShort_t(digits[3]),   // Strip #
+                   Float_t(digits[4]),    // Edep
+                   UShort_t(digits[5]),   // ADC Count1 
+                   Short_t(digits[6]),    // ADC Count2 
+                   Short_t(digits[7]),    // ADC Count3 
+                   Short_t(digits[8]),    // ADC Count4
+                   UShort_t(digits[9]),   // N particles
+                   UShort_t(digits[10])); // N primaries
 }
 
 //____________________________________________________________________
 void 
-AliFMD::AddSDigitByFields(UShort_t detector, 
-                         Char_t   ring, 
-                         UShort_t sector, 
-                         UShort_t strip, 
-                         Float_t  edep,
-                         UShort_t count1, 
-                         Short_t  count2,
-                         Short_t  count3, 
-                         Short_t  count4)
+AliFMD::AddSDigitByFields(UShort_t       detector, 
+                         Char_t         ring, 
+                         UShort_t       sector, 
+                         UShort_t       strip, 
+                         Float_t        edep,
+                         UShort_t       count1, 
+                         Short_t        count2,
+                         Short_t        count3, 
+                         Short_t        count4, 
+                         UShort_t       ntot, 
+                         UShort_t       nprim,
+                         const TArrayI& refs)
 {
   // add a summable digit
   // 
@@ -982,10 +793,11 @@ AliFMD::AddSDigitByFields(UShort_t detector,
   //    count3    ADC count (a 10-bit word), or -1 if not used
   //
   TClonesArray& a = *(SDigitsArray());
+  // AliFMDDebug(0, ("Adding sdigit # %d", fNsdigits));
   
   new (a[fNsdigits++]) 
     AliFMDSDigit(detector, ring, sector, strip, edep, 
-                count1, count2, count3, count4);
+                count1, count2, count3, count4, ntot, nprim, refs);
 }
 
 //____________________________________________________________________
@@ -1046,15 +858,9 @@ AliFMD::Hits2Digits()
   // Create AliFMDDigit's from AliFMDHit's.  This is done by making a
   // AliFMDDigitizer, and executing that code.
   // 
-  Warning("Hits2Digits", "Try not to use this method.\n"
-         "Instead, use AliSimulator");
-  AliRunDigitizer* manager = new AliRunDigitizer(1, 1);
-  manager->SetInputStream(0, "galice.root");
-  manager->SetOutputFile("H2Dfile");
-  
-  /* AliDigitizer* dig =*/ CreateDigitizer(manager);
-  manager->Exec("");
-  delete manager;
+  AliFMDHitDigitizer digitizer(this, AliFMDHitDigitizer::kDigits);
+  digitizer.Init();
+  digitizer.Exec("");
 }
 
 //____________________________________________________________________
@@ -1064,9 +870,9 @@ AliFMD::Hits2SDigits()
   // Create AliFMDSDigit's from AliFMDHit's.  This is done by creating
   // an AliFMDSDigitizer object, and executing it. 
   // 
-  AliFMDSDigitizer* digitizer = new AliFMDSDigitizer("galice.root");
-  digitizer->Exec("");
-  delete digitizer;
+  AliFMDHitDigitizer digitizer(this, AliFMDHitDigitizer::kSDigits);
+  digitizer.Init();
+  digitizer.Exec("");
 }
 
   
@@ -1075,7 +881,20 @@ AliDigitizer*
 AliFMD::CreateDigitizer(AliRunDigitizer* manager) const
 {
   // Create a digitizer object 
-  AliFMDDigitizer* digitizer = new AliFMDDigitizer(manager);
+  
+  /* This is what we probably _should_ do */
+  AliFMDBaseDigitizer* digitizer = 0;
+  
+#ifdef USE_SSDIGITIZER
+  digitizer = new AliFMDSSDigitizer(manager);
+#else 
+  /* This is what we actually do, and will work */
+#if 0
+  AliInfo("SDigit->Digit conversion not really supported, "
+         "doing Hit->Digit conversion instead");
+#endif
+  digitizer = new AliFMDDigitizer(manager);
+#endif
   return digitizer;
 }
 
@@ -1095,6 +914,45 @@ AliFMD::Digits2Raw()
   writer.Exec();
 }
 
+//====================================================================
+//
+// Raw data reading 
+//
+//__________________________________________________________________
+Bool_t
+AliFMD::Raw2SDigits(AliRawReader* reader) 
+{
+  // Turn digits into raw data. 
+  // 
+  // This uses the class AliFMDRawWriter to do the job.   Please refer
+  // to that class for more information. 
+  AliFMDParameters::Instance()->Init();
+  MakeTree("S");
+  MakeBranch("S");
+  
+  TClonesArray*       sdigits = SDigits();
+  AliFMDReconstructor rec;
+  
+  // The two boolean arguments
+  //   Make sdigits instead of digits 
+  //   Subtract the pedestal off the signal
+  rec.Digitize(reader, sdigits);
+  // 
+  // Bool_t ret = fmdReader.ReadAdcs(sdigits, kTRUE, kTRUE);
+  // sdigits->ls();
+  UShort_t ns = sdigits->GetEntriesFast();
+  for (UShort_t i = 0; i < ns; i++) 
+    sdigits->At(i)->Print("pl");
+  
+  AliFMDDebug(1, ("Got a total of %d SDigits", ns));
+
+  fLoader->TreeS()->Fill();
+  ResetSDigits();
+  fLoader->WriteSDigits("OVERWRITE");
+
+  return kTRUE;
+}
+
 
 //====================================================================
 //