Update timestamps for new AMANDA simulation (17/02/2015)
[u/mrichter/AliRoot.git] / EVGEN / AliGenCorrHF.h
CommitLineData
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
17class AliPythia;
18class TParticle;
9fd56238 19class TH2F;
2c890605 20class TFile;
21class TString;
826fd892 22class TClonesArray;
2c890605 23
24//-------------------------------------------------------------
25class 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