/**
 * Stone of Orthanc
 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
 * Department, University Hospital of Liege, Belgium
 * Copyright (C) 2017-2018 Osimis S.A., Belgium
 *
 * This program is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 **/


#include "SdlEngine.h"

#if ORTHANC_ENABLE_SDL == 1

#include <Core/Logging.h>

#include <SDL.h>

namespace OrthancStone
{
  void SdlEngine::SetSize(unsigned int width,
                          unsigned int height)
  {
    context_.GetCentralViewport().SetSize(width, height);
    surface_.SetSize(width, height);
  }
    

  void SdlEngine::RenderFrame()
  {
    if (viewportChanged_)
    {
      BasicSdlApplicationContext::GlobalMutexLocker locker(context_);
      surface_.Render(context_.GetCentralViewport());

      viewportChanged_ = false;
    }
  }


  KeyboardModifiers SdlEngine::GetKeyboardModifiers(const uint8_t* keyboardState,
                                                    const int scancodeCount)
  {
    int result = KeyboardModifiers_None;

    if (keyboardState != NULL)
    {
      if (SDL_SCANCODE_LSHIFT < scancodeCount &&
          keyboardState[SDL_SCANCODE_LSHIFT])
      {
        result |= KeyboardModifiers_Shift;
      }

      if (SDL_SCANCODE_RSHIFT < scancodeCount &&
          keyboardState[SDL_SCANCODE_RSHIFT])
      {
        result |= KeyboardModifiers_Shift;
      }

      if (SDL_SCANCODE_LCTRL < scancodeCount &&
          keyboardState[SDL_SCANCODE_LCTRL])
      {
        result |= KeyboardModifiers_Control;
      }

      if (SDL_SCANCODE_RCTRL < scancodeCount &&
          keyboardState[SDL_SCANCODE_RCTRL])
      {
        result |= KeyboardModifiers_Control;
      }

      if (SDL_SCANCODE_LALT < scancodeCount &&
          keyboardState[SDL_SCANCODE_LALT])
      {
        result |= KeyboardModifiers_Alt;
      }

      if (SDL_SCANCODE_RALT < scancodeCount &&
          keyboardState[SDL_SCANCODE_RALT])
      {
        result |= KeyboardModifiers_Alt;
      }
    }

    return static_cast<KeyboardModifiers>(result);
  }


  SdlEngine::SdlEngine(SdlWindow& window,
                       BasicSdlApplicationContext& context) :
    window_(window),
    context_(context),
    surface_(window),
    viewportChanged_(true)
  {
  }
  

  SdlEngine::~SdlEngine()
  {
  }


  void SdlEngine::Run()
  {
    int scancodeCount = 0;
    const uint8_t* keyboardState = SDL_GetKeyboardState(&scancodeCount);

    {
      BasicSdlApplicationContext::GlobalMutexLocker locker(context_);
      SetSize(window_.GetWidth(), window_.GetHeight());
      context_.GetCentralViewport().SetDefaultView();
    }
    
    bool stop = false;
    while (!stop)
    {
      RenderFrame();

      SDL_Event event;

      while (!stop &&
             SDL_PollEvent(&event))
      {
        BasicSdlApplicationContext::GlobalMutexLocker locker(context_);

        if (event.type == SDL_QUIT) 
        {
          stop = true;
          break;
        }
        else if (event.type == SDL_MOUSEBUTTONDOWN)
        {
          KeyboardModifiers modifiers = GetKeyboardModifiers(keyboardState, scancodeCount);

          switch (event.button.button)
          {
            case SDL_BUTTON_LEFT:
              context_.GetCentralViewport().MouseDown(MouseButton_Left, event.button.x, event.button.y, modifiers);
              break;
            
            case SDL_BUTTON_RIGHT:
              context_.GetCentralViewport().MouseDown(MouseButton_Right, event.button.x, event.button.y, modifiers);
              break;
            
            case SDL_BUTTON_MIDDLE:
              context_.GetCentralViewport().MouseDown(MouseButton_Middle, event.button.x, event.button.y, modifiers);
              break;

            default:
              break;
          }
        }
        else if (event.type == SDL_MOUSEMOTION)
        {
          context_.GetCentralViewport().MouseMove(event.button.x, event.button.y);
        }
        else if (event.type == SDL_MOUSEBUTTONUP)
        {
          context_.GetCentralViewport().MouseUp();
        }
        else if (event.type == SDL_WINDOWEVENT)
        {
          switch (event.window.event)
          {
            case SDL_WINDOWEVENT_LEAVE:
              context_.GetCentralViewport().MouseLeave();
              break;

            case SDL_WINDOWEVENT_ENTER:
              context_.GetCentralViewport().MouseEnter();
              break;

            case SDL_WINDOWEVENT_SIZE_CHANGED:
              SetSize(event.window.data1, event.window.data2);
              break;

            default:
              break;
          }
        }
        else if (event.type == SDL_MOUSEWHEEL)
        {
          KeyboardModifiers modifiers = GetKeyboardModifiers(keyboardState, scancodeCount);

          int x, y;
          SDL_GetMouseState(&x, &y);

          if (event.wheel.y > 0)
          {
            context_.GetCentralViewport().MouseWheel(MouseWheelDirection_Up, x, y, modifiers);
          }
          else if (event.wheel.y < 0)
          {
            context_.GetCentralViewport().MouseWheel(MouseWheelDirection_Down, x, y, modifiers);
          }
        }
        else if (event.type == SDL_KEYDOWN &&
                 event.key.repeat == 0 /* Ignore key bounce */)
        {
          KeyboardModifiers modifiers = GetKeyboardModifiers(keyboardState, scancodeCount);

          switch (event.key.keysym.sym)
          {
            case SDLK_a:    context_.GetCentralViewport().KeyPressed('a', modifiers);  break;
            case SDLK_b:    context_.GetCentralViewport().KeyPressed('b', modifiers);  break;
            case SDLK_c:    context_.GetCentralViewport().KeyPressed('c', modifiers);  break;
            case SDLK_d:    context_.GetCentralViewport().KeyPressed('d', modifiers);  break;
            case SDLK_e:    context_.GetCentralViewport().KeyPressed('e', modifiers);  break;
            case SDLK_f:    window_.ToggleMaximize();                         break;
            case SDLK_g:    context_.GetCentralViewport().KeyPressed('g', modifiers);  break;
            case SDLK_h:    context_.GetCentralViewport().KeyPressed('h', modifiers);  break;
            case SDLK_i:    context_.GetCentralViewport().KeyPressed('i', modifiers);  break;
            case SDLK_j:    context_.GetCentralViewport().KeyPressed('j', modifiers);  break;
            case SDLK_k:    context_.GetCentralViewport().KeyPressed('k', modifiers);  break;
            case SDLK_l:    context_.GetCentralViewport().KeyPressed('l', modifiers);  break;
            case SDLK_m:    context_.GetCentralViewport().KeyPressed('m', modifiers);  break;
            case SDLK_n:    context_.GetCentralViewport().KeyPressed('n', modifiers);  break;
            case SDLK_o:    context_.GetCentralViewport().KeyPressed('o', modifiers);  break;
            case SDLK_p:    context_.GetCentralViewport().KeyPressed('p', modifiers);  break;
            case SDLK_q:    stop = true;                                      break;
            case SDLK_r:    context_.GetCentralViewport().KeyPressed('r', modifiers);  break;
            case SDLK_s:    context_.GetCentralViewport().KeyPressed('s', modifiers);  break;
            case SDLK_t:    context_.GetCentralViewport().KeyPressed('t', modifiers);  break;
            case SDLK_u:    context_.GetCentralViewport().KeyPressed('u', modifiers);  break;
            case SDLK_v:    context_.GetCentralViewport().KeyPressed('v', modifiers);  break;
            case SDLK_w:    context_.GetCentralViewport().KeyPressed('w', modifiers);  break;
            case SDLK_x:    context_.GetCentralViewport().KeyPressed('x', modifiers);  break;
            case SDLK_y:    context_.GetCentralViewport().KeyPressed('y', modifiers);  break;
            case SDLK_z:    context_.GetCentralViewport().KeyPressed('z', modifiers);  break;
            case SDLK_KP_0: context_.GetCentralViewport().KeyPressed('0', modifiers);  break;
            case SDLK_KP_1: context_.GetCentralViewport().KeyPressed('1', modifiers);  break;
            case SDLK_KP_2: context_.GetCentralViewport().KeyPressed('2', modifiers);  break;
            case SDLK_KP_3: context_.GetCentralViewport().KeyPressed('3', modifiers);  break;
            case SDLK_KP_4: context_.GetCentralViewport().KeyPressed('4', modifiers);  break;
            case SDLK_KP_5: context_.GetCentralViewport().KeyPressed('5', modifiers);  break;
            case SDLK_KP_6: context_.GetCentralViewport().KeyPressed('6', modifiers);  break;
            case SDLK_KP_7: context_.GetCentralViewport().KeyPressed('7', modifiers);  break;
            case SDLK_KP_8: context_.GetCentralViewport().KeyPressed('8', modifiers);  break;
            case SDLK_KP_9: context_.GetCentralViewport().KeyPressed('9', modifiers);  break;

            case SDLK_PLUS:
            case SDLK_KP_PLUS:
              context_.GetCentralViewport().KeyPressed('+', modifiers);  break;

            case SDLK_MINUS:
            case SDLK_KP_MINUS:
              context_.GetCentralViewport().KeyPressed('-', modifiers);  break;

            default:
              break;
          }
        }
      }

      // Small delay to avoid using 100% of CPU
      SDL_Delay(1);
    }
  }
}

#endif