1 #if !defined(__CINT__) || defined(__MAKECINT__)
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"
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="");
19 Int_t AliITSHits2Digits(TString hitFile = "galice.root",
20 TString digFile = "galiceD.root",TString opt=""){
21 // Standeard ITS Hits to Digits, excluding creation of SDigits.
23 // Dynamically link some shared libs
24 if (gClassTable->GetID("AliRun") < 0) {
25 gROOT->LoadMacro("loadlibs.C");
29 // Connect the Root Galice file containing Geometry, Kine and Hits
31 TFile *hitfile = 0; // pointer to input file.
32 TFile *digfile = 0; // possible output file for TreeD
33 if(digFile.CompareTo(hitFile) == 0){// write output to same file as input.
34 hitfile = AccessFile(hitFile,"U"); // input file open for update.
35 }else{ // different output file then input file.
36 hitfile = AccessFile(hitFile,"R"); // input file open as read only
37 // open output file and create TreeR on it
38 digfile = gAlice->InitTreeFile("D",digFile);
39 } // end if digFile == hitFile.
41 AliITS *ITS = (AliITS*)gAlice->GetDetector("ITS");
43 cerr<<"AliITSHits2DigitsDefault.C : AliITS object not found on file"
47 if(!(ITS->GetITSgeom())){
48 cerr << " AliITSgeom not found. Can't digitize with out it." << endl;
52 ChangeITSDefaults(hitfile,ITS,opt);
53 // write the AliRun object to the output file if different from input file.
54 if(digfile) writeAR(hitfile,digfile);
58 Int_t evNumber2 = gAlice->GetEventsPerRun();
60 for(Int_t nevent = evNumber1; nevent < evNumber2; nevent++){
61 // cout<<"Producing Digits for event n."<<nevent<<endl;
62 gAlice->GetEvent(nevent);
63 if(!gAlice->TreeD() && digfile == 0){
64 cout << "Having to create the Digits Tree." << endl;
65 gAlice->MakeTree("D");
66 } // end if creating digits tree
67 if(digfile) gAlice->MakeTree("D",digfile);
69 ITS->SetTreeAddress();
75 cout << digFile << " size =" << digfile->GetSize() << endl;
77 cout << hitFile << " size =" << hitfile->GetSize() << endl;
78 } // end if sdigfile!=0
80 delete gAlice; // digfile is closed by deleting gAlice if != hitfile.
82 hitfile->Close(); hitfile = 0;
84 //______________________________________________________________________
85 TFile * AccessFile(TString FileName, TString acctype){
86 // Function used to open the input file and fetch the AliRun object
89 TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(FileName);
95 if(acctype.Contains("U")){
96 file = new TFile(FileName,"UPDATE");
97 } // end if open for update
98 if(acctype.Contains("N") && !file){
99 file = new TFile(FileName,"RECREATE");
100 } // end if open a new file
101 if(!file) file = new TFile(FileName,"READ"); // default readonly
102 if (!file->IsOpen()) {
103 cerr << "Can't open " << FileName << " !" << endl;
105 } // end if error opeing file
107 // Get AliRun object from file or return if not on file
108 if (gAlice) {delete gAlice; gAlice = 0;}
109 gAlice = (AliRun*)file->Get("gAlice");
111 cerr << "AliRun object not found on file "<< FileName << "!" << endl;
112 file->Close(); // close file and return error.
117 //______________________________________________________________________
118 void writeAR(TFile * fin, TFile *fou) {
119 TDirectory *current = gDirectory;
122 AliHeader *alhe = new AliHeader();
123 TeOld = (TTree*)fin->Get("TE");
124 TeOld->SetBranchAddress("Header",&alhe);
125 TeOld->SetBranchStatus("*",1);
127 TeNew = TeOld->CloneTree();
128 TeNew->Write(0,TObject::kOverwrite);
129 gAlice->Write(0,TObject::kOverwrite);
133 cout << "AliRun object written to file" << endl;
136 //______________________________________________________________________
137 Int_t ChangeITSDefaults(TFile *hitfile,AliITS *ITS,TString opt){
139 TDatime *ct0 = new TDatime(2002,04,26,00,00,00);
140 TDatime ct = hitfile->GetCreationDate();
142 if(ct0->GetDate()>ct.GetDate()){
143 // For old files, must change SDD noise.
144 AliITSresponseSDD *resp1 = (AliITSresponseSDD*)ITS->DetType(1)->
146 resp1 = new AliITSresponseSDD();
147 ITS->SetResponseModel(1,resp1);
148 cout << "Changed response class for SDD:" << endl;
152 if(opt.Contains("Dubna")){
153 AliITSresponseSPDdubna *resp0 = new AliITSresponseSPDdubna();
154 if(ITS->DetType(0)->GetResponseModel() !=0){
155 delete ((AliITSresponse*)ITS->DetType(0)->GetResponseModel());
156 ITS->DetType(0)->ResponseModel(0);
158 ITS->DetType(0)->ResponseModel(resp0);
159 AliITSsegmentationSPD *seg0 = (AliITSsegmentationSPD*)ITS->
160 DetType(0)->GetSegmentationModel();
161 AliITSsimulationSPDdubna *sim0 = new AliITSsimulationSPDdubna(seg0,
163 if(ITS->DetType(0)->GetSimulationModel() !=0){
164 delete ((AliITSsimulation*)ITS->DetType(0)->GetSimulationModel());
165 ITS->DetType(0)->SimulationModel(0);
167 ITS->DetType(0)->SimulationModel(sim0);