#include "TGo4PolyCondPainter.h"

#include "TVirtualPad.h"
#include "TList.h"
#include "TROOT.h"

#include "TGo4PolyCondView.h"
#include "TGo4PolyCond.h"

TGo4PolyCondPainter::TGo4PolyCondPainter() :
   TGo4ConditionPainter(),
   fxCutView(0)
{
}

TGo4PolyCondPainter::TGo4PolyCondPainter(const char* name, const char* title) :
   TGo4ConditionPainter(name,title),
   fxCutView(0)
{
}

TGo4PolyCondPainter::~TGo4PolyCondPainter()
{
   UnPaintCondition();
   if (fxCutView!=0) {
      delete fxCutView;
      fxCutView = 0;
   }
}


void TGo4PolyCondPainter::PaintCondition(Option_t* opt)
{
   if(gPad==0) return;
   TGo4PolyCond* pconny=dynamic_cast<TGo4PolyCond*>(fxCondition);
   if(pconny && pconny->IsVisible()) {
      TCutG* cutg=pconny->GetCut(kFALSE);
      if(cutg==0) return; // case of empty polygon condition
      if(fxCutView==0 || gPad->GetListOfPrimitives()->FindObject(fxCutView)==0) {
         // Only set up new view object if not already there
         // necessary to change cut interactively (points, colors, etc...)
         // Since cutg may be deleted from canvas by user.
         fxCutView = new TGo4PolyCondView(cutg);
         fxCutView->SetLineWidth(pconny->GetLineWidth());
         fxCutView->SetLineColor(pconny->GetLineColor());
         fxCutView->SetLineStyle(pconny->GetLineStyle());
         fxCutView->SetFillColor(pconny->GetFillColor());
         fxCutView->SetFillStyle(pconny->GetFillStyle());
      } else
         fxCutView->SetCut(cutg); // update view if condition was changed manually
      fxCutView->SetCondition(pconny); // backreference for execute event
      if(!strcmp(opt,"fixstyle")) {
         // reproduce condition colors always
         // this mode will prevent the user from changing style interactively
         fxCutView->SetLineWidth(pconny->GetLineWidth());
         fxCutView->SetLineColor(pconny->GetLineColor());
         fxCutView->SetLineStyle(pconny->GetLineStyle());
         fxCutView->SetFillColor(pconny->GetFillColor());
         fxCutView->SetFillStyle(pconny->GetFillStyle());
      }
      if(gPad->GetListOfPrimitives()->FindObject(fxCutView)==0)
      {
         fxCutView->AppendPad();
      }
   }// if(pconny && pconny->IsVisible())
else
   UnPaintCondition();
}

void TGo4PolyCondPainter::UnPaintCondition(Option_t* opt)
{
   gROOT->GetListOfCanvases()->RecursiveRemove(fxCutView);
   // we do not delete view, but restore graphics properties though invisible
   TString option(opt);
   if(option.Contains("reset")) {
      delete fxCutView;
      fxCutView=0;
   }
}

void TGo4PolyCondPainter::DisplayToFront(Option_t* opt)
{
if(fxCutView) fxCutView->Pop(); // box to front
TGo4LabelPainter::DisplayToFront(opt); // label itself will be frontmost
}



void TGo4PolyCondPainter::SetCondition(TGo4Condition* con)
{
    fxCondition=con;
    if(fxCutView)
       fxCutView->SetCondition(dynamic_cast<TGo4PolyCond*>(con));
}