#ifndef CBM_RICH_SMALL_PROTOTYPE_QA
#define CBM_RICH_SMALL_PROTOTYPE_QA

#include "FairTask.h"
class TClonesArray;
class CbmRichRing;
class TCanvas;
class CbmHistManager;

#include <vector>
#include <map>

using namespace std;

class CbmRichSmallPrototypeQa : public FairTask
{
    
public:
    /**
     * \brief Standard constructor.
     */
    CbmRichSmallPrototypeQa();
    
    /**
     * \brief Standard destructor.
     */
    virtual ~CbmRichSmallPrototypeQa() {}
    
    /**
     * \brief Inherited from FairTask.
     */
    virtual InitStatus Init();
    
    /**
     * \brief Inherited from FairTask.
     */
    virtual void Exec(
                      Option_t* option);
    
    /**
     * \brief Inherited from FairTask.
     */
    virtual void Finish();
    
    
    /**
     * \brief Set output directory where you want to write results (figures and json).
     * \param[in] dir Path to the output directory.
     */
    void SetOutputDir(const string& dir) {fOutputDir = dir;}
    
    
private:
    
    /**
     * \brief Initialize histograms.
     */
    void InitHistograms();
    
    /**
     *  \brief Draw histograms.
     */
    void DrawHist();
    
    TCanvas* CreateCanvas(
                          const string& name,
                          const string& title,
                          int width,
                          int height);
    
    void SaveCanvasToImage();

    
    /**
     * \brief Copy constructor.
     */
    CbmRichSmallPrototypeQa(const CbmRichSmallPrototypeQa&);
    
    /**
     * \brief Assignment operator.
     */
    CbmRichSmallPrototypeQa& operator=(const CbmRichSmallPrototypeQa&);
    
    CbmHistManager* fHM;
    
    Int_t fEventNum;
    
    string fOutputDir; // output dir for results
    
  

    TClonesArray* fMCTracks;


	TClonesArray* fSTSPoints;

	TClonesArray* fSTSDigis;

	TClonesArray* fSTSHits;

	TClonesArray* fSTSTrack;


	TClonesArray* fRichPoints;

	TClonesArray* fRichDigis;
	
	TClonesArray* fRichHits;

	TClonesArray* fRichRings;


	TClonesArray* fTRDPoints;

	TClonesArray* fTRDHits;

	TClonesArray* fTRDTrack;


	TClonesArray* fToFPoints;

	TClonesArray* fToFHits;

	TClonesArray* fToFTrack;

    
    vector<TCanvas*> fCanvas;
    
    ClassDef(CbmRichSmallPrototypeQa,1)
};

#endif