ParticleSet.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef PARTICLESET_H
#define PARTICLESET_H
 
#include <cmath>
#include <map>
 
class BinValue
{
public:
  BinValue():
    value(0),
    sumOfWeightsSquared(0)
  {;}
 
  inline void Fill(double weight = 1)
  {
    value += weight;
    sumOfWeightsSquared += weight*weight;
  };
 
  inline void Reset() {value = 0; sumOfWeightsSquared = 0;}
  inline double Error() const {return std::sqrt(sumOfWeightsSquared);}
 
  double value;
  double sumOfWeightsSquared;
};
 
class ParticleSet
{
public:
  void Fill(long int x,
            double   weight = 1)
  {data[x].Fill(weight);}
  
  std::map<long int,BinValue>::iterator       begin()        {return data.begin();}
  std::map<long int,BinValue>::iterator       end()          {return data.end();}
  std::map<long int,BinValue>::const_iterator begin()  const {return data.begin();}
  std::map<long int,BinValue>::const_iterator end()    const {return data.end();}
  bool                                   empty()  const {return data.empty();}
  
  BinValue& operator[](const long int key)             {return data[key];}
  const BinValue& operator[](const long int key) const {return data.at(key);}
 
  std::map<long int, BinValue> data;
  unsigned long nEntries;
};
  
#endif