ITS/AliITSDigits2RecPoints.C

   1 #if !defined(__CINT__) || defined(__MAKECINT__)
   2
   3 #include "iostream.h"
   4 #include "TDatetime.h"
   5 #include "STEER/AliRun.h"
   6 #include "STEER/AliRunDigitizer.h"
   7 #include "ITS/AliITSDigitizer.h"
   8 #include "ITS/AliITS.h"
   9 #include "ITS/AliITSDetType.h"
  10 #include "ITS/AliITSresponseSDD.h"
  11 #include "TStopwatch.h"
  12
  13 #endif
  14
  15 TFile* AccessFile(TString inFile="galice.root", TString acctype="R");
  16 void writeAR(TFile * fin, TFile *fou);
  17 Int_t ChangeITSDefaults(TFile *hitfile,AliITS *ITS,TString opt="");
  18 #define DEBUG
  19 Int_t AliITSDigits2RecPoints(TString digFile="galice.root",
  20                             TString recFile="galice.root",TString opt=""){
  21     // Standard ITS Digits to RecPoints.
  22
  23     // Dynamically link some shared libs
  24     if (gClassTable->GetID("AliRun") < 0) {
  25         gROOT->LoadMacro("loadlibs.C");
  26         loadlibs();
  27     } // end if
  28
  29     // Connect the Root Galice file containing Geometry, Kine and Hits
  30
  31     TFile *digfile = 0;   // pointer to input file.
  32     TFile *recfile = 0;  // possible output file for TreeD
  33     if(recFile.CompareTo(digFile) == 0) { //write output to same file as input.
  34         digfile = AccessFile(digFile,"U");  // input file open for update.
  35     }else{ // different output file then input file.
  36         digfile = AccessFile(digFile,"R"); // input file open as read only
  37         recfile = new TFile(recFile,"NEW");
  38     } // end if sdigFile == hitFile.
  39 /*
  40     AliITS *ITS = (AliITS*)gAlice->GetDetector("ITS");
  41     if (!ITS) {
  42         cerr<<"AliITSHits2DigitsDefault.C : AliITS object not found on file"
  43             << endl;
  44         return 3;
  45     }  // end if !ITS
  46     if(!(ITS->GetITSgeom())){
  47         cerr << " AliITSgeom not found. Can't digitize with out it." << endl;
  48         return 4;
  49     } // end if
  50
  51     ChangeITSDefaults(digfile,ITS,opt);
  52 */
  53     // write the AliRun object to the output file if different from input file.
  54     if(recfile) writeAR(digfile,recfile);
  55     if(recfile){ recfile->Close(); recfile = 0;}
  56 //    if(digfile){ digfile->Close(); digfile = 0;}
  57 //    delete gAlice;
  58 //    gAlice = 0;
  59
  60     TStopwatch timer;
  61
  62 #ifdef DEBUG
  63     cout << "Creating reconstructed points from digits for the ITS..." << endl;
  64 #endif
  65 //    const char *nulptr=0;
  66     AliITSreconstruction *itsr = new AliITSreconstruction(gAlice);
  67     if(digFile.CompareTo(recFile)!=0) {
  68 #ifdef DEBUG
  69         cout << itsr << " filename="  << recFile << " compare="<<
  70             digFile.CompareTo(recFile) << endl;
  71 #endif
  72         itsr->SetOutputFile(recFile);
  73     } // end if
  74     timer.Start();
  75     itsr->Init();
  76     itsr->Exec();
  77     timer.Stop();
  78     timer.Print();
  79     delete itsr;
  80 //    delete gAlice;
  81 //    gAlice = 0;
  82     if(digfile){ digfile->Close(); digfile = 0;}
  83     if(recfile){ recfile->Close(); recfile = 0;}
  84     return 0;
  85 }
  86 //______________________________________________________________________
  87 TFile * AccessFile(TString FileName, TString acctype){
  88     // Function used to open the input file and fetch the AliRun object
  89
  90     TFile *retfil = 0;
  91     TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(FileName);
  92     if(file) {
  93         file->Close();
  94         delete file;
  95         file = 0;
  96     } // end if file
  97     if(acctype.Contains("U")){
  98         file = new TFile(FileName,"UPDATE");
  99     } // end if open for update
 100     if(acctype.Contains("N") && !file){
 101         file = new TFile(FileName,"RECREATE");
 102     } // end if open a new file
 103     if(!file) file = new TFile(FileName,"READ");   // default readonly
 104     if (!file->IsOpen()) {
 105         cerr << "Can't open " << FileName << " !" << endl;
 106         return retfil;
 107     } // end if error opeing file
 108
 109     // Get AliRun object from file or return if not on file
 110     if (gAlice) {delete gAlice; gAlice = 0;}
 111     gAlice = (AliRun*)file->Get("gAlice");
 112     if (!gAlice) {
 113         cerr << "AliRun object not found on file "<< FileName << "!" << endl;
 114         file->Close();  // close file and return error.
 115         return retfil;
 116     } // end if !gAlice
 117     return file;
 118 }
 119 //______________________________________________________________________
 120 void writeAR(TFile * fin, TFile *fou) {
 121     TDirectory *current = gDirectory;
 122 /*
 123     TTree *TeOld;
 124     TTree *TeNew;
 125     AliHeader *alhe = new AliHeader();
 126     TeOld = (TTree*)fin->Get("TE");
 127     TeOld->SetBranchAddress("Header",&alhe);
 128     TeOld->SetBranchStatus("*",1);
 129     fou->cd();
 130     TeNew = TeOld->CloneTree();
 131     TeNew->Write(0,TObject::kOverwrite);
 132 */
 133     fou->cd();
 134     gAlice->TreeE()->SetBranchStatus("*",1);
 135     gAlice->TreeE()->Write(0,TObject::kOverwrite);
 136     gAlice->Write(0,TObject::kOverwrite);
 137     current->cd();
 138 //    delete alhe;
 139 #ifdef DEBUG
 140     cout << "AliRun object written to file" << endl;
 141 #endif
 142 }
 143 //______________________________________________________________________
 144 Int_t ChangeITSDefaults(TFile *hitfile,AliITS *ITS,TString opt){
 145
 146     TDatime *ct0 = new TDatime(2002,04,26,00,00,00);
 147     TDatime ct = hitfile->GetCreationDate();
 148
 149     if(ct0->GetDate()>ct.GetDate()){
 150         // For old files, must change SDD noise.
 151         AliITSresponseSDD *resp1 = (AliITSresponseSDD*)ITS->DetType(1)->
 152             GetResponseModel();
 153         resp1 = new AliITSresponseSDD();
 154         ITS->SetResponseModel(1,resp1);
 155         cout << "Changed response class for SDD:" << endl;
 156         resp1->Print();
 157     } // end if
 158
 159     if(opt.Contains("Dubna")){
 160         AliITSresponseSPDdubna *resp0 = new AliITSresponseSPDdubna();
 161         if(ITS->DetType(0)->GetResponseModel() !=0){
 162             delete ((AliITSresponse*)ITS->DetType(0)->GetResponseModel());
 163             ITS->DetType(0)->ResponseModel(0);
 164         } // end if
 165         ITS->DetType(0)->ResponseModel(resp0);
 166         AliITSsegmentationSPD *seg0 = (AliITSsegmentationSPD*)ITS->
 167             DetType(0)->GetSegmentationModel();
 168         AliITSsimulationSPDdubna *sim0 = new AliITSsimulationSPDdubna(seg0,
 169                                                                       resp0);
 170         if(ITS->DetType(0)->GetSimulationModel() !=0){
 171             delete ((AliITSsimulation*)ITS->DetType(0)->GetSimulationModel());
 172             ITS->DetType(0)->SimulationModel(0);
 173         } // end if
 174         ITS->DetType(0)->SimulationModel(sim0);
 175     } // end if Dubna
 176 }