#include "testlights.h"
#include <dtCore/globals.h>
#include <dtCore/camera.h>
#include <dtCore/scene.h>
#include <dtCore/transform.h>

#include <osgGA/GUIEventAdapter>

using namespace dtABC;
using namespace dtCore;

IMPLEMENT_MANAGEMENT_LAYER(TestLightsApp)

float TestLightsApp::countOne   = 0.0f;
float TestLightsApp::countTwo   = 0.0f;
float TestLightsApp::countThree = 0.0f;

TestLightsApp::TestLightsApp(const std::string& configFilename)
   : Application(configFilename)
{}

void TestLightsApp::Config()
{
   // turn off scene light so we only see the lights we create
   GetScene()->UseSceneLight(false);

   // load up a warehouse
   mWarehouse = new Object("Warehouse");
   mWarehouse->LoadFile("models/warehouse.ive");
   AddDrawable(mWarehouse.get());

   Transform trans;

   // create a global spot light.
   mGlobalSpot = new SpotLight(1, "GlobalSpotlight");
   trans.Set(5.0f, 8.0f, 2.0f, 0.0f, 0.0f, 0.0f);
   mGlobalSpot->SetTransform(trans);
   mGlobalSpot->SetSpotCutoff(20.0f);
   mGlobalSpot->SetSpotExponent(50.0f);
   GetScene()->AddDrawable(mGlobalSpot.get());

   // add a child to the local light
   mSphere = new Object("HappySphere");
   mSphere->LoadFile("models/physics_happy_sphere.ive");

   // we want the sphere 1 unit below light so we can see effect of local light and
   trans.Set(0.0f, 0.0f, -1.0f, 0.0f, 0.0f, 0.0f);
   mSphere->SetTransform(trans);

   // create a global positional light.
   mPositional = new PositionalLight(3, "PositionalLight");
   mPositional->SetDiffuse(1.0f, 1.0f, 0.0f, 1.0f); // yellow light
   mPositional->AddChild(mSphere.get()); //move sphere along with light
   GetScene()->AddDrawable(mPositional.get());
   mPositional->SetEnabled(false);

   // create an infinite light
   mGlobalInfinite = new InfiniteLight(4, "GlobalInfiniteLight");
   GetScene()->AddDrawable(mGlobalInfinite.get());
   mGlobalInfinite->SetEnabled(true);

   // set camera stuff
   trans.Set(30.0f, -20.0f, 25.0f, 40.0f, -33.0f, 0.0f);
   GetCamera()->SetTransform(trans);

   osg::Vec3 camLoc, origin;
   trans.GetTranslation(camLoc);

   mOmm = new OrbitMotionModel(GetKeyboard(), GetMouse());
   mOmm->SetTarget(GetCamera());

   float distance(0.0f);
   for (int i = 0; i < 3; i++)
   {
      distance += osg::square(camLoc[i] - origin[i]);
   }
   distance = sqrt(distance);

   mOmm->SetDistance(distance);

   CreateHelpLabel();
}

bool TestLightsApp::KeyPressed(const Keyboard* keyboard, int key)
{
   bool verdict(false);
   switch (key)
   {
   case osgGA::GUIEventAdapter::KEY_Escape:
      Quit();
      verdict = true;
      break;
   case '1':
      mGlobalSpot->SetEnabled(!mGlobalSpot->GetEnabled());
      verdict = true;
      break;
   case '2':
      mPositional->SetEnabled(!mPositional->GetEnabled());
      verdict = true;
      break;
   case '3':
   {
      if (mPositional->GetLightingMode() == Light::GLOBAL)
      {
         mPositional->SetLightingMode(Light::LOCAL);
      }
      else
      {
         mPositional->SetLightingMode(Light::GLOBAL);
      }

      verdict = true;
      break;
   }
   case '4':
      mGlobalInfinite->SetEnabled(!mGlobalInfinite->GetEnabled());
      verdict = true;
      break;
   case osgGA::GUIEventAdapter::KEY_F1:
      mLabel->SetActive(!mLabel->GetActive());
      break;
   default:
      break;
   }

   return verdict;
}

void TestLightsApp::PreFrame(const double deltaFrameTime)
{
   // increment some values at different rates
   countOne   += 50.0f * deltaFrameTime;
   countTwo   += 60.0f * deltaFrameTime;
   countThree += 70.0f * deltaFrameTime;

   // cap at 360
   if (countOne   > 360.0f) countOne   -= 360.0f;
   if (countTwo   > 360.0f) countTwo   -= 360.0f;
   if (countThree > 360.0f) countThree -= 360.0f;

   // scale values to 0.0-1.0
   float redValue   = (cos(osg::DegreesToRadians(countOne))   + 1.0f) / 2.0f;
   float greenValue = (cos(osg::DegreesToRadians(countTwo))   + 1.0f) / 2.0f;
   float blueValue  = (cos(osg::DegreesToRadians(countThree)) + 1.0f) / 2.0f;

   // modify all global lights
   mGlobalSpot->SetDiffuse(redValue, greenValue, blueValue, 1.0f); // change color

   // rotate the spotlight
   Transform trans;
   mGlobalSpot->GetTransform(trans);
   trans.SetRotation(countOne, 0.0f, 0.0f);
   mGlobalSpot->SetTransform(trans);

   mPositional->SetAttenuation(1.0f, greenValue / 2.0f, blueValue / 2.0f); // change attenutation

   // move the global positional light in a circle
   float tx = 1.5 * cos(osg::DegreesToRadians(countOne)) + 2.0f;
   float ty = 1.5 * sin(osg::DegreesToRadians(countOne)) + 5.0f;
   trans.Set( tx, ty, 2.0f, 0.0f, 0.0f, 0.0f);
   mPositional->SetTransform(trans);

   mGlobalInfinite->SetDiffuse(redValue, greenValue, blueValue, 1.0f); //change color

   float th = countOne;
   float tp = countTwo;
   
   mGlobalInfinite->SetTransform(dtCore::Transform(0.0f, 0.0f, 0.0f, th, tp));
}

void TestLightsApp::CreateHelpLabel()
{
   mLabel = new dtABC::LabelActor();
   osg::Vec2 testSize(23.5f, 5.5f);
   mLabel->SetBackSize(testSize);
   mLabel->SetFontSize(0.8f);
   mLabel->SetTextAlignment(dtABC::LabelActor::AlignmentEnum::LEFT_CENTER);
   mLabel->SetText(CreateHelpLabelText());
   mLabel->SetEnableDepthTesting(false);
   mLabel->SetEnableLighting(false);

   GetCamera()->AddChild(mLabel.get());
   dtCore::Transform labelOffset(-17.0f, 50.0f, 10.5f, 0.0f, 90.0f, 0.0f);
   mLabel->SetTransform(labelOffset, dtCore::Transformable::REL_CS);
   AddDrawable(GetCamera());
}

std::string TestLightsApp::CreateHelpLabelText()
{
   std::string testString("");
   testString += "F1: Toggle Help Screen\n";
   testString += "\n";
   testString += "1: Toggle Spot Light\n";
   testString += "2: Toggle Positional Light\n";
   testString += "3: Toggle Positional Light's Mode\n";
   testString += "4: Toggle Infinite Light\n";

   return testString;
}

int main(int argc, const char* argv[])
{
   std::string dataPath = dtCore::GetDeltaDataPathList();
   dtCore::SetDataFilePathList(dataPath + ";" +
      dtCore::GetDeltaRootPath() + "/examples/data" + ";");


   RefPtr<TestLightsApp> app = new TestLightsApp("config.xml");
   app->Config();
   app->Run();

   return 0;
}