/*
 * 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 "static_model.hpp"

#include "vector_3d.hpp"
#include "rotation_3d.hpp"
#include "static_mesh.hpp"
#include "texture.hpp"
#include "vk/static_model.hpp"

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

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

const struct mrb_data_type cg_static_model_type = {
  "CG_StaticModel", cg_free_static_model };

static mrb_value
cg_cStaticModel_initialize(mrb_state *mrb, mrb_value self)
{
  std::shared_ptr<VK::StaticMesh> *static_mesh;
  std::shared_ptr<VK::Texture> *texture;
  std::shared_ptr<glm::vec3> *position;
  std::shared_ptr<glm::vec3> *rotation;
  std::shared_ptr<VK::StaticModel> *ptr;

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

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

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

static mrb_value
cg_cStaticModel_set_rotation(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::StaticModel>*)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_cStaticModel_set_position(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::StaticModel>*)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_cStaticModel_draw(mrb_state *mrb, mrb_value self)
{
  auto ptr = (std::shared_ptr<VK::StaticModel>*)DATA_PTR(self);

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

  return self;
}

void
cg_static_model_init(mrb_state *mrb)
{
  struct RClass *cg_m, *cg_cStaticModel;

  cg_m = mrb_module_get(mrb, "CandyGear");
  cg_cStaticModel = mrb_define_class_under(
    mrb, cg_m, "StaticModel", mrb->object_class);
  MRB_SET_INSTANCE_TT(cg_cStaticModel, MRB_TT_DATA);
  mrb_define_method(
    mrb, cg_cStaticModel, "initialize", cg_cStaticModel_initialize,
    MRB_ARGS_REQ(4));
  mrb_define_method(
    mrb, cg_cStaticModel, "position=", cg_cStaticModel_set_position,
    MRB_ARGS_REQ(1));
  mrb_define_method(
    mrb, cg_cStaticModel, "rotation=", cg_cStaticModel_set_rotation,
    MRB_ARGS_REQ(1));
  mrb_define_method(
    mrb, cg_cStaticModel, "draw", cg_cStaticModel_draw, MRB_ARGS_NONE());
}