source: cpp/frams/model/geometry/modelgeoclass.cpp @ 1280

// This file is a part of Framsticks SDK.  http://www.framsticks.com/
// Copyright (C) 1999-2021  Maciej Komosinski and Szymon Ulatowski.
// See LICENSE.txt for details.

#include "modelgeoclass.h"
#include "modelgeometryinfo.h"
#include <frams/vm/classes/collectionobj.h>
#include <frams/vm/classes/3dobject.h>

#define FIELDSTRUCT ModelGeometry
static ParamEntry modelgeo_paramtab[] =
{
        { "Creature: Geometry", 1, 6, "ModelGeometry",
        "Approximately estimates sizes, volume, and area of a Model based on the geometry of its parts.\n"
        "Example usage:\n"
        "Simulator.print(ModelGeometry.forModel(Model.newFromString(\"//0\\np:sh=1\\n\")).area());\n\n"
        "ModelGeometry.geom_density refers to the global simulator parameter (also available in GUI).\n"
        "To set geom_density for individual ModelGeometry objects:\n"
        "var mg=ModelGeometry.forModel(GenePools[0][0].getModel()); mg.geom_density=2; GenePools[0][0].data->area=mg.area();\n" },
        { "geom_density", 0, 0, "Density", "f 0.01 100.0 3.0", FIELD(density), "The number of samples (per unit length in one dimension) that affects the precision of estimation of geometrical properties." }, //Note #1: we used 'geom_density' instead of 'density' to make the name more unique - because sim_params merges all configuration fields in a single namespace. Note #2: "density" needs better, more interpretable and more reliable parametrization for surface sampling...
        { "forModel", 0, PARAM_USERHIDDEN, "", "p oModelGeometry(oModel)", PROCEDURE(p_formodel), "The returned ModelGeometry object can be used to calculate geometric properties (volume, area, sizes) of the associated model. The density is copied from the current global ModelGeometry.geom_density on object creation." },
        { "volume", 0, PARAM_NOSTATIC | PARAM_USERHIDDEN, "volume", "p f()", PROCEDURE(p_volume), },
        { "area", 0, PARAM_NOSTATIC | PARAM_USERHIDDEN, "area", "p f()", PROCEDURE(p_area), },
        { "voxels", 0, PARAM_NOSTATIC | PARAM_USERHIDDEN, "voxels", "p oVector()", PROCEDURE(p_voxels), "Returns a Vector of Pt3D objects from a regular 3D grid (sampled according to ModelGeometry.geom_density) that are inside of the Model body (Parts and Joints)."},
        { "sizesAndAxes", 0, PARAM_NOSTATIC | PARAM_USERHIDDEN, "sizesAndAxes", "p oVector()", PROCEDURE(p_sizesandaxes), "The returned vector contains XYZ (sizes) and Orient (axes) objects." },
        { 0, 0, 0, },
};
#undef FIELDSTRUCT

ExtObject ModelGeometry::makeDynamicObject(ModelGeometry* mg)
{
        return ExtObject(&mg->par, mg);
}

ModelGeometry::ModelGeometry(ModelObj *mo)
        :par(modelgeo_paramtab, this)
{
        cached_for_density = -1; //invalid value, will be updated on first request
        invalidateAllCached();
        model = mo;
        if (model != NULL)
                model->incref();
}

ModelGeometry::~ModelGeometry()
{
        if (model != NULL)
                model->decref();
}

// Mark all 3 results as invalid.
// Validity of these 3 values must be maintained independently,
// as each of them is calculated by an individual call.
void ModelGeometry::invalidateAllCached()
{
        cached_volume = -1;
        cached_area = -1;
        cached_sizes.x = -1;
        cached_voxels.setEmpty();
}

// Invalidates cached results if a new density is requested
// (called in all geometry calculation functions)
void ModelGeometry::onDensityChanged()
{
        if (cached_for_density != density)
        {
                invalidateAllCached();
                cached_for_density = density;
        }
}

void ModelGeometry::p_formodel(ExtValue *args, ExtValue *ret)
{
        Model *m = ModelObj::fromObject(*args);
        if (m != NULL)
        {
                if (m->getShapeType() == Model::SHAPETYPE_BALL_AND_STICK)
                {
                        Model *converted = new Model;
                        converted->open();
                        converted->buildUsingSolidShapeTypes(*m, Part::SHAPE_CYLINDER);
                        converted->close();
                        m = converted;
                }
                ModelGeometry *mg = new ModelGeometry((ModelObj*)m);
                mg->density = density;
                ret->setObject(ModelGeometry::makeDynamicObject(mg));
        }
        else
                ret->setEmpty();
}

void ModelGeometry::p_volume(ExtValue *args, ExtValue *ret)
{
        onDensityChanged();
        if (cached_volume < 0) //calculate if invalid
                cached_volume = ModelGeometryInfo::volume(*model, density);
        ret->setDouble(cached_volume);
}

void ModelGeometry::p_area(ExtValue *args, ExtValue *ret)
{
        onDensityChanged();
        if (cached_area < 0) //calculate if invalid
                cached_area = ModelGeometryInfo::area(*model, density);
        ret->setDouble(cached_area);
}

void ModelGeometry::p_voxels(ExtValue *args, ExtValue *ret)
{
        onDensityChanged();
        if (cached_voxels.isEmpty()) //calculate if invalid
                cached_voxels = ModelGeometryInfo::getVoxels(*model, density);
        ret->setObject(cached_voxels);
}

void ModelGeometry::p_sizesandaxes(ExtValue *args, ExtValue *ret)
{
        onDensityChanged();
        if (cached_sizes.x < 0) //calculate if invalid
                ModelGeometryInfo::findSizesAndAxes(*model, density, cached_sizes, cached_axes);

        VectorObject* n = new VectorObject;
        n->data += new ExtValue(Pt3D_Ext::makeDynamicObject(cached_sizes));
        n->data += new ExtValue(Orient_Ext::makeDynamicObject(new Orient_Ext(cached_axes)));
        ret->setObject(n->makeObject());
}