[u/mrichter/AliRoot.git] / EMCAL / AliEMCALFastRecParticle.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//_________________________________________________________________________
//  A  Particle modified by EMCAL response and produced by AliEMCALvFast
//*--
//  To become a general class of AliRoot ?    
//*--               
//*-- Author: Yves Schutz (SUBATECH)
//*--
/////////////////////////////////////////////////////////////////////////////

// --- ROOT system ---

// --- Standard library ---

// --- AliRoot header files ---

#include "AliEMCALFastRecParticle.h"
#include "TPad.h"
#include "TPaveText.h"

ClassImp(AliEMCALFastRecParticle) ; 

//____________________________________________________________________________
AliEMCALFastRecParticle::AliEMCALFastRecParticle() : TParticle()
{
  // ctor
  fType = 0 ; 
}

//____________________________________________________________________________
AliEMCALFastRecParticle::AliEMCALFastRecParticle(const AliEMCALFastRecParticle & rp)
  : TParticle(rp)
{
  // copy ctor

  fType        = rp.fType ;
  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; 
}

//____________________________________________________________________________
 AliEMCALFastRecParticle::AliEMCALFastRecParticle(const TParticle & pp)
{
  // ctor from a TParticle (crummy?!)
 
  TParticle & pnoconst = (TParticle &)(pp) ;
  AliEMCALFastRecParticle & p = (AliEMCALFastRecParticle &)(pnoconst) ;
  fType        = 0  ;
  fPdgCode     = p.fPdgCode;
  fStatusCode  = p.fStatusCode;
  fMother[0]   = p.fMother[0];
  fMother[1]   = p.fMother[1];
  fDaughter[0] = p.fDaughter[0];
  fDaughter[1] = p.fDaughter[1];
  fWeight      = p.fWeight;
  fCalcMass    = p.fCalcMass;
  fPx          = p.fPx;
  fPy          = p.fPy;
  fPz          = p.fPz;
  fE           = p.fE;
  fVx          = p.fVx;
  fVy          = p.fVy;
  fVz          = p.fVz;
  fVt          = p.fVt;
  fPolarTheta  = p.fPolarTheta;
  fPolarPhi    = p.fPolarPhi;
  fParticlePDG = p.fParticlePDG; 

}

//____________________________________________________________________________
Int_t AliEMCALFastRecParticle::DistancetoPrimitive(Int_t px, Int_t py)
{
  //  Compute distance from point px,py to a AliEMCALFastRecParticle 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;
}

//___________________________________________________________________________
 void AliEMCALFastRecParticle::Draw(Option_t *option)
 {
   // Draw this AliEMCALFastRecParticle with its current attributes
    
   AppendPad(option);
 }

//______________________________________________________________________________
void AliEMCALFastRecParticle::ExecuteEvent(Int_t event, Int_t /*px*/, Int_t /*py*/)
{
  //  Execute action corresponding to one event
  //  This member function is called when a AliEMCALFastRecParticle is clicked with the locator
  
  if (!gPad->IsEditable()) 
    return ;
  
  static TPaveText * clustertext = 0 ; 
  
  switch (event) {
    
  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 ;
  }
  
  case kButton1Up: {
    delete clustertext ; 
    clustertext = 0 ; 
    gPad->Update() ; 
    break ;
  }
  }
  
}

//____________________________________________________________________________
TString AliEMCALFastRecParticle::Name()const
{
  // Returns the name of the particle type (only valid if PIDv1 is employed)

  TString  name ; 
  
  if(fType == 127)
    name = "PHOTON_LOPU_HIEF" ;    //PCA = 001 TOF = 111 CPV = 111
  
  if(fType == 511)
    name = "PHOTON_HIPU_LOEF" ;    //PCA = 011 TOF = 111 CPV = 111
  
  if(fType == 255)
	name = "PHOTON_MED_PU_EF" ;    //PCA = 111 TOF = 111 CPV = 111
  
  if((fType == 383)||(fType == 447)) 
    name = "PHOTON_STRANGE" ;      //PCA = 101 or 110 TOF = 111 CPV = 111
  
  if(fType == 63)
    name = "NEUTRAL_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 111
  
  if((fType == 504) || (fType == 505) ||(fType == 248)||(fType == 249)||(fType == 120)||(fType == 121))
    name = "CHARGED_FAST_EM" ;     //PCA = 111, 011 or 001 TOF =111 CPV = 000 or 001  
  
  if((fType == 56)||(fType == 57))
    name = "CHARGED_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 000 or 001 
  
  if((fType < 8)&&(fType > 0))
    name = "NEUTRAL_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 001 or 011 or 111
  
  if((fType == 0))
    name = "CHARGED_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 000
  
  if((fType == 448) || (fType == 449) ||(fType == 192)||(fType == 193)||(fType == 64)||(fType == 64))
    name = "CHARGED_SLOW_EM" ;    //PCA = 111, 011 or 001 TOF =000 CPV = 000 or 001  
  
  return name ; 
}


//______________________________________________________________________________
void AliEMCALFastRecParticle::SetType(Int_t type) { 
  // sets the particle type 
  fType = type ; 
  
  if((type == 127) || (fType == 511) || (fType == 255) ||(fType == 383)||(fType == 447)){
    fPdgCode = 22 ; 
    return ;
  }
  
  if ((fType == 63)|| ((fType < 8)&&(fType > 0)) ){
    fPdgCode = 2112 ; 
    return ;
  }
  if ( ((fType == 504) || (fType == 505) ||(fType == 248)||(fType == 249)||(fType == 120)||(fType == 121)) ){
    fPdgCode = 11 ; 
    return ;
  }
  if ((fType == 448) || (fType == 449) ||(fType == 192)||(fType == 193)||(fType == 64)||(fType == 64)){
    fPdgCode = 13 ; 
    return ;
  }
  if((fType == 56)||(fType == 57)){
    fPdgCode = 211 ; 
    return ;
  }
  if (fType == 0){
    fPdgCode = 2212 ; 
    return ;
  }

}	    

//______________________________________________________________________________
void AliEMCALFastRecParticle::Paint(Option_t *)
{
  // 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,"") ;
}

//____________________________________________________________________________
void AliEMCALFastRecParticle::Print(Option_t * /*opt*/)const
{
  // Print the type, energy and momentum of the reconstructed particle
  
  printf("Print Summary:") ; 
  printf("AliEMCALFastRecParticle > type is  %s\n", Name().Data()) ; 
  printf("                      Energy = %f\n", fE) ; 
  printf("                         Px     = %f\n", fPx) ; 
  printf("                         Py     = %f\n", fPy) ;
  printf("                         Pz     = %f\n", fPz) ; 
}
Commit	Line	Data
72177635	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
	18	//_________________________________________________________________________
	19	// A Particle modified by EMCAL response and produced by AliEMCALvFast
fdebddeb	20	//*--
72177635	21	// To become a general class of AliRoot ?
fdebddeb	22	//*--
72177635	23	//*-- Author: Yves Schutz (SUBATECH)
fdebddeb	24	//*--
fdebddeb	25	/////////////////////////////////////////////////////////////////////////////
72177635	26
	27	// --- ROOT system ---
	28
	29	// --- Standard library ---
	30
	31	// --- AliRoot header files ---
	32
	33	#include "AliEMCALFastRecParticle.h"
	34	#include "TPad.h"
	35	#include "TPaveText.h"
	36
	37	ClassImp(AliEMCALFastRecParticle) ;
	38
	39	//____________________________________________________________________________
	40	AliEMCALFastRecParticle::AliEMCALFastRecParticle() : TParticle()
	41	{
	42	// ctor
	43	fType = 0 ;
	44	}
	45
	46	//____________________________________________________________________________
9e5d2067	47	AliEMCALFastRecParticle::AliEMCALFastRecParticle(const AliEMCALFastRecParticle & rp)
9e5d2067	48	: TParticle(rp)
72177635	49	{
	50	// copy ctor
	51
	52	fType = rp.fType ;
	53	fPdgCode = rp.fPdgCode;
	54	fStatusCode = rp.fStatusCode;
	55	fMother[0] = rp.fMother[0];
	56	fMother[1] = rp.fMother[1];
	57	fDaughter[0] = rp.fDaughter[0];
	58	fDaughter[1] = rp.fDaughter[1];
	59	fWeight = rp.fWeight;
	60	fCalcMass = rp.fCalcMass;
	61	fPx = rp.fPx;
	62	fPy = rp.fPy;
	63	fPz = rp.fPz;
	64	fE = rp.fE;
	65	fVx = rp.fVx;
	66	fVy = rp.fVy;
	67	fVz = rp.fVz;
	68	fVt = rp.fVt;
	69	fPolarTheta = rp.fPolarTheta;
	70	fPolarPhi = rp.fPolarPhi;
	71	fParticlePDG = rp.fParticlePDG;
	72	}
	73
	74	//____________________________________________________________________________
	75	AliEMCALFastRecParticle::AliEMCALFastRecParticle(const TParticle & pp)
	76	{
	77	// ctor from a TParticle (crummy?!)
	78
	79	TParticle & pnoconst = (TParticle &)(pp) ;
	80	AliEMCALFastRecParticle & p = (AliEMCALFastRecParticle &)(pnoconst) ;
	81	fType = 0 ;
	82	fPdgCode = p.fPdgCode;
	83	fStatusCode = p.fStatusCode;
	84	fMother[0] = p.fMother[0];
	85	fMother[1] = p.fMother[1];
	86	fDaughter[0] = p.fDaughter[0];
	87	fDaughter[1] = p.fDaughter[1];
	88	fWeight = p.fWeight;
	89	fCalcMass = p.fCalcMass;
	90	fPx = p.fPx;
	91	fPy = p.fPy;
	92	fPz = p.fPz;
	93	fE = p.fE;
	94	fVx = p.fVx;
	95	fVy = p.fVy;
	96	fVz = p.fVz;
	97	fVt = p.fVt;
	98	fPolarTheta = p.fPolarTheta;
	99	fPolarPhi = p.fPolarPhi;
	100	fParticlePDG = p.fParticlePDG;
	101
	102	}
	103
	104	//____________________________________________________________________________
	105	Int_t AliEMCALFastRecParticle::DistancetoPrimitive(Int_t px, Int_t py)
	106	{
	107	// Compute distance from point px,py to a AliEMCALFastRecParticle considered as a Tmarker
	108	// Compute the closest distance of approach from point px,py to this marker.
	109	// The distance is computed in pixels units.
	110
	111	Double_t kRADDEG = 180. / TMath::Pi() ;
	112	Coord_t x = Phi() * kRADDEG ;
113	Coord_t y = Theta() * kRADDEG ;
114	const Int_t kMaxDiff = 10;
115	Int_t pxm = gPad->XtoAbsPixel(x);
116	Int_t pym = gPad->YtoAbsPixel(y);
117	Int_t dist = (px-pxm)(px-pxm) + (py-pym)(py-pym);
118
119	if (dist > kMaxDiff) return 9999;
120	return dist;
121	}
122
123	//___________________________________________________________________________
124	void AliEMCALFastRecParticle::Draw(Option_t *option)
125	{
126	// Draw this AliEMCALFastRecParticle with its current attributes
127
128	AppendPad(option);
129	}
130
131	//______________________________________________________________________________
9e5d2067	132	void AliEMCALFastRecParticle::ExecuteEvent(Int_t event, Int_t /px/, Int_t /py/)
72177635	133	{
	134	// Execute action corresponding to one event
	135	// This member function is called when a AliEMCALFastRecParticle is clicked with the locator
	136
	137	if (!gPad->IsEditable())
	138	return ;
	139
	140	static TPaveText * clustertext = 0 ;
	141
	142	switch (event) {
	143
	144	case kButton1Down: {
	145	Double_t kRADDEG = 180. / TMath::Pi() ;
	146	Coord_t x = Phi() * kRADDEG ;
	147	Coord_t y = Theta() * kRADDEG ;
	148	clustertext = new TPaveText(x-1, y+1, x+5, y+3, "") ;
	149	Text_t line1[40] ;
	150	Text_t line2[40] ;
	151	sprintf( line1, "PID: %s ", (const char*)Name() ) ;
	152	sprintf( line2, "ENERGY: %f ", Energy() ) ;
	153	clustertext ->AddText(line1) ;
	154	clustertext ->AddText(line2) ;
	155	clustertext ->Draw("");
	156	gPad->Update() ;
	157	break ;
	158	}
	159
	160	case kButton1Up: {
	161	delete clustertext ;
	162	clustertext = 0 ;
	163	gPad->Update() ;
	164	break ;
	165	}
	166	}
	167
	168	}
	169
	170	//____________________________________________________________________________
	171	TString AliEMCALFastRecParticle::Name()const
	172	{
	173	// Returns the name of the particle type (only valid if PIDv1 is employed)
	174
	175	TString name ;
	176
	177	if(fType == 127)
	178	name = "PHOTON_LOPU_HIEF" ; //PCA = 001 TOF = 111 CPV = 111
	179
	180	if(fType == 511)
	181	name = "PHOTON_HIPU_LOEF" ; //PCA = 011 TOF = 111 CPV = 111
	182
	183	if(fType == 255)
	184	name = "PHOTON_MED_PU_EF" ; //PCA = 111 TOF = 111 CPV = 111
	185
	186	if((fType == 383)\|\|(fType == 447))
	187	name = "PHOTON_STRANGE" ; //PCA = 101 or 110 TOF = 111 CPV = 111
	188
	189	if(fType == 63)
	190	name = "NEUTRAL_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 111
	191
	192	if((fType == 504) \|\| (fType == 505) \|\|(fType == 248)\|\|(fType == 249)\|\|(fType == 120)\|\|(fType == 121))
	193	name = "CHARGED_FAST_EM" ; //PCA = 111, 011 or 001 TOF =111 CPV = 000 or 001
	194
	195	if((fType == 56)\|\|(fType == 57))
	196	name = "CHARGED_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 000 or 001
197
198	if((fType < 8)&&(fType > 0))
199	name = "NEUTRAL_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 001 or 011 or 111
200
201	if((fType == 0))
202	name = "CHARGED_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 000
203
204	if((fType == 448) \|\| (fType == 449) \|\|(fType == 192)\|\|(fType == 193)\|\|(fType == 64)\|\|(fType == 64))
205	name = "CHARGED_SLOW_EM" ; //PCA = 111, 011 or 001 TOF =000 CPV = 000 or 001
206
207	return name ;
208	}
209
210
211	//______________________________________________________________________________
212	void AliEMCALFastRecParticle::SetType(Int_t type) {
213	// sets the particle type
214	fType = type ;
215
216	if((type == 127) \|\| (fType == 511) \|\| (fType == 255) \|\|(fType == 383)\|\|(fType == 447)){
217	fPdgCode = 22 ;
218	return ;
219	}
220
221	if ((fType == 63)\|\| ((fType < 8)&&(fType > 0)) ){
222	fPdgCode = 2112 ;
223	return ;
224	}
225	if ( ((fType == 504) \|\| (fType == 505) \|\|(fType == 248)\|\|(fType == 249)\|\|(fType == 120)\|\|(fType == 121)) ){
226	fPdgCode = 11 ;
227	return ;
228	}
229	if ((fType == 448) \|\| (fType == 449) \|\|(fType == 192)\|\|(fType == 193)\|\|(fType == 64)\|\|(fType == 64)){
230	fPdgCode = 13 ;
231	return ;
232	}
233	if((fType == 56)\|\|(fType == 57)){
234	fPdgCode = 211 ;
235	return ;
236	}
237	if (fType == 0){
238	fPdgCode = 2212 ;
239	return ;
240	}
241
242	}
243
244	//______________________________________________________________________________
245	void AliEMCALFastRecParticle::Paint(Option_t *)
246	{
247	// Paint this ALiRecParticle in theta,phi coordinate as a TMarker with its current attributes
248
249	Double_t kRADDEG = 180. / TMath::Pi() ;
250	Coord_t x = Phi() * kRADDEG ;
251	Coord_t y = Theta() * kRADDEG ;
252	Color_t markercolor = 1 ;
253	Size_t markersize = 1. ;
254	Style_t markerstyle = 5 ;
255
256	if (!gPad->IsBatch()) {
257	gVirtualX->SetMarkerColor(markercolor) ;
258	gVirtualX->SetMarkerSize (markersize) ;
259	gVirtualX->SetMarkerStyle(markerstyle) ;
260	}
261	gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
262	gPad->PaintPolyMarker(1,&x,&y,"") ;
263	}
264
265	//____________________________________________________________________________
9e5d2067	266	void AliEMCALFastRecParticle::Print(Option_t * /opt/)const
72177635	267	{
	268	// Print the type, energy and momentum of the reconstructed particle
	269
fdebddeb	270	printf("Print Summary:") ;
72177635	271	printf("AliEMCALFastRecParticle > type is %s\n", Name().Data()) ;
	272	printf(" Energy = %f\n", fE) ;
	273	printf(" Px = %f\n", fPx) ;
	274	printf(" Py = %f\n", fPy) ;
	275	printf(" Pz = %f\n", fPz) ;
	276	}