Updated version.
[u/mrichter/AliRoot.git] / EVGEN / AliGenCorrHF.h
1 #ifndef ALIGENCORRHF_H
2 #define ALIGENCORRHF_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //
9 // Class to generate correlated Heavy Flavor hadron pairs using paramtrized
10 // kinematics of quark pairs from some generator and quark fragmentation functions.
11 // Is a generalisation of AliGenParam class for correlated pairs of hadrons.
12 // Author: S. Grigoryan, LPC Clermont-Fd & YerPhI, Smbat.Grigoryan@cern.ch
13 //
14
15 #include "AliGenMC.h"
16
17 class AliPythia;
18 class TParticle;
19 class TH2F;
20 class TFile;
21 class TString;
22
23 //-------------------------------------------------------------
24 class AliGenCorrHF : public AliGenMC
25 {
26  public:
27     AliGenCorrHF();
28     AliGenCorrHF(Int_t npart, Int_t idquark, Int_t energy);
29     AliGenCorrHF(char* tname, Int_t npart, Int_t idquark, Int_t energy);
30      
31     virtual ~AliGenCorrHF();
32     virtual void Generate();
33     virtual void Init();
34     // force decay type
35     virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;}
36
37     // Hadron pair composition functions, needed by GetHadronPair
38     static void IpCharm(TH2F *hProbHH, Int_t &pdg3, Int_t &pdg4);
39     static void IpBeauty(TH2F *hProbHH, Int_t &pdg3, Int_t &pdg4);
40
41     // Computation of cumulative sums of cell wght-s, needed by GetQuarkPair
42     Double_t ComputeIntegral(TFile* fG);
43
44     // fG - input file with QQbar kinematical grid (TTree) and fragm. functions (24 TH2-s)
45     static void GetQuarkPair(TFile* fG, Double_t* fInt, Double_t &y1, Double_t &y2, Double_t &pt1, Double_t &pt2, Double_t &dphi);              
46     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); 
47
48  protected:
49     TString     fFileName;    // Name of the input file
50     TFile*      fFile;        //! Pointer to input file
51     Int_t       fQuark;       // Quark type
52     Int_t       fEnergy;      // p-p c.m.s. energy 
53     Float_t     fBias;        // Biasing factor
54     Int_t       fTrials;      // Number of trials
55     AliDecayer* fDecayer;     //! Pointer to pythia object for decays
56
57  private:
58     AliGenCorrHF(const AliGenCorrHF &CorrHF);
59     AliGenCorrHF & operator=(const AliGenCorrHF & rhs);
60
61     Double_t* fgIntegral; //! Pointer to array of cumulative sums of wght-s
62     static Int_t  fgnptbins;             // =12 Number of bins for the fragm. 
63                                          //   function dependence on quark pt
64     // Number of the grid bins in deltaphi, y and pt:  18, 30 and 50
65     static Double_t fgdph[19];           // deltaphi bin coordinates  
66     static Double_t fgy[31];             // y bin coordinates
67     static Double_t fgpt[51];            // pt bin coordinates
68     static Double_t fgptbmin[12];        // min & max coordinates of pt bins for
69     static Double_t fgptbmax[12];        // the fragm. function
70
71     ClassDef(AliGenCorrHF,1)  // Generator using parameterized QQbar & fragm. functions
72 };
73 #endif
74
75
76
77
78
79
80
81
82
83
84