// //
///////////////////////////////////////////////////////////////////////////////
-
+#include "Riostream.h"
#include "AliITSReconstructor.h"
-#include "AliRunLoader.h"
-#include "AliITSclustererV2.h"
+#include "AliRun.h"
+#include "AliRawReader.h"
+#include "AliITSDetTypeRec.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 "AliITSVertexerCosmics.h"
+#include "AliESDEvent.h"
#include "AliITSpidESD.h"
-#include "AliV0vertexer.h"
-#include "AliCascadeVertexer.h"
-#include "AliRun.h"
-#include "AliITS.h"
+#include "AliITSpidESD1.h"
+#include "AliITSpidESD2.h"
+#include "AliITSInitGeometry.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),
+fDetTypeRec(ob.fDetTypeRec)
+
+{
+ // Copy constructor
+}
+
+//______________________________________________________________________
+AliITSReconstructor& AliITSReconstructor::operator=(const AliITSReconstructor& ob ){
+ // Assignment operator
+ this->~AliITSReconstructor();
+ new(this) AliITSReconstructor(ob);
+ return *this;
+}
+
+//______________________________________________________________________
+void AliITSReconstructor::Init() {
+ // Initalize this constructor bet getting/creating the objects
+ // nesseary for a proper ITS reconstruction.
+ // Inputs:
+ // none.
+ // Output:
+ // none.
+ // Return:
+ // none.
+
+ AliITSInitGeometry initgeom;
+ AliITSgeom *geom = initgeom.CreateAliITSgeom();
+ AliInfo(Form("Geometry name: %s",(initgeom.GetGeometryName()).Data()));
+
+ 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
- AliLoader* loader = runLoader->GetLoader("ITSLoader");
- if (!loader) {
- Error("Reconstruct", "ITS loader not found");
- return;
- }
- loader->LoadRecPoints("recreate");
- loader->LoadDigits("read");
- runLoader->LoadKinematics();
-
- AliITSgeom* geom = GetITSgeom(runLoader);
- if (!geom) return;
- AliITSclustererV2 clusterer(geom);
- Int_t nEvents = runLoader->GetNumberOfEvents();
-
- for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
- runLoader->GetEvent(iEvent);
-
- TTree* treeClusters = loader->TreeR();
- if (!treeClusters) {
- loader->MakeTree("R");
- treeClusters = loader->TreeR();
- }
- TTree* treeDigits = loader->TreeD();
- if (!treeDigits) {
- Error("Reconstruct", "Can't get digits tree !");
- return;
- }
+ Int_t cluFindOpt = GetRecoParam()->GetClusterFinder();
+ Bool_t useV2=kTRUE; // Default: V2 cluster finder
+ if(cluFindOpt==1) useV2=kFALSE;
- clusterer.Digits2Clusters(treeDigits, treeClusters);
-
- loader->WriteRecPoints("OVERWRITE");
- }
+ fDetTypeRec->SetTreeAddressD(digitsTree);
+ fDetTypeRec->MakeBranch(clustersTree,"R");
+ fDetTypeRec->SetTreeAddressR(clustersTree);
+ fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,"All",useV2);
+}
- loader->UnloadRecPoints();
- loader->UnloadDigits();
- runLoader->UnloadKinematics();
+//_________________________________________________________________
+void AliITSReconstructor::Reconstruct(AliRawReader* rawReader, TTree *clustersTree) const
+{
+ // reconstruct clusters from raw data
+
+ fDetTypeRec->SetDefaultClusterFindersV2(kTRUE);
+ fDetTypeRec->DigitsToRecPoints(rawReader,clustersTree);
}
//_____________________________________________________________________________
-AliTracker* AliITSReconstructor::CreateTracker(AliRunLoader* runLoader) const
+AliTracker* AliITSReconstructor::CreateTracker() const
{
// create a ITS tracker
- AliITSgeom* geom = GetITSgeom(runLoader);
- if (!geom) return NULL;
- return new AliITStrackerSA(geom);
+ Int_t trackerOpt = GetRecoParam()->GetTracker();
+ AliTracker* tracker;
+ 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");
+ sat->SetOuterStartLayer(GetRecoParam()->GetOuterStartLayerSA());
+ }
+
+ Int_t pidOpt = GetRecoParam()->GetPID();
+
+ AliITSReconstructor* nc = const_cast<AliITSReconstructor*>(this);
+ if(pidOpt==1){
+ Info("FillESD","ITS LandauFitPID option has been selected\n");
+ nc->fItsPID = new AliITSpidESD2((AliITStrackerMI*)tracker);
+ }
+ else{
+ Info("FillESD","ITS default PID\n");
+ Double_t parITS[] = {79.,0.13, 5.}; //IB: this is "pp tuning"
+ nc->fItsPID = 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")){
+ Int_t vtxOpt = GetRecoParam()->GetVertexer();
+ if(vtxOpt==3){
Info("CreateVertexer","a AliITSVertexerIons object has been selected\n");
- return new AliITSVertexerIons("null");
+ return new AliITSVertexerIons();
}
- if(selectedVertexer.Contains("smear") || selectedVertexer.Contains("SMEAR")){
+ if(vtxOpt==4){
Double_t smear[3]={0.005,0.005,0.01};
Info("CreateVertexer","a AliITSVertexerFast object has been selected\n");
return new AliITSVertexerFast(smear);
}
- if(selectedVertexer.Contains("ppz") || selectedVertexer.Contains("PPZ")){
- Info("CreateVertexer","a AliITSVertexerPPZ object has been selected\n");
- return new AliITSVertexerPPZ("null");
+ if(vtxOpt==1){
+ Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
+ return new AliITSVertexerZ();
}
- // 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 cascades
-
- Double_t parITS[] = {34., 0.15, 10.};
- AliITSpidESD itsPID(parITS);
- itsPID.MakePID(esd);
-
- // V0 finding
- Double_t cuts[]={33, // max. allowed chi2
- 0.16,// min. allowed negative daughter's impact parameter
- 0.05,// min. allowed positive daughter's impact parameter
- 0.08,// max. allowed DCA between the daughter tracks
- 0.99,// max. allowed cosine of V0's pointing angle
- 0.9, // min. radius of the fiducial volume
- 2.9 // max. radius of the fiducial volume
- };
- AliV0vertexer vtxer(cuts);
- Double_t vtx[3], cvtx[6];
- esd->GetVertex()->GetXYZ(vtx);
- esd->GetVertex()->GetSigmaXYZ(cvtx);
- vtxer.SetVertex(vtx);
- vtxer.Tracks2V0vertices(esd);
-
- // cascade finding
- Double_t cts[]={33., // max. allowed chi2
- 0.05, // min. allowed V0 impact parameter
- 0.008, // window around the Lambda mass
- 0.035, // min. allowed bachelor's impact parameter
- 0.10, // max. allowed DCA between a V0 and a track
- 0.9985, //max. allowed cosine of the cascade pointing angle
- 0.9, // min. radius of the fiducial volume
- 2.9 // max. radius of the fiducial volume
- };
- AliCascadeVertexer cvtxer=AliCascadeVertexer(cts);
- cvtxer.SetVertex(vtx);
- cvtxer.V0sTracks2CascadeVertices(esd);
+ if(vtxOpt==2){
+ Info("CreateVertexer","a AliITSVertexerCosmics object has been selected\n");
+ return new AliITSVertexerCosmics();
+ }
+ if(vtxOpt==5){
+ Info("CreateVertexer","vertex is fixed in the position of the TDI\n");
+ return new AliITSVertexerFixed("TDI");
+ }
+ if(vtxOpt==6){
+ Info("CreateVertexer","vertex is fixed in the position of the TED\n");
+ return new AliITSVertexerFixed("TED");
+ }
+ // by default an AliITSVertexer3D object is instatiated
+ Info("CreateVertexer","a AliITSVertexer3D object has been selected\n");
+ AliITSVertexer3D* vtxr = new AliITSVertexer3D();
+ 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);
+ return vtxr;
}
-
//_____________________________________________________________________________
-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;
- }
- AliITS* its = (AliITS*) runLoader->GetAliRun()->GetDetector("ITS");
- if (!its) {
- Error("GetITSgeom", "couldn't get ITS detector");
- return NULL;
+// make PID, find V0s and cascade
+ if(fItsPID!=0) {
+ Int_t pidOpt = GetRecoParam()->GetPID();
+ if(pidOpt==1){
+ fItsPID->MakePID(clustersTree,esd);
+ }else{
+ fItsPID->MakePID(esd);
+ }
}
- if (!its->GetITSgeom()) {
- Error("GetITSgeom", "no ITS geometry available");
- return NULL;
+ else {
+ Error("FillESD","!! cannot do the PID !!\n");
}
- return its->GetITSgeom();
}