]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRD.cxx
Simulation of RAW data (T.Kuhr)
[u/mrichter/AliRoot.git] / TRD / AliTRD.cxx
index 4a8421072796a51c315af46a24dc0d1025c468b9..736a417d3d64bf058ff56057a2022eae86158059 100644 (file)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-Revision 1.36  2002/02/11 14:25:27  cblume
-Geometry update, compressed hit structure
-
-Revision 1.35  2001/11/14 12:08:44  cblume
-Remove unneccessary header files
-
-Revision 1.34  2001/11/14 10:50:45  cblume
-Changes in digits IO. Add merging of summable digits
-
-Revision 1.33  2001/11/06 17:19:41  cblume
-Add detailed geometry and simple simulator
-
-Revision 1.32  2001/10/08 06:57:33  hristov
-Branches for  TRD digits are created only during the digitisation
-
-Revision 1.31  2001/08/30 09:30:30  hristov
-The split level of branches is set to 99
-
-Revision 1.30  2001/05/28 17:07:58  hristov
-Last minute changes; ExB correction in AliTRDclusterizerV1; taking into account of material in G10 TEC frames and material between TEC planes (C.Blume,S.Sedykh)
-
-Revision 1.29  2001/05/21 16:45:47  hristov
-Last minute changes (C.Blume)
-
-Revision 1.28  2001/05/16 14:57:27  alibrary
-New files for folders and Stack
-
-Revision 1.27  2001/05/08 07:05:02  hristov
-Loop variable declared once (HP, Sun)
-
-Revision 1.26  2001/05/07 08:03:22  cblume
-Generate also hits in the amplification region
-
-Revision 1.25  2001/03/13 09:30:35  cblume
-Update of digitization. Moved digit branch definition to AliTRD
-
-Revision 1.24  2001/01/26 19:56:49  hristov
-Major upgrade of AliRoot code
-
-Revision 1.23  2000/11/01 14:53:20  cblume
-Merge with TRD-develop
-
-Revision 1.17.2.6  2000/10/15 23:29:08  cblume
-Introduced more detailed geometry for the display
-
-Revision 1.17.2.5  2000/10/06 16:49:46  cblume
-Made Getters const
-
-Revision 1.17.2.4  2000/10/04 16:34:57  cblume
-Replace include files by forward declarations
-
-Revision 1.17.2.3  2000/09/22 14:45:17  cblume
-Included changes for the tracking
-
-Revision 1.17.2.2  2000/09/18 13:25:13  cblume
-Included LoadPoints() method to display the TR photons
-
-Revision 1.22  2000/10/02 21:28:19  fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.21  2000/06/09 11:10:07  cblume
-Compiler warnings and coding conventions, next round
-
-Revision 1.20  2000/06/08 18:32:57  cblume
-Make code compliant to coding conventions
-
-Revision 1.19  2000/06/07 16:25:37  cblume
-Try to remove compiler warnings on Sun and HP
-
-Revision 1.18  2000/05/08 16:17:27  cblume
-Merge TRD-develop
-
-Revision 1.21  2000/06/09 11:10:07  cblume
-Compiler warnings and coding conventions, next round
-
-Revision 1.20  2000/06/08 18:32:57  cblume
-Make code compliant to coding conventions
-
-Revision 1.19  2000/06/07 16:25:37  cblume
-Try to remove compiler warnings on Sun and HP
-
-Revision 1.18  2000/05/08 16:17:27  cblume
-Merge TRD-develop
-
-Revision 1.17.2.1  2000/05/08 14:28:59  cblume
-Introduced SetPHOShole() and SetRICHhole(). AliTRDrecPoint container is now a TObjArray
-
-Revision 1.17  2000/02/28 19:10:26  cblume
-Include the new TRD classes
-
-Revision 1.16.2.2  2000/02/28 17:53:24  cblume
-Introduce TRD geometry classes
-
-Revision 1.16.2.1  2000/02/28 17:04:19  cblume
-Include functions and data members for AliTRDrecPoint
-
-Revision 1.16  2000/01/19 17:17:35  fca
-Introducing a list of lists of hits -- more hits allowed for detector now
-
-Revision 1.15  1999/11/02 17:04:25  fca
-Small syntax change for HP compiler
-
-Revision 1.14  1999/11/02 16:57:02  fca
-Avoid non ansi warnings on HP compilers
-
-Revision 1.13  1999/11/02 16:35:56  fca
-New version of TRD introduced
-
-Revision 1.12  1999/11/01 20:41:51  fca
-Added protections against using the wrong version of FRAME
-
-Revision 1.11  1999/09/29 09:24:34  fca
-Introduction of the Copyright and cvs Log
-
-*/
+/* $Id$ */
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
@@ -140,35 +24,36 @@ Introduction of the Copyright and cvs Log
 ///////////////////////////////////////////////////////////////////////////////
 
 #include <stdlib.h>
-#include <iostream.h>
+#include <Riostream.h>
 
+#include <TFile.h>
+#include <TGeometry.h>
+#include <TLorentzVector.h>
 #include <TMath.h>
 #include <TNode.h>
-#include <TGeometry.h>
-#include <TTree.h>                                                              
 #include <TPGON.h> 
-#include <TFile.h>
-#include <TROOT.h>
 #include <TParticle.h>
-
-#include "AliRun.h"
+#include <TROOT.h>
+#include <TTree.h>
+#include <TVirtualMC.h>
 #include "AliConst.h"
 #include "AliDigit.h"
+#include "AliLoader.h"
 #include "AliMagF.h"
-#include "AliMC.h"                                                              
+#include "AliRun.h"
 #include "AliTRD.h"
-#include "AliTRDhit.h"
-#include "AliTRDpoints.h"
 #include "AliTRDdigit.h"
 #include "AliTRDdigitizer.h"
-#include "AliTRDclusterizer.h"
-#include "AliTRDgeometryHole.h"
-#include "AliTRDgeometryFull.h"
-#include "AliTRDrecPoint.h"
-#include "AliTRDcluster.h"
 #include "AliTRDdigitsManager.h"
+#include "AliTRDgeometryFull.h"
+#include "AliTRDgeometryHole.h"
+#include "AliTRDhit.h"
+#include "AliTRDpoints.h"
 #include "AliTRDtrackHits.h"  
+#include "AliTRDrawData.h"
+#include "AliTrackReference.h"
+#include "AliMC.h"
 
 ClassImp(AliTRD)
  
@@ -184,9 +69,6 @@ AliTRD::AliTRD()
   fHits          = 0;
   fDigits        = 0;
 
-  fRecPoints     = 0;
-  fNRecPoints    = 0;
-
   fGeometry      = 0;
 
   fGasDensity    = 0;
@@ -216,13 +98,9 @@ AliTRD::AliTRD(const char *name, const char *title)
     exit(1);
   } 
 
-  // Define the TRD geometry according to the FRAME geometry
-  if      (frame->IsVersion() == 0) {
-    // Geometry with hole
-    fGeometry = new AliTRDgeometryHole();
-  }
-  else if (frame->IsVersion() == 1) {
-    // Geometry without hole
+  // Define the TRD geometry
+  if ((frame->IsVersion() == 0) ||
+      (frame->IsVersion() == 1)) {
     fGeometry = new AliTRDgeometryFull();
   }
   else {
@@ -230,17 +108,19 @@ AliTRD::AliTRD(const char *name, const char *title)
     exit(1);
   }
 
+  // Save the geometry
+  TDirectory* saveDir = gDirectory;
+  gAlice->GetRunLoader()->CdGAFile();
+  fGeometry->Write("TRDgeometry");
+  saveDir->cd();
+
   // Allocate the hit array
   fHits           = new TClonesArray("AliTRDhit"     ,405);
-  gAlice->AddHitList(fHits);
+  gAlice->GetMCApp()->AddHitList(fHits);
 
   // Allocate the digits array
   fDigits         = 0;
 
-  // Allocate the rec point array
-  fRecPoints     = new TObjArray(400);
-  fNRecPoints    = 0;
-   
   fIshunt        = 0;
   fGasMix        = 1;
 
@@ -258,7 +138,7 @@ AliTRD::AliTRD(const char *name, const char *title)
 }
 
 //_____________________________________________________________________________
-AliTRD::AliTRD(const AliTRD &trd)
+AliTRD::AliTRD(const AliTRD &trd):AliDetector(trd)
 {
   //
   // Copy constructor
@@ -285,10 +165,10 @@ AliTRD::~AliTRD()
     delete fHits;
     fHits      = 0;
   }
-  if (fRecPoints) {
-    delete fRecPoints;
-    fRecPoints = 0;
-  }
+//  if (fRecPoints) {
+//    delete fRecPoints;
+//    fRecPoints = 0;
+//  }
   if (fTrackHits) {
     delete fTrackHits;
     fTrackHits = 0;
@@ -296,89 +176,30 @@ AliTRD::~AliTRD()
 
 }
 
-//_____________________________________________________________________________
-void AliTRD::AddCluster(Float_t *pos, Int_t *digits, Int_t det, Float_t amp
-                       , Int_t *tracks, Float_t sigmaY2, Int_t iType)
-{
-  //
-  // Add a cluster for the TRD
-  //
-
-  Int_t   pla = fGeometry->GetPlane(det);
-  Int_t   cha = fGeometry->GetChamber(det);
-  Int_t   sec = fGeometry->GetSector(det);
-
-  Float_t padRow  = pos[0]; 
-  Float_t padCol  = pos[1]; 
-  Float_t row0    = fGeometry->GetRow0(pla,cha,sec);
-  Float_t col0    = fGeometry->GetCol0(pla);
-  Float_t rowSize = fGeometry->GetRowPadSize(pla,cha,sec);
-  Float_t colSize = fGeometry->GetColPadSize(pla);
-
-  AliTRDcluster *c = new AliTRDcluster();
-
-  c->SetDetector(det);
-  c->AddTrackIndex(tracks);
-
-  c->SetQ(amp);
-
-  c->SetLocalTimeBin(((Int_t) pos[2]));
-  c->SetY(col0 + padCol * colSize);
-  c->SetZ(row0 + padRow * rowSize);
-  
-  c->SetSigmaY2((sigmaY2 + 1./12.) * colSize*colSize);   
-  c->SetSigmaZ2(rowSize * rowSize / 12.);
-
-  switch (iType) {
-  case 0:
-    c->Set2pad();
-    break;
-  case 1:
-    c->Set3pad();
-    break;
-  case 2:
-    c->Set4pad();
-    break;
-  case 3:
-    c->Set5pad();
-    break;
-  case 4:
-    c->SetLarge();
-    break;
-  };
-
-  fRecPoints->Add(c);
-
-}
-
 //_____________________________________________________________________________
 void AliTRD::Hits2Digits()
 {
   //
   // Create digits
   //
-
-  AliTRDdigitizer *digitizer = new AliTRDdigitizer("TRDdigitizer"
-                                                  ,"TRD digitizer class");
-  digitizer->SetDebug(GetDebug());
-  digitizer->SetEvent(gAlice->GetEvNumber());
-
+  AliTRDdigitizer digitizer("TRDdigitizer","TRD digitizer class");
+  digitizer.SetDebug(GetDebug());
+  
   // Initialization
-  digitizer->InitDetector();
+  digitizer.InitDetector();
     
-  // Create the digits
-  digitizer->MakeDigits();
-  
-  // Write the digits into the input file
-  if (digitizer->MakeBranch(fDigitsFile)) {
-
-    digitizer->WriteDigits();
-
-    // Save the digitizer class in the AliROOT 
-    digitizer->Write();
-
+  if (!fLoader->TreeH()) fLoader->LoadHits("read");
+  fLoader->LoadDigits("recreate");
+  AliRunLoader* runLoader = fLoader->GetRunLoader(); 
+
+  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+    digitizer.Open(runLoader->GetFileName().Data(), iEvent);
+    digitizer.MakeDigits();
+    digitizer.WriteDigits();
   }
 
+  fLoader->UnloadHits();
+  fLoader->UnloadDigits();
 }
 
 //_____________________________________________________________________________
@@ -387,31 +208,32 @@ void AliTRD::Hits2SDigits()
   //
   // Create summable digits
   //
-
-  AliTRDdigitizer *digitizer = new AliTRDdigitizer("TRDdigitizer"
-                                                  ,"TRD digitizer class");
-  digitizer->SetDebug(GetDebug());
-
+  AliTRDdigitizer digitizer("TRDdigitizer","TRD digitizer class");
   // For the summable digits
-  digitizer->SetSDigits(kTRUE);
-  digitizer->SetEvent(gAlice->GetEvNumber());
+  digitizer.SetSDigits(kTRUE);
+  digitizer.SetDebug(GetDebug());
 
   // Initialization
-  digitizer->InitDetector();
+  digitizer.InitDetector();
     
-  // Create the TRD s-digits branch
-  digitizer->MakeDigits();
-  
-  // Write the digits into the input file
-  if (digitizer->MakeBranch(fDigitsFile)) {
-
-    digitizer->WriteDigits();
-
-    // Save the digitizer class in the AliROOT 
-    digitizer->Write();
+  if (!fLoader->TreeH()) fLoader->LoadHits("read");
+  fLoader->LoadSDigits("recreate");
+  AliRunLoader* runLoader = fLoader->GetRunLoader(); 
+
+  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+    digitizer.Open(runLoader->GetFileName().Data(), iEvent);
+    digitizer.MakeDigits();
+    digitizer.WriteDigits();
+  }
 
+  fLoader->UnloadHits();
+  fLoader->UnloadSDigits();
   }
 
+//_____________________________________________________________________________
+AliDigitizer* AliTRD::CreateDigitizer(AliRunDigitizer* manager) const
+{
+  return new AliTRDdigitizer(manager);
 }
 
 //_____________________________________________________________________________
@@ -422,39 +244,68 @@ void AliTRD::SDigits2Digits()
   //
 
    // Create the TRD digitizer
-  AliTRDdigitizer *digitizer = new AliTRDdigitizer("TRDdigitizer"
-                                                  ,"TRD digitizer class");  
-  digitizer->SetDebug(GetDebug());
+  AliTRDdigitizer digitizer("TRDdigitizer","TRD digitizer class");  
+  digitizer.SetDebug(GetDebug());
 
   // Set the parameter
-  digitizer->SetEvent(gAlice->GetEvNumber());
+  digitizer.SetEvent(gAlice->GetEvNumber());
 
   // Initialization
-  digitizer->InitDetector();
+  digitizer.InitDetector();
 
   // Read the s-digits via digits manager
-  AliTRDdigitsManager *sdigitsManager = new AliTRDdigitsManager();
-  sdigitsManager->SetDebug(GetDebug());
-  sdigitsManager->SetSDigits(kTRUE);
-  if (fDigitsFile) {
-    sdigitsManager->Open(fDigitsFile);
-  }
-  sdigitsManager->CreateArrays();
-  sdigitsManager->ReadDigits();
+  AliTRDdigitsManager sdigitsManager;
+  sdigitsManager.SetDebug(GetDebug());
+  sdigitsManager.SetSDigits(kTRUE);
+  sdigitsManager.CreateArrays();
+  
+  if (!fLoader->TreeS()) 
+    if (fLoader->LoadSDigits("read"))
+     {
+       Error("SDigits2Digits","Error while reading SDigits for event %d",gAlice->GetEvNumber());
+       return;
+     }
+  if (!fLoader->TreeS()) return;
+  
+  sdigitsManager.ReadDigits(fLoader->TreeS());
 
   // Add the s-digits to the input list 
-  digitizer->AddSDigitsManager(sdigitsManager);
+  digitizer.AddSDigitsManager(&sdigitsManager);
 
   // Convert the s-digits to normal digits
-  digitizer->SDigits2Digits();
+  digitizer.SDigits2Digits();
 
   // Store the digits
-  if (digitizer->MakeBranch(fDigitsFile)) {
+  if (!fLoader->TreeD()) fLoader->MakeTree("D");
+  if (digitizer.MakeBranch(fLoader->TreeD())){
+    digitizer.WriteDigits();
+  }
+
+}
+
+//_____________________________________________________________________________
+void AliTRD::Digits2Raw() 
+{
+  //
+  // convert digits of the current event to raw data
+  //
 
-    digitizer->WriteDigits();
+  fLoader->LoadDigits();
+  TTree* digits = fLoader->TreeD();
+  if (!digits) {
+    Error("Digits2Raw", "no digits tree");
+    return;
+  }
 
+  AliTRDrawData rawWriter;
+//  rawWriter.SetDebug(2);
+  if (!rawWriter.Digits2Raw(digits)) {
+    Error("AliTRD::Digits2Raw","The raw writer could not load the digits tree");
   }
 
+  fLoader->UnloadDigits();
+
 }
 
 //_____________________________________________________________________________
@@ -590,8 +441,6 @@ void AliTRD::Copy(TObject &trd)
 
   ((AliTRD &) trd).fGasMix      = fGasMix;
   ((AliTRD &) trd).fGeometry    = fGeometry;       
-  ((AliTRD &) trd).fRecPoints   = fRecPoints;
-  ((AliTRD &) trd).fNRecPoints  = fNRecPoints;
   ((AliTRD &) trd).fGasDensity  = fGasDensity;
   ((AliTRD &) trd).fFoilDensity = fFoilDensity;
   ((AliTRD &) trd).fDrawTR      = fDrawTR;
@@ -709,7 +558,7 @@ void AliTRD::CreateMaterials()
   AliMixture(14,"Plexiglas",      apg, zpg, dpg, -3, wpg);
 
   // Gas mixtures
-  Char_t namate[21];
+  Char_t namate[21]="";
   // Xe/CO2-mixture
   // Get properties of Xe 
   gMC->Gfmate((*fIdmate)[4], namate, agm[0], zgm[0], d, radl, absl, buf, nbuf);
@@ -814,6 +663,9 @@ void AliTRD::CreateMaterials()
   // Al of cooling pipes
   AliMedium(24, "Cooling" ,   1, 0, isxfld, sxmgmx
                 , tmaxfd, stemax, deemax, epsil, stmin);
+  // Cu in services
+  AliMedium(25, "Serv-Cu" ,   5, 0, isxfld, sxmgmx
+                , tmaxfd, stemax, deemax, epsil, stmin);
 
   // Save the density values for the TRD absorbtion
   fFoilDensity = dmy;
@@ -825,7 +677,7 @@ void AliTRD::CreateMaterials()
 }
 
 //_____________________________________________________________________________
-void AliTRD::DrawModule()
+void AliTRD::DrawModule() const
 {
   //
   // Draw a shaded view of the Transition Radiation Detector version 0
@@ -884,7 +736,7 @@ void AliTRD::DrawModule()
 }
 
 //_____________________________________________________________________________
-Int_t AliTRD::DistancetoPrimitive(Int_t , Int_t )
+Int_t AliTRD::DistancetoPrimitive(Int_t , Int_t ) const
 {
   //
   // Distance between the mouse and the TRD detector on the screen
@@ -911,16 +763,16 @@ void AliTRD::Init()
     printf("\n");
   }
 
-  if      (fGeometry->IsVersion() == 0) {
-    printf("%s: Geometry for spaceframe with holes initialized\n",ClassName());
-  }
-  else if (fGeometry->IsVersion() == 1) {
-    printf("%s: Geometry for spaceframe without holes initialized\n",ClassName());
+  if (fGeometry->IsVersion() == 1) {
+    printf("%s: Full geometry version initialized\n",ClassName());
     if (fGeometry->GetPHOShole())
       printf("%s: Leave space in front of PHOS free\n",ClassName());
     if (fGeometry->GetRICHhole())
       printf("%s: Leave space in front of RICH free\n",ClassName());
   }
+  else {
+    printf("%s: Not a valid geometry\n",ClassName());
+  }
   
   if (fGasMix == 1) {
     printf("%s: Gas Mixture: 85%% Xe + 15%% CO2\n",ClassName());
@@ -932,7 +784,7 @@ void AliTRD::Init()
 }
 
 //_____________________________________________________________________________
-void AliTRD::LoadPoints(Int_t track)
+void AliTRD::LoadPoints(Int_t )
 {
   //
   // Store x, y, z of all hits in memory.
@@ -955,7 +807,7 @@ void AliTRD::LoadPoints(Int_t track)
   } 
   if (nhits == 0) return;
 
-  Int_t tracks = gAlice->GetNtrack();
+  Int_t tracks = gAlice->GetMCApp()->GetNtrack();
   if (fPoints == 0) fPoints = new TObjArray(tracks);
 
   AliTRDhit *ahit;
@@ -1075,7 +927,7 @@ void AliTRD::LoadPoints(Int_t track)
 }
 
 //_____________________________________________________________________________
-void AliTRD::MakeBranch(Option_t* option, const char *file)
+void AliTRD::MakeBranch(Option_t* option)
 {
   //
   // Create Tree branches for the TRD digits.
@@ -1087,14 +939,14 @@ void AliTRD::MakeBranch(Option_t* option, const char *file)
 
   const char *cD = strstr(option,"D");
 
-  AliDetector::MakeBranch(option,file);
+  AliDetector::MakeBranch(option);
 
   if (fDigits && gAlice->TreeD() && cD) {
-    MakeBranchInTree(gAlice->TreeD(),branchname,&fDigits,buffersize,file);
+    MakeBranchInTree(gAlice->TreeD(),branchname,&fDigits,buffersize,0);
   }    
 
   if (fHitType > 1) {
-    MakeBranch2(option,file); 
+    MakeBranch2(option,0); 
   }
 
 }
@@ -1111,21 +963,6 @@ void AliTRD::ResetDigits()
 
 }
 
-//_____________________________________________________________________________
-void AliTRD::ResetRecPoints()
-{
-  //
-  // Reset number of reconstructed points and the point array
-  //
-
-  if (fRecPoints) {
-    fNRecPoints = 0;
-    Int_t nentr = fRecPoints->GetEntriesFast();
-    for (Int_t i = 0; i < nentr; i++) delete fRecPoints->RemoveAt(i);
-  }
-
-}
-
 //_____________________________________________________________________________
 void AliTRD::SetTreeAddress()
 {
@@ -1133,23 +970,9 @@ void AliTRD::SetTreeAddress()
   // Set the branch addresses for the trees.
   //
 
-  Char_t branchname[15];
-
+  if ( fLoader->TreeH() && (fHits == 0x0))  fHits = new TClonesArray("AliTRDhit",405);
   AliDetector::SetTreeAddress();
 
-  TBranch *branch;
-  TTree   *treeR = gAlice->TreeR();
-
-  if (treeR) {
-    sprintf(branchname,"%scluster",GetName());
-    if (fRecPoints) {
-      branch = treeR->GetBranch(branchname);
-      if (branch) {
-        branch->SetAddress(&fRecPoints);
-      }
-    }
-  }
-
   if (fHitType > 0) {
     SetTreeAddress2();    
   }
@@ -1287,7 +1110,7 @@ AliHit* AliTRD::NextHit()
 }
 
 //_____________________________________________________________________________
-AliHit* AliTRD::FirstHit2(Int_t track)
+AliHit* AliTRD::FirstHit2(Int_t track) 
 {
   //
   // Initializes the hit iterator.
@@ -1299,7 +1122,7 @@ AliHit* AliTRD::FirstHit2(Int_t track)
 
   if (track >= 0) {
     gAlice->ResetHits();
-    gAlice->TreeH()->GetEvent(track);
+    TreeH()->GetEvent(track);
   }
   
   if (fTrackHits) {
@@ -1330,7 +1153,7 @@ AliHit* AliTRD::NextHit2()
 }
 
 //_____________________________________________________________________________
-void AliTRD::MakeBranch2(Option_t *option, const char *file)
+void AliTRD::MakeBranch2(Option_t *option, const char)
 {
   //
   // Create a new branch in the current Root tree.
@@ -1349,45 +1172,14 @@ void AliTRD::MakeBranch2(Option_t *option, const char *file)
  
   if (!fTrackHits) {
     fTrackHits = new AliTRDtrackHits();
-  }
+   }
 
-  if (fTrackHits && gAlice->TreeH() && cH) {
-
-    gAlice->TreeH()->Branch(branchname,"AliTRDtrackHits"
-                                      ,&fTrackHits
-                                      ,fBufferSize,99);
-
-    if (GetDebug() > 1) {
-      printf("<AliTRD::MakeBranch2> Making Branch %s for trackhits\n"
-            ,branchname);
-    }
-
-    const char folder[] = "RunMC/Event/Data";
-
-    if (GetDebug()) {
-      printf("<AliTRD::MakeBranch2> %15s: Publishing %s to %s\n"
-            ,ClassName(),branchname,folder);
-    }
-
-    Publish(folder,&fTrackHits,branchname);
-
-    if (file) {
-      TBranch *b = gAlice->TreeH()->GetBranch(branchname);
-      TDirectory *wd = gDirectory;
-      b->SetFile(file);
-      TIter next(b->GetListOfBranches());
-      while ((b = (TBranch*) next())) {
-        b->SetFile(file);
-      }
-      wd->cd();
-      if (GetDebug() > 1) {
-        printf("<AliTRD::MakeBranch2> Diverting branch %s to file %s\n"
-              ,branchname,file);
-      }
-    }
-
-  }
 
+  if (fTrackHits && TreeH() && cH) 
+   {
+    TreeH()->Branch(branchname,"AliTRDtrackHits",&fTrackHits,fBufferSize,99);
+    Info("MakeBranch2","Making Branch %s for trackhits",branchname);
+   }
 }
 
 //_____________________________________________________________________________
@@ -1396,20 +1188,27 @@ void AliTRD::SetTreeAddress2()
   //
   // Set the branch address for the trackHits tree
   //
+  if (GetDebug())  Info("SetTreeAddress2","");
 
   TBranch *branch;
-
   char branchname[20];
-
   sprintf(branchname,"%s2",GetName());
   
   // Branch address for hit tree
-  TTree *treeH = gAlice->TreeH();
+  TTree *treeH = TreeH();
   if ((treeH) && (fHitType > 0)) {
     branch = treeH->GetBranch(branchname);
-    if (branch) {
-      branch->SetAddress(&fTrackHits);
-    }
+    if (branch) 
+     {
+       branch->SetAddress(&fTrackHits);
+       if (GetDebug())
+         Info("SetTreeAddress2","Success.");
+     }
+    else
+     {
+       if (GetDebug())
+         Info("SetTreeAddress2","Can NOT get the branch %s",branchname);
+     }
   }
 
 }
@@ -1425,13 +1224,13 @@ void AliTRD::AddHit2(Int_t track, Int_t det, Float_t *hits, Int_t q
   Int_t rtrack;
 
   if (fIshunt) {
-    Int_t primary = gAlice->GetPrimary(track);
-    gAlice->Particle(primary)->SetBit(kKeepBit);
+    Int_t primary = gAlice->GetMCApp()->GetPrimary(track);
+    gAlice->GetMCApp()->Particle(primary)->SetBit(kKeepBit);
     rtrack = primary;
   } 
   else {
     rtrack = track;
-    gAlice->FlagTrack(track);
+    gAlice->GetMCApp()->FlagTrack(track);
   }
 
   if ((fTrackHits) && (fHitType > 0)) {
@@ -1773,8 +1572,6 @@ void AliTRD::AddHit2(Int_t track, Int_t det, Float_t *hits, Int_t q
 
 
 
-
-