// $Header$

#include "NLTProjectorEditor.h"
#include <Reve/NLTProjector.h>

#include <Reve/RGValuators.h>

#include <TColor.h>
#include <TGNumberEntry.h>
#include <TGColorSelect.h>
#include <TGComboBox.h>
#include <TGLabel.h>
#include <TG3DLine.h>

using namespace Reve;

//______________________________________________________________________
// NLTProjectorEditor
//
// GUI editor for class NLTProjector.
//

ClassImp(NLTProjectorEditor)

NLTProjectorEditor::NLTProjectorEditor(const TGWindow *p,
					 Int_t width, Int_t height,
					 UInt_t options, Pixel_t back) :
    TGedFrame(p, width, height, options | kVerticalFrame, back),
    fM(0),

    fType(0),
    fDistortion(0),
    fFixedRadius(0),
    fCurrentDepth(0),

    fCenterFrame(0),
    fDrawCenter(0),
    fCenterX(0),
    fCenterY(0),
    fCenterZ(0),

    fAxisColor(0),
    fSIMode(0),
    fSILevel(0)
{
  // Constructor.

  MakeTitle("NLTProjection");
  {
    TGHorizontalFrame* f = new TGHorizontalFrame(this);
    TGLabel* lab = new TGLabel(f, "Type");
    f->AddFrame(lab, new TGLayoutHints(kLHintsLeft|kLHintsBottom, 1, 31, 1, 2));
    fType = new TGComboBox(f);
    fType->AddEntry("CFishEye", NLTProjection::PT_CFishEye);
    fType->AddEntry("RhoZ",     NLTProjection::PT_RhoZ);
    TGListBox* lb = fType->GetListBox();
    lb->Resize(lb->GetWidth(), 2*18);
    fType->Resize(80, 20);
    fType->Connect("Selected(Int_t)", "Reve::NLTProjectorEditor",
		   this, "DoType(Int_t)");
    f->AddFrame(fType, new TGLayoutHints(kLHintsTop, 1, 1, 2, 4));
    AddFrame(f);
  }

  Int_t labelW = 60;
  fDistortion = new RGValuator(this, "Distortion:", 90, 0);
  fDistortion->SetNELength(5);
  fDistortion->SetLabelWidth(labelW);
  fDistortion->Build();
  fDistortion->SetLimits(0, 50, 101, TGNumberFormat::kNESRealTwo);
  fDistortion->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
		       this, "DoDistortion()");
  AddFrame(fDistortion, new TGLayoutHints(kLHintsTop, 1, 1, 1, 0));


  fFixedRadius = new RGValuator(this, "FixedR:", 90, 0);
  fFixedRadius->SetNELength(5);
  fFixedRadius->SetLabelWidth(labelW);
  fFixedRadius->Build();
  fFixedRadius->SetLimits(0, 1000, 101, TGNumberFormat::kNESRealOne);
  fFixedRadius->SetToolTip("Radius not scaled by distotion.");
  fFixedRadius->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
			this, "DoFixedRadius()");
  AddFrame(fFixedRadius, new TGLayoutHints(kLHintsTop, 1, 1, 1, 0));


  fCurrentDepth = new RGValuator(this, "CurrentZ:", 90, 0);
  fCurrentDepth->SetNELength(5);
  fCurrentDepth->SetLabelWidth(labelW);
  fCurrentDepth->Build();
  fCurrentDepth->SetLimits(-300, 300, 601, TGNumberFormat::kNESRealOne);
  fCurrentDepth->SetToolTip("Z coordinate of incoming projected object.");
  fCurrentDepth->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
			 this, "DoCurrentDepth()");
  AddFrame(fCurrentDepth, new TGLayoutHints(kLHintsTop, 1, 1, 1, 0));

  /**************************************************************************/
  MakeTitle("Axis");
  {
    TGHorizontalFrame* hf1 = new TGHorizontalFrame(this);

    TGCompositeFrame *labfr =
      new TGHorizontalFrame(hf1, 60, 15, kFixedSize);
    TGLabel* l = new TGLabel(labfr, "Color");
    labfr->AddFrame(l, new TGLayoutHints(kLHintsLeft|kLHintsBottom));
    hf1->AddFrame(labfr, new TGLayoutHints(kLHintsLeft|kLHintsBottom));


    fAxisColor = new TGColorSelect(hf1, 0, -1);
    hf1->AddFrame(fAxisColor, new TGLayoutHints(kLHintsLeft, 2, 0, 1, 1));
    fAxisColor->Connect
      ("ColorSelected(Pixel_t)",
       "Reve::NLTProjectorEditor", this, "DoAxisColor(Pixel_t)");

    AddFrame(hf1);
  }
  {
    TGHorizontalFrame* f = new TGHorizontalFrame(this);
    TGLabel* lab = new TGLabel(f, "StepMode");
    f->AddFrame(lab, new TGLayoutHints(kLHintsLeft|kLHintsBottom, 1, 6, 1, 2));
    fSIMode = new TGComboBox(f, "Position");
    fSIMode->AddEntry("Value", 1);
    fSIMode->AddEntry("Position", 0);
    fSIMode->GetTextEntry()->SetToolTipText("Set tick-marks on equidistant values/screen position.");
    TGListBox* lb = fSIMode->GetListBox();
    lb->Resize(lb->GetWidth(), 2*18);
    fSIMode->Resize(80, 20);
    fSIMode->Connect("Selected(Int_t)", "Reve::NLTProjectorEditor",
		     this, "DoSplitInfoMode(Int_t)");
    f->AddFrame(fSIMode, new TGLayoutHints(kLHintsTop, 1, 1, 1, 1));
    AddFrame(f);
  }
  {
    TGHorizontalFrame* f = new TGHorizontalFrame(this);
    TGLabel* lab = new TGLabel(f, "SplitLevel");
    f->AddFrame(lab, new TGLayoutHints(kLHintsLeft|kLHintsBottom, 1, 8, 1, 2));

    fSILevel = new TGNumberEntry(f, 0, 3, -1,TGNumberFormat::kNESInteger, TGNumberFormat::kNEANonNegative,
				 TGNumberFormat::kNELLimitMinMax, 0, 7);
    fSILevel->GetNumberEntry()->SetToolTipText("Number of tick-marks TMath::Power(2, level).");
    fSILevel->Connect("ValueSet(Long_t)", "Reve::NLTProjectorEditor", this, "DoSplitInfoLevel()");
    f->AddFrame(fSILevel, new TGLayoutHints(kLHintsTop, 1, 1, 1, 2));
    AddFrame(f, new TGLayoutHints(kLHintsTop, 0, 0, 0, 3) );
  }

  /**************************************************************************/
  // center tab
  fCenterFrame = CreateEditorTabSubFrame("Center");

  TGCompositeFrame *title1 = new TGCompositeFrame(fCenterFrame, 180, 10,
						  kHorizontalFrame |
						  kLHintsExpandX   |
						  kFixedWidth      |
						  kOwnBackground);
  title1->AddFrame(new TGLabel(title1, "Distortion Center"),
		   new TGLayoutHints(kLHintsLeft, 1, 1, 0, 0));
  title1->AddFrame(new TGHorizontal3DLine(title1),
		   new TGLayoutHints(kLHintsExpandX, 5, 5, 7, 7));
  fCenterFrame->AddFrame(title1, new TGLayoutHints(kLHintsTop, 0, 0, 2, 0));


  {

    TGHorizontalFrame* hf1 = new TGHorizontalFrame(fCenterFrame);

    fDrawOrigin = new TGCheckButton(hf1, "DrawOrigin");
    hf1->AddFrame(fDrawOrigin, new TGLayoutHints(kLHintsLeft, 2,1,0,4));
    fDrawOrigin->Connect("Toggled(Bool_t)"," Reve::NLTProjectorEditor", this, "DoDrawOrigin()");


    fDrawCenter = new TGCheckButton(hf1, "DrawCenter");
    hf1->AddFrame(fDrawCenter, new TGLayoutHints(kLHintsLeft, 2,1,0,4));
    fDrawCenter->Connect("Toggled(Bool_t)"," Reve::NLTProjectorEditor", this, "DoDrawCenter()");

    fCenterFrame->AddFrame(hf1, new TGLayoutHints(kLHintsTop, 0,0,0,0));

  }

  Int_t nel = 8;
  fCenterX = new RGValuator(fCenterFrame, "CenterX:", 90, 0);
  fCenterX->SetNELength(nel);
  fCenterX->SetLabelWidth(labelW);
  fCenterX->Build();
  fCenterX->SetLimits(-5, 5, 501, TGNumberFormat::kNESRealThree);
  fCenterX->SetToolTip("Origin of the projection.");
  fCenterX->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
		    this, "DoCenter()");
  fCenterFrame->AddFrame(fCenterX, new TGLayoutHints(kLHintsTop, 1, 1, 1, 1));

  fCenterY = new RGValuator(fCenterFrame, "CenterY:", 90, 0);
  fCenterY->SetNELength(nel);
  fCenterY->SetLabelWidth(labelW);
  fCenterY->Build();
  fCenterY->SetLimits(-5, 5, 501, TGNumberFormat::kNESRealThree);
  fCenterY->SetToolTip("Origin of the projection.");
  fCenterY->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
		    this, "DoCenter()");
  fCenterFrame->AddFrame(fCenterY, new TGLayoutHints(kLHintsTop, 1, 1, 1, 1));

  fCenterZ = new RGValuator(fCenterFrame, "CenterZ:", 90, 0);
  fCenterZ->SetNELength(nel);
  fCenterZ->SetLabelWidth(labelW);
  fCenterZ->Build();
  fCenterZ->SetLimits(-25, 25, 501, TGNumberFormat::kNESRealThree);
  fCenterZ->SetToolTip("Origin of the projection.");
  fCenterZ->Connect("ValueSet(Double_t)", "Reve::NLTProjectorEditor",
		    this, "DoCenter()");
  fCenterFrame->AddFrame(fCenterZ, new TGLayoutHints(kLHintsTop, 1, 1, 1, 1));
}

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

  fM = dynamic_cast<NLTProjector*>(obj);

  fAxisColor->SetColor(TColor::Number2Pixel(fM->GetAxisColor()), kFALSE);
  fSIMode->Select(fM->GetSplitInfoMode(), kFALSE);
  fSILevel->SetNumber(fM->GetSplitInfoLevel());

  fType->Select(fM->GetProjection()->GetType(), kFALSE);
  fDistortion->SetValue(1000.0f * fM->GetProjection()->GetDistortion());
  fFixedRadius->SetValue(fM->GetProjection()->GetFixedRadius());
  fCurrentDepth->SetValue(fM->GetCurrentDepth());

  fDrawCenter->SetState(fM->GetDrawCenter()  ? kButtonDown : kButtonUp);
  fDrawOrigin->SetState(fM->GetDrawOrigin()  ? kButtonDown : kButtonUp);
  fCenterX->SetValue(fM->GetCenter().x);
  fCenterY->SetValue(fM->GetCenter().y);
  fCenterZ->SetValue(fM->GetCenter().z);
}

//______________________________________________________________________________
void NLTProjectorEditor::DoType(Int_t type)
{
  // Slot for setting of projection type.

  fM->SetProjection((NLTProjection::PType_e)type, 0.001f * fDistortion->GetValue());
  fM->ProjectChildren();
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoDistortion()
{
  // Slot for setting distortion.

  fM->GetProjection()->SetDistortion(0.001f * fDistortion->GetValue());
  fM->UpdateName();
  fM->ProjectChildren();
  Update();
}

//______________________________________________________________________________
 void NLTProjectorEditor::DoFixedRadius()
{
  // Slot for setting fixed radius.

  fM->GetProjection()->SetFixedRadius(fFixedRadius->GetValue());
  fM->ProjectChildren();
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoCurrentDepth()
{
  // Slot for setting current depth.

  fM->SetCurrentDepth(fCurrentDepth->GetValue());
  fM->ProjectChildren();
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoCenter()
{
  // Slot for setting center of distortion.
 
  fM->SetCenter(fCenterX->GetValue(), fCenterY->GetValue(), fCenterZ->GetValue());
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoDrawOrigin()
{
  // Slot for setting draw of origin.

  fM->SetDrawOrigin(fDrawOrigin->IsOn());
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoDrawCenter()
{
  // Slot for setting draw of center.

  fM->SetDrawCenter(fDrawCenter->IsOn());
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoSplitInfoMode(Int_t type)
{
  // Slot for setting split info mode.

  fM->SetSplitInfoMode(type);
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoSplitInfoLevel()
{
  // Slot for setting tick-mark density.

  fM->SetSplitInfoLevel((Int_t)fSILevel->GetNumber());
  Update();
}

//______________________________________________________________________________
void NLTProjectorEditor::DoAxisColor(Pixel_t pixel)
{
  // Slot for setting axis color.

  fM->SetAxisColor(Color_t(TColor::GetColor(pixel)));
  Update();
}