SpectraAna.h

/*
 * SpectraAna.h
 *
 *  Created on: 21-07-2015
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALSPECTRAANA_H_
#define HALSPECTRAANA_H_
 
#include "Array.h"
#include "HistogramManager.h"
#include "TrackAna.h"
#include <TClonesArray.h>
#include <vector>
namespace Hal {
  class SpectraAna : public TrackAna {
    HistogramManager_1_2D<TH2D>* fPtEta;
    HistogramManager_1_2D<TH2D>* fPtY;
    HistogramManager_1_2D<TH2D>* fMtEta;
    HistogramManager_1_2D<TH2D>* fMtY;
    Int_t fPtBins, fMtBins, fEtaBins, fYBins;
    Double_t fPtMin, fPtMax;
    Double_t fMtMin, fMtMax;
    Double_t fEtaMin, fEtaMax;
    Double_t fYMin, fYMax;
    TClonesArray* fEventCollectionsNames;
    TClonesArray* fTrackCollectionsNames;
    std::vector<Double_t> fMass;
    Bool_t fUseMass;
    void CheckNames();
 
  protected:
    virtual void ProcessTrack();
    virtual Task::EInitFlag Init();
    Package* Report() const;
 
  public:
    SpectraAna();
    void SetPtAxis(Int_t bins, Double_t min, Double_t max);
    void SetMtAxis(Int_t bins, Double_t min, Double_t max);
    void SetYAxis(Int_t bins, Double_t min, Double_t max);
    void SetEtaAxis(Int_t bins, Double_t min, Double_t max);
    void SetEventCollectionName(TString name, Int_t event_collection = 0);
    void SetTrackCollectionName(TString name, Int_t track_collection = 0);
    void SetMass(Double_t mass, Int_t track_collection);
    void SetOption(Option_t* opt);
    virtual ~SpectraAna();
    ClassDef(SpectraAna, 1)
  };
}  // namespace Hal
#endif /* HALSPECTRAANA_H_ */