// $Id$

#include "TGLViewer.h"

namespace Alieve {
class TPCData;
class Event;
}

Alieve::TPCData* g_tpc_data = 0;
Alieve::Event*   g_tpc_last_event = 0;

void tpc_digits(Int_t mode=1)
{
  if (g_tpc_data == 0 || g_tpc_last_event != Alieve::gEvent) {
    AliRunLoader* rl =  Alieve::Event::AssertRunLoader();
    rl->LoadDigits("TPC");
    TTree* dt = rl->GetTreeD("TPC", false);

    g_tpc_data = new Alieve::TPCData;
    g_tpc_data->LoadDigits(dt, kTRUE); // Create all present sectors.

    g_tpc_last_event = Alieve::gEvent;
  }

  // Viewport limits.
  /*
  Float_t left, right, top, bottom;
  right  = di->fOut2Seg.fNMaxPads* di->fOut2Seg.fPadWidth;
  left   = -right;
  bottom = di->fInnSeg.fRlow;
  top    = bottom + di->fOut2Seg.fRlow +
    di->fOut2Seg.fNRows*di->fOut2Seg.fPadLength - di->fInnSeg.fRlow;
  */

  gStyle->SetPalette(1, 0);
  Color_t col = 36;

  switch(mode) {

  case 0: { // Display a single sector
    Alieve::TPCSector2D* s = new Alieve::TPCSector2D();
    s->SetFrameColor(col);
    s->SetDataSource(g_tpc_data);
    gReve->AddRenderElement(s);
    gReve->Redraw3D();

    //TGLViewer* cam = gReve->GetGLViewer();
    //cam->SetCurrentCamera(TGLViewer::kCameraOrthoXOY) ;
    //cam->SetOrthoCamera(TGLViewer::kCameraOrthoXOY, 2*left, 2*right, 2*top, bottom); 
    //printf("%f %f %f %f\n", left, right, top, bottom);

    break;
  }

  case 1: { // Display all sectors
    gReve->DisableRedraw();
    {
      Reve::RenderElementList* l = new Reve::RenderElementList("TPC plate 1");
      l->SetTitle("TPC Plate");
      l->SetMainColor(Color_t(col));
      gReve->AddRenderElement(l);
      
      for(Int_t i = 0; i<18; i++) {
	Alieve::TPCSector2D* s = new Alieve::TPCSector2D(Form("TPCSector2D %d", i));
	s->SetSectorID(i);
	s->SetDataSource(g_tpc_data);
	s->SetFrameColor(col);
	s->SetAutoTrans(kTRUE);
	gReve->AddRenderElement(s, l);
      }
    }
    {
      Reve::RenderElementList* l = new Reve::RenderElementList("TPC plate 2");
      l->SetTitle("TPC Plate");
      l->SetMainColor(Color_t(col));

      gReve->AddRenderElement(l);
      for(Int_t i = 18; i<36; i++) {
	Alieve::TPCSector2D* s = new Alieve::TPCSector2D(Form("TPCSector2D %d", i));
	s->SetSectorID(i);
	s->SetDataSource(g_tpc_data);
	s->SetFrameColor(col);
	s->SetAutoTrans(kTRUE);
	gReve->AddRenderElement(s, l);
      }
    }
    gReve->EnableRedraw();

    break;
  }

  case 2 : { // Display a single sector in 3D
    Alieve::TPCSector3D* s = new Alieve::TPCSector3D();
    s->SetFrameColor(col);
    s->SetDataSource(g_tpc_data);
    gReve->AddRenderElement(s);
    gReve->Redraw3D();
    break;
  }

  case 3: { // Display all sectors in 3D
    gReve->DisableRedraw();
    {
      Reve::RenderElementList* l = new Reve::RenderElementList("TPC plate 1");
      l->SetTitle("TPC Plate");
      l->SetMainColor(Color_t(col));
      gReve->AddRenderElement(l);
      
      for(Int_t i = 0; i<18; i++) {
	Alieve::TPCSector3D* s = new Alieve::TPCSector3D(Form("TPCSector3D %d", i));
	s->SetSectorID(i);
	s->SetDataSource(g_tpc_data);
	s->SetFrameColor(col);
	s->SetAutoTrans(kTRUE);
	gReve->AddRenderElement(s, l);
      }
    }
    {
      Reve::RenderElementList* l = new Reve::RenderElementList("TPC plate 2");
      l->SetTitle("TPC Plate");
      l->SetMainColor(Color_t(col));

      gReve->AddRenderElement(l);
      for(Int_t i = 18; i<36; i++) {
	Alieve::TPCSector3D* s = new Alieve::TPCSector3D(Form("TPCSector3D %d", i));
	s->SetSectorID(i);
	s->SetDataSource(g_tpc_data);
	s->SetFrameColor(col);
	s->SetAutoTrans(kTRUE);
	gReve->AddRenderElement(s, l);
      }
    }
    gReve->EnableRedraw();

    break;
  }

  } // switch
}


void tpc_digits_2drange(Int_t start, Int_t end)
{
  if (start <  0)  start = 0;
  if (end   > 35)  end   = 35;

  if (g_tpc_data == 0 || g_tpc_last_event != Alieve::gEvent) {
    AliRunLoader* rl =  Alieve::Event::AssertRunLoader();
    rl->LoadDigits("TPC");
    TTree* dt = rl->GetTreeD("TPC", false);

    g_tpc_data = new Alieve::TPCData;
    g_tpc_data->LoadDigits(dt, kTRUE); // Create all present sectors.

    g_tpc_last_event = Alieve::gEvent;
  }

  gStyle->SetPalette(1, 0);
  Color_t col = 36;

  gReve->DisableRedraw();
  {
    Reve::RenderElementList* l = new Reve::RenderElementList("TPC sectors");
    l->SetMainColor(Color_t(col));
    gReve->AddRenderElement(l);
      
    for(Int_t i=start; i<=end; i++) {
      Alieve::TPCSector2D* s = new Alieve::TPCSector2D();
      s->SetSectorID(i);
      s->SetDataSource(g_tpc_data);
      s->SetFrameColor(col);
      s->SetAutoTrans(kTRUE);
      gReve->AddRenderElement(s, l);
    }
  }
  gReve->EnableRedraw();
}