#pragma once

#include <fwd_types.hpp>

#include <blobtree.hpp>
#include <implicit_arrangement.hpp>

#include <settings.h>

using grid_region = std::pair<Eigen::Vector<size_t, 3>, Eigen::Vector<size_t, 3>>;

void                    extract_vertex_infos(const s_settings&           settings,
                                             const baked_blobtree_t&     tree,
                                             scene_bg_mesh_info_t&       scene_bg_mesh_info,
                                             Eigen::Ref<Eigen::MatrixXd> vertex_infos);
void                    build_tetrahedron_and_adjacency(const scene_bg_mesh_info_t&                     scene_bg_mesh_info,
                                                        std::vector<std::array<uint32_t, 4>>&           tetrahedrons,
                                                        flat_hash_map<uint32_t, uint32_t>&              vertex_lexigraphical_adjacency,
                                                        flat_hash_map<uint32_t, std::vector<uint32_t>>& reverse_vertex_adjacency,
                                                        btree_map<uint32_t, std::vector<uint32_t>>&     vertex_to_tet_mapping);
void                    filter_tet_by_subface(const btree_map<uint32_t, std::vector<uint32_t>>&     vertex_to_tet_mapping,
                                              Eigen::MatrixXd&                                      vertex_infos,
                                              flat_hash_map<uint32_t, uint32_t>&                    vertex_indices_mapping,
                                              std::vector<std::array<uint32_t, 4>>&                 tetrahedrons,
                                              flat_hash_map<uint32_t, uint32_t>&                    vertex_lexigraphical_adjacency,
                                              const flat_hash_map<uint32_t, std::vector<uint32_t>>& reverse_vertex_adjacency,
                                              std::vector<uint32_t>&                                tetrahedron_active_subface_start_index,
                                              std::vector<uint32_t>&                                active_subface_indices,
                                              flat_hash_map<uint32_t, std::vector<uint32_t>>&       zero_vertex_to_incident_tet_mapping);
// in turn: num_1_func, num_2_func, num_more_func
std::array<uint32_t, 3> compute_implicit_arrangements(const Eigen::Ref<const Eigen::MatrixXd>     vertex_infos,
                                                      const flat_hash_map<uint32_t, uint32_t>&    vertex_indices_mapping,
                                                      const std::vector<std::array<uint32_t, 4>>& tetrahedrons,
                                                      const std::vector<uint32_t>& tetrahedron_active_subface_start_index,
                                                      const std::vector<uint32_t>& active_subface_indices,
                                                      std::vector<arrangement_t>&  tetrahedron_arrangements);
// extract iso-mesh (topology only)
void                    extract_iso_mesh(const std::array<uint32_t, 3>&              tet_active_subface_counts,
                                         const std::vector<arrangement_t>&           tetrahedron_arrangements,
                                         const scene_bg_mesh_info_t&                 scene_bg_mesh_info,
                                         const std::vector<std::array<uint32_t, 4>>& tetrahedrons,
                                         const std::vector<uint32_t>&                tetrahedron_active_subface_start_index,
                                         const std::vector<uint32_t>&                active_subface_indices,
                                         const Eigen::Ref<const Eigen::MatrixXd>     vertex_infos,
                                         const flat_hash_map<uint32_t, uint32_t>&    vertex_indices_mapping,
                                         std::vector<Eigen::Vector3d>&               iso_pts,
                                         std::vector<iso_vertex_t>&                  iso_verts,
                                         std::vector<polygon_face_t>&                iso_faces);