KFPHistogram1D.h

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

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle 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 this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef KFPHISTOGRAM1D
#define KFPHISTOGRAM1D

#include <vector>
#include <string>
#include <cmath>
#include <iostream>
#include <cstdlib>

class KFPHistogram1D
{
 public:
   
  KFPHistogram1D(): fHistogram(0), fSize(0), fName(""), fMinBin(0), fMaxBin(0) {}
  KFPHistogram1D(std::string name, int nBins, float minX, float maxX):
    fHistogram(0), fSize(nBins+2), fName(name), fMinBin(minX), fMaxBin(maxX) {} 
  ~KFPHistogram1D() {}
  
  int* GetHistogram() const { return fHistogram; } 
  std::string Name()  const { return fName; }      
  float MinBin()      const { return fMinBin; }    
  float MaxBin()      const { return fMaxBin; }    
  int   NBins()       const { return (fSize-2); }  
  int   DataSize()    const { return fSize; }      
  int   Size()        const { return fSize; }      
  
  inline void SetBinContent(int iBin, int value) { fHistogram[iBin] = value; } 
  inline void SetHistogramMemory(int* pointer)   { fHistogram = pointer; }     
  
  void Fill(float value)
  {
    int iBin = floor(float(value - fMinBin)/float(fMaxBin - fMinBin) * float(fSize-2)) + 1;
    
    if(iBin > fSize-1)
      iBin = fSize-1;
    if(iBin < 1)
      iBin = 0;
    
    if( !(iBin==iBin) || !(std::isfinite(iBin)) ) iBin = 0;
    
    fHistogram[iBin]++;    
  }
  
  inline void operator += ( const KFPHistogram1D &h )
  {
    if( fSize != h.fSize )
    {
      std::cout << "Size of 1D histogram " << fName << " is incorrect. Stop the program." << std::endl;
    }
    else
    {
      for(int i=0; i<fSize; i++)
        fHistogram[i] += h.fHistogram[i];
    }
  }
  
  KFPHistogram1D(const KFPHistogram1D& h): fHistogram(h.fHistogram), fSize(h.fSize), fName(h.fName), fMinBin(h.fMinBin), fMaxBin(h.fMaxBin)
  {
  }
  
  const KFPHistogram1D& operator=(const KFPHistogram1D& h)
  {
    fHistogram = h.fHistogram;
    fSize = h.fSize;
    fName = h.fName;
    fMinBin = h.fMinBin;
    fMaxBin = h.fMaxBin;
    
    return *this;
  }
  
 private:
  int* fHistogram;   
  int fSize;         
  
  std::string fName; 
  float fMinBin;     
  float fMaxBin;     
};

#endif