Fixed a few problems

[u/mrichter/AliRoot.git] / FMD / Config.C

static Int_t    eventsPerRun = 1;
static Int_t    nParticles   = 100;

enum PprRun_t {
  test50,
  kParam_8000,   
  kParam_4000,  
  kParam_2000,
  kHijing_cent1, 
  kHijing_cent2,
  kHijing_per1,  
  kHijing_per2, 
  kHijing_per3, 
  kHijing_per4,  
  kHijing_per5,
  kHijing_jj25,  
  kHijing_jj50, 
  kHijing_jj75, 
  kHijing_jj100, 
  kHijing_jj200,
  kHijing_gj25,  
  kHijing_gj50, 
  kHijing_gj75, 
  kHijing_gj100, 
  kHijing_gj200,
  kHijing_pA, 
  kPythia6, 
  kPythia6Jets, 
  kD0PbPb5500, 
  kD_TRD, 
  kB_TRD, 
  kJpsi_TRD,
  kU_TRD, 
  kPyJJ, 
  kPyGJ
};

enum PprGeo_t {
  kHoles, 
  kNoHoles
};

enum PprRad_t {
  kGluonRadiation, 
  kNoGluonRadiation
};

enum PprMag_t {
  k2kG, 
  k4kG, 
  k5kG
};

enum MC_t {
  kFLUKA, 
  kGEANT3, 
  kGEANT4
};


// This part for configuration
//static PprRun_t srun = test50;
static PprRun_t srun = kPythia6;
static PprGeo_t sgeo = kHoles;
static PprRad_t srad = kGluonRadiation;
static PprMag_t smag = k5kG;
// static MC_t     smc  = kFLUKA;
static MC_t     smc  = kGEANT3;

// Comment line
static TString  comment;

// Functions
Float_t EtaToTheta(Float_t arg);


void Config()
{
  cout << "==> Config.C..." << endl;
  
  // Set Random Number seed
  gRandom->SetSeed(12345);
  cout<<"Seed for random number generation= "<<gRandom->GetSeed()<<endl;

  switch (smc) {
  case kFLUKA: 
    // 
    // libraries required by fluka21
    // 
    gSystem->Load("libGeom");
    cout << "\t* Loading TFluka..." << endl;  
    gSystem->Load("libTFluka");    
    
    // 
    // FLUKA MC
    //
    cout << "\t* Instantiating TFluka..." << endl;
    new TFluka("C++ Interface to Fluka", 0/*verbosity*/);
    break;
  case kGEANT3: 
    //
    // Libraries needed by GEANT 3.21 
    //
    gSystem->Load("libgeant321");
    
    // 
    // GEANT 3.21 MC 
    // 
    new TGeant3("C++ Interface to Geant3");
    break;
  default:
    gAlice->Fatal("Config.C", "No MC type chosen");
    return;
  }
  
  //
  // Run loader
  //
  cout<<"Config.C: Creating Run Loader ..."<<endl;
  AliRunLoader* rl = AliRunLoader::Open("galice.root",
					AliConfig::GetDefaultEventFolderName(),
					"recreate");
  if (!rl) {
    gAlice->Fatal("Config.C","Can not instatiate the Run Loader");
    return;
  }
  rl->SetCompressionLevel(2);
  rl->SetNumberOfEventsPerFile(3);
  gAlice->SetRunLoader(rl);

  switch (smc) {
  case kFLUKA: 
    {
      //
      // Use kTRUE as argument to generate alice.pemf first
      //
      TString alice_pemf(gSystem->Which(".", "peg/mat17.pemf"));
      if (!alice_pemf.IsNull()) 
	((TFluka*)gMC)->SetGeneratePemf(kFALSE);
      else
	((TFluka*)gMC)->SetGeneratePemf(kTRUE);
    }
    break;
  }
  
     
  //
  // Set External decayer
  // 
  AliDecayer *decayer = new AliDecayerPythia();
  decayer->SetForceDecay(kAll);
  decayer->Init();
  gMC->SetExternalDecayer(decayer);


  //
  // Physics process control
  // 
  gMC->SetProcess("DCAY",1);
  gMC->SetProcess("PAIR",1);
  gMC->SetProcess("COMP",1);
  gMC->SetProcess("PHOT",1);
  gMC->SetProcess("PFIS",0);
  gMC->SetProcess("DRAY",1);
  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);

  //
  //=======================================================================
  // 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
  // 
  if (gSystem->Getenv("CONFIG_NPARTICLES"))
    nParticles = atoi(gSystem->Getenv("CONFIG_NPARTICLES"));

  cout << "\t* Creating and configuring generator for " << nParticles 
       << " particles..." << endl;
  AliGenHIJINGpara *gener = new AliGenHIJINGpara(nParticles);
  gener->SetMomentumRange(0., 999);
  gener->SetPhiRange(0, 360);

  // Set pseudorapidity range from -6 to 6.
  Float_t thmin = EtaToTheta( 6.);   // theta min. <---> eta max
  Float_t thmax = EtaToTheta(-6.);   // theta max. <---> eta min 
  gener->SetThetaRange(thmin,thmax);
  gener->SetOrigin(0, 0, 0);  //vertex position
  gener->SetSigma(0, 0, 0);   //Sigma in (X,Y,Z) (cm) on IP position
  gener->Init();
  // 
  // Activate this line if you want the vertex smearing to happen
  // track by track
  //
  gAlice->SetDebug(10);

  // 
  // Comments 
  // 
  switch (smag) {
  case k2kG: comment = comment.Append(" | L3 field 0.2 T"); break;
  case k4kG: comment = comment.Append(" | L3 field 0.4 T"); break;
  case k5kG: comment = comment.Append(" | L3 field 0.5 T"); break;
  }

  switch (srad) {
  case kGluonRadiation: 
    comment = comment.Append(" | Gluon Radiation On");  break;
  default:
    comment = comment.Append(" | Gluon Radiation Off"); break;
  }

  switch(sgeo) {
  case kHoles: comment = comment.Append(" | Holes for PHOS/RICH"); break;
  default:     comment = comment.Append(" | No holes for PHOS/RICH"); break;
  }

  std::cout << "\n\n Comment: " << comment << "\n" << std::endl;

  // 
  // Field (L3 0.4 T)
  // 
  AliMagFMaps* field = new AliMagFMaps("Maps","Maps", 2, 1., 10., smag);
  field->SetL3ConstField(0); //Using const. field in the barrel
  rl->CdGAFile();
  gAlice->SetField(field);

  // 
  // Used detectors 
  // 
  Bool_t useABSO  = kFALSE; 
  Bool_t useCRT   = kFALSE; 
  Bool_t useDIPO  = kFALSE; 
  Bool_t useFMD   = kTRUE; 
  Bool_t useFRAME = kFALSE; 
  Bool_t useHALL  = kFALSE; 
  Bool_t useITS   = kFALSE; 
  Bool_t useMAG   = kFALSE; 
  Bool_t useMUON  = kFALSE; 
  Bool_t usePHOS  = kFALSE; 
  Bool_t usePIPE  = kFALSE; 
  Bool_t usePMD   = kFALSE; 
  Bool_t useRICH  = kFALSE; 
  Bool_t useSHIL  = kFALSE; 
  Bool_t useSTART = kFALSE; 
  Bool_t useTOF   = kFALSE; 
  Bool_t useTPC   = kFALSE;
  Bool_t useTRD   = kFALSE; 
  Bool_t useZDC   = kFALSE; 
  Bool_t useEMCAL = kFALSE; 
  Bool_t useVZERO = kFALSE;

  cout << "\t* Creating the detectors ..." << endl;
  //=================== Alice BODY parameters =============================
  //=================== Alice BODY parameters =============================
  AliBODY *BODY = new AliBODY("BODY", "Alice envelop");
  
  
  if (useMAG) {
    //=================== MAG parameters ============================
    // Start with Magnet since detector layouts may be depending on
    // the selected Magnet dimensions 
    AliMAG *MAG = new AliMAG("MAG", "Magnet");
  }

  if (useABSO) {
    //=================== ABSO parameters ============================
    AliABSO *ABSO = new AliABSOv0("ABSO", "Muon Absorber");
  }

  if (useDIPO) {
    //=================== DIPO parameters ============================
    
    AliDIPO *DIPO = new AliDIPOv2("DIPO", "Dipole version 2");
  }

  if (useHALL) {
    //=================== HALL parameters ============================
    AliHALL *HALL = new AliHALL("HALL", "Alice Hall");
  }


  if (useFRAME) {
    //=================== FRAME parameters ============================
    AliFRAMEv2 *FRAME = new AliFRAMEv2("FRAME", "Space Frame");
    switch (sgeo) {
    case kHoles: FRAME->SetHoles(1); break;
    default:     FRAME->SetHoles(0); break;
    }
  }

  if (useSHIL) {
    //=================== SHIL parameters ============================
    AliSHIL *SHIL = new AliSHILv2("SHIL", "Shielding Version 2");
  }


  if (usePIPE) {
    //=================== PIPE parameters ============================
    AliPIPE *PIPE = new AliPIPEv0("PIPE", "Beam Pipe");
  }
  
  if (useITS) {
    //=================== ITS parameters ============================
    //
    // As the innermost detector in ALICE, the Inner Tracking System
    // "impacts" on almost all other detectors. This involves the fact
    // that the ITS geometry still has several options to be followed
    // in parallel in order to determine the best set-up which
    // minimizes the induced background. All the geometries available
    // to date are described in the following. Read carefully the
    // comments and use the default version (the only one uncommented)
    // unless you are making comparisons and you know what you are
    // doing. In this case just uncomment the ITS geometry you want to
    // use and run Aliroot.
    //
    // Detailed geometries:
    //
    //
    // AliITS *ITS = 
    //   new AliITSv5symm("ITS", "Updated ITS TDR detailed version "
    //  		  "with symmetric services");
    // AliITS *ITS  = 
    //   new AliITSv5asymm("ITS","Updates ITS TDR detailed version "
    // 			   "with asymmetric services");
    //
    AliITSvPPRasymmFMD *ITS  = 
      new AliITSvPPRasymmFMD("ITS","New ITS PPR detailed version "
			     "with asymmetric services");
     // don't touch this parameter if you're not an ITS developer
    ITS->SetMinorVersion(2); 
    // don't touch this parameter if you're not an ITS developer
    ITS->SetReadDet(kTRUE);
    // don't touch this parameter if you're not an ITS developer
    // ITS->SetWriteDet("$ALICE_ROOT/ITS/ITSgeometry_vPPRasymm2.det");  
    // detector thickness on layer 1 must be in the range [100,300]
    ITS->SetThicknessDet1(200.);   
    // detector thickness on layer 2 must be in the range [100,300]
    ITS->SetThicknessDet2(200.);   
    // chip thickness on layer 1 must be in the range [150,300]
    ITS->SetThicknessChip1(200.);  
    // chip thickness on layer 2 must be in the range [150,300]
    ITS->SetThicknessChip2(200.);
    // 1 --> rails in ; 0 --> rails out
    ITS->SetRails(0);          
    // 1 --> water ; 0 --> freon
    ITS->SetCoolingFluid(1);   

    // Coarse geometries (warning: no hits are produced with these
    // coarse geometries and they unuseful for reconstruction !):
    //
    //
    // AliITSvPPRcoarseasymm *ITS  = 
    //   new AliITSvPPRcoarseasymm("ITS","New ITS PPR coarse version "
    //                             "with asymmetric services");
    // 1 --> rails in ; 0 --> rails out
    // ITS->SetRails(0);
    // 0 --> Copper ; 1 --> Aluminum ; 2 --> Carbon
    // ITS->SetSupportMaterial(0);      
    //
    // AliITS *ITS  = 
    //  new AliITSvPPRcoarsesymm("ITS","New ITS PPR coarse version "
    //                           "with symmetric services");
    // 1 --> rails in ; 0 --> rails out
    // ITS->SetRails(0);                
    // 0 --> Copper ; 1 --> Aluminum ; 2 --> Carbon
    // ITS->SetSupportMaterial(0);      
    //
    // Geant3 <-> EUCLID conversion
    // ============================
    //
    // SetEUCLID is a flag to output (=1) or not to output (=0) both
    // geometry and media to two ASCII files (called by default
    // ITSgeometry.euc and ITSgeometry.tme) in a format understandable
    // to the CAD system EUCLID.  The default (=0) means that you dont
    // want to use this facility.
    //
    ITS->SetEUCLID(0);
  }

  if (useTPC) {
    //============================ TPC parameters ====================
    //
    // This allows the user to specify sectors for the SLOW (TPC
    // geometry 2) Simulator. SecAL (SecAU) <0 means that ALL lower
    // (upper) sectors are specified, any value other than that
    // requires at least one sector (lower or upper)to be specified!
    //
    // Reminder: 
    //   sectors 1-24 are lower sectors (1-12 -> z>0, 13-24 -> z<0)
    //   sectors 25-72 are the upper ones (25-48 -> z>0, 49-72 -> z<0)
    //
    //   SecLows - number of lower sectors specified (up to 6)
    //   SecUps  - number of upper sectors specified (up to 12)
    //   Sens    - sensitive strips for the Slow Simulator !!!
    //
    // This does NOT work if all S or L-sectors are specified, i.e.
    // if SecAL or SecAU < 0
    //
    //
    //----------------------------------------------------------------
    //  gROOT->LoadMacro("SetTPCParam.C");
    //  AliTPCParam *param = SetTPCParam();
    AliTPC *TPC = new AliTPCv2("TPC", "Default");
    
    // All sectors included
    TPC->SetSecAL(-1);
    TPC->SetSecAU(-1);
  }

  if (useTOF) {
    //=================== TOF parameters ============================
    AliTOF *TOF = new AliTOFv4T0("TOF", "normal TOF");
  }

  if (useRICH) {
    //=================== RICH parameters ===========================
    AliRICH *RICH = new AliRICHv1("RICH", "normal RICH");

  }

  if (useZDC) {
    //=================== ZDC parameters ============================
    AliZDC *ZDC = new AliZDCv2("ZDC", "normal ZDC");
  }

  if (useTRD) {
    //=================== TRD parameters ============================
    AliTRD *TRD = new AliTRDv1("TRD", "TRD slow simulator");

    // Select the gas mixture (0: 97% Xe + 3% isobutane, 1: 90% Xe + 10% CO2)
    TRD->SetGasMix(1);
    if (sgeo == kHoles) {
      // With hole in front of PHOS
      TRD->SetPHOShole();
      // With hole in front of RICH
      TRD->SetRICHhole();
    }
    // Switch on TR
    AliTRDsim *TRDsim = TRD->CreateTR();
  }

  if (useFMD) {
    //=================== FMD parameters ============================
    AliFMD *FMD = new AliFMDv1("FMD", "normal FMD");
  }

  if (useMUON) {
    //=================== MUON parameters ===========================
    AliMUON *MUON = new AliMUONv1("MUON", "default");
    MUON->AddGeometryBuilder(new AliMUONSt1GeometryBuilder(MUON));
    MUON->AddGeometryBuilder(new AliMUONSt2GeometryBuilder(MUON));
    MUON->AddGeometryBuilder(new AliMUONSlatGeometryBuilder(MUON));
    MUON->AddGeometryBuilder(new AliMUONTriggerGeometryBuilder(MUON));
  }

  if (usePHOS) {
    //=================== PHOS parameters ===========================
    AliPHOS *PHOS = new AliPHOSv1("PHOS", "IHEP");
  }

  if (usePMD) {
    //=================== PMD parameters ============================
    AliPMD *PMD = new AliPMDv1("PMD", "normal PMD");
  }

  if (useSTART) {
    //=================== START parameters ============================
    AliSTART *START = new AliSTARTv1("START", "START Detector");
  }

  if (useEMCAL) {
    //=================== EMCAL parameters ============================
    AliEMCAL *EMCAL = new AliEMCALv1("EMCAL", "EMCAL_55_25");
  }

  if (useCRT) {
    //=================== CRT parameters ============================
    AliCRT *CRT = new AliCRTv0("CRT", "normal ACORDE");
  }

  if (useVZERO) {
    //=================== CRT parameters ============================
    AliVZERO *VZERO = new AliVZEROv3("VZERO", "normal VZERO");
  }
}

Float_t EtaToTheta(Float_t arg)
{
  return (180./TMath::Pi())*2.*atan(exp(-arg));
}