// One can use the configuration macro in compiled mode by
// root [0] gSystem->Load("libgeant321");
// root [0] gSystem->SetIncludePath("-I$ROOTSYS/include -I$ALICE_ROOT/include\
//                   -I$ALICE_ROOT -I$ALICE/geant3/TGeant3");
// root [0] .x grun.C(1,"Config_PDC07.C++")

#if !defined(__CINT__) || defined(__MAKECINT__)
#include <Riostream.h>
#include <TRandom.h>
#include <TDatime.h>
#include <TSystem.h>
#include <TVirtualMC.h>
#include <TGeant3TGeo.h>
#include "EVGEN/AliGenCocktail.h"
#include "EVGEN/AliGenParam.h"
#include "EVGEN/AliGenMUONlib.h"
#include "STEER/AliRunLoader.h"
#include "STEER/AliRun.h"
#include "STEER/AliConfig.h"
#include "PYTHIA6/AliDecayerPythia.h"
#include "PYTHIA6/AliGenPythia.h"
#include "STEER/AliMagF.h"
#include "STRUCT/AliBODY.h"
#include "STRUCT/AliMAG.h"
#include "STRUCT/AliABSOv3.h"
#include "STRUCT/AliDIPOv3.h"
#include "STRUCT/AliHALLv3.h"
#include "STRUCT/AliFRAMEv2.h"
#include "STRUCT/AliSHILv3.h"
#include "STRUCT/AliPIPEv3.h"
#include "ITS/AliITSv11Hybrid.h"
#include "TPC/AliTPCv2.h"
#include "TOF/AliTOFv6T0.h"
#include "HMPID/AliHMPIDv3.h"
#include "ZDC/AliZDCv3.h"
#include "TRD/AliTRDv1.h"
#include "TRD/AliTRDgeometry.h"
#include "FMD/AliFMDv1.h"
#include "MUON/AliMUONv1.h"
#include "PHOS/AliPHOSv1.h"
#include "PHOS/AliPHOSSimParam.h"
#include "PMD/AliPMDv1.h"
#include "T0/AliT0v1.h"
#include "EMCAL/AliEMCALv2.h"
#include "ACORDE/AliACORDEv1.h"
#include "VZERO/AliVZEROv7.h"
#endif


enum PDC07Proc_t
{
//--- Heavy Flavour Production ---
  kCharmPbPb5500,  kCharmpPb8800,  kCharmpp14000,  kCharmpp14000wmi,
  kD0PbPb5500,     kD0pPb8800,     kD0pp14000,
  kDPlusPbPb5500,  kDPluspPb8800,  kDPluspp14000,
  kBeautyPbPb5500, kBeautypPb8800, kBeautypp14000, kBeautypp14000wmi,
  // -- Pythia Mb
  kPyMbNoHvq, kPyOmegaPlus, kPyOmegaMinus, kPyJetJet, kBeautyppwmiJet,
  kPyGammaJetPHOS, kPyJetJetPHOS, kPyJetJetPHOSv2, kPyGammaBremsPHOS,
  kPyGammaJetEMCAL, kPyJetJetEMCAL, kPyGammaBremsEMCAL, kRunMax
};

const char * pprRunName[] = {
  "kCharmPbPb5500",  "kCharmpPb8800",  "kCharmpp14000",  "kCharmpp14000wmi",
  "kD0PbPb5500",     "kD0pPb8800",     "kD0pp14000",
  "kDPlusPbPb5500",  "kDPluspPb8800",  "kDPluspp14000",
  "kBeautyPbPb5500", "kBeautypPb8800", "kBeautypp14000", "kBeautypp14000wmi",
  "kPyMbNoHvq", "kPyOmegaPlus", "kPyOmegaMinus", "kPyJetJet", "kBeautyppwmiJet",
  "kPyGammaJetPHOS", "kPyJetJetPHOS",  "kPyJetJetPHOSv2", "kPyGammaBremsPHOS",
  "kPyGammaJetEMCAL", "kPyJetJetEMCAL", "kPyGammaBremsEMCAL"
};


//--- Decay Mode ---
enum DecayHvFl_t
{
  kNature,  kHadr, kSemiEl, kSemiMu
};
//--- Magnetic Field ---
enum Mag_t
{
  kNoField, k5kG, kFieldMax
};
//--- Trigger config ---
enum TrigConf_t
{
    kDefaultPPTrig, kDefaultPbPbTrig
};

const char * TrigConfName[] = {
    "p-p","Pb-Pb"
};

//--- Functions ---
class AliGenPythia ;
AliGenPythia *PythiaHVQ(PDC07Proc_t proc);
AliGenPythia *PythiaHard(PDC07Proc_t proc);
AliGenerator *MbCocktail();
AliGenerator *PyMbTriggered(Int_t pdg);
void ProcessEnvironmentVars();

// This part for configuration
static DecayHvFl_t   decHvFl     = kNature;
static Mag_t         mag         = k5kG;
static TrigConf_t    trig        = kDefaultPPTrig; // default pp trigger configuration
static Int_t         runNumber   = 0;
static Int_t         eventNumber = 0;
//========================//
// Set Random Number seed //
//========================//
TDatime dt;
static UInt_t seed    = dt.Get();

// nEvts = -1  : you get 1 QQbar pair and all the fragmentation and
//               decay chain
// nEvts = N>0 : you get N charm / beauty Hadrons
Int_t nEvts = -1;
// stars = kTRUE : all heavy resonances and their decay stored
//       = kFALSE: only final heavy hadrons and their decays stored
Bool_t stars = kTRUE;

// To be used only with kCharmpp14000wmi and kBeautypp14000wmi
// To get a "reasonable" agreement with MNR results, events have to be
// generated with the minimum ptHard set to 2.76 GeV.
// To get a "perfect" agreement with MNR results, events have to be
// generated in four ptHard bins with the following relative
// normalizations:
//  CHARM
// 2.76-3 GeV: 25%
//    3-4 GeV: 40%
//    4-8 GeV: 29%
//     >8 GeV:  6%
//  BEAUTY
// 2.76-4 GeV:  5%
//    4-6 GeV: 31%
//    6-8 GeV: 28%
//     >8 GeV: 36%


Float_t eCMS=5500;
PDC07Proc_t proc = kPyJetJet;
//PDC07Proc_t proc = kPyGammaJetEMCAL;
//PDC07Proc_t proc = kBeautyppwmiJet;
Int_t nPtHardGJBins = 11;
Int_t ptHardGJLo[] = {5 ,10,20,30,40,50,60,70,80,90 ,100};
Int_t ptHardGJHi[] = {10,20,30,40,50,60,70,80,90,100,-1};
Int_t nPtHardJJBins = 17;
Int_t ptHardJJLo[] = {2 ,12,16,20,24,29,35,41,50,60,72,86 ,104,124,149,179,215};
Int_t ptHardJJHi[] = {12,16,20,24,29,35,41,50,60,72,86,104,124,149,179,215,258};
Float_t ptHardMin =  10.;
Float_t ptHardMax =  20.;
Float_t ptGammaPi0Min = 0.;
Int_t iquenching = 0;
Float_t qhat = 50.; //for  iquenching equal 1 and 4
//Float_t medlength = 6.; // for iquenching equal to 4
Int_t year = 0; //ALICE geometry configuration
Bool_t pi0FragPhotonInDet = kFALSE ;

// Comment line
static TString comment;

void Config()
{

	// Get settings from environment variables
	ProcessEnvironmentVars();
	
	gSystem->Load("liblhapdf.so");      // Parton density functions
	gSystem->Load("libEGPythia6.so");   // TGenerator interface
	gSystem->Load("libpythia6.so");     // Pythia
	gSystem->Load("libAliPythia6.so");  // ALICE specific implementations
	
	// libraries required by geant321
#if defined(__CINT__)
	gSystem->Load("liblhapdf");
	gSystem->Load("libEGPythia6");
	gSystem->Load("libpythia6");
	gSystem->Load("libAliPythia6");
	gSystem->Load("libgeant321");
#endif
	
	new TGeant3TGeo("C++ Interface to Geant3");
	
	// Output every 100 tracks
	((TGeant3*)gMC)->SetSWIT(4,100);
	
	//=======================================================================
	
	// Run loader
	AliRunLoader* rl=0x0;
	rl = AliRunLoader::Open("galice.root",
							AliConfig::GetDefaultEventFolderName(),
							"recreate");
	if (rl == 0x0)
    {
		gAlice->Fatal("Config.C","Can not instatiate the Run Loader");
		return;
    }
	rl->SetCompressionLevel(2);
	rl->SetNumberOfEventsPerFile(1000);
	gAlice->SetRunLoader(rl);
	
	// Set the trigger configuration
	gAlice->SetTriggerDescriptor(TrigConfName[trig]);
	cout<<"Trigger configuration is set to  "<<TrigConfName[trig]<<endl;
	
	//
	//=======================================================================
	// ************* STEERING parameters FOR ALICE SIMULATION **************
	// --- Specify event type to be tracked through the ALICE setup
	// --- All positions are in cm, angles in degrees, and P and E in GeV
	
	
    gMC->SetProcess("DCAY",1);
    gMC->SetProcess("PAIR",1);
    gMC->SetProcess("COMP",1);
    gMC->SetProcess("PHOT",1);
    gMC->SetProcess("PFIS",0);
    gMC->SetProcess("DRAY",0);
    gMC->SetProcess("ANNI",1);
    gMC->SetProcess("BREM",1);
    gMC->SetProcess("MUNU",1);
    gMC->SetProcess("CKOV",1);
    gMC->SetProcess("HADR",1);
    gMC->SetProcess("LOSS",2);
    gMC->SetProcess("MULS",1);
    gMC->SetProcess("RAYL",1);
	
    Float_t cut = 1.e-3;        // 1MeV cut by default
    Float_t tofmax = 1.e10;
	
    gMC->SetCut("CUTGAM", cut);
    gMC->SetCut("CUTELE", cut);
    gMC->SetCut("CUTNEU", cut);
    gMC->SetCut("CUTHAD", cut);
    gMC->SetCut("CUTMUO", cut);
    gMC->SetCut("BCUTE",  cut);
    gMC->SetCut("BCUTM",  cut);
    gMC->SetCut("DCUTE",  cut);
    gMC->SetCut("DCUTM",  cut);
    gMC->SetCut("PPCUTM", cut);
    gMC->SetCut("TOFMAX", tofmax);
	
	//======================//
	// Set External decayer //
	//======================//
	TVirtualMCDecayer* decayer = new AliDecayerPythia();
	// DECAYS
	//
	switch(decHvFl) {
		case kNature:
			decayer->SetForceDecay(kNeutralPion);
			break;
		case kHadr:
			decayer->SetForceDecay(kHadronicD);
			break;
		case kSemiEl:
			decayer->SetForceDecay(kSemiElectronic);
			break;
		case kSemiMu:
			decayer->SetForceDecay(kSemiMuonic);
			break;
	}
	decayer->Init();
	gMC->SetExternalDecayer(decayer);
	//if(proc == kPyJetJetPHOSv2) // in this case we decay the pi0
	//	decayer->SetForceDecay(kNeutralPion);
	
	//=========================//
	// Generator Configuration //
	//=========================//
	AliGenPythia* gener = 0x0;
	
	if (proc <=   kBeautypp14000wmi) {
		AliGenPythia *pythia = PythiaHVQ(proc);
		// FeedDown option
		pythia->SetFeedDownHigherFamily(kFALSE);
		// Stack filling option
		if(!stars) pythia->SetStackFillOpt(AliGenPythia::kParentSelection);
		// Set Count mode
		if(nEvts>0) pythia->SetCountMode(AliGenPythia::kCountParents);
		//
		// DECAYS
		//
		switch(decHvFl) {
			case kNature:
				pythia->SetForceDecay(kNeutralPion);
				break;
			case kHadr:
				pythia->SetForceDecay(kHadronicD);
				break;
			case kSemiEl:
				pythia->SetForceDecay(kSemiElectronic);
				break;
			case kSemiMu:
				pythia->SetForceDecay(kSemiMuonic);
				break;
		}
		//
		// GEOM & KINE CUTS
		//
		pythia->SetMomentumRange(0,99999999);
		pythia->SetPhiRange(0., 360.);
		pythia->SetThetaRange(0,180);
		switch(ycut) {
			case kFull:
				pythia->SetYRange(-12,12);
				break;
			case kBarrel:
				pythia->SetYRange(-2,2);
				break;
			case kMuonArm:
				pythia->SetYRange(1,6);
				break;
		}
		gener = pythia;
	} else if (proc == kPyMbNoHvq) {
		gener = MbCocktail();
	} else if (proc == kPyOmegaMinus) {
		gener = PyMbTriggered(3334);
	} else if (proc == kPyOmegaPlus) {
		gener = PyMbTriggered(-3334);
	} else if (proc <= kPyGammaBremsEMCAL) {
		AliGenPythia *pythia = PythiaHard(proc);
		// FeedDown option
		pythia->SetFeedDownHigherFamily(kFALSE);
		// Set Count mode
		if(nEvts>0) pythia->SetCountMode(AliGenPythia::kCountParents);
		
		//
		// GEOM & KINE CUTS
		//
		pythia->SetMomentumRange(0,99999999);
//		if(proc == kPyJetJetPHOSv2)
//			pythia->SetForceDecay(kNeutralPion);
//		else
//			pythia->SetForceDecay(kAll);

		pythia->SetForceDecay(kNeutralPion);
		pythia->SetPycellParameters(2., 274, 432, 0., 4., 5., 1.0);
		pythia->SetGluonRadiation(1,1); //ISR and FRS effects on.
		pythia->SetPtKick(3.); // set the intrinsic kt to about 5.3 GeV/c
		gener = pythia;
	}
	
	
	// PRIMARY VERTEX
	
	gener->SetOrigin(0., 0., 0.);    // vertex position
	
	// Size of the interaction diamond
	// Longitudinal
	Float_t           sigmaz  = 7.55 / TMath::Sqrt(2.); // [cm] 14  TeV (p-p), 5.5 TeV (Pb-Pb)
	if(eCMS == 10000) sigmaz  = 5.4  / TMath::Sqrt(2.); // [cm] 10  TeV
	if(eCMS == 900)   sigmaz  = 10.5 / TMath::Sqrt(2.); // [cm] 0.9 TeV
	cout<<">>> SigmaZ x srqt(2.) "<< sigmaz * TMath::Sqrt(2.) << endl;
	
    // Transverse
	Float_t betast  = 10;                 // beta* [m]
	Float_t eps     = 3.75e-6;            // emittance [m]
	Float_t gamma   = eCMS / 2.0 / 0.938272;   // relativistic gamma [1]
	Float_t sigmaxy = TMath::Sqrt(eps * betast / gamma) / TMath::Sqrt(2.) * 100.;  // [cm]
	printf("\n \n Diamond size x-y: %10.3e z: %10.3e\n \n", sigmaxy, sigmaz);
	
	gener->SetSigma(sigmaxy, sigmaxy, sigmaz);      // Sigma in (X,Y,Z) (cm) on IP position
	gener->SetCutVertexZ(3.);        // Truncate at 3 sigma
	gener->SetVertexSmear(kPerEvent);
	
	gener->Init();
	
	//Quenching
	gener->SetQuench(iquenching);
	if(iquenching == 1){ //quenching weights
		Float_t k = 6e5*(qhat/1.7) ; //qhat=1.7, k = 6e5, default  value
		AliPythia::Instance()->InitQuenching(0.,0.1,k,0,0.95,6);
	}
	else if(iquenching == 4){ //q-pythia
		//Not yet implemented
		gener->SetQhat(qhat); //   !transport coefficient (GeV2/fm)
	}

  // FIELD
  //
  AliMagF* field = 0x0;
  if (mag == kNoField) {
    comment = comment.Append(" | L3 field 0.0 T");
    field = new AliMagF("Maps","Maps", 0., 0., AliMagF::k5kGUniform);
  } else if (mag == k5kG) {
    comment = comment.Append(" | L3 field 0.5 T");
    field = new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG);
  }
  printf("\n \n Comment: %s \n \n", comment.Data());
  TGeoGlobalMagField::Instance()->SetField(field);

  rl->CdGAFile();
  gAlice->SetField(field);

	
	Int_t iABSO   = 1;
	Int_t iACORDE = 0;
	Int_t iDIPO   = 1;
	Int_t iEMCAL  = 1;
	Int_t iFMD    = 1;
	Int_t iFRAME  = 1;
	Int_t iHALL   = 1;
	Int_t iITS    = 1;
	Int_t iMAG    = 1;
	Int_t iMUON   = 1;
	Int_t iPHOS   = 1;
	Int_t iPIPE   = 1;
	Int_t iPMD    = 1;
	Int_t iHMPID  = 1;
	Int_t iSHIL   = 1;
	Int_t iT0     = 1;
	Int_t iTOF    = 1;
	Int_t iTPC    = 1;
	Int_t iTRD    = 1;
	Int_t iVZERO  = 1;
	Int_t iZDC    = 1;
	
	
    //=================== Alice BODY parameters =============================
    AliBODY *BODY = new AliBODY("BODY", "Alice envelop");
	
	
    if (iMAG)
    {
        //=================== MAG parameters ============================
        // --- Start with Magnet since detector layouts may be depending ---
        // --- on the selected Magnet dimensions ---
        AliMAG *MAG = new AliMAG("MAG", "Magnet");
    }
	
	
    if (iABSO)
    {
        //=================== ABSO parameters ============================
        AliABSO *ABSO = new AliABSOv3("ABSO", "Muon Absorber");
    }
	
    if (iDIPO)
    {
        //=================== DIPO parameters ============================
		
        AliDIPO *DIPO = new AliDIPOv3("DIPO", "Dipole version 3");
    }
	
    if (iHALL)
    {
        //=================== HALL parameters ============================
		
        AliHALL *HALL = new AliHALLv3("HALL", "Alice Hall");
    }
	
	
    if (iFRAME)
    {
        //=================== FRAME parameters ============================
		
        AliFRAMEv2 *FRAME = new AliFRAMEv2("FRAME", "Space Frame");
	    FRAME->SetHoles(1);
    }
	
    if (iSHIL)
    {
        //=================== SHIL parameters ============================
		
        AliSHIL *SHIL = new AliSHILv3("SHIL", "Shielding Version 3");
    }
	
	
    if (iPIPE)
    {
        //=================== PIPE parameters ============================
		
        AliPIPE *PIPE = new AliPIPEv3("PIPE", "Beam Pipe");
    }
	
    if (iITS)
	{
		//=================== ITS parameters ============================
		
		AliITS *ITS  = new AliITSv11Hybrid("ITS","ITS v11Hybrid");
    }
	
    if (iTPC)
    {
		//============================ TPC parameters =====================

        AliTPC *TPC = new AliTPCv2("TPC", "Default");
    }
	
	
    if (iTOF) {
        //=================== TOF parameters ============================

		AliTOF *TOF = new AliTOFv6T0("TOF", "normal TOF");
	}
	
	
    if (iHMPID)
    {
        //=================== HMPID parameters ===========================
 
		AliHMPID *HMPID = new AliHMPIDv3("HMPID", "normal HMPID");  
	}
	
	
    if (iZDC)
    {
        //=================== ZDC parameters ============================
		
        AliZDC *ZDC = new AliZDCv3("ZDC", "normal ZDC");
    }
	
    if (iTRD)
    {
        //=================== TRD parameters ============================
		
        AliTRD *TRD = new AliTRDv1("TRD", "TRD slow simulator");
        AliTRDgeometry *geoTRD = TRD->GetGeometry();
		if(year == 2009){
			// Partial geometry: modules at 0,8,9,17
			// Partial geometry: modules at 1,7,10,16 expected for 2009
			// starting at 3h in positive direction
			
			//geoTRD->SetSMstatus(1,0);
			//geoTRD->SetSMstatus(2,0);
			//geoTRD->SetSMstatus(3,0);
			//geoTRD->SetSMstatus(4,0);
			//geoTRD->SetSMstatus(5,0);
			//geoTRD->SetSMstatus(6,0);
			//geoTRD->SetSMstatus(7,0);
			
			
			//geoTRD->SetSMstatus(10,0);
			//geoTRD->SetSMstatus(11,0);
			//geoTRD->SetSMstatus(12,0);
			//geoTRD->SetSMstatus(13,0);
			//geoTRD->SetSMstatus(14,0);
			//geoTRD->SetSMstatus(15,0);
			//geoTRD->SetSMstatus(16,0);
	
			geoTRD->SetSMstatus(2,0);
			geoTRD->SetSMstatus(3,0);
			geoTRD->SetSMstatus(4,0);
      			geoTRD->SetSMstatus(5,0);
			geoTRD->SetSMstatus(6,0);

   		               geoTRD->SetSMstatus(11,0);
     		               geoTRD->SetSMstatus(12,0);
        		               geoTRD->SetSMstatus(13,0);
        		               geoTRD->SetSMstatus(14,0);
        		               geoTRD->SetSMstatus(15,0);
	}
    }
	
    if (iFMD)
    {
        //=================== FMD parameters ============================
		AliFMD *FMD = new AliFMDv1("FMD", "normal FMD");
	}
	
    if (iMUON)
    {
        //=================== MUON parameters ===========================
        // New MUONv1 version (geometry defined via builders)
        AliMUON *MUON = new AliMUONv1("MUON", "default");
    }
    //=================== PHOS parameters ===========================
	
    if (iPHOS)
    {
        AliPHOS *PHOS = new AliPHOSv1("PHOS", "IHEP");
    }
	
	
    if (iPMD)
    {
        //=================== PMD parameters ============================
        AliPMD *PMD = new AliPMDv1("PMD", "normal PMD");
    }
	
    if (iT0)
    {
        //=================== T0 parameters ============================
        AliT0 *T0 = new AliT0v1("T0", "T0 Detector");
    }
	
    if (iEMCAL)
    {
        //=================== EMCAL parameters ============================
        AliEMCAL *EMCAL = new AliEMCALv2("EMCAL", "EMCAL_COMPLETE");
    }
	
	if (iACORDE)
    {
        //=================== ACORDE parameters ============================
        AliACORDE *ACORDE = new AliACORDEv1("ACORDE", "normal ACORDE");
    }
	
	if (iVZERO)
    {
        //=================== VZERO parameters ============================
        AliVZERO *VZERO = new AliVZEROv7("VZERO", "normal VZERO");
    }
	
	AliLog::Message(AliLog::kInfo, "End of Config", "Config.C", "Config.C", "Config()"," Config.C", __LINE__);
}

//           PYTHIA

AliGenPythia *PythiaHVQ(PDC07Proc_t proc) {
//*******************************************************************//
// Configuration file for charm / beauty generation with PYTHIA      //
//                                                                   //
// The parameters have been tuned in order to reproduce the inclusive//
// heavy quark pt distribution given by the NLO pQCD calculation by  //
// Mangano, Nason and Ridolfi.                                       //
//                                                                   //
// For details and for the NORMALIZATION of the yields see:          //
//   N.Carrer and A.Dainese,                                         //
//   "Charm and beauty production at the LHC",                       //
//   ALICE-INT-2003-019, [arXiv:hep-ph/0311225];                     //
//   PPR Chapter 6.6, CERN/LHCC 2005-030 (2005).                     //
//*******************************************************************//
  AliGenPythia * gener = 0x0;

  switch(proc) {
  case kCharmPbPb5500:
      comment = comment.Append(" Charm in Pb-Pb at 5.5 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyCharmPbPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(5500.);
      gener->SetNuclei(208,208);
      break;
  case kCharmpPb8800:
      comment = comment.Append(" Charm in p-Pb at 8.8 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyCharmpPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(8800.);
      gener->SetProjectile("P",1,1);
      gener->SetTarget("Pb",208,82);
      break;
  case kCharmpp14000:
      comment = comment.Append(" Charm in pp at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyCharmppMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(14000.);
      break;
  case kCharmpp14000wmi:
      comment = comment.Append(" Charm in pp at 14 TeV with mult. interactions");
      gener = new AliGenPythia(-1);
      gener->SetProcess(kPyCharmppMNRwmi);
      gener->SetStrucFunc(kCTEQ5L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      gener->SetEnergyCMS(14000.);
      break;
  case kD0PbPb5500:
      comment = comment.Append(" D0 in Pb-Pb at 5.5 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyD0PbPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(5500.);
      gener->SetNuclei(208,208);
      break;
  case kD0pPb8800:
      comment = comment.Append(" D0 in p-Pb at 8.8 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyD0pPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(8800.);
      gener->SetProjectile("P",1,1);
      gener->SetTarget("Pb",208,82);
      break;
  case kD0pp14000:
      comment = comment.Append(" D0 in pp at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyD0ppMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(14000.);
      break;
  case kDPlusPbPb5500:
      comment = comment.Append(" DPlus in Pb-Pb at 5.5 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyDPlusPbPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(5500.);
      gener->SetNuclei(208,208);
      break;
  case kDPluspPb8800:
      comment = comment.Append(" DPlus in p-Pb at 8.8 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyDPluspPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(8800.);
      gener->SetProjectile("P",1,1);
      gener->SetTarget("Pb",208,82);
      break;
  case kDPluspp14000:
      comment = comment.Append(" DPlus in pp at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyDPlusppMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.1,-1.0);
      gener->SetEnergyCMS(14000.);
      break;
  case kBeautyPbPb5500:
      comment = comment.Append(" Beauty in Pb-Pb at 5.5 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyBeautyPbPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.75,-1.0);
      gener->SetEnergyCMS(5500.);
      gener->SetNuclei(208,208);
      break;
  case kBeautypPb8800:
      comment = comment.Append(" Beauty in p-Pb at 8.8 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyBeautypPbMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.75,-1.0);
      gener->SetEnergyCMS(8800.);
      gener->SetProjectile("P",1,1);
      gener->SetTarget("Pb",208,82);
      break;
  case kBeautypp14000:
      comment = comment.Append(" Beauty in pp at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetProcess(kPyBeautyppMNR);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(2.75,-1.0);
      gener->SetEnergyCMS(14000.);
      break;
  case kBeautypp14000wmi:
      comment = comment.Append(" Beauty in pp at 14 TeV with mult. interactions");
      gener = new AliGenPythia(-1);
      gener->SetProcess(kPyBeautyppMNRwmi);
      gener->SetStrucFunc(kCTEQ5L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      gener->SetEnergyCMS(14000.);
      break;
   }

  return gener;
}

AliGenPythia *PythiaHard(PDC07Proc_t proc) {
//*******************************************************************//
// Configuration file for hard QCD processes generation with PYTHIA  //
//                                                                   //
//*******************************************************************//
  AliGenPythia * gener = 0x0;

  switch(proc) {

  case kPyJetJet:
    comment = comment.Append(" pp->jet + jet over at 14 TeV, no restriction");
    AliGenPythia * gener = new AliGenPythia(nEvts);
    gener->SetEnergyCMS(eCMS);//        Centre of mass energy
    gener->SetProcess(kPyJets);//        Process type
    gener->SetJetEtaRange(-1.5, 1.5);//  Final state kinematic cuts
    gener->SetJetPhiRange(0., 360.);
    gener->SetJetEtRange(10., 1000.);
    gener->SetPtHard(ptHardMin, ptHardMax);// Pt transfer of the hard scattering
    gener->SetStrucFunc(kCTEQ4L);
    return gener;
  case kBeautyppwmiJet:
	comment = comment.Append(" Beauty Jets over EMCAL");
	AliGenPythia *gener = new AliGenPythia(-1);
	gener->SetProcess(kPyBeautyppMNRwmi);
	gener->SetStrucFunc(kCTEQ4L);
	//Jet specific stuff below
	gener->SetPtHard(ptHardMin,ptHardMax);
	gener->SetJetEtaRange(-1,1);
	gener->SetJetPhiRange(60.,210.);
	gener->SetJetEtRange(10.,1000.);
	//jet specific stuff above
	gener->SetEnergyCMS(eCMS);
	gener->SetForceDecay(kSemiElectronic);
	gener->SetYRange(-2,2);
	gener->SetFeedDownHigherFamily(kFALSE);
	gener->SetCountMode(AliGenPythia::kCountParents);		
      break;
  case kPyGammaJetPHOS:
      comment = comment.Append(" pp->jet + gamma over PHOS");
      gener = new AliGenPythia(nEvts);
      gener->SetEnergyCMS(eCMS);
      gener->SetProcess(kPyDirectGamma);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      //gener->SetYHard(-1.0,1.0);
      gener->SetGammaEtaRange(-0.13,0.13);
      gener->SetGammaPhiRange(219.,321.);//Over 5 modules +-2 deg
      break;
  case kPyJetJetPHOS:
      comment = comment.Append(" pp->jet + jet over PHOS");
      gener = new AliGenPythia(nEvts);
      gener->SetEnergyCMS(eCMS);
      gener->SetProcess(kPyJets);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      //gener->SetYHard(-1.0,1.0);
      gener->SetJetEtaRange(-1.,1.);
      gener->SetJetPhiRange(200.,340.);
      gener->SetJetEtRange(10., 1000.);
      gener->SetPi0InPHOS(pi0FragPhotonInDet);
      gener->SetFragPhotonOrPi0MinPt(ptGammaPi0Min);

      printf("\n \n Event generator: Minimum pT of particle in calorimeter %f \n \n", ptGammaPi0Min);
      break;
  case kPyGammaBremsPHOS:
      comment = comment.Append(" pp->jet + jet+bremsphoton over PHOS at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetEnergyCMS(eCMS);
      gener->SetProcess(kPyJets);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      //gener->SetYHard(-1.0,1.0);
      gener->SetJetEtaRange(-1.,1.);
      gener->SetJetPhiRange(200.,340.);
      gener->SetFragPhotonInPHOS(pi0FragPhotonInDet);
      gener->SetFragPhotonOrPi0MinPt(ptGammaPi0Min);
	  printf("\n \n Event generator: Minimum pT of particle in calorimeter %f \n \n", ptGammaPi0Min);
	  break;
   case kPyJetJetPHOSv2:
        comment = comment.Append(" pp->jet + jet over PHOS version2 ");
        gener = new AliGenPythia(nEvts);
	gener->SetEnergyCMS(eCMS);
	gener->SetProcess(kPyJets);
	gener->SetStrucFunc(kCTEQ4L);
	gener->SetPtHard(ptHardMin,ptHardMax);
	//gener->SetYHard(-1.0,1.0);
	gener->SetJetEtaRange(-1.,1.);
	gener->SetJetPhiRange(200.,340.);
	//gener->SetPi0InPHOS(kTRUE);
	gener->SetPhotonInPHOSeta(pi0FragPhotonInDet);
	gener->SetPhotonMinPt(ptGammaPi0Min);
	gener->SetForceDecay(kAll);
	break;
  case kPyGammaJetEMCAL:
    comment = comment.Append(" pp->jet + gamma over EMCAL at 14 TeV");
    gener = new AliGenPythia(nEvts);
    gener->SetEnergyCMS(eCMS);
    gener->SetProcess(kPyDirectGamma);
    gener->SetStrucFunc(kCTEQ4L);
    gener->SetPtHard(ptHardMin,ptHardMax);
    //gener->SetYHard(-1.0,1.0);
    gener->SetGammaEtaRange(-0.701,0.701);
    gener->SetGammaPhiRange(79.,191.);//Over 6 supermodules +-2 deg
    break;
  case kPyJetJetEMCAL:
    comment = comment.Append(" pp->jet + jet over EMCAL at 14 TeV");
    gener = new AliGenPythia(nEvts);
    gener->SetEnergyCMS(eCMS);
    gener->SetProcess(kPyJets);
    gener->SetStrucFunc(kCTEQ4L);
    gener->SetPtHard(ptHardMin,ptHardMax);
    //gener->SetYHard(-1.0,1.0);
    gener->SetJetEtaRange(-1,1);
    gener->SetJetPhiRange(60.,210.);
	gener->SetJetEtRange(10., 1000.);
    gener->SetPi0InEMCAL(pi0FragPhotonInDet);
    gener->SetFragPhotonOrPi0MinPt(ptGammaPi0Min);
    printf("\n \n Event generator: Minimum pT of particle in calorimeter %f \n \n", ptGammaPi0Min);
    break;
  case kPyGammaBremsEMCAL:
      comment = comment.Append(" pp->jet + jet+bremsphoton over EMCAL at 14 TeV");
      gener = new AliGenPythia(nEvts);
      gener->SetEnergyCMS(eCMS);
      gener->SetProcess(kPyJets);
      gener->SetStrucFunc(kCTEQ4L);
      gener->SetPtHard(ptHardMin,ptHardMax);
      //gener->SetYHard(-1.0,1.0);
      gener->SetJetEtaRange(-1,1);
      gener->SetJetPhiRange(60.,210.); 
	  gener->SetJetEtRange(10., 1000.);
      gener->SetFragPhotonInEMCAL(pi0FragPhotonInDet);
      gener->SetFragPhotonOrPi0MinPt(ptGammaPi0Min);

      printf("\n \n Event generator: Minimum pT of particle in calorimeter %f \n \n", ptGammaPi0Min);
      break;

  }

  return gener;
}

AliGenerator* MbCocktail()
{
      comment = comment.Append(" pp at 14 TeV: Pythia low-pt, no heavy quarks + J/Psi from parameterisation");
      AliGenCocktail * gener = new AliGenCocktail();
      gener->UsePerEventRates();

//    Pythia

      AliGenPythia* pythia = new AliGenPythia(-1);
      pythia->SetMomentumRange(0, 999999.);
      pythia->SetThetaRange(0., 180.);
      pythia->SetYRange(-12.,12.);
      pythia->SetPtRange(0,1000.);
      pythia->SetProcess(kPyMb);
      pythia->SetEnergyCMS(14000.);
      pythia->SwitchHFOff();

//   J/Psi parameterisation

      AliGenParam* jpsi = new AliGenParam(1, AliGenMUONlib::kJpsi, "CDF scaled", "Jpsi");
      jpsi->SetPtRange(0.,100.);
      jpsi->SetYRange(-8., 8.);
      jpsi->SetPhiRange(0., 360.);
      jpsi->SetForceDecay(kAll);

      gener->AddGenerator(pythia, "Pythia", 1.);
      gener->AddGenerator(jpsi,   "J/Psi", 8.e-4);

      return gener;
}

AliGenerator* PyMbTriggered(Int_t pdg)
{
    AliGenPythia* pythia = new AliGenPythia(-1);
    pythia->SetMomentumRange(0, 999999.);
    pythia->SetThetaRange(0., 180.);
    pythia->SetYRange(-12.,12.);
    pythia->SetPtRange(0,1000.);
    pythia->SetProcess(kPyMb);
    pythia->SetEnergyCMS(14000.);
    pythia->SetTriggerParticle(pdg, 0.9);
    return pythia;
}

void ProcessEnvironmentVars()
{
	cout << "######################################" << endl;
	cout << "## Processing environment variables ##" << endl;
	cout << "######################################" << endl;
	
    // Run Number
    if (gSystem->Getenv("RUN")) {
		runNumber = atoi(gSystem->Getenv("RUN"));
    }
    //cout<<"Run number "<<runNumber<<endl;
	
    // Event Number
    if (gSystem->Getenv("EVENT")) {
		eventNumber = atoi(gSystem->Getenv("EVENT"));
    }
    //cout<<"Event number "<<eventNumber<<endl;
	
	// Random Number seed
    if (gSystem->Getenv("CONFIG_SEED")) {
		seed = atoi(gSystem->Getenv("CONFIG_SEED"));
    }
    else if(gSystem->Getenv("EVENT") && gSystem->Getenv("RUN")){
		seed = runNumber * 100000 + eventNumber;
    }
	
    gRandom->SetSeed(seed);
	
    cout<<"////////////////////////////////////////////////////////////////////////////////////"<<endl;
    cout<<"Seed for random number generation= "<< seed<<"  "<< gRandom->GetSeed()<<endl;
    cout<<"////////////////////////////////////////////////////////////////////////////////////"<<endl;
	
    // Run type
    if (gSystem->Getenv("DC_RUN_TYPE")) {
		cout<<"run type "<<gSystem->Getenv("DC_RUN_TYPE")<<endl;
		for (Int_t iRun = 0; iRun < kRunMax; iRun++) {
			if (strcmp(gSystem->Getenv("DC_RUN_TYPE"), pprRunName[iRun])==0) {
				proc = (PDC07Proc_t)iRun;
			}
		}
    }
    else
		cout << "Environment variable DC_RUN_TYPE is not defined" << endl;
    if (gSystem->Getenv("YEAR"))
		year = atoi(gSystem->Getenv("YEAR"));
	if (gSystem->Getenv("ECMS"))
		eCMS = atof(gSystem->Getenv("ECMS"));
	if (gSystem->Getenv("TRIGGER"))
		trig = atoi(gSystem->Getenv("TRIGGER"));
	if ( gSystem->Getenv("PI0GAMMAINDET") )
		pi0FragPhotonInDet	= atoi(gSystem->Getenv("PI0GAMMAINDET"));
    if (gSystem->Getenv("PTGAMMAPI0MIN"))
		ptGammaPi0Min = atof(gSystem->Getenv("PTGAMMAPI0MIN"));
	if (gSystem->Getenv("QUENCHING"))
		iquenching = atof(gSystem->Getenv("QUENCHING"));
	if (gSystem->Getenv("QHAT"))
		qhat = atof(gSystem->Getenv("QHAT"));
	//if (gSystem->Getenv("MEDLENGHT"))
	//	medlength = atof(gSystem->Getenv("MEDLENGHT"));
	if (gSystem->Getenv("PTHARDMIN"))
		ptHardMin = atof(gSystem->Getenv("PTHARDMIN"));
	if (gSystem->Getenv("PTHARDMAX"))
		ptHardMax = atof(gSystem->Getenv("PTHARDMAX"));
					
    if(gSystem->Getenv("PTHARDMIN") && gSystem->Getenv("PTHARDMAX")){
    	if( strncmp(pprRunName[proc],"kPyJetJet",9)==0 || strncmp(pprRunName[proc],"kBeautyppwmiJet",15)==0 ){
			ptHardMin = ptHardJJLo[runNumber%nPtHardJJBins];
			ptHardMax = ptHardJJHi[runNumber%nPtHardJJBins];
		}
		else if( strncmp(pprRunName[proc],"kPyGammaJet",11)==0){
			ptHardMin = ptHardGJLo[runNumber%nPtHardGJBins];
			ptHardMax = ptHardGJHi[runNumber%nPtHardGJBins];    	
		}	
	}
	
    cout<<">> Run type set to "<<pprRunName[proc]<<endl;
    cout<<">> Collision energy set to "<<eCMS <<endl;
	cout<<">> Collisions trigger type "<<trig<<" "<<TrigConfName[trig]<<endl;
    cout<<">> ptHard limits: "<<ptHardMin<<" to " <<ptHardMax<<" GeV"<<endl;
	if(pi0FragPhotonInDet) cout<<">> pt gamma/pi0 threshold "<< ptGammaPi0Min<<" GeV "<<endl;
    if(iquenching) cout<<">> quenching on? "<< iquenching<<" qhat "<<qhat//<<" medlength "<<medlength
	<<endl;
	cout<<">> Year geometry : "<<year<<endl;	
    cout << "######################################" << endl;
    cout << "######################################" << endl;
}