[u/mrichter/AliRoot.git] / EVGEN / AliGenMC.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$ */

// Base class for generators using external MC generators.
// For example AliGenPythia using Pythia.
// Provides basic functionality: setting of kinematic cuts on 
// decay products and particle selection.
// andreas.morsch@cern.ch

#include <TClonesArray.h>
#include <TMath.h>
#include <TPDGCode.h>
#include <TParticle.h>

#include "AliGenMC.h"
#include "AliRun.h"
#include "AliGeometry.h"

ClassImp(AliGenMC)

AliGenMC::AliGenMC()
    :AliGenerator(),
     fParticles("TParticle", 1000),
     fParentSelect(8),
     fChildSelect(8),
     fCutOnChild(0),
     fChildPtMin(0.),
     fChildPtMax(1.e10),
     fChildPMin(0.),
     fChildPMax(1.e10),
     fChildPhiMin(0.),
     fChildPhiMax(2. * TMath::Pi()),
     fChildThetaMin(0.),
     fChildThetaMax(TMath::Pi()),
     fChildYMin(-12.),
     fChildYMax(12.),
     fXingAngleX(0.),
     fXingAngleY(0.),
     fForceDecay(kAll),
     fMaxLifeTime(1.e-15),
     fDyBoost(0.),
     fGeometryAcceptance(0),
     fPdgCodeParticleforAcceptanceCut(0),
     fNumberOfAcceptedParticles(0),
     fNprimaries(0)
{
// Default Constructor
  fAProjectile = 1;
  fZProjectile = 1;
  fATarget = 1;
  fZTarget = 1;
  fProjectile = "P";
  fTarget = "P";
}

AliGenMC::AliGenMC(Int_t npart)
    :AliGenerator(npart),
     fParticles("TParticle", 1000),
     fParentSelect(8),
     fChildSelect(8),
     fCutOnChild(0),
     fChildPtMin(0.),
     fChildPtMax(1.e10),
     fChildPMin(0.),
     fChildPMax(1.e10),
     fChildPhiMin(0.),
     fChildPhiMax(2. * TMath::Pi()),
     fChildThetaMin(0.),
     fChildThetaMax(TMath::Pi()),
     fChildYMin(-12.),
     fChildYMax(12.),
     fXingAngleX(0.),
     fXingAngleY(0.),
     fForceDecay(kAll),
     fMaxLifeTime(1.e-15),
     fDyBoost(0.),
     fGeometryAcceptance(0),
     fPdgCodeParticleforAcceptanceCut(0),
     fNumberOfAcceptedParticles(0),
     fNprimaries(0)
{
//  Constructor
// 
  fAProjectile = 1;
  fZProjectile = 1;
  fATarget = 1;
  fZTarget = 1;
  fProjectile = "P";
  fTarget = "P";
  for (Int_t i=0; i<8; i++) fParentSelect[i]=fChildSelect[i]=0;
}

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

void AliGenMC::Init()
{
//
//  Initialization
    switch (fForceDecay) {
    case kBSemiElectronic:
    case kSemiElectronic:
    case kDiElectron:
    case kBJpsiDiElectron:
    case kBPsiPrimeDiElectron:
	fChildSelect[0] = kElectron;	
	break;
    case kHardMuons:	
    case kBSemiMuonic:
    case kSemiMuonic:
    case kDiMuon:
    case kBJpsiDiMuon:
    case kBPsiPrimeDiMuon:
    case kPiToMu:
    case kKaToMu:
    case kWToMuon:
    case kWToCharmToMuon:
    case kZDiMuon:
    case kZDiElectron:
	fChildSelect[0]=kMuonMinus;
	break;
    case kWToCharm:
	break;
    case kHadronicD:
    case kHadronicDWithout4Bodies:
	fChildSelect[0]=kPiPlus;
	fChildSelect[1]=kKPlus;
	break;
    case kPhiKK:
	fChildSelect[0]=kKPlus;
	break;
    case kBJpsi:
	fChildSelect[0]= 443;
	break;
   case kChiToJpsiGammaToMuonMuon:
        fChildSelect[0]= 22;
        fChildSelect[1]= 13;
        break;
    case kChiToJpsiGammaToElectronElectron:
        fChildSelect[0]= 22;
        fChildSelect[1]= 11;
        break;
    case kLambda:	
	fChildSelect[0]= kProton;
        fChildSelect[1]= 211;
	break;
    case kOmega:
    case kAll:
    case kAllMuonic:
    case kNoDecay:
    case kNoDecayHeavy:
    case kNoDecayBeauty:
    case kNeutralPion:
	break;
    }

    if (fZTarget != 0 && fAProjectile != 0) 
    {
	fDyBoost    = - 0.5 * TMath::Log(Double_t(fZProjectile) * Double_t(fATarget) / 
					 (Double_t(fZTarget)    * Double_t(fAProjectile)));
    }
}


Bool_t AliGenMC::ParentSelected(Int_t ip) const
{
// True if particle is in list of parent particles to be selected
    for (Int_t i=0; i<8; i++)
    {
	if (fParentSelect.At(i) == ip) return kTRUE;
    }
    return kFALSE;
}

Bool_t AliGenMC::ChildSelected(Int_t ip) const
{
// True if particle is in list of decay products to be selected
    for (Int_t i=0; i<5; i++)
    {
	if (fChildSelect.At(i) == ip) return kTRUE;
    }
    return kFALSE;
}

Bool_t AliGenMC::KinematicSelection(TParticle *particle, Int_t flag) const
{
// Perform kinematic selection
    Double_t pz    = particle->Pz();
    Double_t pt    = particle->Pt();
    Double_t p     = particle->P();
    Double_t theta = particle->Theta();
    Double_t mass  = particle->GetCalcMass();
    Double_t mt2   = pt * pt + mass * mass;
    Double_t phi   = particle->Phi();
    Double_t e     = particle->Energy();
    
    if (e == 0.)     
	e = TMath::Sqrt(p * p + mass * mass);

    
    Double_t y, y0;

    if (TMath::Abs(pz) <  e) {
	y = 0.5*TMath::Log((e+pz)/(e-pz));
    } else {
	y = 1.e10;
    }
    
    if (mt2) {
	y0 = 0.5*TMath::Log((e+TMath::Abs(pz))*(e+TMath::Abs(pz))/mt2);
    } else {
	if (TMath::Abs(y) < 1.e10) {
	    y0 = y;
	} else {
	    y0 = 1.e10;
	}
    }
      
    y = (pz < 0) ? -y0 : y0;
    
    if (flag == 0) {
//
// Primary particle cuts
//
//  transverse momentum cut    
	if (pt > fPtMax || pt < fPtMin) {
//	    printf("\n failed pt cut %f %f %f \n",pt,fPtMin,fPtMax);
	    return kFALSE;
	}
//
// momentum cut
	if (p > fPMax || p < fPMin) {
//	    printf("\n failed p cut %f %f %f \n",p,fPMin,fPMax);
	    return kFALSE;
	}
//
// theta cut
	if (theta > fThetaMax || theta < fThetaMin) {
//	    printf("\n failed theta cut %f %f %f \n",theta,fThetaMin,fThetaMax);
	    return kFALSE;
	}
//
// rapidity cut
	if (y > fYMax || y < fYMin) {
//	    printf("\n failed y cut %f %f %f \n",y,fYMin,fYMax);
	    return kFALSE;
	}
//
// phi cut
	if (phi > fPhiMax || phi < fPhiMin) {
//	    printf("\n failed phi cut %f %f %f \n",phi,fPhiMin,fPhiMax);
	    return kFALSE;
	}
    } else {
//
// Decay product cuts
//
//  transverse momentum cut    
	if (pt > fChildPtMax || pt < fChildPtMin) {
//	    printf("\n failed pt cut %f %f %f \n",pt,fChildPtMin,fChildPtMax);
	    return kFALSE;
	}
//
// momentum cut
	if (p > fChildPMax || p < fChildPMin) {
//	    printf("\n failed p cut %f %f %f \n",p,fChildPMin,fChildPMax);
	    return kFALSE;
	}
//
// theta cut
	if (theta > fChildThetaMax || theta < fChildThetaMin) {
//	    printf("\n failed theta cut %f %f %f \n",theta,fChildThetaMin,fChildThetaMax);
	    return kFALSE;
	}
//
// rapidity cut
	if (y > fChildYMax || y < fChildYMin) {
//	    printf("\n failed y cut %f %f %f \n",y,fChildYMin,fChildYMax);
	    return kFALSE;
	}
//
// phi cut
	if (phi > fChildPhiMax || phi < fChildPhiMin) {
//	    printf("\n failed phi cut %f %f %f \n",phi,fChildPhiMin,fChildPhiMax);
	    return kFALSE;
	}
    }
    
    return kTRUE;
}

Bool_t AliGenMC::CheckAcceptanceGeometry(Int_t np, TClonesArray* particles)
{
// Check the geometrical acceptance for particle.

  Bool_t check ;  
  Int_t numberOfPdgCodeParticleforAcceptanceCut = 0;
  Int_t numberOfAcceptedPdgCodeParticleforAcceptanceCut = 0;
  TParticle * particle;
  Int_t i;
  for (i = 0; i < np; i++) {
    particle =  (TParticle *) particles->At(i);
    if( TMath::Abs( particle->GetPdgCode() ) == TMath::Abs( fPdgCodeParticleforAcceptanceCut ) ) {
      numberOfPdgCodeParticleforAcceptanceCut++;
      if (fGeometryAcceptance->Impact(particle)) numberOfAcceptedPdgCodeParticleforAcceptanceCut++;
    }   
  }
  if ( numberOfAcceptedPdgCodeParticleforAcceptanceCut > (fNumberOfAcceptedParticles-1) )
    check = kTRUE;
  else
    check = kFALSE;

  return check;
}

Int_t AliGenMC::CheckPDGCode(Int_t pdgcode) const
{
//
//  If the particle is in a diffractive state, then take action accordingly
  switch (pdgcode) {
  case 91:
    return 92;
  case 110:
    //rho_diff0 -- difficult to translate, return rho0
    return 113;
  case 210:
    //pi_diffr+ -- change to pi+
    return 211;
  case 220:
    //omega_di0 -- change to omega0
    return 223;
  case 330:
    //phi_diff0 -- return phi0
    return 333;
  case 440:
    //J/psi_di0 -- return J/psi
    return 443;
  case 2110:
    //n_diffr -- return neutron
    return 2112;
  case 2210:
    //p_diffr+ -- return proton
    return 2212;
  }
  //non diffractive state -- return code unchanged
  return pdgcode;
}

void AliGenMC::Boost()
{
//
// Boost cms into LHC lab frame
//

    Double_t beta  = TMath::TanH(fDyBoost);
    Double_t gamma = 1./TMath::Sqrt((1.-beta)*(1.+beta));
    Double_t gb    = gamma * beta;

    //    printf("\n Boosting particles to lab frame %f %f %f", fDyBoost, beta, gamma);
    
    Int_t i;
    Int_t np = fParticles.GetEntriesFast();
    for (i = 0; i < np; i++) 
    {
	TParticle* iparticle = (TParticle*) fParticles.At(i);

	Double_t e   = iparticle->Energy();
	Double_t px  = iparticle->Px();
	Double_t py  = iparticle->Py();
	Double_t pz  = iparticle->Pz();

	Double_t eb  = gamma * e -      gb * pz;
	Double_t pzb =   -gb * e +   gamma * pz;

	iparticle->SetMomentum(px, py, pzb, eb);
    }
}

void AliGenMC::AddHeader(AliGenEventHeader* header)
{
    // Passes header either to the container or to gAlice
    if (fContainer) {
	fContainer->AddHeader(header);
    } else {
	gAlice->SetGenEventHeader(header);	
    }
}
Commit	Line	Data
e36044d6	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
88cb7938	16	/* $Id$ */
e36044d6	17
62d0ae06	18	// Base class for generators using external MC generators.
	19	// For example AliGenPythia using Pythia.
	20	// Provides basic functionality: setting of kinematic cuts on
	21	// decay products and particle selection.
	22	// andreas.morsch@cern.ch
	23
ac3faee4	24	#include <TClonesArray.h>
65de9f86	25	#include <TMath.h>
116cbefd	26	#include <TPDGCode.h>
e36044d6	27	#include <TParticle.h>
e36044d6	28
116cbefd	29	#include "AliGenMC.h"
d97fb2bb	30	#include "AliRun.h"
ac3faee4	31	#include "AliGeometry.h"
116cbefd	32
198bb1c7	33	ClassImp(AliGenMC)
e36044d6	34
e36044d6	35	AliGenMC::AliGenMC()
1c56e311	36	:AliGenerator(),
969a210b	37	fParticles("TParticle", 1000),
1c56e311	38	fParentSelect(8),
	39	fChildSelect(8),
	40	fCutOnChild(0),
	41	fChildPtMin(0.),
	42	fChildPtMax(1.e10),
	43	fChildPMin(0.),
	44	fChildPMax(1.e10),
	45	fChildPhiMin(0.),
	46	fChildPhiMax(2. * TMath::Pi()),
	47	fChildThetaMin(0.),
	48	fChildThetaMax(TMath::Pi()),
	49	fChildYMin(-12.),
	50	fChildYMax(12.),
	51	fXingAngleX(0.),
	52	fXingAngleY(0.),
	53	fForceDecay(kAll),
	54	fMaxLifeTime(1.e-15),
1c56e311	55	fDyBoost(0.),
	56	fGeometryAcceptance(0),
	57	fPdgCodeParticleforAcceptanceCut(0),
d97fb2bb	58	fNumberOfAcceptedParticles(0),
e7c989e4	59	fNprimaries(0)
e36044d6	60	{
e36044d6	61	// Default Constructor
e7c989e4	62	fAProjectile = 1;
	63	fZProjectile = 1;
	64	fATarget = 1;
	65	fZTarget = 1;
	66	fProjectile = "P";
	67	fTarget = "P";
e36044d6	68	}
	69
	70	AliGenMC::AliGenMC(Int_t npart)
1c56e311	71	:AliGenerator(npart),
ea24f998	72	fParticles("TParticle", 1000),
1c56e311	73	fParentSelect(8),
	74	fChildSelect(8),
	75	fCutOnChild(0),
	76	fChildPtMin(0.),
	77	fChildPtMax(1.e10),
	78	fChildPMin(0.),
	79	fChildPMax(1.e10),
	80	fChildPhiMin(0.),
	81	fChildPhiMax(2. * TMath::Pi()),
	82	fChildThetaMin(0.),
	83	fChildThetaMax(TMath::Pi()),
	84	fChildYMin(-12.),
	85	fChildYMax(12.),
	86	fXingAngleX(0.),
	87	fXingAngleY(0.),
	88	fForceDecay(kAll),
	89	fMaxLifeTime(1.e-15),
1c56e311	90	fDyBoost(0.),
	91	fGeometryAcceptance(0),
	92	fPdgCodeParticleforAcceptanceCut(0),
d97fb2bb	93	fNumberOfAcceptedParticles(0),
e7c989e4	94	fNprimaries(0)
e36044d6	95	{
e36044d6	96	// Constructor
e36044d6	97	//
e7c989e4	98	fAProjectile = 1;
	99	fZProjectile = 1;
	100	fATarget = 1;
	101	fZTarget = 1;
	102	fProjectile = "P";
	103	fTarget = "P";
	104	for (Int_t i=0; i<8; i++) fParentSelect[i]=fChildSelect[i]=0;
e36044d6	105	}
e36044d6	106
e36044d6	107	AliGenMC::~AliGenMC()
	108	{
	109	// Destructor
	110	}
	111
	112	void AliGenMC::Init()
	113	{
	114	//
	115	// Initialization
	116	switch (fForceDecay) {
e345c619	117	case kBSemiElectronic:
e36044d6	118	case kSemiElectronic:
	119	case kDiElectron:
	120	case kBJpsiDiElectron:
	121	case kBPsiPrimeDiElectron:
47fc6bd5	122	fChildSelect[0] = kElectron;
e36044d6	123	break;
2dcb5874	124	case kHardMuons:
e345c619	125	case kBSemiMuonic:
e36044d6	126	case kSemiMuonic:
	127	case kDiMuon:
	128	case kBJpsiDiMuon:
	129	case kBPsiPrimeDiMuon:
	130	case kPiToMu:
	131	case kKaToMu:
cd159249	132	case kWToMuon:
cd159249	133	case kWToCharmToMuon:
2f299526	134	case kZDiMuon:
f5b581a1	135	case kZDiElectron:
e36044d6	136	fChildSelect[0]=kMuonMinus;
e36044d6	137	break;
cd159249	138	case kWToCharm:
cd159249	139	break;
e36044d6	140	case kHadronicD:
af33df34	141	case kHadronicDWithout4Bodies:
e36044d6	142	fChildSelect[0]=kPiPlus;
	143	fChildSelect[1]=kKPlus;
	144	break;
00cb2f27	145	case kPhiKK:
027677ae	146	fChildSelect[0]=kKPlus;
56d47239	147	break;
56d47239	148	case kBJpsi:
00935af2	149	fChildSelect[0]= 443;
56d47239	150	break;
00935af2	151	case kChiToJpsiGammaToMuonMuon:
	152	fChildSelect[0]= 22;
	153	fChildSelect[1]= 13;
	154	break;
	155	case kChiToJpsiGammaToElectronElectron:
	156	fChildSelect[0]= 22;
	157	fChildSelect[1]= 11;
	158	break;
d3553100	159	case kLambda:
199786f7	160	fChildSelect[0]= kProton;
	161	fChildSelect[1]= 211;
	162	break;
	163	case kOmega:
e36044d6	164	case kAll:
2b1f8257	165	case kAllMuonic:
e36044d6	166	case kNoDecay:
d5f86442	167	case kNoDecayHeavy:
1fefb2d4	168	case kNoDecayBeauty:
847af604	169	case kNeutralPion:
e36044d6	170	break;
e36044d6	171	}
36d770d9	172
	173	if (fZTarget != 0 && fAProjectile != 0)
	174	{
	175	fDyBoost = - 0.5 * TMath::Log(Double_t(fZProjectile) * Double_t(fATarget) /
	176	(Double_t(fZTarget) * Double_t(fAProjectile)));
	177	}
e36044d6	178	}
	179
	180
62d0ae06	181	Bool_t AliGenMC::ParentSelected(Int_t ip) const
e36044d6	182	{
	183	// True if particle is in list of parent particles to be selected
	184	for (Int_t i=0; i<8; i++)
	185	{
62d0ae06	186	if (fParentSelect.At(i) == ip) return kTRUE;
e36044d6	187	}
	188	return kFALSE;
	189	}
	190
62d0ae06	191	Bool_t AliGenMC::ChildSelected(Int_t ip) const
e36044d6	192	{
	193	// True if particle is in list of decay products to be selected
	194	for (Int_t i=0; i<5; i++)
	195	{
62d0ae06	196	if (fChildSelect.At(i) == ip) return kTRUE;
e36044d6	197	}
	198	return kFALSE;
	199	}
	200
62d0ae06	201	Bool_t AliGenMC::KinematicSelection(TParticle *particle, Int_t flag) const
e36044d6	202	{
e36044d6	203	// Perform kinematic selection
43a333ec	204	Double_t pz = particle->Pz();
43a333ec	205	Double_t pt = particle->Pt();
	206	Double_t p = particle->P();
	207	Double_t theta = particle->Theta();
	208	Double_t mass = particle->GetCalcMass();
	209	Double_t mt2 = pt * pt + mass * mass;
	210	Double_t phi = particle->Phi();
d643e262	211	Double_t e = particle->Energy();
	212
	213	if (e == 0.)
	214	e = TMath::Sqrt(p * p + mass * mass);
	215
95450c6d	216
95450c6d	217	Double_t y, y0;
95450c6d	218
fef61e14	219	if (TMath::Abs(pz) < e) {
95450c6d	220	y = 0.5*TMath::Log((e+pz)/(e-pz));
	221	} else {
	222	y = 1.e10;
	223	}
e36044d6	224
95450c6d	225	if (mt2) {
95450c6d	226	y0 = 0.5TMath::Log((e+TMath::Abs(pz))(e+TMath::Abs(pz))/mt2);
e36044d6	227	} else {
95450c6d	228	if (TMath::Abs(y) < 1.e10) {
	229	y0 = y;
	230	} else {
	231	y0 = 1.e10;
	232	}
e36044d6	233	}
95450c6d	234
95450c6d	235	y = (pz < 0) ? -y0 : y0;
e36044d6	236
	237	if (flag == 0) {
	238	//
	239	// Primary particle cuts
	240	//
	241	// transverse momentum cut
	242	if (pt > fPtMax \|\| pt < fPtMin) {
	243	// printf("\n failed pt cut %f %f %f \n",pt,fPtMin,fPtMax);
	244	return kFALSE;
	245	}
	246	//
	247	// momentum cut
	248	if (p > fPMax \|\| p < fPMin) {
	249	// printf("\n failed p cut %f %f %f \n",p,fPMin,fPMax);
	250	return kFALSE;
	251	}
	252	//
	253	// theta cut
	254	if (theta > fThetaMax \|\| theta < fThetaMin) {
	255	// printf("\n failed theta cut %f %f %f \n",theta,fThetaMin,fThetaMax);
	256	return kFALSE;
	257	}
	258	//
	259	// rapidity cut
	260	if (y > fYMax \|\| y < fYMin) {
	261	// printf("\n failed y cut %f %f %f \n",y,fYMin,fYMax);
	262	return kFALSE;
	263	}
	264	//
	265	// phi cut
	266	if (phi > fPhiMax \|\| phi < fPhiMin) {
	267	// printf("\n failed phi cut %f %f %f \n",phi,fPhiMin,fPhiMax);
	268	return kFALSE;
	269	}
	270	} else {
	271	//
	272	// Decay product cuts
	273	//
	274	// transverse momentum cut
	275	if (pt > fChildPtMax \|\| pt < fChildPtMin) {
	276	// printf("\n failed pt cut %f %f %f \n",pt,fChildPtMin,fChildPtMax);
	277	return kFALSE;
	278	}
	279	//
	280	// momentum cut
	281	if (p > fChildPMax \|\| p < fChildPMin) {
	282	// printf("\n failed p cut %f %f %f \n",p,fChildPMin,fChildPMax);
	283	return kFALSE;
	284	}
	285	//
	286	// theta cut
	287	if (theta > fChildThetaMax \|\| theta < fChildThetaMin) {
	288	// printf("\n failed theta cut %f %f %f \n",theta,fChildThetaMin,fChildThetaMax);
	289	return kFALSE;
	290	}
	291	//
	292	// rapidity cut
	293	if (y > fChildYMax \|\| y < fChildYMin) {
	294	// printf("\n failed y cut %f %f %f \n",y,fChildYMin,fChildYMax);
	295	return kFALSE;
	296	}
	297	//
	298	// phi cut
	299	if (phi > fChildPhiMax \|\| phi < fChildPhiMin) {
300	// printf("\n failed phi cut %f %f %f \n",phi,fChildPhiMin,fChildPhiMax);
301	return kFALSE;
302	}
303	}
304
e36044d6	305	return kTRUE;
	306	}
	307
65de9f86	308	Bool_t AliGenMC::CheckAcceptanceGeometry(Int_t np, TClonesArray* particles)
65de9f86	309	{
e033748b	310	// Check the geometrical acceptance for particle.
	311
	312	Bool_t check ;
	313	Int_t numberOfPdgCodeParticleforAcceptanceCut = 0;
	314	Int_t numberOfAcceptedPdgCodeParticleforAcceptanceCut = 0;
65de9f86	315	TParticle * particle;
65de9f86	316	Int_t i;
e033748b	317	for (i = 0; i < np; i++) {
65de9f86	318	particle = (TParticle *) particles->At(i);
65de9f86	319	if( TMath::Abs( particle->GetPdgCode() ) == TMath::Abs( fPdgCodeParticleforAcceptanceCut ) ) {
e033748b	320	numberOfPdgCodeParticleforAcceptanceCut++;
e033748b	321	if (fGeometryAcceptance->Impact(particle)) numberOfAcceptedPdgCodeParticleforAcceptanceCut++;
65de9f86	322	}
65de9f86	323	}
e033748b	324	if ( numberOfAcceptedPdgCodeParticleforAcceptanceCut > (fNumberOfAcceptedParticles-1) )
e033748b	325	check = kTRUE;
65de9f86	326	else
e033748b	327	check = kFALSE;
65de9f86	328
e033748b	329	return check;
65de9f86	330	}
65de9f86	331
62d0ae06	332	Int_t AliGenMC::CheckPDGCode(Int_t pdgcode) const
e36044d6	333	{
	334	//
	335	// If the particle is in a diffractive state, then take action accordingly
	336	switch (pdgcode) {
	337	case 91:
	338	return 92;
	339	case 110:
	340	//rho_diff0 -- difficult to translate, return rho0
	341	return 113;
	342	case 210:
	343	//pi_diffr+ -- change to pi+
	344	return 211;
	345	case 220:
	346	//omega_di0 -- change to omega0
	347	return 223;
	348	case 330:
	349	//phi_diff0 -- return phi0
	350	return 333;
	351	case 440:
	352	//J/psi_di0 -- return J/psi
	353	return 443;
	354	case 2110:
	355	//n_diffr -- return neutron
	356	return 2112;
	357	case 2210:
	358	//p_diffr+ -- return proton
	359	return 2212;
	360	}
	361	//non diffractive state -- return code unchanged
	362	return pdgcode;
	363	}
3b945a60	364
36d770d9	365	void AliGenMC::Boost()
3b945a60	366	{
	367	//
	368	// Boost cms into LHC lab frame
	369	//
	370
36d770d9	371	Double_t beta = TMath::TanH(fDyBoost);
60e55aee	372	Double_t gamma = 1./TMath::Sqrt((1.-beta)*(1.+beta));
3b945a60	373	Double_t gb = gamma * beta;
3b945a60	374
7cdba479	375	// printf("\n Boosting particles to lab frame %f %f %f", fDyBoost, beta, gamma);
3b945a60	376
3b945a60	377	Int_t i;
ea24f998	378	Int_t np = fParticles.GetEntriesFast();
3b945a60	379	for (i = 0; i < np; i++)
3b945a60	380	{
ea24f998	381	TParticle* iparticle = (TParticle*) fParticles.At(i);
3b945a60	382
	383	Double_t e = iparticle->Energy();
	384	Double_t px = iparticle->Px();
	385	Double_t py = iparticle->Py();
	386	Double_t pz = iparticle->Pz();
	387
	388	Double_t eb = gamma * e - gb * pz;
	389	Double_t pzb = -gb * e + gamma * pz;
	390
	391	iparticle->SetMomentum(px, py, pzb, eb);
	392	}
	393	}
d97fb2bb	394
	395	void AliGenMC::AddHeader(AliGenEventHeader* header)
	396	{
	397	// Passes header either to the container or to gAlice
	398	if (fContainer) {
	399	fContainer->AddHeader(header);
	400	} else {
	401	gAlice->SetGenEventHeader(header);
	402	}
	403	}