]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - FMD/AliFMD.cxx
Changes to compile with Root6 on macosx64
[u/mrichter/AliRoot.git] / FMD / AliFMD.cxx
index 0f404539c22b25408ab6f8ccc42b93dd6167cbb9..9f950a956cba442687465bfe4bc5e96e1eb5fe67 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 <TGeoGlobalMagField.h> // ROOT_TGeoGlobalMagField
+#include <TGeoManager.h>        // ROOT_TGeoManager
 #include <TRotMatrix.h>                // ROOT_TRotMatrix
 #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 <AliDigitizationInput.h>      // ALIRUNDIGITIZER_H
 #include <AliLoader.h>         // ALILOADER_H
 #include <AliRun.h>            // ALIRUN_H
 #include <AliMC.h>             // ALIMC_H
 //#endif
 // #include "AliFMDGeometryBuilder.h"
 #include "AliFMDRawWriter.h"   // ALIFMDRAWWRITER_H
+#include "AliFMDRawReader.h"   // ALIFMDRAWREADER_H
 #include "AliTrackReference.h" 
 #include "AliFMDStripIndex.h"
+#include "AliFMDEncodedEdx.h"
+#include "AliFMDParameters.h"
+#include "AliFMDReconstructor.h"
 
 //____________________________________________________________________
 ClassImp(AliFMD)
@@ -141,7 +146,7 @@ AliFMD::AliFMD()
   fHits        = 0;
   fDigits      = 0;
   fIshunt      = 0;
-  fBad         = new TClonesArray("AliFMDHit");
+  // fBad         = new TClonesArray("AliFMDHit");
 }
 
 //____________________________________________________________________
@@ -158,15 +163,15 @@ AliFMD::AliFMD(const char *name, const char *title)
   // Standard constructor for Forward Multiplicity Detector
   //
   AliFMDDebug(10, ("\tStandard CTOR"));
-  fBad         = new TClonesArray("AliFMDHit");
+  // fBad         = new TClonesArray("AliFMDHit");
   
   // Initialise Hit array
-  HitsArray();
-  gAlice->GetMCApp()->AddHitList(fHits);
+  // HitsArray();
+  // gAlice->GetMCApp()->AddHitList(fHits);
 
   // (S)Digits for the detectors disk
-  DigitsArray();
-  SDigitsArray();
+  // DigitsArray();
+  // SDigitsArray();
   
   // CHC: What is this?
   fIshunt = 0;
@@ -261,8 +266,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
@@ -426,6 +431,60 @@ void AliFMD::CreateMaterials()
 
 }
 
+#if 0
+//____________________________________________________________________
+void  
+AliFMD::SetTrackingParameters(Int_t imed, 
+                             Float_t gamma,                 
+                             Float_t electron, 
+                             Float_t neutral_hadron, 
+                             Float_t charged_hadron, 
+                             Float_t muon,
+                             Float_t electron_bremstrahlung, 
+                             Float_t muon__bremstrahlung, 
+                             Float_t electron_delta,
+                             Float_t muon_delta,
+                             Float_t muon_pair,
+                             Int_t   annihilation, 
+                             Int_t   bremstrahlung, 
+                             Int_t   compton_scattering, 
+                             Int_t   decay,
+                             Int_t   delta_ray, 
+                             Int_t   hadronic, 
+                             Int_t   energy_loss, 
+                             Int_t   multiple_scattering, 
+                             Int_t   pair_production, 
+                             Int_t   photon_production, 
+                             Int_t   rayleigh_scattering)
+{
+  // Disabled by request of FCA, kept for reference only
+  if (!TVirtualMC::GetMC()) return;
+  TArrayI& idtmed = *(GetIdtmed());
+  Int_t    iimed  = idtmed[imed];
+  // TVirtualMC::GetMC()->Gstpar(iimed, "CUTGAM",      gamma);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "CUTELE",      electron);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "CUTNEU",      neutral_hadron);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "CUTHAD",      charged_hadron);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "CUTMUO",      muon);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "BCUTE",       electron_bremstrahlung);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "BCUTM",       muon__bremstrahlung);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "DCUTE",       electron_delta);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "DCUTM",       muon_delta);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "PPCUTM",      muon_pair);
+  // TVirtualMC::GetMC()->Gstpar(iimed, "ANNI",        Float_t(annihilation));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "BREM",        Float_t(bremstrahlung));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "COMP",        Float_t(compton_scattering));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "DCAY",        Float_t(decay));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "DRAY",        Float_t(delta_ray));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "HADR",        Float_t(hadronic));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "LOSS",        Float_t(energy_loss));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "MULS",        Float_t(multiple_scattering));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "PAIR",        Float_t(pair_production));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "PHOT",        Float_t(photon_production));
+  // TVirtualMC::GetMC()->Gstpar(iimed, "RAYL",        Float_t(rayleigh_scattering));
+}
+#endif
+
 //____________________________________________________________________
 void  
 AliFMD::Init()
@@ -435,9 +494,28 @@ AliFMD::Init()
   AliFMDDebug(1, ("Initialising FMD detector object"));
   TVirtualMC*      mc     = TVirtualMC::GetMC();
   AliFMDGeometry*  fmd    = AliFMDGeometry::Instance();
-  const TArrayI&   actGeo = fmd->ActiveIds();
+  TArrayI          actGeo = fmd->ActiveIds();
+  bool             valid  = true;
+  if (actGeo.fN <= 0) valid = false;
+  else { 
+    for (int i = 0; i < actGeo.fN; i++) {
+      if (actGeo[i] < 0) { 
+       valid = false;
+       break;
+      }
+    }
+  }
+  if (!valid) { 
+    AliFMDDebug(1, ("Extracting geometry info from loaded geometry"));
+    fmd->ExtractGeomInfo();
+    actGeo = fmd->ActiveIds();
+  }
   TArrayI          actVmc(actGeo.fN);
   for (Int_t i = 0; i < actGeo.fN; i++) {
+    if (actGeo[i] < 0) { 
+      AliError(Form("Invalid id: %d", actGeo[i]));
+      continue;
+    }
     TGeoVolume *sens = gGeoManager->GetVolume(actGeo[i]);
     if (!sens) {
       AliError(Form("No TGeo volume for sensitive volume ID=%d",actGeo[i]));
@@ -460,7 +538,7 @@ AliFMD::FinishEvent()
   // 
   if (AliLog::GetDebugLevel("FMD", "AliFMD") < 10) return;
   if (fBad && fBad->GetEntries() > 0) {
-    AliWarning((Form("EndEvent", "got %d 'bad' hits", fBad->GetEntries())));
+    AliWarning(Form("got %d 'bad' hits", fBad->GetEntries()));
     TIter next(fBad);
     AliFMDHit* hit;
     while ((hit = static_cast<AliFMDHit*>(next()))) hit->Print("D");
@@ -652,18 +730,21 @@ 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);
+  // TVirtualMC::GetMC()->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);
-  
+  AliTrackReference* trackRef = 
+    AddTrackReference(mcApplication->GetCurrentTrackNumber(), 
+                     AliTrackReference::kFMD); 
+  UInt_t stripId = AliFMDStripIndex::Pack(detector,ring,sector,strip);  
+  UInt_t dedx    = AliFMDEncodedEdx::Encode(edep, l);
   
+  trackRef->SetUserId((dedx << 19) | stripId);
+
   
   return hit;
 }
@@ -696,14 +777,16 @@ 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,
+                        UShort_t       nrefs,
+                        Int_t*         refs)
 {
   // add a real digit - as coming from data
   // 
@@ -718,12 +801,14 @@ AliFMD::AddDigitByFields(UShort_t detector,
   //    count3    ADC count (a 10-bit word), or -1 if not used
   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)",
+  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, nrefs));
+  new (a[fNdigits++]) 
+    AliFMDDigit(detector, ring, sector, strip, 
+               count1, count2, count3, count4, nrefs, refs);
   
 }
 
@@ -770,7 +855,7 @@ AliFMD::AddSDigitByFields(UShort_t       detector,
                          Short_t        count4, 
                          UShort_t       ntot, 
                          UShort_t       nprim,
-                         const TArrayI& refs)
+                         Int_t*         refs)
 {
   // add a summable digit
   // 
@@ -788,6 +873,11 @@ AliFMD::AddSDigitByFields(UShort_t       detector,
   TClonesArray& a = *(SDigitsArray());
   // AliFMDDebug(0, ("Adding sdigit # %d", fNsdigits));
   
+  AliFMDDebug(15, ("Adding sdigit # %5d/%5d for FMD%d%c[%2d,%3d]"
+                  "=(%d,%d,%d,%d) with %d tracks %d primaries (%p)",
+                  fNsdigits-1, a.GetEntriesFast(),
+                  detector, ring, sector, strip, 
+                  count1, count2, count3, count4, ntot, nprim, refs));
   new (a[fNsdigits++]) 
     AliFMDSDigit(detector, ring, sector, strip, edep, 
                 count1, count2, count3, count4, ntot, nprim, refs);
@@ -812,6 +902,8 @@ AliFMD::HitsArray()
   if (!fHits) { 
     fHits = new TClonesArray("AliFMDHit", 1000);
     fNhits = 0;
+    if (gAlice && gAlice->GetMCApp() && gAlice->GetMCApp()->GetHitLists()) 
+      gAlice->GetMCApp()->AddHitList(fHits);
   }
   return fHits;
 }
@@ -853,7 +945,7 @@ AliFMD::Hits2Digits()
   // 
   AliFMDHitDigitizer digitizer(this, AliFMDHitDigitizer::kDigits);
   digitizer.Init();
-  digitizer.Exec("");
+  digitizer.Digitize("");
 }
 
 //____________________________________________________________________
@@ -865,13 +957,13 @@ AliFMD::Hits2SDigits()
   // 
   AliFMDHitDigitizer digitizer(this, AliFMDHitDigitizer::kSDigits);
   digitizer.Init();
-  digitizer.Exec("");
+  digitizer.Digitize("");
 }
 
   
 //____________________________________________________________________
 AliDigitizer* 
-AliFMD::CreateDigitizer(AliRunDigitizer* manager) const
+AliFMD::CreateDigitizer(AliDigitizationInput* digInput) const
 {
   // Create a digitizer object 
   
@@ -879,14 +971,14 @@ AliFMD::CreateDigitizer(AliRunDigitizer* manager) const
   AliFMDBaseDigitizer* digitizer = 0;
   
 #ifdef USE_SSDIGITIZER
-  digitizer = new AliFMDSSDigitizer(manager);
+  digitizer = new AliFMDSSDigitizer(digInput);
 #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);
+  digitizer = new AliFMDDigitizer(digInput);
 #endif
   return digitizer;
 }
@@ -907,6 +999,46 @@ 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();
+  if (AliLog::GetDebugLevel("FMD", 0) > 5) {
+    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;
+}
+
 
 //====================================================================
 //