spectrum.h

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

/*
    <one line to give the program's name and a brief idea of what it does.>
    Copyright (C) <year>  <name of author>

    This program 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.

    This program 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 <http://www.gnu.org/licenses/>.

*/

#ifndef SPECTRUM_H
#define SPECTRUM_H

#include <vector>
#include "randomgenerator.h"
class beamBeamSystem;
//class randomGenerator;

class spectrum
{
public:

    spectrum(const randomGenerator &randy, beamBeamSystem *bbs);

    int generateKsingle();

    int generateKdouble();

    double drawKsingle();

    void drawKdouble(float &egamma1, float &egamma2);

    void setBeamBeamSystem(beamBeamSystem *bbs) {
        _beamBeamSystem = bbs;
    }

    void setMinGammaEnergy(double energy) { _eGammaMin = energy; }

    void setMaxGammaEnergy(double energy) { _eGammaMax = energy; }

    void setBmin(double bmin) { _bMin = bmin; }

    void setBmax(double bmax) { _bMax = bmax; }
    
protected:

    virtual bool generateBreakupProbabilities();

    virtual double getSigma(double /*egamma*/) const {
        return 1.05;
    }
    
    virtual double getTransformedNofe(double egamma, double b);
    
    double _bMin;

    double _bMax;
    
    int _nBbins;
    
    std::vector<double> _probOfBreakup;
    
    beamBeamSystem *_beamBeamSystem;

private:
    double getFnSingle(double egamma) const;

    double getFnDouble(double egamma1, double egamma2) const;

    int _nK;

    std::vector<double> _fnSingle;

    std::vector<std::vector<double> > _fnDouble;

    std::vector<double> _fnSingleCumulative;

    std::vector<std::vector<double> > _fnDoubleCumulative;

    std::vector<double> _fnDoubleInt;

    std::vector<double> _fnDoubleIntCumulative;
    
    std::vector<double> _eGamma;

    double _eGammaMin;

    double _eGammaMax;

    int _zTarget;

    int _aTarget;

    bool _hadBreakProbCalculated;

    randomGenerator _randy;

    spectrum();

};

#endif // SPECTRUM_H