added back some of the analysis scripts

Author: tomvanmele

scripts/dem_arch_cra.py

@@ -0,0 +1,50 @@
+"""Compute the equilibrium of an arch structure using the CRA method.
+
+To run this script, install `compas_dem` and its dependencies using the preconfigured
+"dem-dev" environment in the `compas_dem` repo.
+
+    $ conda env create -f environment.yml
+    $ conda activate dem-dev
+
+"""
+
+from compas_dem.analysis.cra import cra_penalty_solve
+from compas_dem.models import BlockModel
+from compas_dem.templates import ArchTemplate
+from compas_dem.viewer import DEMViewer
+
+# =============================================================================
+# Template
+# =============================================================================
+
+template = ArchTemplate(rise=3, span=10, thickness=0.25, depth=0.5, n=50)
+
+# =============================================================================
+# Model and interactions
+# =============================================================================
+
+model = BlockModel.from_template(template)
+
+model.compute_contacts(tolerance=0.001)
+
+# =============================================================================
+# Supports
+# =============================================================================
+
+for element in model.elements():
+    if model.graph.degree(element.graphnode) == 1:
+        element.is_support = True
+
+# =============================================================================
+# Equilibrium
+# =============================================================================
+
+cra_penalty_solve(model)
+
+# =============================================================================
+# Viz
+# =============================================================================
+
+viewer = DEMViewer(model)
+viewer.setup()
+viewer.show()

scripts/dem_dome.py

@@ -1,3 +1,6 @@
+import pathlib
+
+import compas
 from compas.datastructures import Mesh
 from compas.geometry import SphericalSurface
 
@@ -56,6 +59,8 @@
 # Export
 # =============================================================================
 
+compas.json_dump(model, pathlib.Path(__file__).parent / "dome.json")
+
 # =============================================================================
 # Viz
 # =============================================================================

scripts/dem_dome_cra.py

@@ -0,0 +1,50 @@
+"""Compute the equilibrium of an arch structure using the CRA method.
+
+To run this script, install `compas_dem` and its dependencies using the preconfigured
+"dem-dev" environment in the `compas_dem` repo.
+
+    $ conda env create -f environment.yml
+    $ conda activate dem-dev
+
+To generate the input file, run the `dem_dome.py` script first.
+
+"""
+
+import pathlib
+
+import compas
+
+# from compas_dem.analysis.cra import cra_penalty_solve
+from compas_dem.analysis.cra import rbe_solve
+from compas_dem.elements import Block
+from compas_dem.models import BlockModel
+from compas_dem.viewer import DEMViewer
+
+# =============================================================================
+# Import
+# =============================================================================
+
+model: BlockModel = compas.json_load(pathlib.Path(__file__).parent / "dome.json")  # type: ignore
+
+# =============================================================================
+# Supports
+# =============================================================================
+
+bottom: list[Block] = sorted(model.elements(), key=lambda e: e.point.z)[:16]
+for block in bottom:
+    block.is_support = True
+
+# =============================================================================
+# Equilibrium
+# =============================================================================
+
+# cra_penalty_solve(model)
+rbe_solve(model)
+
+# =============================================================================
+# Viz
+# =============================================================================
+
+viewer = DEMViewer(model)
+viewer.setup()
+viewer.show()

scripts/dem_stack_cra.py

@@ -0,0 +1,69 @@
+"""Compute the equilibrium of an arch structure using the CRA method.
+
+To run this script, install `compas_dem` and its dependencies using the preconfigured
+"dem-dev" environment in the `compas_dem` repo.
+
+    $ conda env create -f environment.yml
+    $ conda activate dem-dev
+
+"""
+
+import math
+import random
+
+from compas.geometry import Box
+
+from compas_dem.analysis.cra import cra_penalty_solve
+from compas_dem.models import BlockModel
+from compas_dem.viewer import DEMViewer
+
+# =============================================================================
+# Block Geometry
+# =============================================================================
+
+box = Box.from_corner_corner_height([0, 0, 0], [1, 1, 0], 1)
+
+blocks: list[Box] = []
+for i in range(10):
+    block: Box = box.copy()
+    block.translate(
+        [
+            random.choice([-0.1, +0.1]) * random.random(),
+            random.choice([-0.1, +0.1]) * random.random(),
+            i * box.zsize,
+        ]
+    )
+    block.rotate(math.radians(random.choice([-5, +5])), box.frame.zaxis, box.frame.point)
+    blocks.append(block)
+
+# =============================================================================
+# Model and interactions
+# =============================================================================
+
+model = BlockModel.from_boxes(blocks)
+
+model.compute_contacts()
+
+# =============================================================================
+# Supports
+# =============================================================================
+
+bottom = sorted(model.elements(), key=lambda e: e.point.z)[0]
+bottom.is_support = True
+
+# =============================================================================
+# Equilibrium
+# =============================================================================
+
+cra_penalty_solve(model)
+
+for contact in model.contacts():
+    print(contact.forces)
+
+# =============================================================================
+# Viz
+# =============================================================================
+
+viewer = DEMViewer(model)
+viewer.setup()
+viewer.show()

scripts/dem_vault_barrel_cra.py

@@ -0,0 +1,43 @@
+"""Compute the equilibrium of an arch structure using the CRA method.
+
+To run this script, install `compas_dem` and its dependencies using the preconfigured
+"dem-dev" environment in the `compas_dem` repo.
+
+    $ conda env create -f environment.yml
+    $ conda activate dem-dev
+
+"""
+
+from compas_dem.analysis.cra import cra_penalty_solve
+from compas_dem.models import BlockModel
+from compas_dem.templates import BarrelVaultTemplate
+from compas_dem.viewer import DEMViewer
+
+# =============================================================================
+# Template
+# =============================================================================
+
+template = BarrelVaultTemplate()
+
+# =============================================================================
+# Model and interactions
+# =============================================================================
+
+model = BlockModel.from_barrelvault(template)
+
+model.compute_contacts(tolerance=0.001)
+
+# =============================================================================
+# Equilibrium
+# =============================================================================
+
+cra_penalty_solve(model)
+
+# =============================================================================
+# Viz
+# =============================================================================
+
+viewer = DEMViewer(model)
+
+viewer.setup()
+viewer.show()