//-------------------------------------------------------------------------
//
// File name:       hcontourmanager.cc
// Subject:         GUI class for contour management.
// Author:          D.Vasiliev (INFN-LNS, Catania) "vassili@lns.infn.it"
// Date:            April 2000
//
//-------------------------------------------------------------------------
//
// Description:
//
// This file contains the source code for the member functions of 
// the class HContourManager.
// This is a GUI class which provides interface for contour definition.
//
//-------------------------------------------------------------------------

#include "hcontourmanager.h"
#include "honlinecontrol.h"
#include "hcontourcont.h"
#include "hcontour.h"
#include "honlmainwindow.h"
#include "honlobjbrowser.h"
#include "herror.h"
#include <TGLabel.h>
#include <TPad.h>
#include <TGMsgBox.h>
#include <TIterator.h>
#include <TList.h>
#include <TH2.h>

ClassImp(HContourManager)


//---------------------------------------------------------------------
//******  HContourManager::HContourManager(const TGWindow,UInt_t w,UInt_t h,
//                                         TGMainFrame *pp = 0)
//
// Constructor of the class HContourManager.
//
//---------------------------------------------------------------------

HContourManager::HContourManager(const TGWindow *p, UInt_t w, UInt_t h,
				 TGMainFrame *pp) : 
  TGMainFrame(p, w, h) {

//---------------------------------------------------------------
//********* frames
//---------------------------------------------------------------

fFrame = new TGCompositeFrame(this, 500, 500, kHorizontalFrame);
AddFrame(fFrame, new TGLayoutHints(kLHintsNormal,2,2,10,10));

fF1 = new TGCompositeFrame(fFrame, 250, 500, kVerticalFrame);
fFrame->AddFrame(fF1, new TGLayoutHints(kLHintsNormal));

fF2 = new TGCompositeFrame(fFrame, 250, 500, kVerticalFrame);
fFrame->AddFrame(fF2, new TGLayoutHints(kLHintsNormal, 10,0,0,0));

fF10 = new TGGroupFrame(fF1, "contour definition", kVerticalFrame);
fF1->AddFrame(fF10, new TGLayoutHints(kLHintsNormal));

fF11 = new TGCompositeFrame(fF10, 250, 500, kVerticalFrame);
fF10->AddFrame(fF11, new TGLayoutHints(kLHintsNormal,0,3,0,0));
fF11->SetLayoutManager(new TGMatrixLayout(fF11,0,2,15));

fF12 = new TGCompositeFrame(fF1, 250, 100, kHorizontalFrame);
fF1->AddFrame(fF12, new TGLayoutHints(kLHintsNormal));

fF13 = new TGCompositeFrame(fF1, 250, 100, kHorizontalFrame);
fF1->AddFrame(fF13, new TGLayoutHints(kLHintsNormal));

fF14 = new TGCompositeFrame(fF1, 250, 100, kHorizontalFrame);
fF1->AddFrame(fF14, new TGLayoutHints(kLHintsNormal));

fF21 = new TGCompositeFrame(fF2, 250, 100, kHorizontalFrame);
fF2->AddFrame(fF21, new TGLayoutHints(kLHintsNormal,5,0,0,0));
fF21->SetLayoutManager(new TGMatrixLayout(fF21,0,2,3));

fF22 = new TGGroupFrame(fF2, "list of contours", kHorizontalFrame);
fF2->AddFrame(fF22, new TGLayoutHints(kLHintsNormal));

fF23 = new TGCompositeFrame(fF2, 250, 100, kHorizontalFrame);
fF2->AddFrame(fF23, new TGLayoutHints(kLHintsNormal,5,0,5,0));
fF23->SetLayoutManager(new TGMatrixLayout(fF23,0,2,3));

//---------------------------------------------------------------
//********* help, undo, set and clear buttons
//---------------------------------------------------------------

fInit = new TGTextButton(fF11, "Init", M__C_INIT);
fInit->Associate(this);
fInit->SetToolTipText("Catch the current active pad");
fF11->AddFrame(fInit, new TGLayoutHints(kLHintsNormal,30,30,3,5)); 
fInit->Resize(50,20);

fClear = new TGTextButton(fF11, "Clear", M__C_CLEAR);
fClear->Associate(this);
fClear->SetToolTipText("Remove contour from the pad");
fF11->AddFrame(fClear, new TGLayoutHints(kLHintsNormal,30,30,3,1)); 
fClear->Resize(50,20);

fUndo = new TGTextButton(fF11, "Undo", M__C_UNDO);
fUndo->Associate(this);
fUndo->SetToolTipText("One step back");
fF11->AddFrame(fUndo, new TGLayoutHints(kLHintsNormal,30,30,3,5)); 
fUndo->Resize(50,20);

fSet = new TGTextButton(fF11, "Set", M__C_SET);
fSet->Associate(this);
fSet->SetToolTipText("Connect the first and last points of the contour");
fF11->AddFrame(fSet, new TGLayoutHints(kLHintsNormal,30,30,3,5)); 
fSet->Resize(50,20);

fHelp = new TGTextButton(fF11, "Help", M__C_HELP);
fHelp->Associate(this);
fHelp->SetToolTipText("Click me to learn how to work with this panel");
fF11->AddFrame(fHelp, new TGLayoutHints(kLHintsNormal,30,30,3,5)); 
fHelp->Resize(50,20);

//---------------------------------------------------------------
//********* contour name
//---------------------------------------------------------------

fF12->AddFrame(new TGLabel(fF12, new TGHotString("contour name")),
	       new TGLayoutHints(kLHintsNormal,5,2,6,2));

fName = new TGTextEntry(fF12, fNameBuf = new TGTextBuffer(100), -1);
fF12->AddFrame(fName, new TGLayoutHints(kLHintsNormal,4,0,5,2));
fName->Resize(70,fName->GetDefaultHeight());

//---------------------------------------------------------------
//********* inside and outside hits
//---------------------------------------------------------------

fF13->AddFrame(new TGLabel(fF13, new TGHotString("hits inside")),
	       new TGLayoutHints(kLHintsNormal,5,2,5,2));
fInLB = new TGListBox(fF13, -1);
fF13->AddFrame(fInLB, new TGLayoutHints(kLHintsNormal,24,0,4,2)); 
fInLB->Resize(70,20);

fF14->AddFrame(new TGLabel(fF14, new TGHotString("hits outside")),
	       new TGLayoutHints(kLHintsNormal,5,2,5,2));
fOutLB = new TGListBox(fF14, -1);
fF14->AddFrame(fOutLB, new TGLayoutHints(kLHintsNormal,17,0,4,2)); 
fOutLB->Resize(70,20);

//---------------------------------------------------------------
//********* add and remove buttons
//---------------------------------------------------------------

fAdd = new TGTextButton(fF21, "Add", M__C_ADD);
fAdd->Associate(this);
fAdd->SetToolTipText("Add new contour to the list");
fAdd->Resize(55,20);
fF21->AddFrame(fAdd);

fRemove = new TGTextButton(fF21, "Remove", M__C_REMOVE);
fRemove->Associate(this);
fRemove->SetToolTipText("Remove contour from the list");
fRemove->Resize(55,20);
fF21->AddFrame(fRemove);

//---------------------------------------------------------------
//********* list of contours
//---------------------------------------------------------------

fList = new TGListBox(fF22, M__C_LIST);
fList->Associate(this);
fF22->AddFrame(fList, new TGLayoutHints(kLHintsNormal,2,2,2,2));
fList->Resize(100,145);

//---------------------------------------------------------------
//********* previous and next buttons
//---------------------------------------------------------------

fPrev = new TGTextButton(fF23, "Previous", M__C_PREV);
fPrev->Associate(this);
fPrev->SetToolTipText("One entry up in the list");
fPrev->Resize(60,20);
fF23->AddFrame(fPrev);

fNext = new TGTextButton(fF23, "Next", M__C_NEXT);
fNext->Associate(this);
fNext->SetToolTipText("One entry down in the list");
fNext->Resize(60,20);
fF23->AddFrame(fNext);

fWindow = pp;
countIn = countOut = 0;

countContour++;

MapSubwindows();

SetWindowName("Contour Manager");

Resize(GetDefaultSize());
MapWindow();

updateList();
SendMessage(this,MK_MSG(kC_COMMAND,kCM_BUTTON),M__C_CLEAR,0);

}


//--------------------------------------------------------------------
//******  HContourManager::~HContourManager()
//
// Destructor of the class HContourManager.
//
//--------------------------------------------------------------------

HContourManager::~HContourManager() {

clearContour();
delete fNext; delete fPrev;
delete fList;
delete fRemove; delete fAdd;
delete fOutLB; delete fInLB;
delete fName;
delete fHelp; delete fSet; delete fUndo; delete fClear; delete fInit;
delete fF23; delete fF22; delete fF21; 
delete fF14; delete fF13; delete fF12; delete fF11; delete fF10;
delete fF2; delete fF1;
delete fFrame;

countContour--;

}


//--------------------------------------------------------------------
//******  void HContourManager::updateList()
//
// This function updates content of the list of contours.
//
//--------------------------------------------------------------------

void HContourManager::updateList() {

  if(!gOnline) return;
  TGTextLBEntry *pEntry = 0;
  HContour *pContour = 0;
  Int_t i=1;

  while((pEntry = (TGTextLBEntry*) fList->Select(i,kFALSE))) {

    fList->RemoveEntry(i);
    i++;
  }

  Int_t j = 0;
  for(i=0; i<gOnline->getContour()->getEntries(); i++) {

    pContour = gOnline->getContour()->at(i);
    if(pContour) fList->AddEntry(pContour->GetName(),++j);
  }

  fList->MapSubwindows();
  fList->Layout();

}


//--------------------------------------------------------------------
//******  void HContourManager::updateCountBoxes()
//
// This function updates content of the list boxes fInLB and fOutLB.
//
//--------------------------------------------------------------------

void HContourManager::updateCountBoxes() {

  fInLB->RemoveEntry(1);
  fOutLB->RemoveEntry(1);
  char entry[20];

  sprintf(entry,"%i",countIn);
  fInLB->AddEntry(entry,1);

  sprintf(entry,"%i",countOut);
  fOutLB->AddEntry(entry,1);

  fInLB->MapSubwindows();
  fInLB->Layout();

  fOutLB->MapSubwindows();
  fOutLB->Layout();

}


//--------------------------------------------------------------------
//******  void HContourManager::clearContour()
//
// This function removes contour from the active pad.
//
//--------------------------------------------------------------------

void HContourManager::clearContour() {

  TIterator *it = 0;
  TObject *select = 0;
  if(gPad) { 
    TList *list = ((TPad*)gPad)->GetListOfPrimitives();
    if(list) {
      it = list->MakeIterator();
      while((select = it->Next()) != NULL) {
	if(!strcmp(select->ClassName(),"HContour")) {
	  list->Remove(select);
	  if(select->IsOnHeap()) { delete select; select = 0; }
	}
      }
    }
    gPad->Modified();
    gPad->Update();
  }

}


//--------------------------------------------------------------------
//******  void HContourManager::countHits(TH2F *hist, HContour *cont)
//
// This function counts number of hits inside and outside contour.
//
//--------------------------------------------------------------------

void HContourManager::countHits(TH2F *hist, HContour *cont) {
  
  countIn = countOut = 0;
  if(!hist || !cont) return;

  Float_t xmin = hist->GetXaxis()->GetXmin();
  Float_t xmax = hist->GetXaxis()->GetXmax();
  Float_t ymin = hist->GetYaxis()->GetXmin();
  Float_t ymax = hist->GetYaxis()->GetXmax();

  if(!cont->checkLimits(xmin,xmax,ymin,ymax)) return;

  Int_t binsx = hist->GetXaxis()->GetNbins();
  Int_t binsy = hist->GetYaxis()->GetNbins();

  Int_t i, j;
  Float_t lowx, upx, lowy, upy;

  for(i=1; i<=binsx; i++) {
    for(j=1; j<=binsy; j++) {

      lowx = hist->GetXaxis()->GetBinLowEdge(i);
      upx  = hist->GetXaxis()->GetBinUpEdge(i);
      lowy = hist->GetYaxis()->GetBinLowEdge(j);
      upy  = hist->GetYaxis()->GetBinUpEdge(j);

      if(cont->isInside(lowx,lowy) || cont->isInside(lowx,upy) ||
	 cont->isInside(upx,lowy)  || cont->isInside(upx,upy)) {
	countIn += (Int_t) (hist->GetCellContent(i,j));
      }
      else countOut += (Int_t) (hist->GetCellContent(i,j));

    }
  }

}


//---------------------------------------------------------------------
//******  Bool_t HContourManager::ProcessMessage(Long_t msg, Long_t parm1,
//                                                           Long_t parm2)
//
// This function processes all GUI items.
//
//---------------------------------------------------------------------

Bool_t HContourManager::ProcessMessage(Long_t msg,Long_t parm1,Long_t parm2) {

Int_t retval;
HContour *pCont = 0;
TGTextLBEntry *pEntry = 0;
Int_t idx = -1;
TH2F *hist2d = 0;
TIterator *it = 0;
TObject *select = 0;
HOnlMainWindow *pWind;

switch(GET_MSG(msg)) {

 case kC_COMMAND:

  switch(GET_SUBMSG(msg)) {

  case kCM_BUTTON:

    switch(parm1) {

    case M__C_HELP:
      new TGMsgBox(fClient->GetRoot(),this,"Help Message",
		   "A contour can be defined by clicking the left button of the mouse on the current active pad 
(important that in the pad there should be drawn a 2d histogram which serves as a 
coordinate system for the contour).

To make the active pad visible for the panel click 'Init'.
To connect the first and last points make a double click on the left button or click 'Set'.
To undo one step click 'Undo'.
To release the pad and remove the contour from it click 'Clear'.

Important notes:

The program takes care of the selfcross check during the contour definition. If a selfcross 
found the contour is shown in red. To be back one step click 'Undo'.

During the contour definition or display it is not possible to pick up with the mouse 
other objects drawn in the pad. So, to reactivate this facility release the pad by clicking 
'Clear'.",0,0,&retval);
      break;

    case M__C_UNDO:
      if(!gPad || !fContour) break;
      fContour->undo();
      break;

    case M__C_SET:
      if(!gPad || !fContour) break;
      fContour->setAll();
      break;

    case M__C_INIT:
      clearContour();
      countIn = countOut = 0;
      updateCountBoxes();
      fNameBuf->Clear();
      fClient->NeedRedraw(fName);
      if(gPad) {
	fContour = new HContour();
	fContour->Draw();
	gPad->Modified();
	gPad->Update();
      }
      break;

    case M__C_CLEAR:
      clearContour();
      countIn = countOut = 0;
      updateCountBoxes();
      fNameBuf->Clear();
      fClient->NeedRedraw(fName);
      break;

    case M__C_ADD:
      if(!gOnline) {
	HError::message("Container of contours not initialized");
	break;
      }
      if(!fContour) {
	HError::message("No contour defined");
	break;
      }
      if(!strcmp(fNameBuf->GetString(),"")) {
	HError::message("Contour name not defined");
	break;
      }
      pCont = new HContour();
      fContour->copy(pCont);
      pCont->SetName(fNameBuf->GetString());
      gOnline->getContour()->add(pCont);
      SendMessage(this,MK_MSG(kC_COMMAND,kCM_BUTTON),M__C_INIT,0);
      updateList();
      if(fWindow && fWindow->IsOnHeap()) {
	pWind = (HOnlMainWindow*) fWindow;
	if(pWind->getBrowser() && pWind->getBrowser()->IsOnHeap()) {
	  pWind->getBrowser()->createContourList();
	}
      }
      break;      

    case M__C_REMOVE:
      if(!gOnline) {
	HError::message("Container of contours not initialized");
	break;
      }
      pEntry = (TGTextLBEntry*) fList->GetSelectedEntry();
      if(pEntry) {
	idx = gOnline->getContour()->find(pEntry->GetText()->GetString());
	gOnline->getContour()->removeAt(idx);
      }
      SendMessage(this,MK_MSG(kC_COMMAND,kCM_BUTTON),M__C_CLEAR,0);
      updateList();
      if(fWindow && fWindow->IsOnHeap()) {
	pWind = (HOnlMainWindow*) fWindow;
	if(pWind->getBrowser() && pWind->getBrowser()->IsOnHeap()) {
	  pWind->getBrowser()->createContourList();
	}
      }
      break;

    case M__C_NEXT:
      idx = fList->GetSelected();
      if(idx < 0) idx = 0;
      pEntry = (TGTextLBEntry*) fList->Select(++idx,kFALSE);
      if(pEntry) {
	fList->Select(idx,kTRUE);
	SendMessage(this,MK_MSG(kC_COMMAND,kCM_LISTBOX),M__C_LIST,idx);
      }
      else SendMessage(this,MK_MSG(kC_COMMAND,kCM_BUTTON),M__C_CLEAR,0);
      break;

    case M__C_PREV:
      idx = fList->GetSelected();
      if(idx < 0) idx = 0;
      pEntry = (TGTextLBEntry*) fList->Select(--idx,kFALSE);
      if(pEntry) {
	fList->Select(idx,kTRUE);
	SendMessage(this,MK_MSG(kC_COMMAND,kCM_LISTBOX),M__C_LIST,idx);
      }
      else SendMessage(this,MK_MSG(kC_COMMAND,kCM_BUTTON),M__C_CLEAR,0);
      break;

    default:
      break;
    }
    break;

  case kCM_LISTBOX:

    if(parm1 == M__C_LIST) {

      if(!gOnline) break;
      if(!gPad) break;
      pEntry = (TGTextLBEntry*) fList->GetSelectedEntry();
      if(pEntry) {

	idx = gOnline->getContour()->find(pEntry->GetText()->GetString());
	pCont = gOnline->getContour()->at(idx);
	if(pCont) {
	  clearContour();
	  fContour = new HContour();
	  pCont->copy(fContour);
	  fContour->Draw();
	  fNameBuf->Clear();
	  fNameBuf->AddText(0,fContour->GetName());
	  fClient->NeedRedraw(fName);
	  gPad->Modified();
	  gPad->Update();
	}
      }
      if(it) { delete it; it = 0; }
      it = gPad->GetListOfPrimitives()->MakeIterator();
      while((select = it->Next()) != NULL) {
	if(!strcmp(select->ClassName(),"TH2F") && 
	   !strcmp(it->GetOption(),"")) {
	  hist2d = (TH2F*) select;
	  break;
	}
      }
      if(hist2d && pCont) {
	countHits(hist2d,pCont);
	updateCountBoxes();
      }
    }
    break;

  default:
    break;
  }

 default:
   break;
}

return kTRUE;

}