implement method to filter faces by area (#1258)

* implement method to filter faces by area

* fix test aedt logger

Author: gmalinve

_unittest/test_07_Object3D.py

@@ -2,13 +2,19 @@
 import math
 
 from _unittest.conftest import BasisTest
+from pyaedt.generic.general_methods import is_ironpython
 from pyaedt.generic.general_methods import isclose
 from pyaedt.generic.general_methods import time_fn
 from pyaedt.modeler.Object3d import _to_boolean
 from pyaedt.modeler.Object3d import _uname
 from pyaedt.modeler.Object3d import EdgePrimitive
 from pyaedt.modeler.Object3d import FacePrimitive
 
+try:
+    import pytest
+except ImportError:
+    import _unittest_ironpython.conf_unittest as pytest
+
 
 class TestClass(BasisTest, object):
     def setup_class(self):
@@ -410,3 +416,53 @@ def test_23_volume(self):
         assert box3.volume == 100
         rec = self.aedtapp.modeler.create_rectangle(0, [0, 0, 0], [5, 10])
         assert rec.volume == 0.0
+
+    def test_24_filter_faces_by_area(self):
+        faces_equal = []
+        for obj in self.aedtapp.modeler.object_list:
+            if obj.faces_by_area(100):
+                faces_equal.append(obj.faces_by_area(100))
+        if faces_equal:
+            for face_object in faces_equal:
+                for face in face_object:
+                    assert abs(face.area - 100) < 1e-12
+
+        faces_greater_equal = []
+        for obj in self.aedtapp.modeler.object_list:
+            if obj.faces_by_area(100):
+                faces_greater_equal.append(obj.faces_by_area(100, ">="))
+        if faces_greater_equal:
+            for face_object in faces_greater_equal:
+                for face in face_object:
+                    assert (face.area - 100) >= 1e-12
+
+        faces_smaller_equal = []
+        for obj in self.aedtapp.modeler.object_list:
+            if obj.faces_by_area(100):
+                faces_smaller_equal.append(obj.faces_by_area(100, "<="))
+        if faces_smaller_equal:
+            for face_object in faces_smaller_equal:
+                for face in face_object:
+                    assert (face.area - 100) <= 1e-12
+
+        faces_greater = []
+        for obj in self.aedtapp.modeler.object_list:
+            if obj.faces_by_area(100):
+                faces_greater.append(obj.faces_by_area(100, ">"))
+        if faces_greater:
+            for face_object in faces_greater:
+                for face in face_object:
+                    assert (face.area - 100) > 1e-12
+
+        faces_smaller = []
+        for obj in self.aedtapp.modeler.object_list:
+            if obj.faces_by_area(100):
+                faces_smaller.append(obj.faces_by_area(100, "<"))
+        if faces_smaller:
+            for face_object in faces_smaller:
+                for face in face_object:
+                    assert (face.area - 100) < 1e-12
+
+        if not is_ironpython:
+            with pytest.raises(ValueError):
+                self.aedtapp.modeler.object_list[0].faces_by_area(100, "<<")

_unittest/test_14_AedtLogger.py

@@ -112,6 +112,12 @@ def test_01_formatter(self):
         logger = AedtLogger(filename=path)
         assert logger.formatter == settings.formatter
         settings.formatter = None
+        logger.disable_log_on_file()
+
+        for handler in logger._global.handlers:
+            if isinstance(handler, logging.FileHandler):
+                handler.close()
+                logger.removeHandler(handler)
 
     def test_02_output_file_with_app_filter(self):
         content = None

pyaedt/modeler/Object3d.py

@@ -2600,6 +2600,48 @@ def delete(self):
         self._primitives.cleanup_objects()
         self.__dict__ = {}
 
+    @pyaedt_function_handler()
+    def faces_by_area(self, area, area_filter="==", tolerance=1e-12):
+        """Filter faces by area.
+
+        Parameters
+        ----------
+        area : float
+            Value of the area to filter.
+        area_filter : str, optional
+            Comparer symbol.
+            Default value is "==".
+        tolerance : float, optional
+            tolerance for comparison.
+
+        Returns
+        -------
+        list of :class:`pyaedt.modeler.Object3d.FacePrimitive`
+        """
+        filters = ["==", "<=", ">=", "<", ">"]
+        if area_filter not in filters:
+            raise ValueError('Symbol not valid, enter one of the following: "==", "<=", ">=", "<", ">"')
+
+        faces = []
+        for face in self.faces:
+            if area_filter is "==":
+                if abs(face.area - area) < tolerance:
+                    faces.append(face)
+            if area_filter is ">=":
+                if (face.area - area) >= tolerance:
+                    faces.append(face)
+            if area_filter is "<=":
+                if (face.area - area) <= tolerance:
+                    faces.append(face)
+            if area_filter is ">":
+                if (face.area - area) > tolerance:
+                    faces.append(face)
+            if area_filter is "<":
+                if (face.area - area) < tolerance:
+                    faces.append(face)
+
+        return faces
+
     @pyaedt_function_handler()
     def _change_property(self, vPropChange):
         return self._primitives._change_geometry_property(vPropChange, self._m_name)