fix: Reaction calculation (#7)

Author: angelsolaorbaiceta

.github/workflows/python-package.yml

@@ -5,35 +5,34 @@ name: Python package
 
 on:
   push:
-    branches: [ master ]
+    branches: [master]
   pull_request:
-    branches: [ master ]
+    branches: [master]
 
 jobs:
   build:
-
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["^3.6", "^3.7", "^3.8", "^3.9", "^3.10"]
+        python-version: ['^3.9', '^3.10', '^3.11', '^3.12']
 
     steps:
-    - uses: actions/checkout@v2
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
-      with:
-        python-version: ${{ matrix.python-version }}
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install flake8 pytest
-        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
-    - name: Lint with flake8
-      run: |
-        # stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
-        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
-    - name: Test with pytest
-      run: |
-        pytest
+      - uses: actions/checkout@v2
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install flake8
+          if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+      - name: Lint with flake8
+        run: |
+          # stop the build if there are Python syntax errors or undefined names
+          flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+          # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+          flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+      - name: Test with unittest
+        run: |
+          python -m unittest discover -p '*_test.py'

.gitignore

@@ -7,4 +7,7 @@ __pycache__/
 
 # truss structure solution
 result.svg
-result.txt
+result.txt
+
+node_modules/
+.svelte-kit

.vscode/settings.json

@@ -3,8 +3,11 @@
   "[python]": {
     "editor.defaultFormatter": "charliermarsh.ruff",
     "editor.codeActionsOnSave": {
-      "source.fixAll": true,
-      "source.organizeImports": true
+      "source.fixAll": "explicit",
+      "source.organizeImports": "explicit"
     }
-  }
+  },
+  "python.testing.unittestArgs": ["-v", "-s", ".", "-p", "*_test.py"],
+  "python.testing.pytestEnabled": false,
+  "python.testing.unittestEnabled": true
 }

geom2d/vector.py

@@ -216,6 +216,9 @@ def __eq__(self, other):
     def __str__(self):
         return f"({self.u}, {self.v}) with norm {self.norm}"
 
+    def __repr__(self):
+        return str(self)
+
     def to_formatted_str(self, decimals: int):
         """
         Returns a string of the form: '(u, v) with norm N', where

graphic/simulation/draw.py

@@ -1,7 +1,7 @@
 from functools import reduce
 from tkinter import Canvas
 
-from geom2d import Circle, Polygon, Segment, Rect, AffineTransform
+from geom2d import AffineTransform, Circle, Polygon, Rect, Segment
 
 
 class CanvasDrawing:

run_tests.sh

@@ -1,6 +1,6 @@
 from eqs import Matrix
 from geom2d import Segment
-from .node import StrNode
+from structures.model.node import StrNode
 
 
 class StrBar:

structures/model/bar.py

@@ -5,13 +5,12 @@
 from eqs import Vector as EqVector
 from eqs import cholesky_solve
 from geom2d import Vector
+from structures.model.bar import StrBar
+from structures.model.node import StrNode
 from structures.solution.bar import StrBarSolution
 from structures.solution.node import StrNodeSolution
 from structures.solution.structure import StructureSolution
 
-from .bar import StrBar
-from .node import StrNode
-
 
 class Structure:
     """

structures/model/structure.py

@@ -44,6 +44,14 @@ def is_constrained(self):
             self.__original_node.dx_constrained or self.__original_node.dy_constrained
         )
 
+    @property
+    def dx_constrained(self) -> bool:
+        return self.__original_node.dx_constrained
+
+    @property
+    def dy_constrained(self) -> bool:
+        return self.__original_node.dy_constrained
+
     @property
     def loads(self):
         """

structures/solution/node.py

@@ -35,7 +35,7 @@ def bounds_rect(self, margin: float, scale=1):
         d_pos = [node.displaced_pos_scaled(scale) for node in self.nodes]
         return make_rect_containing_with_margin(d_pos, margin)
 
-    def reaction_for_node(self, node: StrNodeSolution):
+    def reaction_for_node(self, node: StrNodeSolution) -> Vector:
         """
         Computes the external reaction force for a given node.
 
@@ -57,4 +57,10 @@ def reaction_for_node(self, node: StrNodeSolution):
         if node.is_loaded:
             forces.append(node.net_load.opposite())
 
-        return reduce(operator.add, forces)
+        net_force = reduce(operator.add, forces)
+
+        # The reaction can only have the components of the constrained directions
+        return Vector(
+            u=net_force.u if node.dx_constrained else 0.0,
+            v=net_force.v if node.dy_constrained else 0.0,
+        )