all examples working

Author: franaudo

src/compas_fea2/model/elements.py

@@ -210,7 +210,7 @@ def __init__(self, nodes, section, frame, implementation=None, rigid=False, **kw
         self._frame = frame
         self._curve = Line(nodes[0].point, nodes[-1].point)
         # self._shape = Brep.from_extrusion(curve=self.section._shape, vector=Vector.from_start_end(nodes[0].point, nodes[-1].point))
-        self._shape = Box(self.length, self.section._w, self.section._h, frame=Frame(self.nodes[0].point, [1,0,0], [0,1,0]))
+        self._shape = section._shape #Box(self.length, self.section._w, self.section._h, frame=Frame(self.nodes[0].point, [1,0,0], [0,1,0]))
 
     @property
     def frame(self):

src/compas_fea2/model/model.py

@@ -393,7 +393,7 @@ def find_nodes_by_name(self, name):
 
     @get_docstring(_Part)
     @part_method
-    def find_nodes_by_location(self, point, distance, plane=None):
+    def find_nodes_around_point(self, point, distance, plane=None, single=False):
         pass
 
     @get_docstring(_Part)

src/compas_fea2/model/parts.py

@@ -266,7 +266,7 @@ def from_compas_lines(cls, lines, element_model="BeamElement", section=None, nam
         # nodes = [Node(n) for n in set([list(p) for l in lines for p in list(l)])]
         for line in lines:
             #FIXME change tolerance
-            nodes = [prt.find_nodes_by_location(list(p), 1, single=True) or Node(list(p)) for p in list(line)]
+            nodes = [prt.find_nodes_around_point(list(p), 1, single=True) or Node(list(p)) for p in list(line)]
             prt.add_nodes(nodes)
             element = getattr(compas_fea2.model, element_model)(nodes=nodes, section=section)
             if not isinstance(element, _Element1D):
@@ -568,7 +568,7 @@ def find_nodes_by_name(self, name):
         """
         return [node for node in self.nodes if node.name == name]
 
-    def find_nodes_by_location(self, point, distance, plane=None, report=False, single=False, **kwargs):
+    def find_nodes_around_point(self, point, distance, plane=None, report=False, single=False, **kwargs):
         # type: (Point, float, Plane, bool, bool, bool) -> list[Node]
         """Find all nodes within a distance of a given geometrical location.
 
@@ -623,7 +623,7 @@ def find_closest_nodes_to_point(self, point, distance, number_of_nodes=1, plane=
         list[:class:`compas_fea2.model.Node`]
 
         """
-        nodes = self.find_nodes_by_location(point, distance, plane, report=True)
+        nodes = self.find_nodes_around_point(point, distance, plane, report=True)
         if number_of_nodes > len(nodes):
             number_of_nodes = len(nodes)
         return [k for k, v in sorted(nodes.items(), key=lambda item: item[1])][:number_of_nodes]
@@ -647,7 +647,7 @@ def find_nodes_around_node(self, node, distance, plane=None):
             The nodes around the given node
 
         """
-        nodes = self.find_nodes_by_location(node.xyz, distance, plane, report=True)
+        nodes = self.find_nodes_around_point(node.xyz, distance, plane, report=True)
         if node in nodes:
             del nodes[node]
         return nodes
@@ -672,7 +672,7 @@ def find_closest_nodes_to_node(self, node, distance, number_of_nodes=1, plane=No
         list[:class:`compas_fea2.model.Node`]
 
         """
-        nodes = self.find_nodes_by_location(node.xyz, distance, plane, report=True)
+        nodes = self.find_nodes_around_point(node.xyz, distance, plane, report=True)
         if number_of_nodes > len(nodes):
             number_of_nodes = len(nodes)
         return [k for k, v in sorted(nodes.items(), key=lambda item: item[1])][:number_of_nodes]
@@ -832,7 +832,7 @@ def add_node(self, node):
             return
 
         if not compas_fea2.POINT_OVERLAP:
-            if self.find_nodes_by_location(node.xyz, distance=compas_fea2.GLOBAL_TOLERANCE):
+            if self.find_nodes_around_point(node.xyz, distance=compas_fea2.GLOBAL_TOLERANCE):
                 if compas_fea2.VERBOSE:
                     print("NODE SKIPPED: Part {!r} has already a node at {}.".format(self, node.xyz))
                 return
@@ -1339,7 +1339,7 @@ def get_average_displacement_at_point(self, problem, point, distance, step=None,
 
         """
         step = step or problem._steps_order[-1]
-        nodes = self.find_nodes_by_location(point=point, distance=distance, report=True)
+        nodes = self.find_nodes_around_point(point=point, distance=distance, report=True)
         if nodes:
             displacements = [getattr(Vector(*node.results[problem][step].get("U", None)), component) for node in nodes]
             return point, sum(displacements) / len(displacements)

src/compas_fea2/problem/problem.py

@@ -788,7 +788,10 @@ def show_deformed(self, step=None, scale_factor=1.0, original=0.5, opacity=0.5,
 
         if original:
             for part in step.model.parts:
-                v.draw_mesh(part.discretized_boundary_mesh, opacity=original)
+                try:
+                    v.draw_mesh(part.discretized_boundary_mesh, opacity=original)
+                except:
+                    print("No mesh found")
                 # v.draw_nodes_field_vector(step=step, field_name='U', vector_sf=scale_factor)
 
         v.draw_deformed(step, scale_factor=scale_factor, opacity=opacity)

src/compas_fea2/results/fields.py

@@ -253,13 +253,17 @@ def get_results_at_point(self, point, distance, plane=None, steps=None):
         Returns
         -------
         dict
-            Dictionary with {'part':..; 'node':..; 'vector':...}
+            Dictionary with {step: result}
 
         """
         nodes = self.model.find_nodes_around_point(point, distance, plane)
-        results = []
-        for step in steps:
-            results.append(self.get_results(nodes, steps)[step])
+        if not nodes:
+            print(f"WARNING: No nodes found at {point} within {distance}")
+        else:
+            results = []
+            steps = steps or self.problem.steps
+            results = self.get_results(nodes, steps)
+            return results
 
 
 class DisplacementFieldResults(FieldResults):