#include "TGLViewer.h"
#include "TGLOrthoCamera.h"
#include "TTimer.h"
#include "TRandom.h"
#include "TVirtualPad.h"

TGLViewer::ECameraType camera;
TTimer timer(25);
TRandom randGen(0);

Int_t moveCount = 0;

void AnimateCamera()
{
   // initialization
   static Double_t fov = 30;
   static Double_t zoom = 0.78;
   static Double_t dolly = 1500.0;
   static Double_t center[3] = {-164.0, -164.0, -180.0};
   static Double_t hRotate = 0.0;
   static Double_t vRotate = 0.0;
   // steps
   static Double_t fovStep = randGen.Rndm() * 3.0 - 0.5;
   static Double_t zoomStep = (20 - randGen.Rndm()) / 1000.;
   static Double_t dollyStep = randGen.Rndm() * 5.0 - 1.0;
   static Double_t centerStep[3] = {randGen.Rndm() * 4, randGen.Rndm() * 4, randGen.Rndm() * 4};
   static Double_t hRotateStep = randGen.Rndm() * 0.025;
   static Double_t vRotateStep = randGen.Rndm() * 0.05;

   // move center
   center[0] += centerStep[0];
   center[1] += centerStep[1];
   center[2] += centerStep[2];
   Double_t mag = TMath::Sqrt(center[0] * center[0] + center[1] * center[1] + center[2] * center[2]);
   if (mag > 500) {
      centerStep[0] = -centerStep[0];
      centerStep[1] = -centerStep[1];
      centerStep[2] = -centerStep[2];
   }

   // rotate
   hRotate += hRotateStep;
   vRotate += vRotateStep;
   if (vRotate >= TMath::TwoPi() || vRotate <= 0.0)
      vRotateStep = -vRotateStep;

   if (hRotate >= (TMath::PiOver2() - 0.02f) || hRotate <= (0.07f - TMath::PiOver2())) {
      hRotateStep = -hRotateStep;
   }

   // dolly
   dolly += dollyStep;
   if (dolly >= 2000.0 || dolly <= 1500.0)
      dollyStep = -dollyStep;

   // modify frustum
   TGLViewer *v = gPad ? (TGLViewer *)gPad->GetViewer3D() : nullptr;
   if (!v) return;
   if (camera < 3) {
      fov += fovStep;
      if (fov > 130.0 || fov < 5.0)
         fovStep = -fovStep;
   } else {
      zoom += zoomStep;
      if (zoom > 4.0 || zoom < 0.25)
         zoomStep = -zoomStep;
   }

   // apply
   if (camera < 3)
      v->SetPerspectiveCamera(camera, fov, dollyStep, center, hRotateStep, vRotateStep);
   else
      v->SetOrthoCamera(camera, zoom, dollyStep, center, hRotateStep, vRotateStep);

   if (++moveCount % 10 == 0)
      v->RefreshPadEditor(v);
}

void glViewerExercise()
{
   gROOT->ProcessLine(".x nucleus.C");
   TGLViewer *v = gPad ? (TGLViewer *)gPad->GetViewer3D() : nullptr;
   if (!v) return;

   // Random draw style
   Int_t style = randGen.Integer(3);
   switch (style) {
   case 0: v->SetStyle(TGLRnrCtx::kFill); break;
   case 1: v->SetStyle(TGLRnrCtx::kOutline); break;
   case 2: v->SetStyle(TGLRnrCtx::kWireFrame); break;
   }

   // Lights - turn some off randomly
   TGLLightSet *ls = v->GetLightSet();
   if (randGen.Integer(2) == 0)
      ls->SetLight(TGLLightSet::kLightLeft, kFALSE);
   if (randGen.Integer(2) == 0)
      ls->SetLight(TGLLightSet::kLightRight, kFALSE);
   if (randGen.Integer(2) == 0)
      ls->SetLight(TGLLightSet::kLightTop, kFALSE);
   if (randGen.Integer(2) == 0)
      ls->SetLight(TGLLightSet::kLightBottom, kFALSE);

   // Random camera type
   Int_t id = randGen.Integer(6);
   camera = (TGLViewer::ECameraType)id;
   v->SetCurrentCamera(camera);
   v->CurrentCamera().SetExternalCenter(kTRUE);
   if (id > 2) {
      // 0, 1, and 2 - are different 'perspective' camers.
      TGLOrthoCamera &o = static_cast<TGLOrthoCamera &>(v->CurrentCamera());
      o.SetEnableRotate(kTRUE);
   }

   // Now animate the camera
   TGLSAViewer *sav = dynamic_cast<TGLSAViewer *>(v);
   if (sav)
      sav->GetFrame()->Connect("CloseWindow()", "TTimer", &timer, "TurnOff()");
   timer.SetCommand("AnimateCamera()");
   timer.TurnOn();
}