Forester.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Forester.h

#ifndef INCLUDE_EICSMEAR_ERHIC_FORESTER_H_
#define INCLUDE_EICSMEAR_ERHIC_FORESTER_H_

// C(++) headers
#include <cmath>
#include <ctime>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <string>

// ROOT headers
#include <Rtypes.h>
#include <TFile.h>
#include <TStopwatch.h>
#include <TTree.h>

// Other headers
#include "eicsmear/erhic/EventMC.h"

namespace erhic {

class FileType;
class VirtualEventFactory;


class Forester : public TObject {
 public:
  Forester();

  virtual ~Forester();

  Long64_t Plant();

  void SetInputFileName(const std::string&);

  void SetOutputFileName(const std::string&);

  void SetTreeName(const std::string& = "EICTree");

  void SetBranchName(const std::string& = "event");

  std::string GetInputFileName() const;

  std::string GetOutputFileName() const;

  std::string GetTreeName() const;

  std::string GetBranchName() const;

  void SetMaxNEvents(Long64_t = 0);

  Long64_t GetMaxNEvents() const;

  void SetMessageInterval(Long64_t = 10000);

  void Print(std::ostream& stream) const;

  void Print(Option_t* = "not used") const;

  void SetBeVerbose(bool = false);

  bool BeVerbose() const;

  const erhic::FileType* GetFileType() const;

  class Status{
  public:
    Status();
    virtual ~Status();
    virtual std::ostream& Print(std::ostream& os = std::cout) const;

  protected:
    virtual void StartTimer();
    virtual void StopTimer();
    virtual void ModifyEventCount(Long64_t count);
    virtual void ModifyParticleCount(Long64_t count);

    time_t mStartTime;
    time_t mEndTime;
    Long64_t mNEvents;
    Long64_t mNParticles;

    // The TStopwatch is mutable as "GetRealTime()" is non-const.
    mutable TStopwatch mTimer;

    friend class Forester;

    ClassDef(Status, 1);
  };

 protected:
  const Status& GetGetStatus() const {
    return mStatus;
  }

  bool OpenInput();

  bool SetupOutput();

  void Finish();

  bool AllocateEvent();

  bool FindFirstEvent();

  void PrintStatus() const;

  bool MustQuit() const;

  void SetMustQuit(bool quit);

  // Member variables.
  // Those with comment //! will be treated as transient members by ROOT
  // and won't be written to a file.
  // IMPORTANT: There MUST be a space after //!

  Bool_t mQuit;  
  Bool_t mVerbose;  
  TTree* mTree;  
  VirtualEvent* mEvent;  
  const erhic::FileType* mFile;  
  TFile* mRootFile;  
  Long64_t mMaxNEvents;  
  Long64_t mInterval;  

  std::shared_ptr<std::istream> mTextFile;  
  std::string mInputName;  
  std::string mOutputName;  
  std::string mTreeName;  
  std::string mBranchName;  
  std::string mLine;  
  Status mStatus;  
  VirtualEventFactory* mFactory;  

  ClassDef(Forester, 3)
};

inline void Forester::SetInputFileName(const std::string& name) {
  mInputName = name;
}

inline void Forester::SetOutputFileName(const std::string& name) {
  mOutputName = name;
}

inline void Forester::SetTreeName(const std::string& name) {
  mTreeName = name;
}

inline void Forester::SetBranchName(const std::string& name) {
  mBranchName = name;
}

inline std::string Forester::GetInputFileName() const {
  return mInputName;
}

inline std::string Forester::GetOutputFileName() const {
  return mOutputName;
}

inline std::string Forester::GetTreeName() const {
  return mTreeName;
}

inline std::string Forester::GetBranchName() const {
  return mBranchName;
}

inline void Forester::SetMaxNEvents(Long64_t number) {
  mMaxNEvents = number;
}

inline Long64_t Forester::GetMaxNEvents() const {
  return mMaxNEvents;
}

inline void Forester::SetMessageInterval(Long64_t number) {
  mInterval = number;
}

inline bool Forester::MustQuit() const {
  return mQuit;
}

inline void Forester::SetMustQuit(bool flag) {
  mQuit = flag;
}

inline void Forester::SetBeVerbose(bool flag) {
  mVerbose = flag;
}

inline bool Forester::BeVerbose() const {
  return mVerbose;
}

inline const FileType* Forester::GetFileType() const {
  return mFile;
}

}  // namesce erhic

#endif  // INCLUDE_EICSMEAR_ERHIC_FORESTER_H_