Changeset 1111 for cpp/frams/model/geometry/modelgeometryinfo.cpp

Timestamp:

03/16/21 23:59:15 (3 years ago)

Author:

Maciej Komosinski

Message:

Added a function to get voxels of sampled Model geometry from script

File:

• cpp/frams/model/geometry/modelgeometryinfo.cpp (modified) (13 diffs)

cpp/frams/model/geometry/modelgeometryinfo.cpp

@@ -1 +1 @@
 // This file is a part of Framsticks SDK.  http://www.framsticks.com/
-// Copyright (C) 1999-2015  Maciej Komosinski and Szymon Ulatowski.
+// Copyright (C) 1999-2021  Maciej Komosinski and Szymon Ulatowski.
 // See LICENSE.txt for details.
 
@@ -6 +6 @@
 #include <frams/model/geometry/geometryutils.h>
 #include <frams/model/geometry/meshbuilder.h>
+#include <frams/vm/classes/collectionobj.h>
+#include <frams/vm/classes/3dobject.h>
 
 void ModelGeometryInfo::findSizesAndAxes(Model &input_model, const double density,
@@ -33 +35 @@
                 return;
         }
-        
+
         boundingBox(model.getPart(0), lowerBoundary, upperBoundary);
-        
+
         for (int i = 1; i < model.getPartCount(); i++)
         {
                 Pt3D partLowerBoundary, partUpperBoundary;
                 boundingBox(model.getPart(i), partLowerBoundary, partUpperBoundary);
-                
+
                 lowerBoundary.getMin(partLowerBoundary);
                 upperBoundary.getMax(partUpperBoundary);
@@ -51 +53 @@
         lowerBoundary.y = upperBoundary.y = part->p.y;
         lowerBoundary.z = upperBoundary.z = part->p.z;
-        
-        for (Octants::Octant o = Octants::FIRST; o < Octants::NUMBER; o = Octants::Octant(o+1))
+
+        for (Octants::Octant o = Octants::FIRST; o < Octants::NUMBER; o = Octants::Octant(o + 1))
         {
                 Pt3D vertex = part->scale;
@@ -58 +60 @@
                 vertex.y *= Octants::isPositiveY(o) ? +1 : -1;
                 vertex.z *= Octants::isPositiveZ(o) ? +1 : -1;
-                
+
                 vertex = part->o.transform(vertex) + part->p;
-                
+
                 lowerBoundary.getMin(vertex);
                 upperBoundary.getMax(vertex);
@@ -71 +73 @@
         Pt3D lowerBoundary, upperBoundary;
         boundingBox(model, lowerBoundary, upperBoundary);
-        
+
         MeshBuilder::BoundingBoxVolume iterator(density);
         iterator.initialize(lowerBoundary, upperBoundary);
-        
+
         Pt3D point;
         int allPoints = 0, pointsInsideModel = 0;
-        
+
         while (iterator.tryGetNext(point))
         {
@@ -83 +85 @@
                 pointsInsideModel += GeometryUtils::isPointInsideModel(point, model) ? 1 : 0;
         }
-        
+
         double boundingBoxVolume
                 = (upperBoundary.x - lowerBoundary.x)
                 * (upperBoundary.y - lowerBoundary.y)
                 * (upperBoundary.z - lowerBoundary.z);
-        
+
         return boundingBoxVolume * (double)pointsInsideModel / (double)allPoints;
 }
 
+ExtObject ModelGeometryInfo::getVoxels(Model& input_model, const double density)
+{
+        SolidsShapeTypeModel model(input_model);
+        Pt3D lowerBoundary, upperBoundary;
+        boundingBox(model, lowerBoundary, upperBoundary);
+
+        MeshBuilder::BoundingBoxVolume iterator(density);
+        iterator.initialize(lowerBoundary, upperBoundary);
+
+        Pt3D point;
+        VectorObject *voxels = new VectorObject;
+
+        while (iterator.tryGetNext(point))
+        {
+                if (GeometryUtils::isPointInsideModel(point, model))
+                {
+                        voxels->data += new ExtValue(Pt3D_Ext::makeDynamicObject(point));
+                }
+        }
+
+        return voxels->makeObject();
+}
+
 double ModelGeometryInfo::area(Model &input_model, const double density)
 {
         SolidsShapeTypeModel model(input_model);
         double area = 0.0;
-        
-        for (int partIndex = 0; partIndex < model.getModel().getPartCount(); partIndex+=1)
+
+        for (int partIndex = 0; partIndex < model.getModel().getPartCount(); partIndex += 1)
         {
                 area += externalAreaOfPart(model, partIndex, density);
         }
-        
+
         return area;
 }
@@ -110 +135 @@
         switch (part->shape)
         {
-                case Part::SHAPE_ELLIPSOID:
-                        return externalAreaOfEllipsoid(model, partIndex, density);
-                        
-                case Part::SHAPE_CUBOID:
-                        return externalAreaOfCuboid(model, partIndex, density);
-                        
-                case Part::SHAPE_CYLINDER:
-                        return externalAreaOfCylinder(model, partIndex, density);
+        case Part::SHAPE_ELLIPSOID:
+                return externalAreaOfEllipsoid(model, partIndex, density);
+
+        case Part::SHAPE_CUBOID:
+                return externalAreaOfCuboid(model, partIndex, density);
+
+        case Part::SHAPE_CYLINDER:
+                return externalAreaOfCylinder(model, partIndex, density);
         }
         logPrintf("ModelGeometryInfo", "externalAreaOfPart", LOG_ERROR, "Part shape=%d not supported", part->shape);
@@ -127 +152 @@
         Pt3D point;
         int all = 0, sur = 0;
-        
+
         while (surface.tryGetNext(point))
         {
@@ -133 +158 @@
                 sur += GeometryUtils::isPointInsideModelExcludingPart(point, &model, partIndex) ? 0 : 1;
         }
-        
+
         return sur * area / all;
 }
@@ -140 +165 @@
 {
         Part *part = model.getPart(partIndex);
-        
+
         MeshBuilder::EllipsoidSurface ellipsoid(density);
         ellipsoid.initialize(part);
-        
+
         double area = GeometryUtils::ellipsoidArea(part->scale);
-        
+
         return externalAreaOfSurface(model, partIndex, ellipsoid, area);
 }
@@ -152 +177 @@
 {
         Part *part = model.getPart(partIndex);
-        
+
         double externalArea = 0.0;
-        
+
         MeshBuilder::RectangleSurface rectangle(density);
-        
-        for (CuboidFaces::Face f = CuboidFaces::FIRST; f < CuboidFaces::NUMBER; f = CuboidFaces::Face(f+1))
+
+        for (CuboidFaces::Face f = CuboidFaces::FIRST; f < CuboidFaces::NUMBER; f = CuboidFaces::Face(f + 1))
         {
                 rectangle.initialize(part, f);
-                
+
                 double area = 4.0;
                 area *= !CuboidFaces::isX(f) ? part->scale.x : 1.0;
                 area *= !CuboidFaces::isY(f) ? part->scale.y : 1.0;
                 area *= !CuboidFaces::isZ(f) ? part->scale.z : 1.0;
-                
+
                 externalArea += externalAreaOfSurface(model, partIndex, rectangle, area);
         }
-        
+
         return externalArea;
 }
@@ -175 +200 @@
 {
         Part *part = model.getPart(partIndex);
-        
+
         double externalArea = 0.0;
-        
+
         MeshBuilder::EllipseSurface ellipse(density);
-        
+
         double area = M_PI * part->scale.y * part->scale.z;
-        
-        for (CylinderBases::Base b = CylinderBases::FIRST; b < CylinderBases::NUMBER; b = CylinderBases::Base(b+1))
+
+        for (CylinderBases::Base b = CylinderBases::FIRST; b < CylinderBases::NUMBER; b = CylinderBases::Base(b + 1))
         {
                 ellipse.initialize(part, b);
-                
+
                 externalArea += externalAreaOfSurface(model, partIndex, ellipse, area);
         }
-        
+
         MeshBuilder::CylinderWallSurface cylinderWall(density);
         cylinderWall.initialize(part);
-        
-        area = 2.0*part->scale.x * GeometryUtils::ellipsePerimeter(part->scale.y, part->scale.z);
-        
+
+        area = 2.0 * part->scale.x * GeometryUtils::ellipsePerimeter(part->scale.y, part->scale.z);
+
         externalArea += externalAreaOfSurface(model, partIndex, cylinderWall, area);
-        
+
         return externalArea;
 }