X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITSReconstructor.cxx;h=9e61af29d029dbc166af4c0985eb2bd5c17e0a45;hb=f9cdb612f270a574140df4a57f10497a65ea21c0;hp=932968794a7b4bb47e08aa0b271ebeda5c9d9e6e;hpb=108bd0fecdfc9619510aaf5d0a4fd46a4e8c7ce4;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITSReconstructor.cxx b/ITS/AliITSReconstructor.cxx index 932968794a7..9e61af29d02 100644 --- a/ITS/AliITSReconstructor.cxx +++ b/ITS/AliITSReconstructor.cxx @@ -24,57 +24,59 @@ #include "Riostream.h" #include "AliITSReconstructor.h" #include "AliRun.h" -#include "AliRunLoader.h" #include "AliRawReader.h" -#include "AliITSDetTypeRec.h" +#include "AliESDEvent.h" +#include "AliESDpid.h" +#include "AliITSgeom.h" #include "AliITSLoader.h" #include "AliITStrackerMI.h" +#include "AliITStrackerV2.h" #include "AliITStrackerSA.h" -#include "AliITSVertexerIons.h" #include "AliITSVertexerFast.h" -#include "AliITSVertexerPPZ.h" +#include "AliITSVertexerFixed.h" +#include "AliITSVertexer3D.h" #include "AliITSVertexerZ.h" -#include "AliESD.h" -#include "AliITSpidESD.h" -#include "AliITSpidESD1.h" -#include "AliITSpidESD2.h" +#include "AliITSVertexerZD.h" +#include "AliITSVertexerCosmics.h" #include "AliITSInitGeometry.h" +#include "AliITSTrackleterSPDEff.h" +#include "AliITSMultReconstructor.h" ClassImp(AliITSReconstructor) //___________________________________________________________________________ AliITSReconstructor::AliITSReconstructor() : AliReconstructor(), -fItsPID(0) +fDetTypeRec(0) { // Default constructor } //___________________________________________________________________________ AliITSReconstructor::~AliITSReconstructor(){ // destructor - delete fItsPID; + if(fDetTypeRec) delete fDetTypeRec; } -//______________________________________________________________________ -AliITSReconstructor::AliITSReconstructor(const AliITSReconstructor &ob) :AliReconstructor(ob), -fItsPID(ob.fItsPID) -{ - // Copy constructor -} +//____________________________________________________________________________ +void AliITSReconstructor::GetPidSettings(AliESDpid *ESDpid) { + // + // pass PID settings from AliITSRecoParam to AliESDpid + // + Int_t pidOpt = GetRecoParam()->GetPID(); -//______________________________________________________________________ -AliITSReconstructor& AliITSReconstructor::operator=(const AliITSReconstructor& ob ){ - // Assignment operator - this->~AliITSReconstructor(); - new(this) AliITSReconstructor(ob); - return *this; + if(pidOpt==1){ + AliDebug(1,"ITS LandauFitPID option has been selected\n"); + ESDpid->SetITSPIDmethod(AliESDpid::kITSLikelihood); + } + else{ + AliDebug(1,"ITS default PID\n"); + ESDpid->SetITSPIDmethod(AliESDpid::kITSTruncMean); + } } //______________________________________________________________________ -void AliITSReconstructor::Init(AliRunLoader *runLoader) const{ +void AliITSReconstructor::Init() { // Initalize this constructor bet getting/creating the objects // nesseary for a proper ITS reconstruction. // Inputs: - // AliRunLoader *runLoader Pointer to the run loader to allow - // the getting of files/folders data - // needed to do reconstruction + // none. // Output: // none. // Return: @@ -83,177 +85,178 @@ void AliITSReconstructor::Init(AliRunLoader *runLoader) const{ AliITSInitGeometry initgeom; AliITSgeom *geom = initgeom.CreateAliITSgeom(); AliInfo(Form("Geometry name: %s",(initgeom.GetGeometryName()).Data())); - AliITSLoader* loader = static_cast - (runLoader->GetLoader("ITSLoader")); - if (!loader) { - Error("Init", "ITS loader not found"); - return; - } - loader->SetITSgeom(geom); + + fDetTypeRec = new AliITSDetTypeRec(); + fDetTypeRec->SetITSgeom(geom); + fDetTypeRec->SetDefaults(); + return; } + //_____________________________________________________________________________ -void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader) const +void AliITSReconstructor::Reconstruct(TTree *digitsTree, TTree *clustersTree) const { // reconstruct clusters - - AliITSLoader* loader = static_cast(runLoader->GetLoader("ITSLoader")); - if (!loader) { - Error("Reconstruct", "ITS loader not found"); - return; - } - AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader); - rec->SetDefaults(); - - loader->LoadRecPoints("recreate"); - loader->LoadDigits("read"); - runLoader->LoadKinematics(); - TString option = GetOption(); - Bool_t clusfinder=kTRUE; // Default: V2 cluster finder - if(option.Contains("OrigCF"))clusfinder=kFALSE; - - Int_t nEvents = runLoader->GetNumberOfEvents(); - - for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) { - runLoader->GetEvent(iEvent); - if(loader->TreeR()==0x0) loader->MakeTree("R"); - rec->MakeBranch("R"); - rec->SetTreeAddress(); - rec->DigitsToRecPoints(iEvent,0,"All",clusfinder); - } - - loader->UnloadRecPoints(); - loader->UnloadDigits(); - runLoader->UnloadKinematics(); + TString recopt = GetRecoParam()->GetOptReco(); + fDetTypeRec->SetTreeAddressD(digitsTree); + fDetTypeRec->MakeBranch(clustersTree,"R"); + fDetTypeRec->SetTreeAddressR(clustersTree); + fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,recopt.Data(),1); } //_________________________________________________________________ -void AliITSReconstructor::Reconstruct(AliRunLoader* runLoader, - AliRawReader* rawReader) const +void AliITSReconstructor::Reconstruct(AliRawReader* rawReader, TTree *clustersTree) const { -// reconstruct clusters - - - AliITSLoader* loader = static_cast(runLoader->GetLoader("ITSLoader")); - if (!loader) { - Error("Reconstruct", "ITS loader not found"); - return; - } - - AliITSDetTypeRec* rec = new AliITSDetTypeRec(loader); - rec->SetDefaults(); - rec->SetDefaultClusterFindersV2(kTRUE); - - loader->LoadRecPoints("recreate"); - - Int_t iEvent = 0; + // reconstruct clusters from raw data + TString recopt = GetRecoParam()->GetOptReco(); + fDetTypeRec->SetDefaultClusterFindersV2(kTRUE,kTRUE); + fDetTypeRec->DigitsToRecPoints(rawReader,clustersTree,recopt.Data()); +} - while(rawReader->NextEvent()) { - runLoader->GetEvent(iEvent++); - if(loader->TreeR()==0x0) loader->MakeTree("R"); - rec->DigitsToRecPoints(rawReader); - } +//_____________________________________________________________________________ +AliTrackleter* AliITSReconstructor::CreateMultFinder() const +{ + // create the SPD trackeleter for mult. reconstruction + AliITSMultReconstructor* multReco = new AliITSMultReconstructor(); + multReco->SetHistOn(kFALSE); + multReco->SetDetTypeRec(fDetTypeRec); + return multReco; +} - loader->UnloadRecPoints(); +//_____________________________________________________________________________ +AliTracker* AliITSReconstructor::CreateTrackleter() const +{ +// create the SPD trackeleter (for SPD PlaneEfficiency evaluation) + if(!GetRecoParam()->GetComputePlaneEff() || !GetRecoParam()->GetUseTrackletsPlaneEff()) return NULL; + //Int_t trackerOpt = GetRecoParam()->GetTracker(); + AliTracker* trackleter; + trackleter = new AliITSTrackleterSPDEff(); + AliITSTrackleterSPDEff *spdtrackleter=(AliITSTrackleterSPDEff*)trackleter; + // here set cuts (from RecoParam) + if(GetRecoParam()->GetBkgTrackletsPlaneEff()) spdtrackleter->SetReflectClusterAroundZAxisForLayer(1,kTRUE); + if(GetRecoParam()->GetMCTrackletsPlaneEff()) spdtrackleter->SetMC(); + spdtrackleter->SetHistOn(); + spdtrackleter->SetPhiWindowL2(GetRecoParam()->GetTrackleterPhiWindowL2()); + spdtrackleter->SetZetaWindowL2(GetRecoParam()->GetTrackleterZetaWindowL2()); + spdtrackleter->SetPhiWindowL1(GetRecoParam()->GetTrackleterPhiWindowL1()); + spdtrackleter->SetZetaWindowL1(GetRecoParam()->GetTrackleterZetaWindowL1()); + if(GetRecoParam()->GetUpdateOncePerEventPlaneEff()) spdtrackleter->SetUpdateOncePerEventPlaneEff(); + spdtrackleter->SetMinContVtx(GetRecoParam()->GetMinContVtxPlaneEff()); + return trackleter; } //_____________________________________________________________________________ -AliTracker* AliITSReconstructor::CreateTracker(AliRunLoader* runLoader)const +AliTracker* AliITSReconstructor::CreateTracker() const { // create a ITS tracker - - TString selectedTracker = GetOption(); + Int_t trackerOpt = GetRecoParam()->GetTracker(); AliTracker* tracker; - if (selectedTracker.Contains("MI")) { + if (trackerOpt==1) { tracker = new AliITStrackerMI(0); + AliITStrackerMI *mit=(AliITStrackerMI*)tracker; + mit->SetDetTypeRec(fDetTypeRec); + } + else if (trackerOpt==2) { + tracker = new AliITStrackerV2(0); } else { tracker = new AliITStrackerSA(0); // inherits from AliITStrackerMI + AliITStrackerSA *sat=(AliITStrackerSA*)tracker; + sat->SetDetTypeRec(fDetTypeRec); + if(GetRecoParam()->GetTrackerSAOnly()) sat->SetSAFlag(kTRUE); + if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n"); + if(GetRecoParam()->GetInwardFindingSA()){ + sat->SetInwardFinding(); + sat->SetInnerStartLayer(GetRecoParam()->GetInnerStartLayerSA()); + }else{ + sat->SetOutwardFinding(); + sat->SetOuterStartLayer(GetRecoParam()->GetOuterStartLayerSA()); + } + sat->SetMinNPoints(GetRecoParam()->GetMinNPointsSA()); } - TString selectedPIDmethod = GetOption(); - AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader"); - if (!loader) { - Error("CreateTracker", "ITS loader not found"); - } - if(selectedPIDmethod.Contains("LandauFitPID")){ - loader->AdoptITSpid(new AliITSpidESD2((AliITStrackerMI*)tracker,loader)); - } - else{ - Double_t parITS[] = {34., 0.15, 10.}; - loader->AdoptITSpid(new AliITSpidESD1(parITS)); - } return tracker; } //_____________________________________________________________________________ -AliVertexer* AliITSReconstructor::CreateVertexer(AliRunLoader* /*runLoader*/) const +AliVertexer* AliITSReconstructor::CreateVertexer() const { // create a ITS vertexer - TString selectedVertexer = GetOption(); - if(selectedVertexer.Contains("ions") || selectedVertexer.Contains("IONS")){ - Info("CreateVertexer","a AliITSVertexerIons object has been selected\n"); - return new AliITSVertexerIons("null"); + AliITSVertexer *vptr = NULL; + Int_t vtxOpt = GetRecoParam()->GetVertexer(); + if(vtxOpt==3){ + AliFatal("Option AliITSVertexerIons is no longer supported"); + return vptr; } - if(selectedVertexer.Contains("smear") || selectedVertexer.Contains("SMEAR")){ - Double_t smear[3]={0.005,0.005,0.01}; - Info("CreateVertexer","a AliITSVertexerFast object has been selected\n"); - return new AliITSVertexerFast(smear); + else if(vtxOpt==4){ + Double_t smear[3]={GetRecoParam()->GetVertexerFastSmearX(), + GetRecoParam()->GetVertexerFastSmearY(), + GetRecoParam()->GetVertexerFastSmearZ()}; + AliDebug(1,"AliITSVertexerFast has been selected"); + vptr = new AliITSVertexerFast(smear); } - if(selectedVertexer.Contains("ppz") || selectedVertexer.Contains("PPZ")){ - Info("CreateVertexer","a AliITSVertexerPPZ object has been selected\n"); - return new AliITSVertexerPPZ("null"); + else if(vtxOpt==1){ + AliDebug(1,"AliITSVertexerZ has been selected"); + AliITSVertexerZ* vtxrz = new AliITSVertexerZ(); + Int_t pileupAlgo=GetRecoParam()->GetSPDVertexerPileupAlgo(); + if(pileupAlgo==3) vtxrz->SetSearchForPileup(kFALSE); + vptr = vtxrz; } - // by default an AliITSVertexerZ object is instatiated - Info("CreateVertexer","a AliITSVertexerZ object has been selected\n"); - return new AliITSVertexerZ("null"); -} - -//_____________________________________________________________________________ -void AliITSReconstructor::FillESD(AliRunLoader* runLoader, - AliESD* esd) const -{ -// make PID, find V0s and cascade - AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader"); - AliITSpidESD *pidESD = 0; - TString selectedPIDmethod = GetOption(); - if(selectedPIDmethod.Contains("LandauFitPID")){ - Info("FillESD","ITS LandauFitPID option has been selected\n"); - pidESD=loader->GetITSpid(); + else if(vtxOpt==2){ + AliDebug(1,"AliITSVertexerCosmics has been selected"); + vptr = new AliITSVertexerCosmics(); } - else{ - Info("FillESD","ITS default PID\n"); - pidESD=loader->GetITSpid(); + else if(vtxOpt==5){ + AliDebug(1,"Vertex is fixed in the position of the TDI\n"); + vptr = new AliITSVertexerFixed("TDI"); + } + else if(vtxOpt==6){ + AliDebug(1,"Vertex is fixed in the position of the TED\n"); + vptr = new AliITSVertexerFixed("TED"); } - if(pidESD!=0){ - pidESD->MakePID(esd); + else if(vtxOpt==7){ + AliDebug(1,"VertexerZD: reconstruction of the Z coordinate with SDD \n"); + AliITSVertexerZD*vtxrz = new AliITSVertexerZD(); + vtxrz->SetSearchForPileup(kFALSE); + vptr = vtxrz; } else { - Error("FillESD","!! cannot do the PID !!\n"); + // by default an AliITSVertexer3D object is instatiated + AliITSVertexer3D* vtxr = + new AliITSVertexer3D(GetRecoParam()->GetVertexer3DWideFiducialRegionZ()); + Float_t dzw=GetRecoParam()->GetVertexer3DWideFiducialRegionZ(); + Float_t drw=GetRecoParam()->GetVertexer3DWideFiducialRegionR(); + vtxr->SetWideFiducialRegion(dzw,drw); + Float_t dzn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionZ(); + Float_t drn=GetRecoParam()->GetVertexer3DNarrowFiducialRegionR(); + vtxr->SetNarrowFiducialRegion(dzn,drn); + Float_t dphil=GetRecoParam()->GetVertexer3DLooseDeltaPhiCut(); + Float_t dphit=GetRecoParam()->GetVertexer3DTightDeltaPhiCut(); + vtxr->SetDeltaPhiCuts(dphil,dphit); + Float_t dcacut=GetRecoParam()->GetVertexer3DDCACut(); + vtxr->SetDCACut(dcacut); + Int_t pileupAlgo=GetRecoParam()->GetSPDVertexerPileupAlgo(); + vtxr->SetPileupAlgo(pileupAlgo); + UChar_t highmultAlgo=GetRecoParam()->GetSPDVertexerHighMultAlgo(); + vtxr->SetHighMultAlgo(highmultAlgo); + AliDebug(1,Form("AliITSVertexer3D with pileup algo %d has been selected",pileupAlgo)); + vptr = vtxr; } + vptr->SetDetTypeRec(fDetTypeRec); + return vptr; } - //_____________________________________________________________________________ -AliITSgeom* AliITSReconstructor::GetITSgeom(AliRunLoader* runLoader) const +void AliITSReconstructor::FillESD(TTree * /*digitsTree*/, TTree * /*clustersTree*/, + AliESDEvent* /* esd */) const { -// get the ITS geometry - - if (!runLoader->GetAliRun()) runLoader->LoadgAlice(); - if (!runLoader->GetAliRun()) { - Error("GetITSgeom", "couldn't get AliRun object"); - return NULL; - } - AliITSLoader *loader = (AliITSLoader*)runLoader->GetLoader("ITSLoader"); - AliITSgeom* geom = (AliITSgeom*)loader->GetITSgeom(); - if(!geom){ - Error("GetITSgeom","no ITS geometry available"); - return NULL; - } - - return geom; +// make PID, find V0s and cascade +/* Now done in AliESDpid + if(fItsPID!=0) fItsPID->MakePID(esd); + else AliError("!! cannot do the PID !!"); +*/ }