// Copyright 2021 DeepMind Technologies Limited
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef MUJOCO_SIMULATE_H_
#define MUJOCO_SIMULATE_H_

#include <mutex>
#include <thread>

#include "uitools.h"

namespace mujoco {

//-------------------------------- global -----------------------------------------------

// Simulate states not contained in MuJoCo structures
class Simulate {
 public:
  // create object and initialize the simulate ui
  Simulate(void);

  // Start the Simulate UI thread
  void startthread(void);

  // Stop the Simulate UI thread
  void stopthread(void);

  // Apply UI pose perturbations to model and data
  void applyposepertubations(int flg_paused);

  // Apply UI force perturbations to model and data
  void applyforceperturbations(void);

  // Request that the Simulate UI thread render a new model
  // optionally delete the old model and data when done
  void load(const char* file, mjModel* m, mjData* d, bool delete_old_m_d);

  // functions below are used by the renderthread
  // load mjb or xml model that has been requested by load()
  void loadmodel(void);

  // prepare to render
  void prepare(void);

  // render the ui to the window
  void render(void);

  // clear callbacks registered in external structures
  void clearcallback(void);

  // thread to render the UI
  void renderthread(void);

  // constants
  static constexpr int kMaxFilenameLength = 1000;

  // the UI rendering thread
  std::thread renderthreadhandle;

  // model and data to be visualized
  mjModel* mnew = nullptr;
  mjData* dnew = nullptr;
  bool delete_old_m_d = false;

  mjModel* m = nullptr;
  mjData* d = nullptr;
  std::mutex mtx;

  // file
  int exitrequest = 0;

  // option
  int spacing = 0;
  int color = 0;
  int font = 0;
  int ui0_enable = 1;
  int ui1_enable = 1;
  int help = 0;
  int info = 0;
  int profiler = 0;
  int sensor = 0;
  int fullscreen = 0;
  int vsync = 1;
  int busywait = 0;

  // simulation
  int run = 1;
  int key = 0;
  int uiloadrequest = 0;
  int droploadrequest = 0;
  int loadrequest = 0;
  // strings
  char loadError[kMaxFilenameLength] = "";
  char dropfilename[kMaxFilenameLength] = "";
  char filename[kMaxFilenameLength] = "";
  char previous_filename[kMaxFilenameLength] = "";
  int slow_down = 1;
  bool speed_changed = true;
  double ctrlnoisestd = 0.0;
  double ctrlnoiserate = 0.0;

  // watch
  char field[mjMAXUITEXT] = "qpos";
  int index = 0;

  // physics: need sync
  int disable[mjNDISABLE];
  int enable[mjNENABLE];

  // rendering: need sync
  int camera = 0;

  // abstract visualization
  mjvScene scn;
  mjvCamera cam;
  mjvOption vopt;
  mjvPerturb pert;
  mjvFigure figconstraint;
  mjvFigure figcost;
  mjvFigure figtimer;
  mjvFigure figsize;
  mjvFigure figsensor;

  // OpenGL rendering and UI
  GLFWvidmode vmode;
  int windowpos[2];
  int windowsize[2];
  mjrContext con;
  GLFWwindow* window;
  mjuiState uistate;
  mjUI ui0, ui1;

  // Constant arrays needed for the option section of UI and the UI interface
  // TODO setting the size here is not ideal
  const mjuiDef defOption[14] = {
    {mjITEM_SECTION,   "Option",        1, nullptr,              "AO"},
    {mjITEM_SELECT,    "Spacing",       1, &this->spacing,       "Tight\nWide"},
    {mjITEM_SELECT,    "Color",         1, &this->color,         "Default\nOrange\nWhite\nBlack"},
    {mjITEM_SELECT,    "Font",          1, &this->font,          "50 %\n100 %\n150 %\n200 %\n250 %\n300 %"},
    {mjITEM_CHECKINT,  "Left UI (Tab)", 1, &this->ui0_enable,    " #258"},
    {mjITEM_CHECKINT,  "Right UI",      1, &this->ui1_enable,    "S#258"},
    {mjITEM_CHECKINT,  "Help",          2, &this->help,          " #290"},
    {mjITEM_CHECKINT,  "Info",          2, &this->info,          " #291"},
    {mjITEM_CHECKINT,  "Profiler",      2, &this->profiler,      " #292"},
    {mjITEM_CHECKINT,  "Sensor",        2, &this->sensor,        " #293"},
  #ifdef __APPLE__
    {mjITEM_CHECKINT,  "Fullscreen",    0, &this->fullscreen,    " #294"},
  #else
    {mjITEM_CHECKINT,  "Fullscreen",    1, &this->fullscreen,    " #294"},
  #endif
    {mjITEM_CHECKINT,  "Vertical Sync", 1, &this->vsync,         ""},
    {mjITEM_CHECKINT,  "Busy Wait",     1, &this->busywait,      ""},
    {mjITEM_END}
  };


  // simulation section of UI
  const mjuiDef defSimulation[12] = {
    {mjITEM_SECTION,   "Simulation",    1, nullptr,              "AS"},
    {mjITEM_RADIO,     "",              2, &this->run,           "Pause\nRun"},
    {mjITEM_BUTTON,    "Reset",         2, nullptr,              " #259"},
    {mjITEM_BUTTON,    "Reload",        2, nullptr,              "CL"},
    {mjITEM_BUTTON,    "Align",         2, nullptr,              "CA"},
    {mjITEM_BUTTON,    "Copy pose",     2, nullptr,              "CC"},
    {mjITEM_SLIDERINT, "Key",           3, &this->key,           "0 0"},
    {mjITEM_BUTTON,    "Load key",      3},
    {mjITEM_BUTTON,    "Save key",      3},
    {mjITEM_SLIDERNUM, "Noise scale",   2, &this->ctrlnoisestd,  "0 2"},
    {mjITEM_SLIDERNUM, "Noise rate",    2, &this->ctrlnoiserate, "0 2"},
    {mjITEM_END}
  };


  // watch section of UI
  const mjuiDef defWatch[5] = {
    {mjITEM_SECTION,   "Watch",         0, nullptr,              "AW"},
    {mjITEM_EDITTXT,   "Field",         2, this->field,          "qpos"},
    {mjITEM_EDITINT,   "Index",         2, &this->index,         "1"},
    {mjITEM_STATIC,    "Value",         2, nullptr,              " "},
    {mjITEM_END}
  };

  // info strings
  char info_title[Simulate::kMaxFilenameLength];
  char info_content[Simulate::kMaxFilenameLength];
};

}  // namespace mujoco

#endif