]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/AliITSHits2SDigits.C
Transition to newIO
[u/mrichter/AliRoot.git] / ITS / AliITSHits2SDigits.C
index 58e594c564ed58876a8867cba07cc3162770dab5..5ca9fffa7749b17fe8ce3779cf747361234418e0 100644 (file)
 #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 "Riostream.h"
+#include "TDatime.h"
+#include "TClassTable.h"
+#include "AliRun.h"
+#include "AliRunDigitizer.h"
+#include "AliITSDigitizer.h"
+#include "AliITS.h"
+#include "AliITSDetType.h"
+#include "AliITSLoader.h"
+#include "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 AliITSHits2SDigits(TString  hitFile = "galice.root",
-                        TString sdigFile = "galice.root",TString opt=""){
-    // Standeard ITS Hits to SDigits.
+Int_t AliITSHits2SDigits(TString  filename = "galice.root")
+ {
+    // Standard ITS Hits to SDigits.
 
     // Dynamically link some shared libs
     if (gClassTable->GetID("AliRun") < 0) {
-       gROOT->LoadMacro("loadlibs.C");
-       loadlibs();
-    } // end if
+      gROOT->ProcessLine(".x $(ALICE_ROOT)/macros/loadlibs.C");
+    }else if (gAlice){
+      delete gAlice->GetRunLoader();
+      delete gAlice;
+      gAlice=0;
+     } 
 
     // Connect the Root Galice file containing Geometry, Kine and Hits
 
-    TFile *hitfile = 0;   // pointer to input file.
-    TFile *sdigfile = 0;  // possible output file for TreeD
-    if(sdigFile.CompareTo(hitFile) == 0){// write output to same file as input.
-       hitfile = AccessFile(hitFile,"U");  // input file open for update.
-    }else{ // different output file then input file.
-       hitfile = AccessFile(hitFile,"R"); // input file open as read only
-       // open output file and create TreeR on it
-       sdigfile = gAlice->InitTreeFile("S",sdigFile);
-    } // end if sdigFile == hitFile.
-
+    AliRunLoader* rl = AliRunLoader::Open(filename);
+    if (rl == 0x0)
+     {
+      cerr<<"AliITSHits2SDigits.C : Can not open session RL=NULL"
+           << endl;
+       return 3;
+     }
+     
+    Int_t retval = rl->LoadgAlice();
+    if (retval)
+     {
+      cerr<<"AliITSHits2SDigits.C : LoadgAlice returned error"
+           << endl;
+       return 3;
+     }
+    gAlice=rl->GetAliRun();
+    AliITSLoader* gime = (AliITSLoader*) rl->GetLoader("ITSLoader");
+    if (gime == 0x0)
+     {
+      cerr<<"AliITSHits2SDigits.C : can not get ITS loader"
+           << endl;
+     }
     AliITS *ITS = (AliITS*)gAlice->GetDetector("ITS");      
     if (!ITS) {
-       cerr<<"AliITSHits2DigitsDefault.C : AliITS object not found on file"
+       cerr<<"AliITSHits2SDigits.C : AliITS object not found on file"
            << endl;
        return 3;
     }  // end if !ITS
     if(!(ITS->GetITSgeom())){
-       cerr << " AliITSgeom not found. Can't digitize with out it." << endl;
-       return 4;
+       cerr << " AliITSgeom not found. Can't digitize without it." << endl;
+       return 4;
     } // end if
 
-    ChangeITSDefaults(hitfile,ITS,opt);
-    // write the AliRun object to the output file if different from input file.
-    if(sdigfile) writeAR(hitfile,sdigfile);
-
     TStopwatch timer;
     Int_t evNumber1 = 0;
     Int_t evNumber2 = gAlice->GetEventsPerRun();
     timer.Start();
+    retval = gime->LoadHits();
+    if (retval)
+     {
+      cerr<<"AliITSHits2SDigits.C : ITSLoader::LoadHits returned error"
+           << endl;
+       return 3;
+     }
+    retval = gime->LoadSDigits("recreate");
+    if (retval)
+     {
+      cerr<<"AliITSHits2SDigits.C : ITSLoader::LoadSDigits returned error"
+           << endl;
+       return 3;
+     }
     for(Int_t event = evNumber1; event < evNumber2; event++){
-       gAlice->GetEvent(event);
-       if(!gAlice->TreeS() && sdigfile == 0){ 
-           cout << "Having to create the SDigits Tree." << endl;
-           gAlice->MakeTree("S");
-       } // end if !gAlice->TreeS()
-       if(sdigfile) gAlice->MakeTree("S",sdigfile);
-       //    make branch
-       ITS->MakeBranch("S");
-       ITS->SetTreeAddress();
-#ifdef DEBUG
-       cout<<"Making ITS SDigits for event "<<event<<endl;
-#endif
-       ITS->Hits2SDigits();
+       rl->GetEvent(event);
+       if(!gime->TreeS()){ 
+           cout << "Having to create the SDigits Tree." << endl;
+           gime->MakeTree("S");
+       } // end 
+
+       ITS->MakeBranch("S");
+       ITS->SetTreeAddress();
+       cout<<"Making ITS SDigits for event "<<event<<endl;
+       ITS->Hits2SDigits();
     } // end for event
     timer.Stop();
     timer.Print();
-    if(sdigfile!=0){
-       cout << sdigFile << " size =" << sdigfile->GetSize() << endl;
-    }else{
-       cout << hitFile << " size =" << hitfile->GetSize() << endl;
-    } // end if sdigfile!=0
-
-    delete gAlice; // sdigfile is closed by deleting gAlice if != hitfile.
-    gAlice = 0;
-    hitfile->Close(); hitfile=0;
-}
-//______________________________________________________________________
-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 " << FileName << " !" << endl;
-       return retfil;
-    } // end if error opeing file
-
-    // Get AliRun object from file or return if not on file
-    if (gAlice) {delete gAlice; gAlice = 0;}
-    gAlice = (AliRun*)file->Get("gAlice");
-    if (!gAlice) {
-       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){
-
-    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
+    delete rl; // sdigfile is closed by deleting gAlice if != hitfile.
+    return 0;
 }