* about the suitability of this software for any purpose. It is *
* provided "as is" without express or implied warranty. *
**************************************************************************/
-
+/* $Id$ */
//_________________________________________________________________________
-// Reconstructed Particle
-//*-- Y. Schutz: SUBATECH
-//////////////////////////////////////////////////////////////////////////////
+// A Reconstructed Particle in PHOS
+// To become a general class of AliRoot ?
+// Why should I put meaningless comments
+// just to satisfy
+// the code checker
+//
+//*-- Author: Yves Schutz (SUBATECH)
+
// --- ROOT system ---
// --- Standard library ---
-// --- AliRoot header files ---
+// --- AliRoot header files ---
+#include "AliStack.h"
+#include "AliPHOSHit.h"
+#include "AliPHOSDigit.h"
+#include "AliPHOSTrackSegment.h"
+#include "AliPHOSEmcRecPoint.h"
#include "AliPHOSRecParticle.h"
-#include "TPad.h"
-
-ClassImp(AliPHOSRecParticle)
-
+#include "AliPHOSLoader.h"
+#include "AliPHOSGeometry.h"
+#include "AliLog.h"
//____________________________________________________________________________
- AliPHOSRecParticle::AliPHOSRecParticle(AliPHOSTrackSegment * ts)
+AliPHOSRecParticle::AliPHOSRecParticle():
+ fPHOSTrackSegment(0),
+ fDebug(kFALSE),
+ fPos()
{
// ctor
-
- fPHOSTrackSegment = new AliPHOSTrackSegment(*ts) ;
- fE = ts->GetEnergy() ;
- TVector3 momdir = ts->GetMomentumDirection() ;
- fPx = fE * momdir.X() ;
- fPy = fE * momdir.Y() ;
- fPz = fE * momdir.Z() ;
- fType = kUNDEFINED ;
-
+ const Int_t nSPECIES = AliPID::kSPECIESN;
+ for(Int_t i = 0; i<nSPECIES ; i++)
+ fPID[i]=0.;
}
+
//____________________________________________________________________________
-Int_t AliPHOSRecParticle::DistancetoPrimitive(Int_t px, Int_t py)
+AliPHOSRecParticle::AliPHOSRecParticle(const AliPHOSRecParticle & rp):
+ AliPHOSFastRecParticle(rp),
+ fPHOSTrackSegment(rp.fPHOSTrackSegment),
+ fDebug(kFALSE),
+ fPos()
{
- // Compute distance from point px,py to a AliPHOSRecParticle considered as a Tmarker
- // Compute the closest distance of approach from point px,py to this marker.
- // The distance is computed in pixels units.
-
- Double_t kRADDEG = 180. / TMath::Pi() ;
- Coord_t x = Phi() * kRADDEG ;
- Coord_t y = Theta() * kRADDEG ;
- const Int_t kMaxDiff = 10;
- Int_t pxm = gPad->XtoAbsPixel(x);
- Int_t pym = gPad->YtoAbsPixel(y);
- Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
-
- if (dist > kMaxDiff) return 9999;
- return dist;
+ // copy ctor
+ fType = rp.fType ;
+ fIndexInList = rp.fIndexInList ;
+
+ fPdgCode = rp.fPdgCode;
+ fStatusCode = rp.fStatusCode;
+ fMother[0] = rp.fMother[0];
+ fMother[1] = rp.fMother[1];
+ fDaughter[0] = rp.fDaughter[0];
+ fDaughter[1] = rp.fDaughter[1];
+ fWeight = rp.fWeight;
+ fCalcMass = rp.fCalcMass;
+ fPx = rp.fPx;
+ fPy = rp.fPy;
+ fPz = rp.fPz;
+ fE = rp.fE;
+ fVx = rp.fVx;
+ fVy = rp.fVy;
+ fVz = rp.fVz;
+ fVt = rp.fVt;
+ fPolarTheta = rp.fPolarTheta;
+ fPolarPhi = rp.fPolarPhi;
+ fParticlePDG = rp.fParticlePDG;
+ const Int_t nSPECIES = AliPID::kSPECIESN;
+ for(Int_t i = 0; i<nSPECIES ; i++)
+ fPID[i]=rp.fPID[i];
}
-//___________________________________________________________________________
- void AliPHOSRecParticle::Draw(Option_t *option)
- {
- // Draw this AliPHOSRecParticle with its current attributes
-
- AppendPad(option);
- }
+//____________________________________________________________________________
+Int_t AliPHOSRecParticle::GetNPrimaries() const
+{
+ return -1;
+}
-//______________________________________________________________________________
-void AliPHOSRecParticle::ExecuteEvent(Int_t event, Int_t px, Int_t py)
-{
- // Execute action corresponding to one event
- // This member function is called when a AliPHOSRecParticle is clicked with the locator
- //
-
- if (!gPad->IsEditable())
- return ;
+//____________________________________________________________________________
+Int_t AliPHOSRecParticle::GetNPrimariesToRecParticles() const
+{
+ // Get the number of primaries at the origine of the RecParticle
+ Int_t rv = 0 ;
+ AliRunLoader* rl = AliRunLoader::Instance() ;
+ AliPHOSLoader * phosLoader = dynamic_cast<AliPHOSLoader*>(rl->GetLoader("PHOSLoader"));
+ Int_t emcRPindex = dynamic_cast<AliPHOSTrackSegment*>(phosLoader->TrackSegments()->At(GetPHOSTSIndex()))->GetEmcIndex();
+ dynamic_cast<AliPHOSEmcRecPoint*>(phosLoader->EmcRecPoints()->At(emcRPindex))->GetPrimaries(rv) ;
+ return rv ;
+}
- static TPaveText * clustertext = 0 ;
+//____________________________________________________________________________
+const TParticle * AliPHOSRecParticle::GetPrimary() const
+{
+ // Get the primary particle at the origine of the RecParticle and
+ // which has deposited the largest energy in SDigits
+ AliRunLoader* rl = AliRunLoader::Instance() ;
+ AliPHOSLoader * phosLoader = dynamic_cast<AliPHOSLoader*>(rl->GetLoader("PHOSLoader"));
+ rl->GetEvent(rl->GetEventNumber()) ;
+ rl->LoadKinematics("READ");
+ rl->LoadSDigits("READ");
+
+ if(GetNPrimaries() == 0)
+ return 0 ;
+ if(GetNPrimaries() == 1)
+ return GetPrimary(0) ;
+ Int_t AbsId = 0;
+ Int_t module ;
- switch (event) {
+ // Get PHOS Geometry object
+ AliPHOSGeometry *geom;
+ if (!(geom = AliPHOSGeometry::GetInstance()))
+ geom = AliPHOSGeometry::GetInstance("IHEP","");
+ Double_t x,z ;
+//DP to be fixed: Why do we use this method here??? M.B. use RecPoint???
+ Double_t vtx[3]={0.,0.,0.} ;
+ geom->ImpactOnEmc(vtx,static_cast<double>(Theta()),static_cast<double>(Phi()), module,z,x);
+ Int_t amp = 0 ;
+ Int_t iPrim=-1 ;
+ if(module != 0){
+ geom->RelPosToAbsId(module,x,z,AbsId) ;
+ TClonesArray * sdigits = phosLoader->SDigits() ;
+ AliPHOSDigit * sdig ;
- case kButton1Down: {
- Double_t kRADDEG = 180. / TMath::Pi() ;
- Coord_t x = Phi() * kRADDEG ;
- Coord_t y = Theta() * kRADDEG ;
- clustertext = new TPaveText(x-1, y+1, x+5, y+3, "") ;
- Text_t line1[40] ;
- Text_t line2[40] ;
- sprintf( line1, "PID: %s ", (const char*)Name() ) ;
- sprintf( line2, "ENERGY: %f ", Energy() ) ;
- clustertext ->AddText(line1) ;
- clustertext ->AddText(line2) ;
- clustertext ->Draw("");
- gPad->Update() ;
- break ;
+ for(Int_t i = 0 ; i < sdigits->GetEntriesFast() ; i++){
+ sdig = static_cast<AliPHOSDigit *>(sdigits->At(i)) ;
+ if((sdig->GetId() == AbsId)){
+ if((sdig->GetAmp() > amp) && (sdig->GetNprimary())){
+ amp = sdig->GetAmp() ;
+ iPrim = sdig->GetPrimary(1) ;
+ }
+ // do not scan rest of list
+ if(sdig->GetId() > AbsId)
+ continue ;
+ }
+ }
}
+ if(iPrim >= 0)
+ return rl->Stack()->Particle(iPrim) ;
+ else
+ return 0 ;
+}
- case kButton1Up: {
- delete clustertext ;
- clustertext = 0 ;
- gPad->Update() ;
- break ;
- }
+//____________________________________________________________________________
+Int_t AliPHOSRecParticle::GetPrimaryIndex() const
+{
+ // Get the primary track index in TreeK which deposits the most energy
+ // in Digits which forms EmcRecPoint, which produces TrackSegment,
+ // which the RecParticle is created from
+
+
+ AliRunLoader* rl = AliRunLoader::Instance() ;
+ AliPHOSLoader * phosLoader = dynamic_cast<AliPHOSLoader*>(rl->GetLoader("PHOSLoader"));
+ rl->GetEvent(rl->GetEventNumber()) ;
+ rl->LoadHits("READ");
+ rl->LoadDigits("READ");
+ rl->LoadRecPoints("READ");
+ rl->LoadTracks("READ");
+ // Get TrackSegment corresponding to this RecParticle
+ const AliPHOSTrackSegment *ts = phosLoader->TrackSegment(fPHOSTrackSegment);
+
+ // Get EmcRecPoint corresponding to this TrackSegment
+ Int_t emcRecPointIndex = ts->GetEmcIndex();
+
+ const AliPHOSEmcRecPoint *emcRecPoint = phosLoader->EmcRecPoint(emcRecPointIndex);
+
+ // Get the list of digits forming this EmcRecParticle
+ Int_t nDigits = emcRecPoint->GetDigitsMultiplicity();
+ Int_t *digitList = emcRecPoint->GetDigitsList();
+
+ // Find the digit with maximum amplitude
+ Int_t maxAmp = 0;
+ Int_t bestDigitIndex = -1;
+ for (Int_t iDigit=0; iDigit<nDigits; iDigit++) {
+ const AliPHOSDigit * digit = phosLoader->Digit(digitList[iDigit]);
+ if (digit->GetAmp() > maxAmp) {
+ maxAmp = digit->GetAmp();
+ bestDigitIndex = iDigit;
+ }
+ }
+ if (bestDigitIndex>-1) {
+ const AliPHOSDigit * digit = phosLoader->Digit(digitList[bestDigitIndex]);
+ if (digit==0) return -12345;
}
+
+ // Get the list of primary tracks producing this digit
+ // and find which track has more track energy.
+ //Int_t nPrimary = digit->GetNprimary();
+ //TParticle *track = 0;
+ //Double_t energyEM = 0;
+ //Double_t energyHadron = 0;
+ //Int_t trackEM = -1;
+ //Int_t trackHadron = -1;
+ //for (Int_t iPrim=0; iPrim<nPrimary; iPrim++) {
+ // Int_t iPrimary = digit->GetPrimary(iPrim+1);
+ // if (iPrimary == -1 || TMath::Abs(iPrimary)>10000000)
+ // continue ; //PH 10000000 is the shift of the track
+ // //PH indexes in the underlying event
+ // track = gime->Primary(iPrimary);
+ // Int_t pdgCode = track->GetPdgCode();
+ // Double_t energy = track->Energy();
+ // if (pdgCode==22 || TMath::Abs(pdgCode)==11) {
+ // if (energy > energyEM) {
+ // energyEM = energy;
+ // trackEM = iPrimary;
+ // }
+ // }
+ // else {
+ // if (energy > energyHadron) {
+ // energyHadron = energy;
+ // trackHadron = iPrimary;
+ // }
+ // }
+ //}
+ // Preferences are given to electromagnetic tracks
+ //if (trackEM != -1) return trackEM; // track is gamma or e+-
+ //if (trackHadron != -1) return trackHadron; // track is hadron
+ //return -12345; // no track found :(
-}
-//____________________________________________________________________________
-TString AliPHOSRecParticle::Name()
-{
- TString name ;
- switch (fType) {
- case kGAMMA:
- name = "PHOTON" ;
- break ;
- case kELECTRON:
- name = "ELECTRON" ;
- break ;
- case kNEUTRAL:
- name = "NEUTRAL" ;
- break ;
- case kCHARGEDHADRON:
- name = "CHARGED HADRON" ;
- break ;
- case kNEUTRALHADRON:
- name = "NEUTRAL HADRON" ;
- break ;
- case kNEUTRALEM:
- name = "NEUTRAL EM" ;
- break ;
- case kGAMMAHADRON:
- name = "PHOTON HADRON" ;
- break ;
+ // Get the list of hits producing this digit,
+ // find which hit has deposited more energy
+ // and find the primary track.
+
+ TClonesArray *hits = phosLoader->Hits();
+ if (hits==0) return -12345;
+
+ Double_t maxedep = 0;
+ Int_t maxtrack = -1;
+ Int_t nHits = hits ->GetEntries();
+ Int_t id = (phosLoader->Digit(digitList[bestDigitIndex]))->GetId();
+
+ for (Int_t iHit=0; iHit<nHits; iHit++) {
+ const AliPHOSHit * hit = phosLoader->Hit(iHit);
+ if(hit->GetId() == id){
+ Double_t edep = hit->GetEnergy();
+ Int_t track = hit->GetPrimary();
+ if(edep > maxedep){
+ maxedep = edep;
+ maxtrack = track;
+ }
+ }
}
- return name ;
+
+ if (maxtrack != -1) return maxtrack; // track is hadron
+ return -12345; // no track found :(
}
-//______________________________________________________________________________
-void AliPHOSRecParticle::Paint(Option_t *)
+//____________________________________________________________________________
+const TParticle * AliPHOSRecParticle::GetPrimary(Int_t index) const
{
-// Paint this ALiRecParticle in theta,phi coordinate as a TMarker with its current attributes
-
- Double_t kRADDEG = 180. / TMath::Pi() ;
- Coord_t x = Phi() * kRADDEG ;
- Coord_t y = Theta() * kRADDEG ;
- Color_t markercolor = 1 ;
- Size_t markersize = 1. ;
- Style_t markerstyle = 5 ;
-
- if (!gPad->IsBatch()) {
- gVirtualX->SetMarkerColor(markercolor) ;
- gVirtualX->SetMarkerSize (markersize) ;
- gVirtualX->SetMarkerStyle(markerstyle) ;
- }
- gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
- gPad->PaintPolyMarker(1,&x,&y,"") ;
+ // Get one of the primary particles at the origine of the RecParticle
+ if ( index > GetNPrimariesToRecParticles() ) {
+ if (fDebug)
+ Warning("GetPrimary", "AliPHOSRecParticle::GetPrimary -> %d is larger that the number of primaries %d",
+ index, GetNPrimaries()) ;
+ return 0 ;
+ }
+ else {
+ Int_t dummy ;
+ AliRunLoader* rl = AliRunLoader::Instance() ;
+ AliPHOSLoader * phosLoader = dynamic_cast<AliPHOSLoader*>(rl->GetLoader("PHOSLoader"));
+
+ Int_t emcRPindex = dynamic_cast<AliPHOSTrackSegment*>(phosLoader->TrackSegments()->At(GetPHOSTSIndex()))->GetEmcIndex();
+ Int_t primaryindex = dynamic_cast<AliPHOSEmcRecPoint*>(phosLoader->EmcRecPoints()->At(emcRPindex))->GetPrimaries(dummy)[index] ;
+ return rl->Stack()->Particle(primaryindex) ;
+ }
+ // return 0 ;
}
//____________________________________________________________________________
-void AliPHOSRecParticle::Print()
+void AliPHOSRecParticle::SetPID(Int_t type, Float_t weight)
{
- cout << "AliPHOSRecParticle > " << "type is " << Name() << endl
- << " " << "Energy = " << fE << endl
- << " " << "Px = " << fPx << endl
- << " " << "Py = " << fPy << endl
- << " " << "Pz = " << fPz << endl ;
+ // Set the probability densities that this reconstructed particle
+ // has a type of i:
+ // i particle types
+ // ----------------------
+ // 0 electron
+ // 1 muon
+ // 2 pi+-
+ // 3 K+-
+ // 4 p/pbar
+ // 5 photon
+ // 6 pi0 at high pt
+ // 7 neutron
+ // 8 K0L
+ // 9 Conversion electron
+
+ fPID[type] = weight ;
}