Armadillo Vault

"""Load and visualise armadillo vault CRA result."""

import os

import compas
import compas_cra
from compas_cra.algorithms import assembly_interfaces_numpy
from compas_cra.viewers import cra_view

# if True show precalculated result, False to identify interfaces
load_from_result = True


def load_result():
    """Load armadillo vault with cra result."""
    armadillo_assembly = compas.json_load(os.path.join(compas_cra.SAMPLE, "armadillo_cra.json"))
    return armadillo_assembly


def identify_interfaces():
    """Load armadillo vault blocks."""
    armadillo_assembly = compas.json_load(os.path.join(compas_cra.SAMPLE, "armadillo.json"))
    assembly_interfaces_numpy(armadillo_assembly, tmax=0.05, amin=0.0001)
    return armadillo_assembly


if load_from_result:
    assembly = load_result()
else:
    assembly = identify_interfaces()

print(assembly)

cra_view(
    assembly,
    resultant=True,
    nodal=False,
    weights=False,
    grid=False,
    interfaces=not load_from_result,
    forcesline=True,
    forcesdirect=False,
    displacements=False,
    dispscale=0,
    scale=300,
)