KVSimReader_ELIE.cpp

//Created by KVClassFactory on Sun Jan 20 13:31:10 2013
//Author: John Frankland,,,
 
#include "KVSimReader_ELIE.h"
 
ClassImp(KVSimReader_ELIE)
 
 
 
void KVSimReader_ELIE::define_output_filename()
{
   // ROOT file called either
   //
   //       ELIE_[PROJ]_[TARG]_[EBEAM]AMeV_PRIM.root
   //
   // or
   //
   //       ELIE_[PROJ]_[TARG]_[EBEAM]AMeV_Run[xx...]_PRIM.root
   //
   // if run_number is defined
   //
   // Call after reading file header
 
   if (run_number >= 0)
      SetROOTFileName(Form("ELIE_%s_%s_%.1fAMeV_Run%d_PRIM",
                           proj.GetSymbol(), targ.GetSymbol(), ebeam, run_number));
   else
      SetROOTFileName(Form("ELIE_%s_%s_%.1fAMeV_PRIM",
                           proj.GetSymbol(), targ.GetSymbol(), ebeam));
   SetTreeTitle(Form("ELIE primary events %s + %s %.1f MeV/nuc.",
                   proj.GetSymbol(), targ.GetSymbol(), ebeam));
}
 
 
 
 
void KVSimReader_ELIE::transform_to_cm(KVSimEvent* evt)
{
   // transform all particle kinematics to CM frame from lab
 
   KVNucleus compound = proj + targ;
   evt->ChangeFrame(compound.GetV(), "CM");
}
 
 
 
 
KVSimReader_ELIE::KVSimReader_ELIE()
{
   // Default constructor
   init();
}
 
 
 
 
KVSimReader_ELIE::KVSimReader_ELIE(KVString filename) : KVSimReader()
{
   // Read file
   init();
   ConvertEventsInFile(filename);
}
 
 
 
 
Bool_t KVSimReader_ELIE::ReadFileHeader()
{
   // File header contains info on simulation
   //
   // 54 129 50 119 32
   // 20000
   //   ievt=50
   //   projectile=Ni
   //   target=Au
   //   a_proj=58
   //   z_proj=28
   //   a_targ=197
   //   z_targ=79
   //   [...]
   //   geometry=0
   //   free_nucleon=0
   //   lambda_ex=1.3
   //
   // The simulation parameters are stored in a KVNameValueList
 
   auto res = ReadLineAndCheck(5, " ");
   switch (res) {
      case KVFileReader::ReadStatus::EmptyLine:
         Info("ReadHeader", "Can't read system");
         return kFALSE;
      case KVFileReader::ReadStatus::OK:
         AddInfo({"Aproj", GetIntReadPar(1)});
         AddInfo({"Zproj", GetIntReadPar(0)});
         AddInfo({"Atarg", GetIntReadPar(3)});
         AddInfo({"Ztarg", GetIntReadPar(2)});
         AddInfo({"Ebeam", GetDoubleReadPar(4)});
         proj.SetZAandE(GetIntReadPar(0), GetIntReadPar(1), GetIntReadPar(1)*GetDoubleReadPar(4));
         targ.SetZandA(GetIntReadPar(2), GetIntReadPar(3));
         ebeam = GetDoubleReadPar(4);
         break;
      default:
         return kFALSE;
   }
 
   res = ReadLineAndCheck(1, " ");
   switch (res) {
      case KVFileReader::ReadStatus::EmptyLine:
         Info("ReadHeader", "Can't read events");
         return kFALSE;
      case KVFileReader::ReadStatus::OK:
         AddInfo({"Nevents", GetIntReadPar(0)});
         break;
 
      default:
         return kFALSE;
   }
   read_elie_params(*this);
 
   define_output_filename();
 
   return kTRUE;
}
 
 
 
 
void KVSimReader_ELIE::read_elie_params(KVFileReader& input_file_reader)
{
   reuse_last_line=true;
   elie_params->Clear();
   elie_params->SetName("ELIE Parameters");
   auto res = input_file_reader.ReadLineAndCheck(2, "=");
   while (res != KVFileReader::ReadStatus::EndOfFile) {
      if (res == KVFileReader::ReadStatus::OK) {
         if (input_file_reader.GetReadPar(1).IsDigit()) elie_params->SetValue(input_file_reader.GetReadPar(0), input_file_reader.GetIntReadPar(1));
         else if (input_file_reader.GetReadPar(1).IsFloat()) elie_params->SetValue(input_file_reader.GetReadPar(0), input_file_reader.GetDoubleReadPar(1));
         else elie_params->SetValue(input_file_reader.GetReadPar(0), input_file_reader.GetReadPar(1));
      }
      res = input_file_reader.ReadLineAndCheck(2, "=");
   }
   if (elie_params->GetEntries()) AddObject(elie_params);
}
 
 
 
 
Bool_t KVSimReader_ELIE::ReadEventHeader(KVSimEvent* evt)
{
   // Event structure: (primary events) -> OLD WAY, see bellow for recent ELIE calculations
   //   nevt multiplicite b reduit
   //   0 2 0.998654688551
   //   particules dans l'evt
   //   indice_particule  charge_particule masse_particule teta_particule(degres) phi_particule(degres) indice_particule energie d'excitation totale (MeV)
   //   0 27 57 3.27897003986 230.52425244 1109.37002505 0 0.00499304975261
   //   1 80 198 176.726338776 129.481056376 319.364098122 1 0.017344278088
 
   //   nevt multiplicite b reduit
   //1 11 0.665404278608
   //   indice_particule  charge_particule masse_particule teta_particule(degres) phi_particule(degres) E(MeV) E*(MeV) L(hbar)
   //0 21 45 2.49354294869 78.2453656442 2180.55536792 91.0936851491 1.0
   //1 20 46 5.25547324063 79.8035650615 7.5872385507 93.1179892635 1.0
 
   // first line of first event will have been read already by ReadFileHeader,
   // therefore we do not read a new line in this case
   auto res = (reuse_last_line ? ReuseLineAndCheck(3, " ") : ReadLineAndCheck(3, " "));
   reuse_last_line=false;
   mult = 0;
   switch (res) {
      case KVFileReader::ReadStatus::EmptyLine:
      case KVFileReader::ReadStatus::EndOfFile:
         return kFALSE;
      case KVFileReader::ReadStatus::OK:
         evt->SetNumber(GetIntReadPar(0));
         mult = GetIntReadPar(1);
         evt->GetParameters()->SetValue("bred", GetDoubleReadPar(2));
         evt->GetParameters()->SetValue("mult", mult);
         return kTRUE;
      default:
         return kFALSE;
   }
   return kFALSE;
}
 
 
 
 
Bool_t KVSimReader_ELIE::ReadEvent(KVSimEvent*evt)
{
   for (Int_t mm = 0; mm < mult; mm++) {
      auto nuc = evt->AddParticle();
      if (!ReadNucleus(nuc)) return kFALSE;
   }
 
   // if in lab, transform to CM frame
   if (elie_params->GetIntValue("lab_frame") > 0) transform_to_cm(evt);
   else evt->SetFrameName("CM");
 
   return kTRUE;
}
 
 
 
 
Bool_t KVSimReader_ELIE::ReadNucleus(KVSimNucleus* nuc)
{
   // Event structure: (primary events) -> OLD WAY, see bellow for recent ELIE calculations
   //   nevt multiplicite b reduit
   //   0 2 0.998654688551
   //   particules dans l'evt
   //   indice_particule  charge_particule masse_particule teta_particule(degres) phi_particule(degres) indice_particule energie d'excitation totale (MeV)
   //   0 27 57 3.27897003986 230.52425244 1109.37002505 0 0.00499304975261
   //   1 80 198 176.726338776 129.481056376 319.364098122 1 0.017344278088
 
   //   nevt multiplicite b reduit
   //1 11 0.665404278608
   //   indice_particule  charge_particule masse_particule teta_particule(degres) phi_particule(degres) E(MeV) E*(MeV) L(hbar)
   //0 21 45 2.49354294869 78.2453656442 2180.55536792 91.0936851491 1.0
   //1 20 46 5.25547324063 79.8035650615 7.5872385507 93.1179892635 1.0
 
 
 
   auto res = ReadLineAndCheck(8, " ");
   switch (res) {
      case KVFileReader::ReadStatus::EmptyLine:
         Info("ReadNucleus", "case 0 line est vide");
         return kFALSE;
 
      case KVFileReader::ReadStatus::OK:
         nuc->SetZ(GetIntReadPar(1));
         nuc->SetA(GetIntReadPar(2));
         if (GetDoubleReadPar(6) >= 0.) nuc->SetExcitEnergy(GetDoubleReadPar(6));
         nuc->SetEnergy(GetDoubleReadPar(5));
         nuc->SetTheta(GetDoubleReadPar(3));
         nuc->SetPhi(GetDoubleReadPar(4));
         if (nuc->GetExcitEnergy() > 0.) nuc->SetAngMom(GetDoubleReadPar(7), 0, 0);
 
         return kTRUE;
         break;
 
      default:
         return kFALSE;
   }
 
   return kFALSE;
}