]>
Commit | Line | Data |
---|---|---|
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; | |
826fd892 | 22 | class TClonesArray; |
2c890605 | 23 | |
24 | //------------------------------------------------------------- | |
25 | class AliGenCorrHF : public AliGenMC | |
26 | { | |
27 | public: | |
28 | AliGenCorrHF(); | |
b33adf51 | 29 | AliGenCorrHF(Int_t npart, Int_t idquark, Int_t energy); |
30 | AliGenCorrHF(char* tname, Int_t npart, Int_t idquark, Int_t energy); | |
2c890605 | 31 | |
32 | virtual ~AliGenCorrHF(); | |
33 | virtual void Generate(); | |
34 | virtual void Init(); | |
35 | // force decay type | |
36 | virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;} | |
37 | ||
9fd56238 | 38 | // Hadron pair composition functions, needed by GetHadronPair |
39 | static void IpCharm(TH2F *hProbHH, Int_t &pdg3, Int_t &pdg4); | |
40 | static void IpBeauty(TH2F *hProbHH, Int_t &pdg3, Int_t &pdg4); | |
2c890605 | 41 | |
42 | // Computation of cumulative sums of cell wght-s, needed by GetQuarkPair | |
5cc3bcd2 | 43 | Double_t ComputeIntegral(TFile* fG); |
2c890605 | 44 | |
b33adf51 | 45 | // fG - input file with QQbar kinematical grid (TTree) and fragm. functions (24 TH2-s) |
2c890605 | 46 | static void GetQuarkPair(TFile* fG, Double_t* fInt, Double_t &y1, Double_t &y2, Double_t &pt1, Double_t &pt2, Double_t &dphi); |
47 | 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); | |
826fd892 | 48 | |
49 | //Loading tracks in the stack | |
50 | void LoadTracks(Int_t iquark, Float_t *pq, Int_t iPart, | |
51 | Float_t *p, Int_t np, TClonesArray* particles, | |
52 | Float_t *origine0, Float_t *polar, Float_t wgtp, | |
53 | Float_t wgtch, Int_t &nt, Int_t ncsel, Int_t *pSelected, | |
54 | Int_t *trackIt); | |
55 | //Setting the flag for Background transportation while using SetForceDecay() | |
18e09c20 | 56 | void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;} |
2c890605 | 57 | |
58 | protected: | |
59 | TString fFileName; // Name of the input file | |
60 | TFile* fFile; //! Pointer to input file | |
b33adf51 | 61 | Int_t fQuark; // Quark type |
62 | Int_t fEnergy; // p-p c.m.s. energy | |
2c890605 | 63 | Float_t fBias; // Biasing factor |
64 | Int_t fTrials; // Number of trials | |
18e09c20 | 65 | Bool_t fSelectAll; // Flag for transportation of Background while using SetForceDecay() |
2c890605 | 66 | AliDecayer* fDecayer; //! Pointer to pythia object for decays |
67 | ||
68 | private: | |
93a2041b | 69 | AliGenCorrHF(const AliGenCorrHF &CorrHF); |
70 | AliGenCorrHF & operator=(const AliGenCorrHF & rhs); | |
71 | ||
826fd892 | 72 | Double_t* fgIntegral; //! Pointer to array of cumulative sums of wght-s |
2c890605 | 73 | static Int_t fgnptbins; // =12 Number of bins for the fragm. |
74 | // function dependence on quark pt | |
b33adf51 | 75 | // Number of the grid bins in deltaphi, y and pt: 18, 30 and 50 |
2c890605 | 76 | static Double_t fgdph[19]; // deltaphi bin coordinates |
77 | static Double_t fgy[31]; // y bin coordinates | |
b33adf51 | 78 | static Double_t fgpt[51]; // pt bin coordinates |
2c890605 | 79 | static Double_t fgptbmin[12]; // min & max coordinates of pt bins for |
80 | static Double_t fgptbmax[12]; // the fragm. function | |
826fd892 | 81 | |
82 | ClassDef(AliGenCorrHF,1) // Generator using parameterized QQbar & fragm. functions | |
2c890605 | 83 | }; |
84 | #endif | |
85 | ||
86 | ||
87 | ||
88 | ||
89 | ||
90 | ||
91 | ||
92 | ||
93 |