| 2c890605 |
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; |
| 9fd56238 |
19 | class TH2F; |
| 2c890605 |
20 | class TFile; |
| 21 | class TString; |
| 22 | |
| 23 | //------------------------------------------------------------- |
| 24 | class AliGenCorrHF : public AliGenMC |
| 25 | { |
| 26 | public: |
| 27 | AliGenCorrHF(); |
| b33adf51 |
28 | AliGenCorrHF(Int_t npart, Int_t idquark, Int_t energy); |
| 29 | AliGenCorrHF(char* tname, Int_t npart, Int_t idquark, Int_t energy); |
| 2c890605 |
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 | |
| 9fd56238 |
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); |
| 2c890605 |
40 | |
| 41 | // Computation of cumulative sums of cell wght-s, needed by GetQuarkPair |
| 5cc3bcd2 |
42 | Double_t ComputeIntegral(TFile* fG); |
| 2c890605 |
43 | |
| b33adf51 |
44 | // fG - input file with QQbar kinematical grid (TTree) and fragm. functions (24 TH2-s) |
| 2c890605 |
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 |
| b33adf51 |
51 | Int_t fQuark; // Quark type |
| 52 | Int_t fEnergy; // p-p c.m.s. energy |
| 2c890605 |
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: |
| 93a2041b |
58 | AliGenCorrHF(const AliGenCorrHF &CorrHF); |
| 59 | AliGenCorrHF & operator=(const AliGenCorrHF & rhs); |
| 60 | |
| 5cc3bcd2 |
61 | Double_t* fgIntegral; //! Pointer to array of cumulative sums of wght-s |
| 2c890605 |
62 | static Int_t fgnptbins; // =12 Number of bins for the fragm. |
| 63 | // function dependence on quark pt |
| b33adf51 |
64 | // Number of the grid bins in deltaphi, y and pt: 18, 30 and 50 |
| 2c890605 |
65 | static Double_t fgdph[19]; // deltaphi bin coordinates |
| 66 | static Double_t fgy[31]; // y bin coordinates |
| b33adf51 |
67 | static Double_t fgpt[51]; // pt bin coordinates |
| 2c890605 |
68 | static Double_t fgptbmin[12]; // min & max coordinates of pt bins for |
| 69 | static Double_t fgptbmax[12]; // the fragm. function |
| cb11fd27 |
70 | |
| 2c890605 |
71 | ClassDef(AliGenCorrHF,1) // Generator using parameterized QQbar & fragm. functions |
| 72 | }; |
| 73 | #endif |
| 74 | |
| 75 | |
| 76 | |
| 77 | |
| 78 | |
| 79 | |
| 80 | |
| 81 | |
| 82 | |
| 83 | |
| 5cc3bcd2 |
84 | |
git://git.uio.no / u / mrichter / AliRoot.git / blame