#ifndef MARS_MJSpectrum
#define MARS_MJSpectrum

#ifndef MARS_MJob
#include <MJob.h>
#endif

class TH1;
class TH1D;
class TH2D;

class MH3;
class MTask;
class MParList;
class MDataSet;
class MHEnergyEst;
class MAlphaFitter;
class MStatusArray;
class MHCollectionArea;
class MMcSpectrumWeight;

class MJSpectrum : public MJob
{
private:
    MTask *fCut0;
    MTask *fCut1;
    MTask *fCut2;
    MTask *fCut3;
    MTask *fEstimateEnergy;

    Bool_t fRefill;
    Bool_t fSimpleMode;
    Bool_t fRawMc;

    // Read Input
    Bool_t  ReadTask(MTask* &task, const char *name) const;
    Float_t ReadInput(MParList &plist, TH1D &h1, TH1D &size);
    Bool_t  ReadOrigMCDistribution(const MDataSet &set, TH1 &h, MMcSpectrumWeight &w) const;
    Bool_t  GetThetaDistribution(TH1D &temp1, TH1D &temp2) const;
    Bool_t  Refill(MParList &plist, TH1D &h) const;
    Bool_t  InitWeighting(const MDataSet &set, MMcSpectrumWeight &w) const;

    // Display Output
    void    PrintSetup(const MAlphaFitter &fit) const;
    void    DisplayResult(const TH2D &mh1) const;
    Bool_t  IntermediateLoop(MParList &plist, MH3 &h1, TH1D &temp1, const MDataSet &set, MMcSpectrumWeight &w) const;
    TArrayD DisplaySpectrum(MHCollectionArea &area, TH1D &excess, MHEnergyEst &hest, Double_t ontime) const;
    Bool_t  DisplaySize(MParList &plist, Double_t scale) const;
    Bool_t  PlotSame(MStatusArray &arr, MParList &plist, const char *name, const char *tab, const char *plot, Double_t scale) const;

public:
    MJSpectrum(const char *name=NULL, const char *title=NULL);
    ~MJSpectrum();

    Bool_t Process(const MDataSet &set);

    void EnableRefilling(Bool_t b=kTRUE)  { fRefill=b; }
    void EnableSimpleMode(Bool_t b=kTRUE) { fSimpleMode=b; }
    void EnableRawMc(Bool_t b=kTRUE)      { fRawMc=b; }

    void SetEnergyEstimator(const MTask *task);

    ClassDef(MJSpectrum, 0) // Proh'gram to calculate spectrum
};

#endif
