[u/mrichter/AliRoot.git] / THydjet / AliGenHydjet.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

// Generator using Hydjet as an external generator
// The main Hydjet options are accessable for the user through this interface.
// Uses the THydjet implementation of TGenerator.
// Author:
// Rafael Diaz Valdes    (rafael.diaz.valdes@cern.ch)
//
#include <Riostream.h>
#include <THydjet.h>
#include <TParticle.h>
#include <TClonesArray.h>

#include "AliGenHydjet.h"
#include "AliLog.h"
#include "AliGenHydjetEventHeader.h"
#include "AliRun.h"
#include "AliPythiaRndm.h"

using std::cout;
using std::endl;

ClassImp(AliGenHydjet)

AliGenHydjet::AliGenHydjet(Int_t npart) :
  AliGenMC(npart),
    //initial parameters
  fFrame("CMS"),      // Reference frame
  fAtomicWeigth(207),        // Projectile-Target atomic weight
  fIfbtype(0),          // centrality type
  fFixImpactParam(0),          // fixed impact parameter
  fMinImpactParam(0.),         // minimum impact parameter
  fMaxImpactParam(1.),         // maximum impact parameter
  fSoftMult(20000),      // mean soft multiplicity
  //hydro parameters
  fJetProd(2),          // flag Jet production (nhsel)
  fYflow(5.),         // max longitudinal flow
  fTflow(1.),         // max transverse flow
  fSoftFract(1.),         // Soft multiplicity fraction
  // PYTHIA parameters
  fDijetProd(1),          // flag dijet production
  fMinPtHard(10.),        // min pt hard
  fStructFunction(7),          // Structure Function (default CTEQ5M)
  fMultipleInt(0),          // flag multiple interaction

  fHydjet(0),

  fSelectAll(0),
  fFlavor(0)
    
{
// Default PbPb collisions at 5. 5 TeV
//
  fEnergyCMS = 5500.;      //Energy cms
  fName = "Hydjet";
  fTitle = "Particle Generator using Hydjet";
  // Set random number generator 
  if (!AliPythiaRndm::GetPythiaRandom()) 
    AliPythiaRndm::SetPythiaRandom(GetRandom());
}

AliGenHydjet::~AliGenHydjet()
{
// Destructor
}

void AliGenHydjet::Init()
{
// Initialisation

    SetMC(new THydjet(fEnergyCMS, fFrame,fAtomicWeigth,
                      fIfbtype,fMinImpactParam, fMaxImpactParam,
                      fFixImpactParam,fSoftMult));

    fHydjet=(THydjet*) fMCEvGen;

    // init hydro common blocks
    fHydjet->SetNHSEL(fJetProd);
    fHydjet->SetYLFL(fYflow);
    fHydjet->SetYTFL(fTflow);
    fHydjet->SetFPART(fSoftFract);
    // init PYTHIA common block
    fHydjet->SetMSEL(fDijetProd);
    fHydjet->SetPTMIN(fMinPtHard);
    fHydjet->SetCKIN(3,fMinPtHard);
    fHydjet->SetMSTP(51,fStructFunction);
    fHydjet->SetMSTP(81,0);
    fHydjet->SetMSTU(21,1); 
    fHydjet->SetPARU(14,1.);
    
//  Initialize Hydjet  
    fHydjet->Initialize();

}

void AliGenHydjet::Generate()
{

///////////////////////////////////////////////////////////////////////////////////////////
  Float_t polar[3]    =   {0,0,0};
  Float_t origin[3]   =   {0,0,0};
  Float_t origin0[3]  =   {0,0,0};
  Float_t time0 = 0.;
  Float_t p[3];
  Float_t tof;

//  converts from mm/c to s
  const Float_t kconv = 0.001/2.999792458e8;
//
  Int_t nt  = 0;
  Int_t j, kf, ks, imo;
  kf = 0;

  for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
  time0 = fTimeOrigin;
  if(fVertexSmear == kPerEvent) {
      Vertex();
      for (j=0; j < 3; j++) origin0[j] = fVertex[j];
      time0 = fTime;
  }

  while(1)
  {
//    Generate one event
      fHydjet->GenerateEvent();
      fHydjet->ImportParticles(&fParticles,"All");

      Int_t np = fParticles.GetEntriesFast();
      Int_t nc = 0;
      if (np == 0 ) continue;
      Int_t i;
      Int_t* pSelected  = new Int_t[np];

      for (i = 0; i < np; i++) {
           pSelected[i] = 0;
      }

//      Get event vertex
      fVertex[0] = origin0[0];
      fVertex[1] = origin0[1];
      fVertex[2] = origin0[2];
      fTime = time0;

//
// Now select the final state particles
//

      for(i = 0; i<np; i++){
         TParticle *  iparticle = (TParticle *) fParticles.At(i);
         // Is this a final state particle ?
         if (!Stable(iparticle)) continue;
         Bool_t selected = kTRUE;
         kf = iparticle->GetPdgCode();
         ks = iparticle->GetStatusCode();
         if(!fSelectAll) selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
         // Put particle on the stack if selected
         if(selected) {
           nc++;
           pSelected[i] = 1;
         } // selected
      } // particle loop final state


//
// Write particles to stack
//

      for(i = 0; i<np; i++) {
         TParticle *  iparticle = (TParticle *) fParticles.At(i);
         Bool_t  hasDaughter = (iparticle->GetFirstDaughter() > 0);
         if(pSelected[i]){
           kf = iparticle->GetPdgCode();
           ks = iparticle->GetStatusCode();
           p[0] = iparticle->Px();
           p[1] = iparticle->Py();
           p[2] = iparticle->Pz();
           origin[0] = fVertex[0]+iparticle->Vx()/10;
           origin[1] = fVertex[1]+iparticle->Vy()/10;
           origin[2] = fVertex[2]+iparticle->Vz()/10;
           tof   = fTime + kconv*iparticle->T();

           imo = -1;
           //imo = iparticle->GetFirstMother();
           //if(imo == 0) imo =-1;
           Bool_t tFlag = (fTrackIt && !hasDaughter);
           PushTrack(tFlag,imo,kf,p,origin,polar,tof,kPNoProcess,nt, 1., ks);
           KeepTrack(nt);
         } // if selected
      } // particle loop
      delete[] pSelected;

      AliInfo(Form("\n I've put %i particles on the stack \n",nc));
      if (nc > 0) break;
  } // event loop
  MakeHeader();
}


Bool_t AliGenHydjet::SelectFlavor(Int_t pid)
{
// Select flavor of particle
// 0: all
// 4: charm and beauty
// 5: beauty
    Bool_t res = 0;
    if (fFlavor == 0) {
	res = kTRUE;
    } else {
	Int_t ifl = TMath::Abs(pid/100);
	if (ifl > 10) ifl/=10;
	res = (fFlavor == ifl);
    }
    return res;
}

Bool_t AliGenHydjet::Stable(TParticle*  particle) const
{
// Return true for a stable particle
    if (particle->GetFirstDaughter() == 0 )
    {
	return kTRUE;
    } else {
	return kFALSE;
    }
}


///////////////////////////////////
void AliGenHydjet::MakeHeader()
{
// Builds the event header, to be called after each event
    AliGenEventHeader* header = new AliGenHydjetEventHeader("Hydjet");
    ((AliGenHydjetEventHeader*) header)->SetNProduced(fHydjet->GetN());
    ((AliGenHydjetEventHeader*) header)->SetImpactParameter(fHydjet->GetBGEN());
    ((AliGenHydjetEventHeader*) header)->SetNbcol(fHydjet->GetNBCOL());
    ((AliGenHydjetEventHeader*) header)->SetNpart(fHydjet->GetNPART());
    ((AliGenHydjetEventHeader*) header)->SetNpyt(fHydjet->GetNPYT());
    ((AliGenHydjetEventHeader*) header)->SetNhyd(fHydjet->GetNHYD());
  // draw generation values
  cout << "*********Hyflow ********** " << endl;
  cout << " YTFL " << fHydjet->GetYTFL() << endl;
  cout << " YLFL " << fHydjet->GetYLFL() << endl;
  cout << " FPART " << fHydjet->GetFPART() << endl;
  cout << "*********Hyjpar ********* " << endl;
  cout << " GetNHSEL() " << fHydjet->GetNHSEL() << endl;
  cout << " GetPTMIN() " << fHydjet->GetPTMIN() << endl;
  cout << " GetNJET() " << fHydjet->GetNJET() << endl;
  cout << "*********Hyfpar ********* " << endl;
  cout << " GetBGEN() " << fHydjet->GetBGEN() << endl;
  cout << " GetNBCOL() " << fHydjet->GetNBCOL() << endl;
  cout << " GetNPART() " << fHydjet->GetNPART() << endl;
  cout << " GetNPYT() " << fHydjet->GetNPYT() << endl;
  cout << " GetNHYD() " << fHydjet->GetNHYD() << endl;
  

// Event Vertex
    header->SetPrimaryVertex(fVertex);
    header->SetInteractionTime(fTime);
    AddHeader(header);
    fCollisionGeometry = (AliGenHydjetEventHeader*)  header;
}

void AliGenHydjet::AddHeader(AliGenEventHeader* header)
{
    // Passes header either to the container or to gAlice
    if (fContainer) {
	fContainer->AddHeader(header);
    } else {
	gAlice->SetGenEventHeader(header);
    }
}


void AliGenHydjet::Copy(TObject &) const
{
  Fatal("Copy","Not implemented!\n");
}

AliGenHydjet& AliGenHydjet::operator=(const  AliGenHydjet& rhs)
{
    rhs.Copy(*this); 
    return (*this);
}
Commit	Line	Data
cb220f83	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
	18	// Generator using Hydjet as an external generator
	19	// The main Hydjet options are accessable for the user through this interface.
	20	// Uses the THydjet implementation of TGenerator.
	21	// Author:
	22	// Rafael Diaz Valdes (rafael.diaz.valdes@cern.ch)
	23	//
	24	#include <Riostream.h>
	25	#include <THydjet.h>
	26	#include <TParticle.h>
	27	#include <TClonesArray.h>
	28
	29	#include "AliGenHydjet.h"
3f4e7ad8	30	#include "AliLog.h"
cb220f83	31	#include "AliGenHydjetEventHeader.h"
	32	#include "AliRun.h"
	33	#include "AliPythiaRndm.h"
	34
dc3e3e91	35	using std::cout;
	36	using std::endl;
	37
cb220f83	38	ClassImp(AliGenHydjet)
	39
	40	AliGenHydjet::AliGenHydjet(Int_t npart) :
	41	AliGenMC(npart),
	42	//initial parameters
cb220f83	43	fFrame("CMS"), // Reference frame
	44	fAtomicWeigth(207), // Projectile-Target atomic weight
	45	fIfbtype(0), // centrality type
	46	fFixImpactParam(0), // fixed impact parameter
	47	fMinImpactParam(0.), // minimum impact parameter
	48	fMaxImpactParam(1.), // maximum impact parameter
	49	fSoftMult(20000), // mean soft multiplicity
	50	//hydro parameters
	51	fJetProd(2), // flag Jet production (nhsel)
	52	fYflow(5.), // max longitudinal flow
	53	fTflow(1.), // max transverse flow
	54	fSoftFract(1.), // Soft multiplicity fraction
	55	// PYTHIA parameters
	56	fDijetProd(1), // flag dijet production
	57	fMinPtHard(10.), // min pt hard
	58	fStructFunction(7), // Structure Function (default CTEQ5M)
	59	fMultipleInt(0), // flag multiple interaction
	60
	61	fHydjet(0),
	62
	63	fSelectAll(0),
	64	fFlavor(0)
	65
	66	{
	67	// Default PbPb collisions at 5. 5 TeV
	68	//
e7c989e4	69	fEnergyCMS = 5500.; //Energy cms
cb220f83	70	fName = "Hydjet";
cb220f83	71	fTitle = "Particle Generator using Hydjet";
9731de01	72	// Set random number generator
	73	if (!AliPythiaRndm::GetPythiaRandom())
	74	AliPythiaRndm::SetPythiaRandom(GetRandom());
cb220f83	75	}
	76
	77	AliGenHydjet::~AliGenHydjet()
	78	{
	79	// Destructor
cb220f83	80	}
	81
	82	void AliGenHydjet::Init()
	83	{
	84	// Initialisation
	85
	86	SetMC(new THydjet(fEnergyCMS, fFrame,fAtomicWeigth,
	87	fIfbtype,fMinImpactParam, fMaxImpactParam,
	88	fFixImpactParam,fSoftMult));
	89
	90	fHydjet=(THydjet*) fMCEvGen;
	91
	92	// init hydro common blocks
	93	fHydjet->SetNHSEL(fJetProd);
	94	fHydjet->SetYLFL(fYflow);
	95	fHydjet->SetYTFL(fTflow);
	96	fHydjet->SetFPART(fSoftFract);
	97	// init PYTHIA common block
	98	fHydjet->SetMSEL(fDijetProd);
	99	fHydjet->SetPTMIN(fMinPtHard);
	100	fHydjet->SetCKIN(3,fMinPtHard);
	101	fHydjet->SetMSTP(51,fStructFunction);
	102	fHydjet->SetMSTP(81,0);
	103	fHydjet->SetMSTU(21,1);
	104	fHydjet->SetPARU(14,1.);
	105
	106	// Initialize Hydjet
	107	fHydjet->Initialize();
	108
	109	}
	110
	111	void AliGenHydjet::Generate()
	112	{
	113
	114	///////////////////////////////////////////////////////////////////////////////////////////
	115	Float_t polar[3] = {0,0,0};
	116	Float_t origin[3] = {0,0,0};
	117	Float_t origin0[3] = {0,0,0};
21391258	118	Float_t time0 = 0.;
cb220f83	119	Float_t p[3];
	120	Float_t tof;
	121
	122	// converts from mm/c to s
	123	const Float_t kconv = 0.001/2.999792458e8;
	124	//
	125	Int_t nt = 0;
	126	Int_t j, kf, ks, imo;
	127	kf = 0;
	128
	129	for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
21391258	130	time0 = fTimeOrigin;
cb220f83	131	if(fVertexSmear == kPerEvent) {
	132	Vertex();
	133	for (j=0; j < 3; j++) origin0[j] = fVertex[j];
21391258	134	time0 = fTime;
cb220f83	135	}
	136
	137	while(1)
	138	{
	139	// Generate one event
	140	fHydjet->GenerateEvent();
9731de01	141	fHydjet->ImportParticles(&fParticles,"All");
cb220f83	142
9731de01	143	Int_t np = fParticles.GetEntriesFast();
cb220f83	144	Int_t nc = 0;
	145	if (np == 0 ) continue;
	146	Int_t i;
	147	Int_t* pSelected = new Int_t[np];
	148
	149	for (i = 0; i < np; i++) {
	150	pSelected[i] = 0;
	151	}
	152
	153	// Get event vertex
	154	fVertex[0] = origin0[0];
	155	fVertex[1] = origin0[1];
	156	fVertex[2] = origin0[2];
21391258	157	fTime = time0;
cb220f83	158
	159	//
	160	// Now select the final state particles
	161	//
	162
	163	for(i = 0; i<np; i++){
9731de01	164	TParticle * iparticle = (TParticle *) fParticles.At(i);
cb220f83	165	// Is this a final state particle ?
	166	if (!Stable(iparticle)) continue;
	167	Bool_t selected = kTRUE;
	168	kf = iparticle->GetPdgCode();
	169	ks = iparticle->GetStatusCode();
	170	if(!fSelectAll) selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
	171	// Put particle on the stack if selected
	172	if(selected) {
	173	nc++;
	174	pSelected[i] = 1;
	175	} // selected
	176	} // particle loop final state
	177
	178
	179	//
	180	// Write particles to stack
	181	//
	182
	183	for(i = 0; i<np; i++) {
9731de01	184	TParticle * iparticle = (TParticle *) fParticles.At(i);
cb220f83	185	Bool_t hasDaughter = (iparticle->GetFirstDaughter() > 0);
	186	if(pSelected[i]){
	187	kf = iparticle->GetPdgCode();
	188	ks = iparticle->GetStatusCode();
	189	p[0] = iparticle->Px();
	190	p[1] = iparticle->Py();
	191	p[2] = iparticle->Pz();
	192	origin[0] = fVertex[0]+iparticle->Vx()/10;
	193	origin[1] = fVertex[1]+iparticle->Vy()/10;
	194	origin[2] = fVertex[2]+iparticle->Vz()/10;
21391258	195	tof = fTime + kconv*iparticle->T();
cb220f83	196
	197	imo = -1;
	198	//imo = iparticle->GetFirstMother();
	199	//if(imo == 0) imo =-1;
	200	Bool_t tFlag = (fTrackIt && !hasDaughter);
	201	PushTrack(tFlag,imo,kf,p,origin,polar,tof,kPNoProcess,nt, 1., ks);
	202	KeepTrack(nt);
	203	} // if selected
	204	} // particle loop
	205	delete[] pSelected;
	206
3f4e7ad8	207	AliInfo(Form("\n I've put %i particles on the stack \n",nc));
cb220f83	208	if (nc > 0) break;
	209	} // event loop
	210	MakeHeader();
	211	}
	212
	213
	214	Bool_t AliGenHydjet::SelectFlavor(Int_t pid)
	215	{
	216	// Select flavor of particle
	217	// 0: all
	218	// 4: charm and beauty
	219	// 5: beauty
	220	Bool_t res = 0;
	221	if (fFlavor == 0) {
	222	res = kTRUE;
	223	} else {
	224	Int_t ifl = TMath::Abs(pid/100);
	225	if (ifl > 10) ifl/=10;
	226	res = (fFlavor == ifl);
	227	}
	228	return res;
	229	}
	230
	231	Bool_t AliGenHydjet::Stable(TParticle* particle) const
	232	{
	233	// Return true for a stable particle
	234	if (particle->GetFirstDaughter() == 0 )
	235	{
	236	return kTRUE;
	237	} else {
	238	return kFALSE;
	239	}
	240	}
	241
	242
	243	///////////////////////////////////
	244	void AliGenHydjet::MakeHeader()
	245	{
	246	// Builds the event header, to be called after each event
	247	AliGenEventHeader* header = new AliGenHydjetEventHeader("Hydjet");
	248	((AliGenHydjetEventHeader*) header)->SetNProduced(fHydjet->GetN());
	249	((AliGenHydjetEventHeader*) header)->SetImpactParameter(fHydjet->GetBGEN());
	250	((AliGenHydjetEventHeader*) header)->SetNbcol(fHydjet->GetNBCOL());
	251	((AliGenHydjetEventHeader*) header)->SetNpart(fHydjet->GetNPART());
	252	((AliGenHydjetEventHeader*) header)->SetNpyt(fHydjet->GetNPYT());
	253	((AliGenHydjetEventHeader*) header)->SetNhyd(fHydjet->GetNHYD());
	254	// draw generation values
	255	cout << "*******Hyflow ******** " << endl;
	256	cout << " YTFL " << fHydjet->GetYTFL() << endl;
	257	cout << " YLFL " << fHydjet->GetYLFL() << endl;
	258	cout << " FPART " << fHydjet->GetFPART() << endl;
	259	cout << "*******Hyjpar ******* " << endl;
	260	cout << " GetNHSEL() " << fHydjet->GetNHSEL() << endl;
	261	cout << " GetPTMIN() " << fHydjet->GetPTMIN() << endl;
	262	cout << " GetNJET() " << fHydjet->GetNJET() << endl;
	263	cout << "*******Hyfpar ******* " << endl;
	264	cout << " GetBGEN() " << fHydjet->GetBGEN() << endl;
	265	cout << " GetNBCOL() " << fHydjet->GetNBCOL() << endl;
	266	cout << " GetNPART() " << fHydjet->GetNPART() << endl;
	267	cout << " GetNPYT() " << fHydjet->GetNPYT() << endl;
	268	cout << " GetNHYD() " << fHydjet->GetNHYD() << endl;
	269
	270
	271	// Event Vertex
272	header->SetPrimaryVertex(fVertex);
21391258	273	header->SetInteractionTime(fTime);
cb220f83	274	AddHeader(header);
	275	fCollisionGeometry = (AliGenHydjetEventHeader*) header;
	276	}
	277
	278	void AliGenHydjet::AddHeader(AliGenEventHeader* header)
	279	{
	280	// Passes header either to the container or to gAlice
	281	if (fContainer) {
	282	fContainer->AddHeader(header);
	283	} else {
	284	gAlice->SetGenEventHeader(header);
	285	}
	286	}
	287
	288
	289	void AliGenHydjet::Copy(TObject &) const
	290	{
	291	Fatal("Copy","Not implemented!\n");
	292	}
	293
	294	AliGenHydjet& AliGenHydjet::operator=(const AliGenHydjet& rhs)
	295	{
	296	rhs.Copy(*this);
	297	return (*this);
	298	}
	299