EVGEN/AliGenReaderSL.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17 //
  18 // Realisations of the AliGenReader interface to be used with AliGenExFile.
  19 // NextEvent() loops over events
  20 // and NextParticle() loops over particles.
  21 // This implementation reads various StarLight output formats
  22 // Author: andreas.morsch@cern.ch
  23
  24 #include <TVirtualMC.h>
  25 #include <TDatabasePDG.h>
  26 #include <TParticle.h>
  27
  28 #include "AliGenReaderSL.h"
  29 #include "AliRun.h"
  30
  31
  32 ClassImp(AliGenReaderSL)
  33
  34
  35 AliGenReaderSL& AliGenReaderSL::operator=(const  AliGenReaderSL& rhs)
  36 {
  37 // Assignment operator
  38     rhs.Copy(*this);
  39     return *this;
  40 }
  41
  42 void AliGenReaderSL::Copy(TObject&) const
  43 {
  44     //
  45     // Copy
  46     //
  47     Fatal("Copy","Not implemented!\n");
  48 }
  49
  50 void AliGenReaderSL::Init()
  51 {
  52     // Initialisation
  53     if( !(fFile = fopen(fFileName,"r")) ) {
  54         printf("Couldn't open input file: %s \n", fFileName);
  55     } else {
  56         printf("File %s opened \n", fFileName);
  57     }
  58
  59
  60 }
  61
  62 Int_t AliGenReaderSL::NextEvent()
  63 {
  64 // Read the next event
  65
  66     if (fFormat == 0) {
  67         fNParticles = 4;
  68         return (fNParticles);
  69     }
  70
  71 // Example
  72 // EVENT: 7 23 1
  73 // GAMMAENERGIES: 27.6431
  74 // VERTEX: 0 0 0 0 1 0 0 23
  75     char linelabel[20];
  76     int i1 = 0;
  77     int i2 = 0;
  78     int i3 = 0;
  79
  80
  81     double x1 = 0.0;
  82     double x2 = 0.0;
  83     double x3 = 0.0;
  84     double x4 = 0.0;
  85
  86     int ntrk = 0;
  87     int nvtx = 0;
  88     //
  89     Float_t eGamma1, eGamma2;
  90     eGamma2 = 0.;
  91
  92     // Event line
  93     fscanf(fFile,"%6s %d %d %d ",linelabel, &i1, &ntrk, &i2);
  94     fNParticles = ntrk;
  95     //printf("Event line: %s i1 = %5d ntrk = %5d i2 = %5d \n", linelabel, i1, ntrk, i2);
  96
  97     // Gamma line
  98     if (fFormat == 1) {
  99         fscanf(fFile, "%14s %f  ",linelabel, &eGamma1);
 100     } else if (fFormat == 2) {
 101         fscanf(fFile, "%14s %f %f ",linelabel, &eGamma1, &eGamma2);
 102     }
 103
 104 //    printf("Gamma line: %s Egamma1 = %13.3f Egamma2 = %13.3f \n", linelabel, eGamma1, eGamma2);
 105
 106     // Vertex line
 107     fscanf(fFile, "%7s %lf %lf %lf %lf %d %d %d %d",
 108            linelabel, &x1, &x2, &x3, &x4, &i1, &i2, &i3, &nvtx);
 109 //    printf("Vertex line: %s (x = %13.3f, y =  %13.3f, z =  %13.3f, t =  %13.3f) i1 = %5d i2 = %5d i3 = %5d nvtx = %5d \n",
 110 //         linelabel, x1, x2, x3, x4, i1, i2, i3, nvtx);
 111     if(ntrk != nvtx) printf("ERROR: ntrk = %5d  nvtx = %5d \n", ntrk, nvtx);
 112
 113     return (fNParticles);
 114
 115 }
 116
 117 TParticle* AliGenReaderSL::NextParticle()
 118 {
 119     // Read next particle
 120
 121     Float_t px, py, pz;
 122     Int_t pdg;
 123     static TParticle particle;
 124
 125     if (fFormat == 0) {
 126         Int_t ievent;
 127         Int_t ipart;
 128         fscanf(fFile, "%d %d %d %f %f %f ", &ievent, &ipart, &pdg, &px, &py, &pz);
 129 //      printf("%5d %5d %5d %13.3f %13.3f %13.3f \n", ievent, ipart, pdg, px, py, pz);
 130
 131         if (pdg == 8) pdg =  211;
 132         if (pdg == 9) pdg = -211;
 133
 134     } else {
 135         char tracklabel[20];
 136         int i1 = 0;
 137         int i2 = 0;
 138         int i3 = 0;
 139         int i4 = 0;
 140         fscanf(fFile,"%6s %d %f %f %f %d %d %d %d",
 141                tracklabel, &i1, &px, &py, &pz, &i2, &i3, &i4, &pdg);
 142 //      printf("Particle %5d %13.3f %13.3f %13.3f \n",  pdg, px, py, pz);
 143     }
 144
 145     Double_t mass = TDatabasePDG::Instance()->GetParticle(pdg)->Mass();
 146     Float_t e = TMath::Sqrt(px * px + py * py + pz * pz + mass * mass);
 147     particle.SetMomentum(px, py, pz, e);
 148     particle.SetPdgCode(pdg);
 149
 150     return &particle;
 151 }
 152
 153 void AliGenReaderSL::RewindEvent()
 154 {
 155 //    fFile->Rewind();
 156 }