/*
 * Copyright 2022-2023 Frederico de Oliveira Linhares
 *
 * 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.
 */

#include "skeletal_model.hpp"

#include "vector_3d.hpp"
#include "rotation_3d.hpp"
#include "skeletal_mesh.hpp"
#include "texture.hpp"
#include "vk/skeletal_model.hpp"

void
cg_free_skeletal_model(mrb_state *mrb, void *obj)
{
  auto ptr = static_cast<std::shared_ptr<VK::SkeletalModel>*>(obj);

  ptr->~shared_ptr();
  mrb_free(mrb, ptr);
}

const struct mrb_data_type cg_skeletal_model_type = {
  "CG_SkeletalModel", cg_free_skeletal_model };

static mrb_value
cg_cSkeletalModel_initialize(mrb_state *mrb, mrb_value self)
{
  std::shared_ptr<VK::SkeletalMesh> *skeletal_mesh;
  std::shared_ptr<VK::Texture> *texture;
  std::shared_ptr<glm::vec3> *position;
  std::shared_ptr<glm::vec3> *rotation;
  std::shared_ptr<VK::SkeletalModel> *ptr;

  mrb_get_args(
    mrb, "dddd", &skeletal_mesh, &cg_skeletal_mesh_type, &texture,
    &cg_texture_type, &position, &cg_vector_3d_type, &rotation,
    &cg_rotation_3d_type);
  ptr = (std::shared_ptr<VK::SkeletalModel>*)DATA_PTR(self);
  if(ptr) mrb_free(mrb, ptr);
  ptr = (std::shared_ptr<VK::SkeletalModel>*)mrb_malloc(
    mrb, sizeof(std::shared_ptr<VK::SkeletalModel>));

  new(ptr)std::shared_ptr<VK::SkeletalModel>(
    std::make_shared<VK::SkeletalModel>(
      *skeletal_mesh, *texture, *position, *rotation));

  mrb_data_init(self, ptr, &cg_skeletal_model_type);
  return self;
}

static mrb_value
cg_cSkeletalModel_set_rotation(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::SkeletalModel>*)DATA_PTR(self);
  std::shared_ptr<glm::vec3> *rotation;

  mrb_get_args(mrb, "d", &rotation, &cg_rotation_3d_type);
  (*ptr)->rotation = *rotation;

  return self;
}

static mrb_value
cg_cSkeletalModel_set_position(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::SkeletalModel>*)DATA_PTR(self);
  std::shared_ptr<glm::vec3> *position;

  mrb_get_args(mrb, "d", &position, &cg_vector_3d_type);
  (*ptr)->position = *position;

  return self;
}

static mrb_value
cg_cSkeletalModel_set_animation(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::SkeletalModel>*)DATA_PTR(self);
  mrb_int animation_index;

  mrb_get_args(mrb, "i", &animation_index);
  (*ptr)->animation_index = animation_index;

  return self;
}

static mrb_value
cg_cSkeletalModel_draw(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::SkeletalModel>*)DATA_PTR(self);

  auto &instances = cg_core.vk_renderer->skeletal_models_to_draw[
    cg_core.vk_swapchain->current_frame][(*ptr)->skeletal_mesh];
  instances.push_back(*ptr);

  return self;
}

void
cg_skeletal_model_init(mrb_state *mrb)
{
  struct RClass *cg_m, *cg_cSkeletalModel;

  cg_m = mrb_module_get(mrb, "CandyGear");
  cg_cSkeletalModel = mrb_define_class_under(
    mrb, cg_m, "SkeletalModel", mrb->object_class);
  MRB_SET_INSTANCE_TT(cg_cSkeletalModel, MRB_TT_DATA);
  mrb_define_method(
    mrb, cg_cSkeletalModel, "initialize", cg_cSkeletalModel_initialize,
    MRB_ARGS_REQ(4));
  mrb_define_method(
    mrb, cg_cSkeletalModel, "position=", cg_cSkeletalModel_set_position,
    MRB_ARGS_REQ(1));
  mrb_define_method(
    mrb, cg_cSkeletalModel, "rotation=", cg_cSkeletalModel_set_rotation,
    MRB_ARGS_REQ(1));
  mrb_define_method(
    mrb, cg_cSkeletalModel, "animation=", cg_cSkeletalModel_set_animation,
    MRB_ARGS_REQ(1));
  mrb_define_method(
    mrb, cg_cSkeletalModel, "draw", cg_cSkeletalModel_draw, MRB_ARGS_NONE());
}