script.h

#pragma once

#include <GLFW/glfw3.h>
#include <imgui.h>

// Salamesh core
#include <app_interface.h>
#include <script.h>
#include <input_states.h>
#include <models/model.h>
#include <models/surface_model.h>
#include <element.h>
// Ultimaille
#include <ultimaille/all.h>
// ImGui
#include "imgui.h"
#include "imgui_internal.h"
// Std
#include <iostream>

// Return nring of a facet for a given model
std::vector<int> n_ring(Model &model, int fi, int n) {
	// Get mesh from model
	auto &m = model.as<PolyModel>().getMesh();
	m.connect();

	std::vector<int> ring;
	std::queue<std::tuple<int, int>> q;
	q.push({fi, 0});
	
	FacetAttribute<bool> processed(m);

	while (!q.empty()) {
		auto [fi, d] = q.front();
		auto f = m.facet(fi);
		q.pop();

		if (processed[fi])
			continue;

		processed[fi] = true;

		if (d >= n)
			continue;

		for (auto &h : f.iter_halfedges()) {
			if (!h.opposite().active())
				continue;
				
			auto o = h.opposite().facet();
			ring.push_back(o);
			q.push({o, d + 1});
		}
	}
	
	return ring;
}

std::tuple<glm::vec3, glm::vec3> getFacetBBox(Model &model, int fi) {
	// Get mesh from model
	auto &m = model.as<PolyModel>().getMesh();
	// auto f = m.facet(fi);

	auto facets = n_ring(model, fi, 4);

	vec3 b{0.};
	for (auto fi : facets) {
		auto f = m.facet(fi);
		vec3 cur_b{0.};
		for (int lv = 0; lv < f.size(); ++lv)
			cur_b += f.vertex(lv).pos();

		cur_b /= f.size();
		b += cur_b;
	}

	b /= facets.size();

	// Compute highest dist from barycenter
	double d = 0.;
	for (auto fi : facets) {
		auto f = m.facet(fi);
		for (int lv = 0; lv < f.size(); ++lv) {
			double cur_d = (f.vertex(lv).pos() - b).norm();
			if (cur_d > d) {
				d = cur_d;
			}
		}
	}

	vec3 min = b - vec3(d, d, d);
	vec3 max = b + vec3(d, d, d);

	return std::make_tuple(glm::vec3(min.x, min.y, min.z), glm::vec3(max.x, max.y, max.z));
}

void jacobian_scaled(Model &model) {
	// Get mesh from model
	auto &poly_model = model.as<PolyModel>();
	auto &m = poly_model.getMesh();

	FacetAttribute<double> scaled;
	scaled.bind("scaled", poly_model.getSurfaceAttributes(), m);
}