1 files changed, 134 insertions, 0 deletions
diff --git a/src/skeletal_model.cpp b/src/skeletal_model.cpp
new file mode 100644
index 0000000..0165e91
--- /dev/null
+++ b/src/skeletal_model.cpp
@@ -0,0 +1,134 @@
+/*
+ * 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());
+}