PhotosHEPEVTEvent.cc

#include "PhotosHEPEVTEvent.h"
#include "Log.h"
#include <generators/kkmc/KKGenInterface.h>
const static int fortranNMXHEP = 4000;
 
/*
struct HEPEVT
{
  int    nevhep;
  int    nhep;
  int    isthep[fortranNMXHEP];
  int    idhep[fortranNMXHEP];
  int    jmohep[fortranNMXHEP][2];
  int    jdahep[fortranNMXHEP][2];
  double phep[fortranNMXHEP][5];
  double vhep[fortranNMXHEP][4];
  //      NEVPHO,NPHO,ISTPHO(NMXPHO),IDPHO(NMXPHO),
  //  JMOPHO(2,NMXPHO),JDAPHO(2,NMXPHO),PPHO(5,NMXPHO),VPHO(4,NMXPHO)
  //   int qedrad[NMXHEP]  was an add up
  //   for  HEPEVT in F77 times. Separate common PH_PHOQED
  //   also phoif_.chkif[NMXPHO] was add up for PHOEVT
  //   now it is pho.qedrad
} hepevt_;
*/
 
struct PHOQED {
  int qedrad[fortranNMXHEP]; // Photos flag
} phoqed_;
 
namespace Photospp {
 
  PhotosHEPEVTEvent::~PhotosHEPEVTEvent()
  {
    for (unsigned int i = 0; i < particle_list.size(); i++) delete particle_list[i];
  }
 
  PhotosHEPEVTEvent::PhotosHEPEVTEvent()
  {
    // NOTE: We set default units to be GEV for HEPEVT events. This can be
    //       overridden by calling Photos::setMomentumUnit(Photos::MEV);
    //       after the event is created.
    Photos::setMomentumUnit(Photos::GEV);
  }
 
  void PhotosHEPEVTEvent::addParticle(PhotosHEPEVTParticle* p)
  {
    p->setEvent(this);
 
    p->setBarcode(particle_list.size());
    particle_list.push_back(p);
  }
 
  PhotosHEPEVTParticle* PhotosHEPEVTEvent::getParticle(int i)
  {
    if (i < 0 || i >= (int)particle_list.size()) return NULL;
    return particle_list[i];
  }
 
  void PhotosHEPEVTEvent::setParticle(int i, PhotosHEPEVTParticle* p)
  {
    if (i < 0 || i >= (int)particle_list.size()) return;
    particle_list[i] = p;
  }
 
  int PhotosHEPEVTEvent::getParticleCount()
  {
    return particle_list.size();
  }
 
  std::vector<PhotosParticle*> PhotosHEPEVTEvent::getParticleList()
  {
    std::vector<PhotosParticle*> ret;
 
    for (unsigned int i = 0; i < particle_list.size(); i++) ret.push_back((PhotosParticle*)particle_list[i]);
 
    return ret;
  }
 
  void PhotosHEPEVTEvent::print()
  {
    Log::Info() << "PhotosHEPEVTEvent" << endl << "-----------------" << endl;
    for (unsigned int i = 0; i < particle_list.size(); i++) particle_list[i]->print();
  }
 
  void PhotosHEPEVTEvent::clear()
  {
    for (unsigned int i = 0; i < particle_list.size(); i++) delete particle_list[i];
    particle_list.clear();
  }
 
  void PhotosHEPEVTEvent::read_event_from_HEPEVT(PhotosHEPEVTEvent* evt)
// vertex info is not needed, but what about info for write_event_to_HEPEVT?
// need to be fixed later.
  {
    if (evt == NULL) return;
 
    for (int i = 0; i < hepevt_.nhep; i++) {
      PhotosHEPEVTParticle* p = new PhotosHEPEVTParticle
      (
        hepevt_.idhep [i],
        hepevt_.isthep[i],
        hepevt_.phep  [i][0],
        hepevt_.phep  [i][1],
        hepevt_.phep  [i][2],
        hepevt_.phep  [i][3],
        hepevt_.phep  [i][4],
        hepevt_.jmohep[i][0] - 1,
        hepevt_.jmohep[i][1] - 1,
        hepevt_.jdahep[i][0] - 1,
        hepevt_.jdahep[i][1] - 1
      );
      evt->addParticle(p);
    }
  }
 
  void PhotosHEPEVTEvent::write_event_to_HEPEVT(PhotosHEPEVTEvent* evt)
// vertex info is needed, for photons it should be as of other sisters
// taken at  read_event_from_HEPEVT
// need to be fixed later.
 
  {
    if (evt == NULL) return;
 
    hepevt_.nhep = evt->getParticleCount();
 
    for (int i = 0; i < hepevt_.nhep; i++) {
      PhotosHEPEVTParticle* p = evt->getParticle(i);
 
      hepevt_.idhep [i]   = p->getPdgID();
      hepevt_.isthep[i]   = p->getStatus();
      hepevt_.phep  [i][0] = p->getPx();
      hepevt_.phep  [i][1] = p->getPy();
      hepevt_.phep  [i][2] = p->getPz();
      hepevt_.phep  [i][3] = p->getE();
      hepevt_.phep  [i][4] = p->getMass();
      hepevt_.jmohep[i][0] = p->getFirstMotherIndex()  + 1;
      hepevt_.jmohep[i][1] = p->getSecondMotherIndex() + 1;
      hepevt_.jdahep[i][0] = p->getDaughterRangeStart() + 1;
      hepevt_.jdahep[i][1] = p->getDaughterRangeEnd()  + 1;
      hepevt_.vhep  [i][0] = 0.0;
      hepevt_.vhep  [i][1] = 0.0;
      hepevt_.vhep  [i][2] = 0.0;
      hepevt_.vhep  [i][3] = 0.0;
    }
  }
 
} // namespace Photospp