o small fix

[u/mrichter/AliRoot.git] / EVGEN / AliGenCorrHF.h

diff --git a/EVGEN/AliGenCorrHF.h b/EVGEN/AliGenCorrHF.h
index 4341e080102be8a7f5af855bc17838a20302a7c9..25db14861e79414c15f77cb4f421c04dc529a608 100644 (file)
--- a/EVGEN/AliGenCorrHF.h
+++ b/EVGEN/AliGenCorrHF.h
@@ -19,6 +19,7 @@ class TParticle;
 class TH2F;
 class TFile;
 class TString;
+class TClonesArray;
 
 //-------------------------------------------------------------
 class AliGenCorrHF : public AliGenMC
@@ -45,6 +46,15 @@ class AliGenCorrHF : public AliGenMC
     static void GetQuarkPair(TFile* fG, Double_t* fInt, Double_t &y1, Double_t &y2, Double_t &pt1, Double_t &pt2, Double_t &dphi);              
     static void GetHadronPair(TFile* fG, Int_t idq, Double_t y1, Double_t y2, Double_t pt1, Double_t pt2, Int_t &id3, Int_t &id4, Double_t &pz3, Double_t &pz4, Double_t &pt3, Double_t &pt4); 
 
+    //Loading tracks in the stack
+    void LoadTracks(Int_t iquark, Float_t *pq, Int_t iPart, 
+                   Float_t *p, Int_t np, TClonesArray* particles,
+                   Float_t *origine0, Float_t *polar, Float_t wgtp, 
+                   Float_t wgtch, Int_t &nt, Int_t ncsel, Int_t *pSelected, 
+                   Int_t *trackIt);
+    //Setting the flag for Background transportation while using SetForceDecay()
+    void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;}
+
  protected:
     TString     fFileName;    // Name of the input file
     TFile*      fFile;        //! Pointer to input file
@@ -52,6 +62,7 @@ class AliGenCorrHF : public AliGenMC
     Int_t       fEnergy;      // p-p c.m.s. energy 
     Float_t     fBias;        // Biasing factor
     Int_t       fTrials;      // Number of trials
+    Bool_t      fSelectAll;   // Flag for transportation of Background while using SetForceDecay()
     AliDecayer* fDecayer;     //! Pointer to pythia object for decays
 
  private:
@@ -80,5 +91,3 @@ class AliGenCorrHF : public AliGenMC
 
 
 
-
-