// ferhue, September 2017
		 

#ifndef TH2VIEWER_H
#define TH2VIEWER_H TH2VIEWER_H

#include "RQ_OBJECT.h"
#include "TGFrame.h"
#include <vector>

//forward declarations
class TRootEmbeddedCanvas;
class TGStatusBar;
class TGCheckButton;
class TGRadioButton;
class TGNumberEntry;
class TGHSlider;
class TGDoubleSlider;
class TGHorizontalFrame;
class TGLabel;
class TextedHSlider;
class TextedDoubleSlider;
class TH1;
class TH2;
class TBox;

class TH2Viewer : public TGMainFrame
{
	RQ_OBJECT("TH2Viewer")	///< for classes not derived from TQObject, RQ_OBJECT macro should be used
	
	public:	
		TH2Viewer(const std::vector<TH2*> h2s);///< Class constructor, generates the GUI. \param h2s a vector of all 2D histograms to analyse. Histograms should have same number of bins.
		~TH2Viewer();///< Destructor

		/**
		 * \brief Handles key press events for going forward or backward
		 * \param event the type of event
		 * \param px the key character
		 * \param py the key symbol/address
		 * \param sel the object selected where event happens
		 * \note It has to be public!
		 */
		void CaptureEvent(Int_t event, Int_t px, Int_t py, TObject *sel);
		
		void DoRadio();///< Handles the radio button click
		
		void Redraw2D(const UShort_t id);///< Redraws the 2D histogram selected (id)
		void RedrawBorderBoxX();///< Redraws the 2D x border box
		void RedrawBorderBoxY();///< Redraws the 2D y border box
		void RedrawX();///< Redraws the 1D x projections 
		void RedrawY();///< Redraws the 1D y projections 
		void ZoomX();///< Zooms x axis
		void ZoomY();///< Zooms y axis
		void MovePalette();///< Moves and resizes color scale palette
		
        void ReplaceData(std::vector<float*> newData, const UShort_t xbins, const UShort_t ybins);///< Replaces the current 2D histograms stored with another set of exactly the same size/shape. Faster than deleting class and regenerating. \param h2s a vector of all new 2D histograms

	protected:
		void CloseWindow();			///< Closes the GUI. Automatically called	
				
	private:
		TRootEmbeddedCanvas* fCanvas2;		///< The main 2D canvas
		TRootEmbeddedCanvas* fCanvasX;		///< The x projection canvas
		TRootEmbeddedCanvas* fCanvasY;		///< The y projection canvas
		
		std::vector<TH2*> fH2;				///< All 2D histograms available
		std::vector<TH1*> fHx;				///< All 1D x projections available
		std::vector<TH1*> fHy;				///< All 1D y projections available
		
		TGStatusBar*   fStatusBar;			///< Event status bar
		TGHorizontalFrame* fRow1;			///< Horizontal container first row
		TGHorizontalFrame* fRow2;			///< Horizontal container second row
		TGVerticalFrame* fVertical;			///< Vertical container
		
		TGVerticalFrame* fHistList;			///< Container of list of histograms
		
		std::vector<TGCheckButton*> fSelect1;///< Check which 1D histograms have to be drawn
		std::vector<TGRadioButton*> fSelect2;///< Check which 2D histogram has to be drawn
		
		TextedDoubleSlider* fBorderX;		///<Horizontal borders in 2D histogram for the projection
		TextedDoubleSlider* fBorderY;		///<Vertical borders in 2D histogram for the projection
		TBox* fBorderBoxX;					///< 2D x projection window visualisation
		TBox* fBorderBoxY;					///< 2D y projection window visualisation
		
		TextedDoubleSlider* fZoomX;			///<X axis zoom in histograms
		TextedDoubleSlider* fZoomY;			///<Y axis zoom in histograms
		TextedDoubleSlider* fZoomZ;			///<Z axis zoom in histograms
		
		Short_t fCurrent2;					///< ID of current 2D histogram
		Short_t fCurrentX;					///< ID of first x projection
		Short_t fCurrentY;					///< ID of first x projection
		
	#if ROOT_VERSION_CODE < ROOT_VERSION(6,0,0)//Just needed for ROOT v5
	ClassDef(TH2Viewer,1);///needed as it derives from TObject
	#endif
};

/**
 * \brief This class is a modified GUI slider with associated number entry, so that you can either choose to move slider or write the number.
 */
class TextedHSlider
{
	public:
		/**
		 * \brief Constructor
		 * \param p the parent GUI window where slider will be located
		 * \param w width
		 * \param type the slider type
		 * \param id a slider-specific code
		 * \param options layout options for the frame
		 * \param back the background color
		 * \param start min allowed value of the calibrating line slider. 
		 * \param end max allowed value of the calibrating line slider. 
		 * \param position the initial position of the calibrating line slider.
		 * \param isReal true if the calibrating line position has non-integer values, false otherwise.
		 */
		TextedHSlider(const TGWindow* p, UInt_t w, UInt_t type, Int_t id, UInt_t options, Pixel_t back, const Double_t start, const Double_t end, const Double_t position, Bool_t isReal);
		virtual ~TextedHSlider();///< Destructor
		
		TGHSlider* GetSlider();///< Gets the horizontal slider (internal class member) \return TGHSlider
		TGNumberEntry* GetNumberEntry();///< Gets the associated number entry of the class \return TGNumberEntry
		void MoveSlider();///< Moves the (integer value) slider after the number entry was modified via the GUI.
		void MoveRealThreeSlider();///< Moves the (decimal value) slider after the number entry was modified via the GUI.
		void MoveNumber(Int_t pos);///< Moves the (integer value) number entry after the slider position was modified via the GUI. \param pos the slider position to be moved to
		void MoveRealThreeNumber(Int_t pos);///< Moves the (decimal value) number entry after the slider position was modified via the GUI. \param pos the slider position that was set in the GUI.
		
	private:
		TGHSlider* fSlider;	///< internal horizontal slider
		TGNumberEntry* fNumber;	///< internal number entry associated to the horizontal slider
};

/**
 * \brief This class is a modified GUI double slider with associated number entries, so that you can either choose to move slider or write the numbers.
 */
class TextedDoubleSlider : public TGHorizontalFrame
{
	public:
		/**
		 * \brief Constructor
		 * \param p the parent GUI window where slider will be located
		 * \param w width
		 * \param type the slider type
		 * \param id a slider-specific code
		 * \param options layout options for the frame
		 * \param back the background color
		 * \param start min allowed value of the double slider. 
		 * \param end max allowed value of the double slider. 
		 * \param minPosition the min initial position of the double slider.
		 * \param minPosition the max initial position of the double slider.
		 * \param isReal true if the position has non-integer values, false otherwise.
		 * \param label the description label, leave as NULL if not wanted
		 */
		TextedDoubleSlider(TGVerticalFrame* const p, UInt_t w, UInt_t type, Int_t id, UInt_t options, Pixel_t back, const Double_t start, const Double_t end, const Double_t minPosition, const Double_t maxPosition, Bool_t isReal, const TString label);

		virtual ~TextedDoubleSlider();///< Destructor
		
		TGDoubleSlider* GetSlider();///< Gets the horizontal double slider (internal class member) \return TGHSlider
		TGNumberEntry* GetMinNumberEntry();///< Gets the associated minimum number entry of the class \return TGNumberEntry
		TGNumberEntry* GetMaxNumberEntry();///< Gets the associated maximum number entry of the class \return TGNumberEntry
		void MoveSlider();///< Moves the (integer value) slider after the number entry was modified via the GUI.
		void MoveRealThreeSlider();///< Moves the (decimal value) slider after the number entry was modified via the GUI.
		void MoveNumbers();///< Moves the (integer value) number entries after the slider position was modified via the GUI.
		void MoveRealThreeNumbers();///< Moves the (decimal value) number entries after the slider position was modified via the GUI.
		
	private:
		TGLabel* fLabel;			///< description label
		TGDoubleSlider* fSlider;	///< internal horizontal slider
		TGNumberEntry* fMinNumber;	///< internal minimum number entry associated to the double slider
		TGNumberEntry* fMaxNumber;	///< internal maximum number entry associated to the double slider
};

#endif