// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/

#include "AliEveTRDModule.h"
#include "AliEveTRDModuleImp.h"

#include "TGButton.h"
#include "TGColorSelect.h"
#include "TGNumberEntry.h"
#include "TG3DLine.h"

#include "TEveManager.h"

ClassImp(AliEveTRDModule)
ClassImp(AliEveTRDModuleEditor)

//______________________________________________________________________________
AliEveTRDModule::AliEveTRDModule(const char *typ, Int_t det) :
  TNamed(Form("%s%03d", typ, det), ""),
  fLoadHits (kFALSE),
  fRnrHits  (kTRUE),

  fLoadDigits (kFALSE),
  fRnrDigits  (kTRUE),
  fDigitsLog  (kFALSE),
  fDigitsBox  (kFALSE),
  fDigitsNeedRecompute (kTRUE),

  fLoadRecPoints (kFALSE),
  fRnrRecPoints  (kTRUE),

  fLoadTracklets (kFALSE),
  fRnrTracklets  (kTRUE),

  fDet             (det),
  fDigitsThreshold (15)
{
  // Constructor.
}

///////////////////////////////////////////////////////////
//////////   AliEveTRDModuleEditor       //////////////////
///////////////////////////////////////////////////////////

//______________________________________________________________________________
AliEveTRDModuleEditor::AliEveTRDModuleEditor(const TGWindow* p, Int_t width, Int_t height,
					     UInt_t options, Pixel_t back) :
  TGedFrame(p, width, height, options | kVerticalFrame, back),
  fM (0),
  fDisplayHits(0),
  fHitsColor(0),
  fDisplayDigits(0), fToggleLog(0), fToggleBox(0), fThreshold(0),
  fThresValue(0),
  fDisplayClusters(0),
  fClustersColor(0),
  fDisplayTracks(0)
{
  // Constructor.

  MakeTitle("TRD Detector");

  // "Hits" group frame
  TGGroupFrame *fGroupFrame615 = new TGGroupFrame(this,"Hits");
  fDisplayHits = new TGCheckButton(fGroupFrame615,"Display");
  fDisplayHits->SetToolTipText("Display hits attached to chamber");
  fDisplayHits->SetState(kButtonDown);
  fGroupFrame615->AddFrame(fDisplayHits, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fDisplayHits->Connect("Clicked()",
			"AliEveTRDModuleEditor", this, "UpdateChamber()");

  // color select widget
  ULong_t lColPar1;
  gClient->GetColorByName("#ff00ff", lColPar1);
  fHitsColor = new TGColorSelect(fGroupFrame615, lColPar1, -1);

  fGroupFrame615->AddFrame(fHitsColor, new TGLayoutHints(kLHintsLeft | kLHintsCenterX | kLHintsTop | kLHintsCenterY,2,2,2,2));
  fHitsColor->Connect("ColorSelected(Pixel_t)",
		      "AliEveTRDModuleEditor", this, "UpdateHits(Pixel_t)");

  fGroupFrame615->SetLayoutManager(new TGVerticalLayout(fGroupFrame615));
  AddFrame(fGroupFrame615, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));

  // "Digits" group frame
  TGGroupFrame *fGroupFrame636 = new TGGroupFrame(this,"Digits");
  TGVerticalFrame *fVerticalFrame538 = new TGVerticalFrame(fGroupFrame636,68,63,kVerticalFrame);
  fDisplayDigits = new TGCheckButton(fVerticalFrame538,"Display");
  fDisplayDigits->SetState(kButtonDown);
  fDisplayDigits->SetToolTipText("Toggle display digits attached to chamber");
  fVerticalFrame538->AddFrame(fDisplayDigits, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fDisplayDigits->Connect("Clicked()",
			  "AliEveTRDModuleEditor", this, "UpdateChamber()");
  fToggleLog = new TGCheckButton(fVerticalFrame538,"Log");
  fToggleLog->SetState(kButtonUp);
  fToggleLog->SetToolTipText("Toggle logarithmic representation of digits");
  fVerticalFrame538->AddFrame(fToggleLog, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fToggleLog->Connect("Clicked()",
		      "AliEveTRDModuleEditor", this, "ModifyDigitsView()");

  fToggleBox = new TGCheckButton(fVerticalFrame538,"Box");
  fToggleBox->SetState(kButtonUp);
  fToggleBox->SetToolTipText("Toggle display digits as boxes");
  fVerticalFrame538->AddFrame(fToggleBox, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fToggleBox->Connect("Clicked()",
		      "AliEveTRDModuleEditor", this, "ModifyDigitsView()");
  fGroupFrame636->AddFrame(fVerticalFrame538, new TGLayoutHints(kLHintsLeft | kLHintsTop,2,2,2,2));

  TGVertical3DLine *fVertical3DLine773 = new TGVertical3DLine(fGroupFrame636,8,63);
  fGroupFrame636->AddFrame(fVertical3DLine773, new TGLayoutHints(kLHintsLeft | kLHintsCenterX | kLHintsTop | kLHintsExpandY,2,2,2,2));

  // vertical frame
  TGVerticalFrame *fVerticalFrame543 = new TGVerticalFrame(fGroupFrame636,81,47,kVerticalFrame);
  fThreshold = new TGCheckButton(fVerticalFrame543, "Threshold");
  fThreshold->SetToolTipText("Toggle threshold updating display");
  fVerticalFrame543->AddFrame(fThreshold, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fThresValue = new TGNumberEntry(fVerticalFrame543, (Int_t)10, 5, -1, TGNumberFormat::kNESInteger, TGNumberFormat::kNEAPositive, TGNumberFormat::kNELLimitMinMax, 0, 1024);
  fThresValue->Connect("ValueSet(Long_t)",
		       "AliEveTRDModuleEditor", this, "SetThreshold(Long_t)");
  fVerticalFrame543->AddFrame(fThresValue, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));

  fGroupFrame636->AddFrame(fVerticalFrame543, new TGLayoutHints(kLHintsLeft | kLHintsTop,2,2,2,2));

  fGroupFrame636->SetLayoutManager(new TGHorizontalLayout(fGroupFrame636));
  AddFrame(fGroupFrame636, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));

  // "Clusters" group frame
  TGGroupFrame *fGroupFrame649 = new TGGroupFrame(this,"Clusters");
  fDisplayClusters = new TGCheckButton(fGroupFrame649,"Display");
  fDisplayClusters->SetToolTipText("Display clusters attached to chamber");
  fGroupFrame649->AddFrame(fDisplayClusters, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fDisplayClusters->SetState(kButtonDown);
  fDisplayClusters->Connect("Clicked()",
			    "AliEveTRDModuleEditor", this, "UpdateChamber()");

  // color select widget
  ULong_t lColPar2;
  gClient->GetColorByName("#ff00ff", lColPar2);
  fClustersColor = new TGColorSelect(fGroupFrame649, lColPar2, -1);

  fGroupFrame649->AddFrame(fClustersColor, new TGLayoutHints(kLHintsLeft | kLHintsCenterX | kLHintsTop | kLHintsCenterY,2,2,2,2));
  fClustersColor->Connect("ColorSelected(Pixel_t)",
			  "AliEveTRDModuleEditor", this, "UpdateClusters(Pixel_t)");

  fGroupFrame649->SetLayoutManager(new TGVerticalLayout(fGroupFrame649));
  AddFrame(fGroupFrame649, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));

  // "Tracks" group frame
  TGGroupFrame *fGroupFrame676 = new TGGroupFrame(this,"Tracklets");
  fDisplayTracks = new TGCheckButton(fGroupFrame676,"Display");
  fDisplayTracks->SetToolTipText("Display MCM tracklets attached to chamber");
  fGroupFrame676->AddFrame(fDisplayTracks, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
  fDisplayTracks->SetState(kButtonDown);
  fDisplayTracks->Connect("Clicked()",
			  "AliEveTRDModuleEditor", this, "UpdateChamber()");

  fGroupFrame676->SetLayoutManager(new TGVerticalLayout(fGroupFrame676));
  AddFrame(fGroupFrame676, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
}

//______________________________________________________________________________
void AliEveTRDModuleEditor::SetModel(TObject* obj)
{
  // Set model object.

  if(!(fM = dynamic_cast<AliEveTRDModule*>(obj))) return;

  if(AliEveTRDNode *node = dynamic_cast<AliEveTRDNode*>(fM)) node->UpdateNode();

  fDisplayHits->SetEnabled(fM->fLoadHits);
  if(fM->fLoadHits) fDisplayHits->SetState( fM->fRnrHits ? kButtonDown : kButtonUp);

  fDisplayDigits->SetEnabled(fM->fLoadDigits);
  fToggleLog->SetEnabled(fM->fLoadDigits);
  fToggleBox->SetEnabled(fM->fLoadDigits);
  fThreshold->SetEnabled(fM->fLoadDigits);
  fThresValue->GetNumberEntry()->SetEnabled(fM->fLoadDigits);
  fThresValue->GetButtonDown()->SetEnabled(fM->fLoadDigits);
  fThresValue->GetButtonUp()->SetEnabled(fM->fLoadDigits);
  fThresValue->SetIntNumber(fM->fDigitsThreshold);
  if(fM->fLoadDigits){
    fDisplayDigits->SetState(fM->fRnrDigits ? kButtonDown : kButtonUp);
    fToggleLog->SetState(fM->fDigitsLog ? kButtonDown : kButtonUp);
    fToggleBox->SetState(fM->fDigitsBox ? kButtonDown : kButtonUp);
    fThreshold->SetState(fM->fDigitsNeedRecompute ? kButtonDown : kButtonUp);
  }

  if(fM->fLoadRecPoints) fDisplayClusters->SetState(fM->fRnrRecPoints ? kButtonDown : kButtonUp);
  else fDisplayClusters->SetEnabled(kFALSE);

  if(fM->fLoadTracklets) fDisplayTracks->SetState(fM->fRnrTracklets ? kButtonDown : kButtonUp);
  else fDisplayTracks->SetEnabled(kFALSE);
}

//______________________________________________________________________________
void AliEveTRDModuleEditor::ModifyDigitsView()
{
  // Slor to modify digit view.

  fM->fDigitsNeedRecompute = kTRUE;
  UpdateChamber();
}

//______________________________________________________________________________
void AliEveTRDModuleEditor::SetThreshold(Long_t tres)
{
  // Slot for SetThreshold.

  if(!fM->fLoadDigits) return;

  fM->fDigitsThreshold = (tres == 0) ? (int)fThresValue->GetNumber():tres;
  fM->fDigitsNeedRecompute = fThreshold->IsDown();
  UpdateChamber();
}


//______________________________________________________________________________
void AliEveTRDModuleEditor::UpdateChamber()
{
  //	Info("UpdateChamber()", "");

  fM->fRnrHits = kFALSE;
  if(fDisplayHits->IsDown()) fM->fRnrHits = kTRUE;

  fM->fRnrDigits = kFALSE;
  if(fDisplayDigits->IsDown()){
    fM->fRnrDigits = kTRUE;
    if(fToggleLog->IsDown()) fM->fDigitsLog = kTRUE;
    else  fM->fDigitsLog = kFALSE;
    if(fToggleBox->IsDown()) fM->fDigitsBox = kTRUE;
    else  fM->fDigitsBox = kFALSE;
  }

  fM->fRnrRecPoints = kFALSE;
  if(fDisplayClusters->IsDown()) fM->fRnrRecPoints = kTRUE;

  fM->fRnrTracklets = kFALSE;
  if(fDisplayTracks->IsDown()) fM->fRnrTracklets = kTRUE;

  AliEveTRDNode *node(NULL);
  if((node = dynamic_cast<AliEveTRDNode*>(fM))) node->UpdateLeaves();

  gEve->Redraw3D();
}

//______________________________________________________________________________
void AliEveTRDModuleEditor::UpdateClusters(Pixel_t /*col*/)
{
  //	Info("UpdateClusters()","");
  /*
    if(fM->fRecPoints){
    fM->fRecPoints->SetMarkerColor(col ? fHitsColor->GetColor() : col);
    gEve->Redraw3D();
    }
  */
}

//______________________________________________________________________________
void AliEveTRDModuleEditor::UpdateHits(Pixel_t /*col*/)
{
  //	Info("UpdateHits()","");
  /*
    if(fM->fHits){
    fM->fHits->SetMarkerColor(col ? fClustersColor->GetColor() : col);
    gEve->Redraw3D();
    }
  */
}