#include <iostream>
#include <iomanip>
#include <sstream>
#include <cmath>
#include <fstream>
using namespace std;

#include <pthread.h>

#include <TRACKING/DMCThrown.h>
#include "hdv_endviewAframe.h"
#include "hdview2.h"
#include "MyProcessor.h"
#include "FDC/DFDCGeometry.h"
#include "FCAL/DFCALGeometry.h"
#include "DVector2.h"
#include "HDGEOMETRY/DGeometry.h"
#include <PID/DNeutralParticle.h>

#include <TPolyMarker.h>
#include <TLine.h>
#include <TMarker.h>
#include <TBox.h>
#include <TVector3.h>
#include <TGeoVolume.h>
#include <TGeoManager.h>
#include <TGLabel.h>
#include <TGComboBox.h>
#include <TGButton.h>
#include <TGButtonGroup.h>
#include <TGTextEntry.h>
#include <TArrow.h>
#include <TLatex.h>
#include <TColor.h>
#include <TG3DLine.h>

extern JApplication *japp;
//TGeoVolume *MOTHER = NULL;
//TGeoCombiTrans *MotherRotTrans = NULL;

extern int GO;


//-------------------
// Constructor
//-------------------
hdv_endviewAframe::hdv_endviewAframe(hdv_mainframe *hdvmf, const TGWindow *p, UInt_t w, UInt_t h):TGMainFrame(p,w,h)
{
	this->hdvmf = hdvmf;

	// First, define all of the of the graphics objects. Below that, make all
	// of the connections to the methods so these things will work!

	// The main GUI window is divided into three sections, top, middle, and bottom.
	// Create those frames here.
	TGLayoutHints *lhints = new TGLayoutHints(kLHintsNormal, 2,2,2,2);
	TGLayoutHints *chints = new TGLayoutHints(kLHintsCenterY|kLHintsCenterX, 2,2,2,2);
	TGLayoutHints *xhints = new TGLayoutHints(kLHintsNormal|kLHintsExpandX, 2,2,2,2);
	TGLayoutHints *yhints = new TGLayoutHints(kLHintsNormal|kLHintsExpandY|kLHintsCenterY, 2,2,2,2);
	TGLayoutHints *thints = new TGLayoutHints(kLHintsTop|kLHintsCenterX|kLHintsExpandX, 2,2,0,0);
	TGHorizontalFrame *topframe = new TGHorizontalFrame(this, w, h);
	TGHorizontalFrame *botframe = new TGHorizontalFrame(this, w, h);
	AddFrame(topframe, lhints);
	AddFrame(botframe, chints);

	TGGroupFrame *canvasframe = new TGGroupFrame(topframe, "BCAL", kVerticalFrame);
	TGGroupFrame *controls = new TGGroupFrame(topframe, "", kVerticalFrame);
	topframe->AddFrame(canvasframe, lhints);
	topframe->AddFrame(controls, lhints);

	TGGroupFrame *viewcontrols = new TGGroupFrame(controls, "Controls", kVerticalFrame);
	controls->AddFrame(viewcontrols, lhints);
	// TGGroupFrame *colorcode = new TGGroupFrame(controls, "Color code", kVerticalFrame);
	// controls->AddFrame(colorcode, lhints);
	TGGroupFrame *bcalColorCodes = new TGGroupFrame(controls, "BCAL colors", kVerticalFrame);
	controls->AddFrame(bcalColorCodes, lhints);

		//------------- View canvas
		ecanvas = new TRootEmbeddedCanvas("endviewA Large Canvas", canvasframe, w, h, kSunkenFrame, GetWhitePixel());
		canvasframe->AddFrame(ecanvas, chints);

		//------------- Pan buttons
		TGGroupFrame *panframe = new TGGroupFrame(viewcontrols, "Pan", kHorizontalFrame);
			TGTextButton *panxneg	= new TGTextButton(panframe,	"-X");
			TGVerticalFrame *panupdn = new TGVerticalFrame(panframe);
				TGTextButton *panypos	= new TGTextButton(panupdn,	"+Y");
				TGTextButton *panyneg	= new TGTextButton(panupdn,	"-Y");
			TGTextButton *panxpos	= new TGTextButton(panframe,	"X+");
			
		viewcontrols->AddFrame(panframe,	yhints);
			panframe->AddFrame(panxneg,	chints);
			panframe->AddFrame(panupdn,	chints);
				panupdn->AddFrame(panypos,	chints);
				panupdn->AddFrame(panyneg,	chints);
			panframe->AddFrame(panxpos,	chints);

		panxneg->Connect("Clicked()","hdv_mainframe",hdvmf,"DoPanXneg()");
		panyneg->Connect("Clicked()","hdv_mainframe",hdvmf,"DoPanYneg()");
		panxpos->Connect("Clicked()","hdv_mainframe",hdvmf,"DoPanXpos()");
		panypos->Connect("Clicked()","hdv_mainframe",hdvmf,"DoPanYpos()");
		
		//------------- Zoom buttons
		TGGroupFrame *zoomframe = new TGGroupFrame(viewcontrols, "ZOOM", kHorizontalFrame);
		viewcontrols->AddFrame(zoomframe,	thints);
			TGTextButton *zoomout = new TGTextButton(zoomframe,	" - ");
			TGTextButton *zoomin	= new TGTextButton(zoomframe,	" + ");
			zoomframe->AddFrame(zoomout,	xhints);
			zoomframe->AddFrame(zoomin,	xhints);

		zoomout->Connect("Clicked()","hdv_mainframe",hdvmf,"DoZoomOut()");
		zoomin->Connect("Clicked()","hdv_mainframe",hdvmf,"DoZoomIn()");

	//------------- Reset button
	TGTextButton *reset	= new TGTextButton(viewcontrols,	"Reset");
	viewcontrols->AddFrame(reset, chints);
	reset->Connect("Clicked()","hdv_mainframe", hdvmf, "DoReset()");

	//------------- SaveAs message
	stringstream ss;
	ss<<"To save the canvas to\n";
	ss<<"a file, right click\n";
	ss<<"and select \"SaveAs\"\n";
	ss<<"from the menu. File type\n";
	ss<<"will be determined by\n";
	ss<<"the suffix of the file\n";
	ss<<"name.\n";
	TGLabel *saveas	= new TGLabel(viewcontrols, ss.str().c_str());
	viewcontrols->AddFrame(saveas, chints);

	// color lables BCAL
	TGLabel* BCCLables[9]; 
	unsigned int BCccodes[9] = {0x0000FF,0x7700FF,0xFF00FF,0xFF0077,0xFF0000,0xFF7700,0xFFFF00,0xFFFF77,0xFFFFFF};
	for (int i=0;i<9;i++) {
	  double e = pow(10,((8-(double)i)/2.0));
	  char str1[128];
	  sprintf(str1,"%7.1f MeV",e);
	  BCCLables[i] =  new TGLabel(bcalColorCodes, (const char*)str1);
	  //BCCLables[i]->SetTextColor(1);
	  BCCLables[i]->SetBackgroundColor(BCccodes[i]);
	  bcalColorCodes->AddFrame(BCCLables[i],lhints);
	}

	// // color code frame
	// TGLabel* FCCLables[9];
	// unsigned int ccodes[9] = {0xFF0033,0xFF2233,0xFF4433,0xFF6633,0xFF8833,0xFFaa33,0xFFcc33,0xFFee33,0xFFFFaa};
 	// for (int i=0;i<9;i++) {
	//   double E = pow(10.,((1. - (double)i*0.11)*log10(1./0.005)));
	//   char str1[128];
	//   sprintf(str1,"%5.1f MeV",E);
	//   FCCLables[i] =  new TGLabel(colorcode, (const char*)str1);
	//   FCCLables[i]->SetBackgroundColor(ccodes[i]);
	//   colorcode->AddFrame(FCCLables[i],lhints);
	// }
	



	//========== Dismiss Button ===========
	TGTextButton *dismiss = new TGTextButton(botframe,	"dismiss");
	botframe->AddFrame(dismiss, chints);

	//&&&&&&&&&&&&&&&& Connections
	dismiss->Connect("Clicked()","hdv_endviewAframe", this, "DoDismiss()");

	// Finish up and map the window
	SetWindowName("Hall-D Event Viewer BCAL View");
	SetIconName("HDView");


	MapSubwindows();
	Resize(GetDefaultSize());
	panframe->Resize();
	saveas->Resize();
	viewcontrols->Resize();
}

//-------------------
// DoDismiss
//-------------------
void hdv_endviewAframe::DoDismiss(void)
{
	UnmapWindow();
}

//-------------------
// SetRange
//-------------------
void hdv_endviewAframe::SetRange(double xlo, double ylo, double xhi, double yhi)
{
	ecanvas->GetCanvas()->cd();
	ecanvas->GetCanvas()->Range(xlo, ylo, xhi, yhi);
}

//-------------------
// DrawObjects
//-------------------
void hdv_endviewAframe::DrawObjects(vector<TObject*> &graphics_endA)
{
	if(!IsMapped())return;

	ecanvas->GetCanvas()->cd(0);
	// for(unsigned int i=0; i<graphics_endA.size(); i++)graphics_endA[i]->Draw("f");
	// for(unsigned int i=0; i<graphics_endA.size(); i++)graphics_endA[i]->Draw();
	for(unsigned int i=0; i<graphics_endA.size(); i++){
		TPolyLine *l = dynamic_cast<TPolyLine*>(graphics_endA[i]);
		if(l && l->GetFillStyle()!=0){
			graphics_endA[i]->Draw("f");
		}else{
			graphics_endA[i]->Draw("");
		}
	}
	ecanvas->GetCanvas()->Update();
}