-void AliITSMerge(Int_t Nfiles=1,const char* file0="galice.root",
- const char* file1="galice_bg.root"){
- //
-
- TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(file0);
- if (file) {file->Close(); delete file;}
- cout << "AliITSMerge" << endl;
- file = new TFile(file0,"UPDATE");
+#if !defined(__CINT__) || defined(__MAKECINT__)
+
+#include "iostream.h"
+#include "TDatetime.h"
+#include "STEER/AliRun.h"
+#include "STEER/AliRunDigitizer.h"
+#include "ITS/AliITSDigitizer.h"
+#include "ITS/AliITS.h"
+#include "ITS/AliITSDetType.h"
+#include "ITS/AliITSresponseSDD.h"
+#include "TStopwatch.h"
+
+#endif
+
+TFile* AccessFile(TString inFile="galice.root", TString acctype="R");
+void writeAR(TFile * fin, TFile *fou);
+Int_t ChangeITSDefaults(TFile *hitfile,AliITS *ITS,TString opt="");
+//#define DEBUG
+Int_t AliITSMerge(TString digFile="galiceMD.root",
+ TString sdigFileSig="galiceS_sig.root",
+ TString sdigFileBg="",TString opt=""){
+ // Standeard ITS SDigits to Digits, with posible merging.
+
+ // Dynamically link some shared libs
+ if (gClassTable->GetID("AliRun") < 0) {
+ gROOT->LoadMacro("loadlibs.C");
+ loadlibs();
+ } // end if
+ // Connect the Root Galice file containing Geometry, Kine and Hits
+ if (gAlice) {delete gAlice; gAlice = 0;}
+
+ TFile *sdigfilesig = 0; // pointer to signal input file.
+ TFile *sdigfilebg = 0; // pointer to background input file.
+ TFile *digfile = 0; // possible output file for TreeD
+
+ // Setup to copy gAlice and the event tree to the output file.
+
+ if(digFile.CompareTo(sdigFileSig)==0){//write output to same file as input.
+ sdigfilesig = AccessFile(sdigFileSig,"U");//input file open for update.
+ }else{ // different output file then input file.
+ sdigfilesig = AccessFile(sdigFileSig,"R");//input file open read only
+ digfile = new TFile(digFile,"NEW");
+ } // end if digFile == hitFile.
+
+ AliITS *ITS = (AliITS*)gAlice->GetDetector("ITS");
+ if (!ITS) {
+ cerr<<"AliITSHits2DigitsDefault.C : AliITS object not found on file"
+ << endl;
+ return 3;
+ } // end if !ITS
+
+ ChangeITSDefaults(sdigfilesig,ITS,opt);
+ // write the AliRun object to the output file if different from input file.
+ if(digfile){
+ writeAR(sdigfilesig,digfile);
+ digfile->Close(); // Manager will open in update mode.
+ digfile = 0;
+ } // end if digfile
+ sdigfilesig->Close();
+ sdigfilesig = 0;
+// delete gAlice; gAlice=0; // there is a problem with deleting gAlice????
+
+ AliRunDigitizer *manager;
+ if(sdigFileBg.CompareTo("")==0) { // do not merge.
+ manager = new AliRunDigitizer(1,1);
+ }else{
+ manager = new AliRunDigitizer(2,1);
+ manager->SetInputStream(0,sdigFileSig.Data());
+ manager->SetInputStream(1,sdigFileBg.Data());
+ } // end if
+ if (digFile.CompareTo(sdigFileSig) !=0) {
+ manager->SetOutputFile(digFile);
+ } // end if
+// manager->SetCopyTreesFromInput(0);
+ AliITSDigitizer *dITS = new AliITSDigitizer(manager);
+ if(opt.Contains("ROI")){
+ cout << "Region of Interest selected" << endl;
+ dITS->SetByRegionOfInterestFlag(1);
+ }else{
+ cout << "Digizing everthing" << endl;
+ dITS->SetByRegionOfInterestFlag(0);
+ } // end if
+
+ TStopwatch timer;
+ timer.Start();
+ manager->Exec("all");
+ timer.Stop();
+ timer.Print();
+
+ if(digfile!=0){
+ cout << digFile << " size =" << digfile->GetSize() << endl;
+ }else if(sdigfilesig!=0){
+ cout << sdigFileSig << " size =" << sdigfilesig->GetSize() << endl;
+ } // end if sdigfile!=0
+
+
+ if(digfile) digfile->Close();
+ if(sdigfilesig) sdigfilesig->Close();
+ if(sdigfilebg) sdigfilebg->Close();
+
+ // There is a problem deleting gAlice. It is related to the destructor
+ // and the fTreeE in AliRun. So for now it is not deleted.
+// delete gAlice; // digfile is closed by deleting gAlice if != hitfile.
+// gAlice = 0;
+ delete manager;
+}
+//______________________________________________________________________
+TFile * AccessFile(TString FileName, TString acctype){
+ // Function used to open the input file and fetch the AliRun object
+
+ TFile *retfil = 0;
+ TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(FileName);
+ if(file) {
+ file->Close();
+ delete file;
+ file = 0;
+ } // end if file
+ if(acctype.Contains("U")){
+ file = new TFile(FileName,"UPDATE");
+ } // end if open for update
+ if(acctype.Contains("N") && !file){
+ file = new TFile(FileName,"RECREATE");
+ } // end if open a new file
+ if(!file) file = new TFile(FileName,"READ"); // default readonly
if (!file->IsOpen()) {
- cerr<<"Can't open "<<file0<<" !" << endl;
- return;
- } // end if !file
- file->ls();
-
+ cerr << "Can't open " << FileName << " !" << endl;
+ return retfil;
+ } // end if error opeing file
+
// Get AliRun object from file or return if not on file
- if (gAlice) delete gAlice;
+ if (gAlice) {delete gAlice; gAlice = 0;}
gAlice = (AliRun*)file->Get("gAlice");
if (!gAlice) {
- cerr << "AliITSMerge.C : AliRun object not found on file" << endl;
- return;
+ cerr << "AliRun object not found on file "<< FileName << "!" << endl;
+ file->Close(); // close file and return error.
+ return retfil;
} // end if !gAlice
+ return file;
+}
+//______________________________________________________________________
+void writeAR(TFile * fin, TFile *fou) {
+ TDirectory *current = gDirectory;
+ TTree *TeOld;
+ TTree *TeNew;
+ AliHeader *alhe = new AliHeader();
+ TeOld = (TTree*)fin->Get("TE");
+ TeOld->SetBranchAddress("Header",&alhe);
+ TeOld->SetBranchStatus("*",1);
+ fou->cd();
+ TeNew = TeOld->CloneTree();
+ TeNew->Write(0,TObject::kOverwrite);
+ gAlice->Write(0,TObject::kOverwrite);
+ current->cd();
+ delete alhe;
+#ifdef DEBUG
+ cout << "AliRun object written to file" << endl;
+#endif
+}
+//______________________________________________________________________
+Int_t ChangeITSDefaults(TFile *hitfile,AliITS *ITS,TString opt){
- if(Nfiles>2) Nfiles = 2;
- if(Nfiles<1) Nfiles = 1;
- AliRunDigitizer *manager = new AliRunDigitizer(Nfiles,1);
- manager->SetInputStream(0,file0);
- if(Nfiles>1) manager->SetInputStream(1,file1);
- manager->SetOutputFile(file0);
- AliITSDigitizer *dITS = new AliITSDigitizer(manager);
- manager->Exec("");
+ TDatime *ct0 = new TDatime(2002,04,26,00,00,00);
+ TDatime ct = hitfile->GetCreationDate();
+
+ if(ct0->GetDate()>ct.GetDate()){
+ // For old files, must change SDD noise.
+ AliITSresponseSDD *resp1 = (AliITSresponseSDD*)ITS->DetType(1)->
+ GetResponseModel();
+ resp1 = new AliITSresponseSDD();
+ ITS->SetResponseModel(1,resp1);
+ cout << "Changed response class for SDD:" << endl;
+ resp1->Print();
+ } // end if
+
+ if(opt.Contains("Dubna")){
+ AliITSresponseSPDdubna *resp0 = new AliITSresponseSPDdubna();
+ if(ITS->DetType(0)->GetResponseModel() !=0){
+ delete ((AliITSresponse*)ITS->DetType(0)->GetResponseModel());
+ ITS->DetType(0)->ResponseModel(0);
+ } // end if
+ ITS->DetType(0)->ResponseModel(resp0);
+ AliITSsegmentationSPD *seg0 = (AliITSsegmentationSPD*)ITS->
+ DetType(0)->GetSegmentationModel();
+ AliITSsimulationSPDdubna *sim0 = new AliITSsimulationSPDdubna(seg0,
+ resp0);
+ if(ITS->DetType(0)->GetSimulationModel() !=0){
+ delete ((AliITSsimulation*)ITS->DetType(0)->GetSimulationModel());
+ ITS->DetType(0)->SimulationModel(0);
+ } // end if
+ ITS->DetType(0)->SimulationModel(sim0);
+ } // end if Dubna
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff