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