HepMCFile.h

// -*- C++ -*-
//
// HepMCFile.h is a part of ThePEG - Toolkit for HEP Event Generation
// Copyright (C) 1999-2019 Leif Lonnblad
//
// ThePEG is licenced under version 3 of the GPL, see COPYING for details.
// Please respect the MCnet academic guidelines, see GUIDELINES for details.
//
#ifndef THEPEG_HepMCFile_H
#define THEPEG_HepMCFile_H
//
// This is the declaration of the HepMCFile class.
//
#include <iostream>
#include <fstream>
#include <config.h>
#include "ThePEG/Handlers/AnalysisHandler.h"
#include "ThePEG/Repository/CurrentGenerator.h"
#include "ThePEG/Repository/EventGenerator.h"
#include "ThePEG/Config/HepMCHelper.h"
namespace ThePEG {
 
class HepMCFile: public AnalysisHandler {
 
public:
 
  HepMCFile();
 
  HepMCFile(const HepMCFile &);
 
public:
 
  virtual void analyze(tEventPtr event, long ieve, int loop, int state);
 
public:
 
  void persistentOutput(PersistentOStream & os) const;
 
  void persistentInput(PersistentIStream & is, int version);
 
  static void Init();
 
protected:
 
  virtual IBPtr clone() const;
 
  virtual IBPtr fullclone() const;
 
protected:
 
   virtual void doinitrun();
 
  virtual void dofinish();
 
private:
 
  static ClassDescription<HepMCFile> initHepMCFile;
 
  HepMCFile & operator=(const HepMCFile &) = delete;
 
private:
 
  long _eventNumber;
 
  int _format;
 
  string _filename;
 
#ifdef HAVE_HEPMC_ROOTIO
  string _ttreename;
 
  string _tbranchname;
 
#endif
#ifdef HAVE_HEPMC3
  HepMC::Writer *_hepmcio;
#else
  HepMC::IO_BaseClass *_hepmcio;
#endif
  ofstream _hepmcdump;
 
  int _unitchoice;
 
  unsigned int _geneventPrecision;
 
  int _addHI;
 
};
 
}
 
#include "ThePEG/Utilities/ClassTraits.h"
 
namespace ThePEG {
 
template <>
struct BaseClassTrait<HepMCFile,1> {
  typedef AnalysisHandler NthBase;
};
 
template <>
struct ClassTraits<HepMCFile>
  : public ClassTraitsBase<HepMCFile> {
  static string className() { return "ThePEG::HepMCFile"; }
  static string library() { return "HepMCAnalysis.so"; }
};
 
}
 
#endif /* THEPEG_HepMCFile_H */