Fun4AllHepMCInputManager.h

Go to the documentation of this file.

#ifndef FUN4ALLHEPMCINPUTMANAGER_H__
#define FUN4ALLHEPMCINPUTMANAGER_H__

#include "PHHepMCGenHelper.h"

#include <fun4all/Fun4AllInputManager.h>
#include <fun4all/Fun4AllReturnCodes.h>

#include <fstream>
#include <iostream>
#include <map>
#include <string>

#ifndef __CINT__
#include <boost/iostreams/filtering_streambuf.hpp>
#endif

#ifndef __CINT__
#include <gsl/gsl_rng.h>
#endif

// forward declaration of classes in namespace
namespace HepMC
{
class IO_GenEvent;
class GenEvent;
};

class PHHepMCGenEvent;
class PHCompositeNode;

class Fun4AllHepMCInputManager : public Fun4AllInputManager
{
 public:
  Fun4AllHepMCInputManager(const std::string &name = "DUMMY", const std::string &nodename = "DST", const std::string &topnodename = "TOP");
  virtual ~Fun4AllHepMCInputManager();
  virtual int fileopen(const std::string &filenam);
  virtual int fileclose();
  virtual int run(const int nevents = 0);
  int isOpen() { return isopen; }
  void ReadOscar(const int i = 1) { readoscar = i; }
  virtual void Print(const std::string &what = "ALL") const;
  virtual int PushBackEvents(const int i);

  // Effectivly turn off the synchronization checking
  //
  int SyncIt(const SyncObject * /*mastersync*/) { return Fun4AllReturnCodes::SYNC_OK; }
  int GetSyncObject(SyncObject ** /*mastersync*/) { return Fun4AllReturnCodes::SYNC_NOOBJECT; }
  int NoSyncPushBackEvents(const int nevt) { return PushBackEvents(nevt); }
  HepMC::GenEvent *ConvertFromOscar();

  void set_vertex_distribution_function(PHHepMCGenHelper::VTXFUNC x, PHHepMCGenHelper::VTXFUNC y, PHHepMCGenHelper::VTXFUNC z, PHHepMCGenHelper::VTXFUNC t)
  {
    hepmc_helper.set_vertex_distribution_function(x, y, z, t);
  }

  void set_vertex_distribution_mean(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_mean(x, y, z, t);
  }

  void set_vertex_distribution_width(const double x, const double y, const double z, const double t)
  {
    hepmc_helper.set_vertex_distribution_width(x, y, z, t);
  }
  //
  void set_reuse_vertex(int src_embedding_id)
  {
    hepmc_helper.set_reuse_vertex(src_embedding_id);
  }

  int get_embedding_id() const { return hepmc_helper.get_embedding_id(); }
  //
  void set_embedding_id(int id) { hepmc_helper.set_embedding_id(id); }
 protected:
  int OpenNextFile();

  int isopen;
  int events_total;
  int events_thisfile;
  int readoscar;

  std::string filename;
  std::string topNodeName;
  PHCompositeNode *topNode;

  HepMC::IO_GenEvent *ascii_in;
  HepMC::GenEvent *evt;
  HepMC::GenEvent *save_evt;

  // some pointers for use in decompression handling
  std::ifstream *filestream;  // holds compressed filestream
  std::istream *unzipstream;  // feed into HepMc
  std::ifstream theOscarFile;

  PHHepMCGenHelper hepmc_helper;

#ifndef __CINT__
  boost::iostreams::filtering_streambuf<boost::iostreams::input> zinbuffer;
#endif
};

#endif /* __FUN4ALLHEPMCINPUTMANAGER_H__ */