Updates needed for full jet reconstruction (charged + emcal) [Magali Estienne]

[u/mrichter/AliRoot.git] / JETAN / AliJet.cxx

diff --git a/JETAN/AliJet.cxx b/JETAN/AliJet.cxx
index cf9cee8a37d37bcd85fa3d7c8d0b6f06f87e234c..23e4e68739ed3c9fc41f115b0a30c15ced6b7c84 100644 (file)
--- a/JETAN/AliJet.cxx
+++ b/JETAN/AliJet.cxx
@@ -12,14 +12,18 @@
  * about the suitability of this software for any purpose. It is          *
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
+
+/* $Id$ */
  
 //---------------------------------------------------------------------
 // Jet class 
 // Stores the output of a jet algorithm
 // Author: jgcn@mda.cinvestav.mx
 //---------------------------------------------------------------------
+
  
 #include <Riostream.h>
+//#include <vector.h>
 #include <TClonesArray.h>
 #include <TLorentzVector.h>
 
@@ -37,7 +41,15 @@ AliJet::AliJet():
   fJets(0),
   fEtaIn(0),
   fPhiIn(0),
-  fPtIn(0)
+  fPtIn(0),
+  fPtChPtCutIn(0),
+  fEnTotChPtCutIn(0),
+  fVectorSizeIn(0),
+  fDetIn(0),
+  fVPx(0),
+  fVPy(0),
+  fVPz(0)
+             //  fVectorIn(0)
 {
   // Default constructor
   fJets = new TClonesArray("TLorentzVector",1000);
@@ -48,6 +60,10 @@ AliJet::AliJet():
   fPtFromSignal = TArrayF();
   fMultiplicities = TArrayI();
   fNCells = TArrayI();
+  fPtChPtCutIn = TArrayF();
+  fEnTotChPtCutIn = TArrayF();
+  fVectorSizeIn = TArrayI();
+  fDetIn = TArrayI();
 } 
 
 ////////////////////////////////////////////////////////////////////////
@@ -242,6 +258,53 @@ void AliJet::SetPhiIn(Float_t* x)
 {
   if (fNInput>0) fPhiIn.Set(fNInput, x);
 }
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetPtChargedPtCutIn(Float_t* x)
+{
+  if (fNInput>0) fPtChPtCutIn.Set(fNInput, x);
+}
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetEnTotChargedPtCutIn(Float_t* x)
+{
+  if (fNInput>0) fEnTotChPtCutIn.Set(fNInput, x);
+}
+
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetDetectorFlagIn(Int_t* x)
+{
+  if (fNInput>0) fDetIn.Set(fNInput, x);
+}
+
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetVectorSizeIn(Int_t* x)
+{
+  if (fNInput>0) fVectorSizeIn.Set(fNInput, x);
+}
+
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetVectorPxIn(vector< vector<Float_t> > pxT)
+{
+  fVPx = pxT;;
+}
+
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetVectorPyIn(vector< vector<Float_t> > pyT)
+{
+  fVPy = pyT;;
+}
+
+////////////////////////////////////////////////////////////////////////
+
+void AliJet::SetVectorPzIn(vector< vector<Float_t> > pzT)
+{
+  fVPz = pzT;;
+}
 
 ////////////////////////////////////////////////////////////////////////
 
@@ -283,6 +346,10 @@ void AliJet::ClearJets(Option_t *option)
   fEtaIn.Set(0);
   fPtIn.Set(0);
   fNCells.Set(0);
+  fPtChPtCutIn.Set(0);
+  fEnTotChPtCutIn.Set(0);
+  fVectorSizeIn.Set(0);
+  fDetIn.Set(0);
 }
 
 ////////////////////////////////////////////////////////////////////////