// //
///////////////////////////////////////////////////////////////////////////////
-
+#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 "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)
+AliITSRecoParam *AliITSReconstructor::fgRecoParamDefault = AliITSRecoParam::GetHighFluxParam();
+AliITSRecoParam *AliITSReconstructor::fgRecoParam =0; // reconstruction parameters
+
+//___________________________________________________________________________
+AliITSReconstructor::AliITSReconstructor() : AliReconstructor(),
+fItsPID(0),
+fDetTypeRec(0)
+{
+ // Default constructor
+ if (!fgRecoParam) {
+ AliWarning("Using default reconstruction parameters");
+ fgRecoParam = fgRecoParamDefault;
+ }
+}
+ //___________________________________________________________________________
+AliITSReconstructor::~AliITSReconstructor(){
+// destructor
+ delete fItsPID;
+ if(!fgRecoParam && (fgRecoParam != fgRecoParamDefault)) delete fgRecoParam;
+ 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;
- }
+ TString option = GetOption();
+ Bool_t clusfinder=kTRUE; // Default: V2 cluster finder
+ if(option.Contains("OrigCF"))clusfinder=kFALSE;
- clusterer.Digits2Clusters(treeDigits, treeClusters);
-
- loader->WriteRecPoints("OVERWRITE");
- }
+ fDetTypeRec->SetTreeAddressD(digitsTree);
+ fDetTypeRec->MakeBranch(clustersTree,"R");
+ fDetTypeRec->SetTreeAddressR(clustersTree);
+ fDetTypeRec->DigitsToRecPoints(digitsTree,clustersTree,0,"All",clusfinder);
+}
- 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);
+
+ TString selectedTracker = GetOption();
+ AliTracker* tracker;
+ if (selectedTracker.Contains("MI")) {
+ tracker = new AliITStrackerMI(0);
+ }
+ else if (selectedTracker.Contains("V2")) {
+ tracker = new AliITStrackerV2(0);
+ }
+ else {
+ tracker = new AliITStrackerSA(0); // inherits from AliITStrackerMI
+ AliITStrackerSA *sat=(AliITStrackerSA*)tracker;
+ if(selectedTracker.Contains("onlyITS"))sat->SetSAFlag(kTRUE);
+ if(sat->GetSAFlag())AliDebug(1,"Tracking Performed in ITS only\n");
+ if(selectedTracker.Contains("cosmics")||selectedTracker.Contains("COSMICS"))
+ sat->SetOuterStartLayer(AliITSgeomTGeo::GetNLayers()-2);
+ }
+
+ TString selectedPIDmethod = GetOption();
+ AliITSReconstructor* nc = const_cast<AliITSReconstructor*>(this);
+ if(selectedPIDmethod.Contains("LandauFitPID")){
+ 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
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(selectedVertexer.Contains("3d") || selectedVertexer.Contains("3D")){
+ Info("CreateVertexer","a AliITSVertexer3D object has been selected\n");
+ return new AliITSVertexer3D("null");
+ }
+ if(selectedVertexer.Contains("cosmics") || selectedVertexer.Contains("COSMICS")){
+ Info("CreateVertexer","a AliITSVertexerCosmics object has been selected\n");
+ return new AliITSVertexerCosmics();
}
// by default an AliITSVertexerZ object is instatiated
Info("CreateVertexer","a AliITSVertexerZ object has been selected\n");
}
//_____________________________________________________________________________
-void AliITSReconstructor::FillESD(AliRunLoader* /*runLoader*/,
- AliESD* esd) const
+void AliITSReconstructor::FillESD(TTree * /*digitsTree*/, TTree *clustersTree,
+ AliESDEvent* 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);
-}
-
-
-//_____________________________________________________________________________
-AliITSgeom* AliITSReconstructor::GetITSgeom(AliRunLoader* runLoader) 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) {
+ TString selectedPIDmethod = GetOption();
+ if(selectedPIDmethod.Contains("LandauFitPID")){
+ 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();
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff