Added ability to store original parton type - by default set to 0
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALParton.cxx
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 //*-- Author: Renan Cabrera (Creighton U.)
19
20 #include "AliEMCALParton.h"
21 #include "Ecommon.h"
22 #include <Riostream.h>
23 ClassImp(AliEMCALParton)   
24     
25 //____________________________________________________________________________
26 AliEMCALParton::AliEMCALParton()
27 {
28   // Default constructor
29 //  fTrackEnergy = 0;
30 //  fTrackEta    = 0;
31 //  fTrackPhi    = 0;
32 //  fTrackPDG    = 0;
33   fNTracks     = 0;
34   fPartonCode = 0;
35 }
36
37 AliEMCALParton::AliEMCALParton(Float_t energy, Float_t phi, Float_t eta)
38 {
39   // Constructor
40   fEnergy = energy;
41   fPhi    = phi;
42   fEta    = eta;
43 //  fTrackEnergy = 0;
44 //  fTrackEta    = 0;
45 //  fTrackPhi    = 0;
46 //  fTrackPDG    = 0;
47   fNTracks     = 0;
48   fPartonCode = 0;
49 }
50
51 void AliEMCALParton::SetTrackList(Int_t NTracks, Float_t* Energy,  Float_t* Eta, Float_t* Phi, Int_t* PDG)
52 {
53 // Set the stored tracklist
54   fNTracks     = NTracks;
55   for (Int_t i=0;i<NTracks;i++)
56   {
57     fTrackEnergy[i] = Energy[i] ;
58     fTrackEta[i]    = Eta[i];
59     fTrackPhi[i]    = Phi[i];
60     fTrackPDG[i]    = PDG[i];
61   }
62 }
63
64 void AliEMCALParton::GetTrackList(Float_t* Energy,  Float_t* Eta, Float_t* Phi, Int_t* PDG) const
65 {
66 // retrieves the stored tracklist       
67   for (Int_t i=0;i<fNTracks;i++)
68   {
69     Energy[i] = fTrackEnergy[i] ;
70     Eta[i]    = fTrackEta[i];
71     Phi[i]    = fTrackPhi[i];
72     PDG[i]    = fTrackPDG[i];
73   } 
74 }
75
76
77 //____________________________________________________________________________
78
79 AliEMCALParton::~AliEMCALParton()
80 {
81   // Destructor
82   
83 }