Update timestamps for new AMANDA simulation (17/02/2015)

[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderHepMC.cxx

diff --git a/EVGEN/AliGenReaderHepMC.cxx b/EVGEN/AliGenReaderHepMC.cxx
index 0bd03677a180bde8cf03e8ba279f49eea7a05832..88636bf657618684e31759b8d51f79f9ec889a63 100644 (file)
--- a/EVGEN/AliGenReaderHepMC.cxx
+++ b/EVGEN/AliGenReaderHepMC.cxx
@@ -5,11 +5,18 @@
 #include "AliGenReaderHepMC.h"
 #include "AliRun.h"
 #include "AliStack.h"
+#include "AliGenHepMCEventHeader.h"
 
+#include "HepMC/IO_BaseClass.h"
+#include "HepMC/GenEvent.h"
+#include "HepMC/IO_GenEvent.h"
 
+ClassImp(AliGenReaderHepMC)
 
+AliGenReaderHepMC::AliGenReaderHepMC():fEventsHandle(0), fGenEvent(0), fParticleArray(0), fParticleIterator(0), fGenEventHeader(0) {;}
 
-ClassImp(AliGenReaderHepMC)
+AliGenReaderHepMC::AliGenReaderHepMC(const AliGenReaderHepMC &reader)
+   :AliGenReader(reader), fEventsHandle(0), fGenEvent(0), fParticleArray(0), fParticleIterator(0), fGenEventHeader(0) {reader.Copy(*this);}
 
 
 AliGenReaderHepMC& AliGenReaderHepMC::operator=(const  AliGenReaderHepMC& rhs)
@@ -19,6 +26,8 @@ AliGenReaderHepMC& AliGenReaderHepMC::operator=(const  AliGenReaderHepMC& rhs)
    return *this;
 }
 
+AliGenReaderHepMC::~AliGenReaderHepMC(){ delete fEventsHandle; delete fGenEvent; delete fParticleArray; delete fParticleIterator;} // not deleting fGenEventHeader as it is returned out
+
 void AliGenReaderHepMC::Copy(TObject&) const
 {
    //
@@ -29,27 +38,58 @@ void AliGenReaderHepMC::Copy(TObject&) const
 
 void AliGenReaderHepMC::Init()
 {
+   // check if file exists, using FILE to avoid (the otherwise faster) POSIX dependencies
+   if (FILE *file = fopen(fFileName,"r"))  {
+      printf("File %s opened \n", fFileName);
+      fclose(file);
+   } else {
+      printf("Couldn't open input file: %s \n", fFileName);
+   }
    // Initialisation
    fEventsHandle = new HepMC::IO_GenEvent(fFileName, std::ios::in);
    fParticleArray = new TClonesArray("TParticle");
    fParticleIterator = new TIter(fParticleArray);
-
-
-
 }
 
 Int_t AliGenReaderHepMC::NextEvent()
 {
+   // Clean memory
+   if (fGenEvent) delete fGenEvent;
    // Read the next event
    if ((fGenEvent = fEventsHandle->read_next_event())) {
-      THepMCParser::ParseGenEvent2TCloneArray(fGenEvent,fParticleArray,false);
+      THepMCParser::ParseGenEvent2TCloneArray(fGenEvent,fParticleArray,"GEV","CM",false);
       fParticleIterator->Reset();
-
-
-      // implement header... somewhere
-
+      THepMCParser::HeavyIonHeader_t heavyIonHeader;
+      THepMCParser::PdfHeader_t pdfHeader;
+      THepMCParser::ParseGenEvent2HeaderStructs(fGenEvent,heavyIonHeader,pdfHeader,true,true);
+      fGenEventHeader = new AliGenHepMCEventHeader(
+            heavyIonHeader.Ncoll_hard,
+            heavyIonHeader.Npart_proj,
+            heavyIonHeader.Npart_targ,
+            heavyIonHeader.Ncoll,
+            heavyIonHeader.spectator_neutrons,
+            heavyIonHeader.spectator_protons,
+            heavyIonHeader.N_Nwounded_collisions,
+            heavyIonHeader.Nwounded_N_collisions,
+            heavyIonHeader.Nwounded_Nwounded_collisions,
+            heavyIonHeader.impact_parameter,
+            heavyIonHeader.event_plane_angle,
+            heavyIonHeader.eccentricity,
+            heavyIonHeader.sigma_inel_NN,
+            pdfHeader.id1,
+            pdfHeader.id2,
+            pdfHeader.pdf_id1,
+            pdfHeader.pdf_id2,
+            pdfHeader.x1,
+            pdfHeader.x2,
+            pdfHeader.scalePDF,
+            pdfHeader.pdf1,
+            pdfHeader.pdf2
+      );
+      printf("Parsed event %d with %d particles.\n", fGenEvent->event_number(), fGenEvent->particles_size());
       return fGenEvent->particles_size();
    }
+   printf("No more events in the file.\n");
    return 0;
 }