BeamHaloParticleBuffer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMHALOPARTICLEBUFFER_H
#define BEAMHALOPARTICLEBUFFER_H
 
#include <fstream>
#include <vector>
#include <map>
 
class BeamHaloParticle;
namespace CLHEP {
  class HepRandomEngine;
}
 
class BinnedInterval {
 public:
  BinnedInterval(double upperBinEdge,
         const std::vector<double>& intervalUpperBinEdges);
  BinnedInterval(const BinnedInterval& binnedInterval);
  ~BinnedInterval();
 
  BinnedInterval & operator=(const BinnedInterval &rhs) 
    {
      if ( this != &rhs ) // Fix Coverity SELF_ASSIGN
      {
        m_upperBinEdge = rhs.m_upperBinEdge;
        std::vector<double>* tmp = new std::vector<double>(*(rhs.m_intervalUpperBinEdges));
        delete m_intervalUpperBinEdges;
        m_intervalUpperBinEdges = tmp;
      }
      return *this;
    }
 
  double upperBinEdge() const { return m_upperBinEdge; }
  const std::vector<double>* intervalUpperBinEdges() const { return m_intervalUpperBinEdges; }
 
 private:
  double m_upperBinEdge;
  std::vector<double>* m_intervalUpperBinEdges;
};
 
class BeamHaloParticleBuffer {
 public:
  BeamHaloParticleBuffer(const std::string& fileName);
  ~BeamHaloParticleBuffer() = default;
  int openForWriting();
  int openForReading();
  int close();

  int writeParticle(BeamHaloParticle *particle);

  BeamHaloParticle* readRandomParticle(CLHEP::HepRandomEngine* engine);
  
 private:

  BeamHaloParticle* readParticle(void);

  std::string m_fileName;

  std::ofstream m_ofstream;

  std::ifstream m_ifstream;

  int m_recordSize;

  unsigned int m_particlesPerInterval;

  long m_numberOfParticles;

  std::vector<BinnedInterval> m_binnedIntervals;

  double m_upperBinEdge;

  std::vector<double> m_intervalUpperBinEdges;

  double m_intervalUpperBinEdge;

  bool m_writeFlag;
};
 
#endif