/********************************************************************************
 *
 * Copyright (c) 2018 ROCm Developer Tools
 *
 * MIT LICENSE:
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is furnished to do
 * so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 *******************************************************************************/
#include "include/rvs_module.h"

#include <stdint.h>
#include <cstddef>

#include "rocm_smi/rocm_smi.h"

#include "include/action.h"
#include "include/rvsloglp.h"
#include "include/worker.h"
#include "include/gpu_util.h"

/**
 * @defgroup GM GM Module
 *
 * @brief GPU Monitor module
 *
 * The GPU monitor tool is capable of running on one, some or all of the GPU(s) 
 *installed and will
 * report various information at regular intervals. The module can be configured 
 * to halt another
 * RVS modules execution if one of the quantities exceeds a specified boundary 
 * value.
 */

Worker* pworker;

extern "C" int rvs_module_has_interface(int iid) {
  int sts = 0;
  switch (iid) {
  case 0:
  case 1:
    sts = 1;
  }
  return sts;
}

extern "C" const char* rvs_module_get_description(void) {
    return "ROCm Validation Suite GM module";
}

extern "C" const char* rvs_module_get_config(void) {
  return "monitor (bool)";
}

extern "C" const char* rvs_module_get_output(void) {
  return "state (string)";
}

extern "C" int   rvs_module_init(void* pMi) {
  rvs::lp::Initialize(static_cast<T_MODULE_INIT*>(pMi));
  RVSTRACE_
  rvs::gpulist::Initialize();
  rsmi_init(0);
  return 0;
}

extern "C" int   rvs_module_terminate(void) {
  RVSTRACE_
  if (pworker) {
    RVSTRACE_
    pworker->set_stop_name("module_terminate");
    pworker->stop();
    delete pworker;
    pworker = nullptr;
  }
  RVSTRACE_
  rsmi_shut_down();

  return 0;
}

extern "C" void* rvs_module_action_create(void) {
  return static_cast<void*>(new gm_action);
}

extern "C" int   rvs_module_action_destroy(void* pAction) {
  delete static_cast<rvs::actionbase*>(pAction);
  return 0;
}

extern "C" int rvs_module_action_property_set(void* pAction, const char* Key,
                const char* Val) {
  return static_cast<rvs::actionbase*>(pAction)->property_set(Key, Val);
}

extern "C" int rvs_module_action_run(void* pAction) {
  return static_cast<rvs::actionbase*>(pAction)->run();
}