// $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 "AliEveTrackFitterEditor.h"
#include <Alieve/AliEveTrackFitter.h>

#include <TGButton.h>


//______________________________________________________________________
// AliEveTrackFitterEditor
//

ClassImp(AliEveTrackFitterEditor)

AliEveTrackFitterEditor::AliEveTrackFitterEditor(const TGWindow *p, Int_t width, Int_t height,
	     UInt_t options, Pixel_t back) :
  TGedFrame(p, width, height, options | kVerticalFrame, back),
  fM(0),
  fFit(0),
  fReset(0),
  fStart(0),
  fStop(0),
  fGraph(0)
{
  // Constructor.

  MakeTitle("AliEveTrackFitter");

  fStart = new TGTextButton(this, "Start");
  AddFrame(fStart, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 4, 1, 1, 1));
  fStart->Connect("Clicked()",
                  "AliEveTrackFitterEditor", this, "DoStart()");

  fFit = new TGTextButton(this, "Fit");
  AddFrame(fFit, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 4, 1, 1, 1));
  fFit->Connect("Clicked()",
                "AliEveTrackFitterEditor", this, "DoFit()");

  fReset = new TGTextButton(this, "Reset");
  AddFrame(fReset, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 4, 1, 1, 1));
  fReset->Connect("Clicked()",
                  "AliEveTrackFitterEditor", this, "DoReset()");

  fStop = new TGTextButton(this, "Stop");
  AddFrame(fStop, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 4, 1, 1, 1));
  fStop->Connect("Clicked()",
                 "AliEveTrackFitterEditor", this, "DoStop()");

  fGraph = new TGTextButton(this, " RiemanGraph ");
  AddFrame(fGraph, new TGLayoutHints(kLHintsLeft, 4, 2, 4, 1));
  fGraph->Connect("Clicked()",
                 "AliEveTrackFitterEditor", this, "DoGraph()");
 }

/**************************************************************************/

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

  fM = dynamic_cast<AliEveTrackFitter*>(obj);

  if(fM->GetConnected())
  {
    fStart->SetState(kButtonDisabled);
    fStop->SetState(kButtonUp);
  }
  else
  { 
    fStop->SetState(kButtonDisabled);
    fStart->SetState(kButtonEngaged);
    fStart->SetState(kButtonUp);
  }
}

void AliEveTrackFitterEditor::DoFit()
{
  // Fit slot.

  fM->FitTrack();
  Update();
}

void AliEveTrackFitterEditor::DoReset()
{
  // Reset slot.

  fM->Reset();
  Update();
}

void AliEveTrackFitterEditor::DoStart()
{
  // Start selection slot.

  fM->Start();
  fStart->SetState(kButtonDisabled);
  fStop->SetState(kButtonUp);
}

void AliEveTrackFitterEditor::DoStop()
{
  // Stop selection slot.

  fM->Stop();
  fStop->SetState(kButtonDisabled);
  fStart->SetState(kButtonUp);
}

/**************************************************************************/

void AliEveTrackFitterEditor::DoGraph()
{
  // Draw graph slot.

  fM->DrawRiemanGraph();
  Update();
}