Transition to NewIO

[u/mrichter/AliRoot.git] / STEER / AliHits2SDigits.C

////////////////////////////////////////////////////////////////////////
//
// name: AliHits2SDigits
// date: 4.4.2002
// last update: 4.4.2002
// author: Jiri Chudoba
// version: 1.0
//
// description: 
//       creates sdigits for several detectors
//       stores sdigits in separate file (or in the source file
//       with hits). Stores gAlice object and copies TE to the
//       file with sdigits
//
// input:
//       TString fileNameSDigits ... output file with sdigits
//       TString fileNameHits ... input file with hits
//       Int_t nEvents  ... how many events to proceed
//       Int_t firstEvent  ... first event number
//       Int_t ITS, TPC, ...   many flags for diff. detectors
//
// History:
//
// 04.04.02 - first version
// 
////////////////////////////////////////////////////////////////////////

#if !defined(__CINT__) || defined(__MAKECINT__)
#include "iostream.h"
#include "TTree.h"
#include "TBranch.h"
#include "TDirectory.h"
#include "TFile.h"
#include "AliRun.h"
#include "TParticle.h"
#include "TPC/AliTPCDigitsArray.h"
#include "AliHeader.h"
#include "TGeometry.h"
#include "TObjArray.h"
#include "TString.h"
#include "ITS/AliITS.h"
#include "TPC/AliTPC.h"
#include "PHOS/AliPHOSSDigitizer.h"
#include "TRD/AliTRDdigitizer.h"
#include "TStopwatch.h"
#include "TRD/AliTRDparameter.h"
#endif

AliRunLoader* Init(TString fileName);

AliTRDdigitizer *InitTRDdigitizer();
void AliCopy(TFile *inputFile, TFile *outputFile);

// global variables

TFile *gFileHits = 0;
Bool_t gSameFiles = kFALSE;
Int_t gDEBUG = 1;


Int_t AliHits2SDigits(TString fileName="rfio:galice.root", 
                  Int_t nEvents = 1, Int_t firstEvent = 0, Int_t iITS = 0,
                  Int_t iTPC = 0, Int_t iTRD = 0,Int_t iPHOS = 0, 
                  Int_t iCopy = 1)
{
//
// Initialization
//
  AliRunLoader* rl = Init(fileName);
  if (!rl) return 1;
  if (iCopy) {
//    AliCopy(gFileHits,fileSDigits);
//    gFileHits->cd();
  }  

// ITS
  AliITS *ITS;
  if (iITS) {
    ITS  = (AliITS*) gAlice->GetModule("ITS");
    if (!ITS) {
      iITS = 0;
      cerr<<"AliITS object not found on file." << endl;
    } else if (!ITS->GetITSgeom()) {
      cerr<<"AliITSgeom not found." << endl;
      iITS = 0;
    }
  }

// TPC
  AliTPC *TPC;
  if (iTPC) {
    TPC = (AliTPC*)gAlice->GetDetector("TPC");
    if (!TPC) {
      iTPC = 0;
      cerr<<"AliTPC object not found"<<endl;
    }
  }

// TRD
  AliTRDdigitizer *sdTRD;
  if (iTRD) {
    sdTRD = InitTRDdigitizer();
  }


// PHOS
  AliPHOSSDigitizer *sdPHOS;
  if (iPHOS) {
    sdPHOS = new AliPHOSSDigitizer(fileName.Data());
  }



//
// loop over events
//
  TStopwatch timer;
  timer.Start();
  for (Int_t iEvent = firstEvent;iEvent<firstEvent+nEvents;iEvent++){
    rl->GetEvent(iEvent);
//    gAlice->MakeTree("S",fileSDigits);
    
// ITS
    if (iITS) {
      if (gDEBUG) {cout<<"  Create ITS sdigits: ";}
      AliLoader* loader = rl->GetLoader("ITSLoader");
      if (loader)
       { 
        loader->LoadHits("read");
        loader->LoadSDigits("update");
        ITS->SetTreeAddress();
        ITS->Hits2SDigits();
        loader->UnloadHits();
        loader->UnloadSDigits();
        if (gDEBUG) {cout<<"done"<<endl;}
       }
      else if (gDEBUG) {cout<<"Did not get loader"<<endl;}
    }

// TPC
    if (iTPC) {
      if (gDEBUG) {cout<<"  Create TPC sdigits: ";}
      AliLoader* loader = rl->GetLoader("TPCLoader");
      if (loader)
       { 
        loader->LoadHits("read");
        loader->LoadSDigits("update");
      
        TPC->SetActiveSectors(1);
        TPC->Hits2SDigits2(iEvent);
        loader->UnloadHits();
        loader->UnloadSDigits();
        if (gDEBUG) {cout<<"done"<<endl;}
       }
      else if (gDEBUG) {cout<<"Did not get loader"<<endl;}
    }

// TRD
    if (iTRD) {
      if (gDEBUG) {cout<<"  Create TRD sdigits: ";}
      AliLoader* loader = rl->GetLoader("TRDLoader");
      if (loader)
       { 
        loader->LoadHits("read");
        loader->LoadSDigits("update");
        sdTRD->MakeDigits();
        sdTRD->WriteDigits();
        loader->UnloadHits();
        loader->UnloadSDigits();
        if (gDEBUG) {cout<<"done"<<endl;}
       }
      else if (gDEBUG) {cout<<"Did not get loader"<<endl;}
    }
    
  } // end of loop over events

// PHOS processes always all events
  if (iPHOS) {
    sdPHOS->ExecuteTask("deb all");
  }

//
// finish 
//
  timer.Stop(); 
  timer.Print();

  delete rl;
}
 

////////////////////////////////////////////////////////////////////////
AliRunLoader* Init(TString fileName) 
 {
// open input file, read in gAlice, prepare output file
  if (gAlice) delete gAlice;
  gAlice = 0;
  AliRunLoader*rl = AliRunLoader::Open(fileName);
  if (rl == 0x0) return 0x0;
  rl->LoadgAlice();
  gAlice = rl->GetAliRun();
  return rl;
 
}


////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////
AliTRDdigitizer *InitTRDdigitizer() {
// initialization of TRD digitizer
  AliTRDdigitizer *sdTRD = new AliTRDdigitizer("TRDdigitizer"
                                     ,"TRD digitizer class");
  sdTRD->SetDebug(0);
  sdTRD->SetSDigits(kTRUE);
  AliTRDparameter *TRDparam = new AliTRDparameter("TRDparameter"
                                      ,"TRD parameter class");

  sdTRD->SetParameter(TRDparam);
  sdTRD->InitDetector();
  if (!sdTRD->MakeBranch()) {
    cerr<<"Problems with TRD digitizer initialization."<<endl;
  }
  return sdTRD;
}
////////////////////////////////////////////////////////////////////////
/*void AliCopy(TFile *inputFile, TFile *outputFile) {
// copy some objects

// copy gAlice
  if (gDEBUG) cout<<"Copy gAlice: ";
  outputFile->cd();
  gAlice->Write();
  if (gDEBUG) cout<<"done"<<endl;

  TTree *treeE  = gAlice->TreeE();
  if (!treeE) {
    cerr<<"No TreeE found "<<endl;
    return;
  }      

// copy TreeE
  if (gDEBUG) cout<<"Copy TreeE: ";
  AliHeader *header = new AliHeader();
  treeE->SetBranchAddress("Header", &header);
  treeE->SetBranchStatus("*",1);
  TTree *treeENew =  treeE->CloneTree();
  treeENew->Write();
  if (gDEBUG) cout<<"done"<<endl;

// copy AliceGeom
  if (gDEBUG) cout<<"Copy AliceGeom: ";
  TGeometry *AliceGeom = static_cast<TGeometry*>(inputFile->Get("AliceGeom"));
  if (!AliceGeom) {
    cerr<<"AliceGeom was not found in the input file "<<endl;
    return;
  }
  AliceGeom->Write();
  if (gDEBUG) cout<<"done"<<endl;

}
*/