]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TTherminator/AliGenTherminator.cxx
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / TTherminator / AliGenTherminator.cxx
index 004564af140bcf38f85f5bb8cac489341e43d314..58b504742d2f0679ac92bef62244782fa52a1c9d 100644 (file)
@@ -31,6 +31,7 @@ AliGenTherminator::AliGenTherminator():
   fEventNumber(0),
   fFileName(""),
   fFreezeOutModel(""),
+  fFOHSlocation(""),
   fTemperature(0.1656),
   fMiuI(-0.0009),
   fMiuS(0.0),
@@ -44,6 +45,15 @@ AliGenTherminator::AliGenTherminator():
   fBWDelay(0.0)
 {
   // Default constructor
+  fFOHSlocation = "";
+
+  fEnergyCMS = 5500.;
+  fAProjectile = 208;
+  fZProjectile = 82;
+  fProjectile = "A";
+  fATarget = 208;
+  fZTarget = 82;
+  fTarget = "A";
 }
 AliGenTherminator::AliGenTherminator(Int_t npart):
   AliGenMC(npart),
@@ -51,6 +61,7 @@ AliGenTherminator::AliGenTherminator(Int_t npart):
   fEventNumber(0),
   fFileName(""),
   fFreezeOutModel(""),
+  fFOHSlocation(""),
   fTemperature(0.1656),
   fMiuI(-0.0009),
   fMiuS(0.0),
@@ -65,6 +76,13 @@ AliGenTherminator::AliGenTherminator(Int_t npart):
 {
   // Constructor specifying the size of the particle table
   fNprimaries = 0;
+  fEnergyCMS = 5500.;
+  fAProjectile = 208;
+  fZProjectile = 82;
+  fProjectile = "A";
+  fATarget = 208;
+  fZTarget = 82;
+  fTarget = "A";
 }
 
 AliGenTherminator::~AliGenTherminator()
@@ -80,7 +98,9 @@ void AliGenTherminator::Generate()
 
   Float_t polar[3]    =   {0,0,0};
   Float_t origin[3]   =   {0,0,0};
+  Float_t time        =   0.;
   Float_t origin0[3]  =   {0,0,0};
+  Float_t time0       =   0.;
   Float_t p[3];
   Float_t mass, energy;
 
@@ -90,6 +110,7 @@ void AliGenTherminator::Generate()
 
   Vertex();
   for (j=0; j < 3; j++) origin0[j] = fVertex[j];
+  time0 = fTime;
 
   // Generate one event
 
@@ -100,6 +121,9 @@ void AliGenTherminator::Generate()
   Int_t np = fParticles.GetEntriesFast();
   AliWarning(Form("Imported %d particles", np));
 
+  Int_t *idsOnStack;
+  idsOnStack = new Int_t[np];
+
   TParticle *iparticle;
   Double_t evrot = gRandom->Rndm()*TMath::Pi();
   
@@ -128,24 +152,26 @@ void AliGenTherminator::Generate()
       
       Double_t vphi = TMath::ATan2(iparticle->Vy(), iparticle->Vx());
       Double_t vrho = TMath::Hypot(iparticle->Vx(), iparticle->Vy());
-      origin[0] = origin0[0]+vrho*TMath::Cos(vphi + evrot);
-      origin[1] = origin0[1]+vrho*TMath::Sin(vphi + evrot);
-      origin[2] = origin0[2]+iparticle->Vz();
-      
+      origin[0] = vrho*TMath::Cos(vphi + evrot);
+      origin[1] = vrho*TMath::Sin(vphi + evrot);
+      origin[2] = iparticle->Vz();
+      time      = iparticle->T();
+
       imo = -1;
-      TParticle* mother = 0;
+      //      TParticle* mother = 0;
       if (hasMother) {
        imo = iparticle->GetFirstMother();
-       mother = (TParticle *) fParticles->At(imo);
+       //      mother = (TParticle *) fParticles.At(imo);
       } // if has mother   
       Bool_t tFlag = (!hasDaughter);
       
-      printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo);
-      PushTrack(tFlag,imo,kf,
+      //      printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo>=0?idsOnStack[imo]:imo);
+      PushTrack(tFlag,imo>=0?idsOnStack[imo]:imo,kf,
                p[0],p[1],p[2],energy,
-               origin[0],origin[1],origin[2],iparticle->T()*0.197327*1e-13/300000000,
+               origin[0],origin[1],origin[2],time,
                polar[0],polar[1],polar[2],
                hasMother ? kPDecay:kPNoProcess,nt);
+      idsOnStack[i] = nt;
       fNprimaries++;
       KeepTrack(nt);
     }
@@ -175,36 +201,40 @@ void AliGenTherminator::Generate()
       origin[0] = vrho*TMath::Cos(vphi + evrot);
       origin[1] = vrho*TMath::Sin(vphi + evrot);
       origin[2] = iparticle->Vz();
-      
+      time      = iparticle->T();
+
       imo = -1;
-      TParticle* mother = 0;
+      //      TParticle* mother = 0;
       if (hasMother) {
        imo = iparticle->GetFirstMother();
-       mother = (TParticle *) fParticles->At(imo);
+       //      mother = (TParticle *) fParticles.At(imo);
       } // if has mother   
       Bool_t tFlag = (hasDaughter);
       
-      printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo);
-      PushTrack(tFlag,imo,kf,
+//       printf("Found mother %i with true id %i\n", imo, imo>=0?idsOnStack[imo]:imo);
+//       printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo>=0?idsOnStack[imo]:imo);
+      PushTrack(tFlag,imo>=0?idsOnStack[imo]:imo,kf,
                p[0],p[1],p[2],energy,
-               origin[0],origin[1],origin[2],iparticle->T()*0.197327*1e-13/300000000,
+               origin[0],origin[1],origin[2],time,
                polar[0],polar[1],polar[2],
                hasMother ? kPDecay:kPNoProcess,nt);
+      idsOnStack[i] = nt;
       fNprimaries++;
       KeepTrack(nt);
 
       origin[0] = origin0[0]+vrho*TMath::Cos(vphi + evrot);
       origin[1] = origin0[1]+vrho*TMath::Sin(vphi + evrot);
       origin[2] = origin0[2]+iparticle->Vz();
+      time      = time0+iparticle->T();
 
       imo = nt;
-      mother = (TParticle *) fParticles->At(nt);
+      //      mother = (TParticle *) fParticles.At(nt);
       tFlag = (!hasDaughter);
       
-      printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo);
+//       printf("Pushing Track %d with status %d mother %d\n", kf, tFlag, imo);
       PushTrack(tFlag,imo,kf,
                p[0],p[1],p[2],energy,
-               origin[0],origin[1],origin[2],iparticle->T()*0.197327*1e-13/300000000,
+               origin[0],origin[1],origin[2],time,
                polar[0],polar[1],polar[2],
                hasMother ? kPDecay:kPNoProcess,nt);
       fNprimaries++;
@@ -212,6 +242,8 @@ void AliGenTherminator::Generate()
     }
   }
 
+
+
   SetHighWaterMark(fNprimaries);
 
   TArrayF eventVertex;
@@ -219,6 +251,7 @@ void AliGenTherminator::Generate()
   eventVertex[0] = origin0[0];
   eventVertex[1] = origin0[1];
   eventVertex[2] = origin0[2];
+  Float_t eventTime = time0;
 
 // Builds the event header, to be called after each event
   AliGenEventHeader* header = new AliGenHijingEventHeader("Therminator");
@@ -231,6 +264,7 @@ void AliGenTherminator::Generate()
 
   ((AliGenHijingEventHeader*) header)->SetNProduced(fNprimaries);
   ((AliGenHijingEventHeader*) header)->SetPrimaryVertex(eventVertex);
+  ((AliGenHijingEventHeader*) header)->SetInteractionTime(eventTime);
   ((AliGenHijingEventHeader*) header)->SetImpactParameter(0.0);
   ((AliGenHijingEventHeader*) header)->SetTotalEnergy(0.0);
   ((AliGenHijingEventHeader*) header)->SetHardScatters(0);
@@ -267,9 +301,12 @@ void AliGenTherminator::Generate()
 //     ((AliGenHijingEventHeader*) header)->SetTrials(fTrials);
 // Event Vertex
   header->SetPrimaryVertex(fVertex);
+  header->SetInteractionTime(fTime);
   AddHeader(header);
   fCollisionGeometry = (AliGenHijingEventHeader*)  header;
 
+  delete [] idsOnStack;
+
   //  gAlice->SetGenEventHeader(header); 
 }
 
@@ -384,6 +421,8 @@ void AliGenTherminator::SetModel(const char *model)
 {
   // Set the freeze-out model to use
   fFreezeOutModel = model;
+  AliWarning(Form("Selected model %s", fFreezeOutModel.Data()));
+  AliWarning(Form("FOHSLocation is %s", fFOHSlocation.Data()));
 }
 
 void AliGenTherminator::SetLhyquidSet(const char *set)
@@ -392,27 +431,112 @@ void AliGenTherminator::SetLhyquidSet(const char *set)
   const char *aroot = gSystem->Getenv("ALICE_ROOT");
   if (strstr(set, "LHC500C0005")) {
     AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
-    AliWarning(Form("  Pb-Pb collisions, centrality 0-5%"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 0-5 percent"));
     AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
     AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
     AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C0005/FO.txt"));
-    fFOHSlocation = aroot;
-    fFOHSlocation += "/TTherminator/data/LHC500C0005";
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C0005");
+  }
+  if (strstr(set, "LHC500C0510")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 5-10 percent"));
+    AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C0510/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C0510");
+  }
+  if (strstr(set, "LHC500C1020")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 10-20 percent"));
+    AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C1020/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C1020");
   }
   else if (strstr(set, "LHC500C2030")) {
     AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
-    AliWarning(Form("  Pb-Pb collisions, centrality 20-30%"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 20-30 percent"));
     AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
     AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
     AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C2030/FO.txt"));
-    fFOHSlocation = aroot;
-    fFOHSlocation += "/TTherminator/data/LHC500C2030";
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C2030");
+  }
+  else if (strstr(set, "LHC500C3040")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 30-40 percent"));
+    AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C3040/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C3040");
+  }
+  else if (strstr(set, "LHC500C4050")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, centrality 40-50 percent"));
+    AliWarning(Form("  initial temperature at tau=1 fm in the center Ti=500 MeV"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC500C4050/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC500C4050");
+  }
+  else if (strstr(set, "LHC276TC0005")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 00-05 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC0005/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC0005");
+  }
+  else if (strstr(set, "LHC276TC0510")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 05-10 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC0510/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC0510");
+  }
+  else if (strstr(set, "LHC276TC1020")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 10-20 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC1020/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC1020");
+  }
+  else if (strstr(set, "LHC276TC2030")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 20-30 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC2030/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC2030");
+  }
+  else if (strstr(set, "LHC276TC3040")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 30-40 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC3040/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC3040");
+  }
+  else if (strstr(set, "LHC276TC4050")) {
+    AliWarning(Form("AliGenTherminator: Selected default Lhyquid hypersurface"));
+    AliWarning(Form("  Pb-Pb collisions, 2.76TeV, centrality 40-50 percent"));
+    AliWarning(Form("  freeze-out criteria Tf=145 MeV"));
+    AliWarning(Form("  for details see $(ALICE_ROOT)/TTherminator/data/LHC276TC4050/FO.txt"));
+    fFOHSlocation.Append(aroot);
+    fFOHSlocation.Append("/TTherminator/data/LHC276TC4050");
   }
   else {
     AliWarning(Form("Did not find Lhyquid set %s", set));
     AliWarning(Form("Reverting to default: current directory"));
-    fFOHSlocation = "";
+    fFOHSlocation += "";
   }
+
 }
 
 void AliGenTherminator::SetLhyquidInputDir(const char *inputdir)