structuralcodes 0.1.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

structuralcodes/geometry/_rectangular.py

@@ -0,0 +1,83 @@
+"""Classes for rectangular geometries.
+
+The class `RectangularGeometry` represents a rectangular SurfaceGeometry with
+homogenous material.
+This class is simply a wrapper of `SurfaceGeometry` class and permits an easy
+input by the user.
+"""
+
+import typing as t
+
+from shapely import Polygon
+
+from structuralcodes.core.base import ConstitutiveLaw, Material
+
+from ._geometry import SurfaceGeometry
+
+
+class RectangularGeometry(SurfaceGeometry):
+    """This is a wrapper class for defining a `SurfaceGeometry` of rectangular
+    shape with a homogeneous material.
+    """
+
+    _width: float
+    _height: float
+
+    def __init__(
+        self,
+        width: float,
+        height: float,
+        material: t.Union[Material, ConstitutiveLaw],
+        density: t.Optional[float] = None,
+        concrete: bool = False,
+    ) -> None:
+        """Initialize a RectangularGeometry.
+
+        Arguments:
+            width (float): The width of the geometry.
+            height (float): The height of the geometry.
+            material (Union(Material, ConstitutiveLaw)): A Material or
+                ConsitutiveLaw class applied to the geometry.
+            density (Optional(float)): When a ConstitutiveLaw is passed as
+                material, the density can be provided by this argument. When
+                material is a Material object the density is taken from the
+                material.
+            concrete (bool): Flag to indicate if the geometry is concrete. When
+                passing a Material as material, this is automatically inferred.
+
+        Note:
+            The RectangularGeometry is simply a wrapper for a SurfaceGeometry
+            object.
+        """
+        # Check that size is strictly positive
+        if width <= 0:
+            raise ValueError('Width must be a positive number.')
+        if height <= 0:
+            raise ValueError('Height must be a positive number.')
+
+        self._width = width
+        self._height = height
+
+        # Create the shapely polygon
+        polygon = Polygon(
+            (
+                (-width / 2, -height / 2),
+                (width / 2, -height / 2),
+                (width / 2, height / 2),
+                (-width / 2, height / 2),
+            )
+        )
+        # Pass everything to the base class
+        super().__init__(
+            poly=polygon, material=material, density=density, concrete=concrete
+        )
+
+    @property
+    def width(self):
+        """Returns the width of the rectangle."""
+        return self._width
+
+    @property
+    def height(self):
+        """Return the height of the rectangle."""
+        return self._height

structuralcodes/geometry/_reinforcement.py

@@ -1,16 +1,14 @@
 """Functions related to reinforcement definition."""
 
+import math
 import typing as t
 
 import numpy as np
 from shapely import Point
 
 from structuralcodes.core.base import ConstitutiveLaw, Material
-from structuralcodes.geometry import (
-    CompoundGeometry,
-    PointGeometry,
-    SurfaceGeometry,
-)
+
+from ._geometry import CompoundGeometry, PointGeometry, SurfaceGeometry
 
 
 def add_reinforcement(
@@ -125,3 +123,132 @@ def add_reinforcement_line(
             group_label=group_label,
         )
     return geo
+
+
+def add_reinforcement_circle(
+    geo: t.Union[SurfaceGeometry, CompoundGeometry],
+    center: t.Tuple[float, float],
+    radius: float,
+    diameter: float,
+    material: t.Union[Material, ConstitutiveLaw],
+    n: int = 0,
+    s: float = 0.0,
+    first: bool = True,
+    last: bool = True,
+    start_angle: float = 0.0,
+    stop_angle: float = 2 * np.pi,
+    group_label: t.Optional[str] = None,
+) -> CompoundGeometry:
+    """Adds a set of bars distributed in a circular arch line.
+    By default the whole circle is considered. If one wants to specify a
+    circular arch, the `start_angle` and `stop_angle` attributes need to be
+    specified.
+
+    Arguments:
+        geo (Union(SurfaceGeometry, CompoundGeometry)): The geometry used as
+            input.
+        center (Tuple(float, float)): Coordinates of the center point of
+            the circle line where reinforcement will be added.
+        radius (float): Radius of the circle line where reinforcement will be
+            added.
+        diameter (float): The diameter of the bars.
+        material (Union(Material, ConstitutiveLaw)): A valid material or
+            constitutive law.
+        n (int): The number of bars to be distributed inside the line (default
+            = 0).
+        s (float): The distance between the bars (default = 0).
+        first (bool): Boolean indicating if placing the first bar (default =
+            True).
+        last (bool): Boolean indicating if placing the last bar (default =
+            True).
+        start_angle (float): Start angle (respect to X axis) for defining the
+            arch where to add bars in radians (default = 0)
+        stop_angle (float): Stop angle (respect to X axis) for defining the
+            arch where to add bars in radians (default = 2pi)
+        group_label (Optional(str)): A label for grouping several objects
+            (default is None).
+
+    Note:
+        At least n or s should be greater than zero.
+        Attribues start_angle and stop_angle by default are 0 and 2pi
+        respectively, so that bars will be distributed along the whole circle.
+        If only a portion of the circle must be used (i.e. an arch of
+        circumference), then set start and stop angles correspondingly.
+        stop_angle must always be larger than start_angle.
+
+    Returns:
+        CompoundGeometry: A compound geometry with the original geometry and
+        the reinforcement.
+    """
+    # Check that difference between stop and start angle is
+    # positive and less than 2pi
+    if stop_angle - start_angle <= 0 or stop_angle - start_angle > 2 * np.pi:
+        raise ValueError(
+            'Stop angle should be larger than start angle and difference \
+            them should be at most 2pi.'
+        )
+    # Calculate length from start_angle to stop_angle
+    length = radius * (stop_angle - start_angle)
+
+    # If the whole circle, than deal with the case that would add an extra bar
+    whole = math.isclose(length - 2 * np.pi * radius, 0, abs_tol=1e-4)
+    add_n = 0 if not whole else 1
+
+    # delta_angle is used if we need to center the set of bars
+    # in the curve.
+    delta_angle = 0
+    if n > 0 and s > 0:
+        # Provided both the number of bars and spacing
+        # Check there is enough space for fitting the bars
+        n += add_n
+        needed_length = (n - 1) * s
+        if needed_length > length:
+            raise ValueError(
+                f'There is not room to fit {n} bars with a spacing of {s} \
+                in {length}'
+            )
+        # Compute delta_angle to make bars centered in the curvilinear segment
+        delta_angle = (length - needed_length) / 2.0 / radius
+    elif n > 0:
+        # Provided the number of bars
+        s = length / (n - 1)
+        # If we are distributing bars i the whole circle add a fictitious extra
+        # bar (than later will be removed).
+        n += add_n
+    elif s > 0:
+        # Provided the spacing
+        # 1. Compute the number of bars
+        n = math.floor(length / s) + 1
+        # 2. Distribute the bars centered in the curvilinear segment
+        needed_length = (n - 1) * s
+        delta_angle = (length - needed_length) / 2.0 / radius
+        # set whole to False bacause in this case we don't need to deal with
+        # the special case
+        whole = False
+    else:
+        raise ValueError('At least n or s should be provided')
+
+    phi_rebars = np.linspace(
+        start_angle + delta_angle, stop_angle - delta_angle, n
+    )
+    if whole:
+        n -= 1
+        phi_rebars = phi_rebars[:-1]
+
+    x = center[0] + radius * np.cos(phi_rebars)
+    y = center[1] + radius * np.sin(phi_rebars)
+
+    # add the bars
+    for i in range(n):
+        if i == 0 and not first:
+            continue
+        if i == n - 1 and not last:
+            continue
+        geo = add_reinforcement(
+            geo,
+            (x[i], y[i]),
+            diameter,
+            material,
+            group_label=group_label,
+        )
+    return geo

structuralcodes/materials/constitutive_laws/__init__.py

@@ -0,0 +1,84 @@
+"""Constitutive laws for materials."""
+
+import typing as t
+
+from ...core.base import ConstitutiveLaw, Material
+from ._bilinearcompression import BilinearCompression
+from ._elastic import Elastic
+from ._elasticplastic import ElasticPlastic
+from ._parabolarectangle import ParabolaRectangle
+from ._popovics import Popovics
+from ._sargin import Sargin
+from ._userdefined import UserDefined
+
+__all__ = [
+    'Elastic',
+    'ElasticPlastic',
+    'ParabolaRectangle',
+    'BilinearCompression',
+    'Popovics',
+    'Sargin',
+    'UserDefined',
+    'get_constitutive_laws_list',
+    'create_constitutive_law',
+]
+
+CONSTITUTIVE_LAWS: t.Dict[str, ConstitutiveLaw] = {
+    'elastic': Elastic,
+    'elasticplastic': ElasticPlastic,
+    'elasticperfectlyplastic': ElasticPlastic,
+    'bilinearcompression': BilinearCompression,
+    'parabolarectangle': ParabolaRectangle,
+    'popovics': Popovics,
+    'sargin': Sargin,
+}
+
+
+def get_constitutive_laws_list() -> t.List[str]:
+    """Returns a list with valid keywords for constitutive law factory."""
+    return list(CONSTITUTIVE_LAWS.keys())
+
+
+def create_constitutive_law(
+    constitutive_law_name: str, material: Material
+) -> ConstitutiveLaw:
+    """A factory function to create the constitutive law.
+
+    Arguments:
+        constitutive_law_name (str): A string defining a valid constitutive law
+            type. The available keys can be get with the method
+            `get_constitutive_laws_list`.
+        material (Material): The material containing the properties needed for
+            the definition of the constitutive law.
+
+    Note:
+        For working with this facotry function, the material class
+        implementations need to provide special dunder methods (__elastic__,
+        __parabolarectangle__, etc.) needed for the specific material that
+        return the kwargs needed to create the corresponding constitutive
+        law object. If the special dunder method is not found an exception
+        will be raised.
+
+        If the consitutive law selected is not available for the specific
+        material, an exception will be raised.
+    """
+    law = None
+    const_law = CONSTITUTIVE_LAWS.get(constitutive_law_name.lower())
+    if const_law is not None:
+        method_name = f'__{constitutive_law_name}__'
+        # check if the material object has the special method needed
+        if hasattr(material, method_name):
+            method = getattr(material, method_name)
+            if callable(method):
+                # get the kwargs from the special dunder method
+                kwargs = method()
+                # create the constitutive law
+                law = const_law(**kwargs)
+        else:
+            raise ValueError(
+                f'Constitutive law {constitutive_law_name} not available for'
+                f' material {material.__class__.__name__}'
+            )
+    else:
+        raise ValueError(f'Unknown constitutive law: {constitutive_law_name}')
+    return law

structuralcodes/materials/constitutive_laws/_bilinearcompression.py

@@ -0,0 +1,183 @@
+"""Bilinear compression constitutive law."""
+
+from __future__ import annotations  # To have clean hints of ArrayLike in docs
+
+import typing as t
+
+import numpy as np
+from numpy.typing import ArrayLike
+
+from ...core.base import ConstitutiveLaw
+
+
+class BilinearCompression(ConstitutiveLaw):
+    """Class for Bilinear Elastic-PerfectlyPlastic Constitutive Law for
+    Concrete (only compression behavior).
+    """
+
+    __materials__: t.Tuple[str] = ('concrete',)
+
+    def __init__(
+        self,
+        fc: float,
+        eps_c: float,
+        eps_cu: t.Optional[float] = None,
+        name: t.Optional[str] = None,
+    ) -> None:
+        """Initialize a BilinearCompression Material.
+
+        Arguments:
+            fc (float): Compressive strength (negative number).
+            eps_c (float): Strain at compressive strength (pure number).
+
+        Keyword Arguments:
+            eps_cu (float): Ultimate strain (pure number).
+            name (str): A descriptive name for the constitutive law.
+        """
+        name = name if name is not None else 'BilinearCompressionLaw'
+        super().__init__(name=name)
+        self._fc = -abs(fc)
+        self._eps_c = -abs(eps_c)
+        self._eps_cu = -abs(eps_cu)
+        self._E = self._fc / self._eps_c
+
+    def get_stress(
+        self, eps: t.Union[float, ArrayLike]
+    ) -> t.Union[float, ArrayLike]:
+        """Return the stress given strain."""
+        eps = eps if np.isscalar(eps) else np.atleast_1d(eps)
+        # Preprocess eps array in order
+        eps = self.preprocess_strains_with_limits(eps=eps)
+        # Compute stress
+        # If it is a scalar
+        if np.isscalar(eps):
+            sig = 0
+            if self._fc / self._E <= eps <= 0:
+                sig = self._E * eps
+            return sig
+        # If it is an array
+        sig = self._E * eps
+        sig[sig < self._fc] = self._fc
+        sig[eps > 0] = 0
+        sig[eps < self._eps_cu] = 0
+        return sig
+
+    def get_tangent(
+        self, eps: t.Union[float, ArrayLike]
+    ) -> t.Union[float, ArrayLike]:
+        """Return the tangent for given strain."""
+        eps = eps if np.isscalar(eps) else np.atleast_1d(eps)
+        # If it is a scalar
+        if np.isscalar(eps):
+            tangent = 0
+            if self._fc / self._E <= eps <= 0:
+                tangent = self._E
+            return tangent
+        # If it is an array
+        tangent = np.ones_like(eps) * self._E
+        tangent[eps < self._eps_c] = 0.0
+
+        return tangent
+
+    def __marin__(
+        self, strain: t.Tuple[float, float]
+    ) -> t.Tuple[t.List[t.Tuple], t.List[t.Tuple]]:
+        """Returns coefficients and strain limits for Marin integration in a
+        simply formatted way.
+
+        Arguments:
+            strain (float, float): Tuple defining the strain profile: eps =
+                strain[0] + strain[1]*y.
+
+        Example:
+            [(0, -0.002), (-0.002, -0.003)]
+            [(a0, a1, a2), (a0)]
+        """
+        strains = []
+        coeff = []
+        if strain[1] == 0:
+            # Uniform strain equal to strain[0]
+            # understand in which branch we are
+            strain[0] = self.preprocess_strains_with_limits(strain[0])
+            if strain[0] > 0:
+                # We are in tensile branch
+                strains = None
+                coeff.append((0.0,))
+            elif strain[0] > self._eps_c:
+                # We are in the linear branch
+                strains = None
+                a0 = self._E * strain[0]
+                a1 = self._E * strain[1]
+                coeff.append((a0, a1))
+            elif strain[0] >= self._eps_cu:
+                # We are in the constant branch
+                strains = None
+                coeff.append((self._fc,))
+            else:
+                # We are in a branch of non-resisting concrete
+                # Too much compression
+                strains = None
+                coeff.append((0.0,))
+        else:
+            # linear part
+            strains.append((self._eps_c, 0))
+            a0 = self._E * strain[0]
+            a1 = self._E * strain[1]
+            coeff.append((a0, a1))
+            # Constant part
+            strains.append((self._eps_cu, self._eps_c))
+            coeff.append((self._fc,))
+        return strains, coeff
+
+    def __marin_tangent__(
+        self, strain: t.Tuple[float, float]
+    ) -> t.Tuple[t.List[t.Tuple], t.List[t.Tuple]]:
+        """Returns coefficients and strain limits for Marin integration of
+        tangent in a simply formatted way.
+
+        Arguments:
+            strain (float, float): Tuple defining the strain profile: eps =
+                strain[0] + strain[1]*y.
+
+        Example:
+            [(0, -0.002), (-0.002, -0.003)]
+            [(a0, a1, a2), (a0)]
+        """
+        strains = []
+        coeff = []
+        if strain[1] == 0:
+            # Uniform strain equal to strain[0]
+            # understand in which branch we are
+            strain[0] = self.preprocess_strains_with_limits(strain[0])
+            if strain[0] > 0:
+                # We are in tensile branch
+                strains = None
+                coeff.append((0.0,))
+            elif strain[0] > self._eps_c:
+                # We are in the linear branch
+                strains = None
+                a0 = self._E
+                coeff.append((a0,))
+            else:
+                # We are in the constant branch or
+                # We are in a branch of non-resisting concrete
+                # Too much compression
+                strains = None
+                coeff.append((0.0,))
+        else:
+            # linear part
+            strains.append((self._eps_c, 0))
+            a0 = self._E
+            coeff.append((a0,))
+            # Constant part
+            strains.append((self._eps_cu, self._eps_c))
+            coeff.append((0.0,))
+        return strains, coeff
+
+    def get_ultimate_strain(
+        self, yielding: bool = False
+    ) -> t.Tuple[float, float]:
+        """Return the ultimate strain (negative and positive)."""
+        if yielding:
+            return (self._eps_c, 100)
+        return (self._eps_cu, 100)

structuralcodes/materials/constitutive_laws/_elastic.py

@@ -0,0 +1,133 @@
+"""Elastic constitutive law."""
+
+from __future__ import annotations  # To have clean hints of ArrayLike in docs
+
+import typing as t
+
+import numpy as np
+from numpy.typing import ArrayLike
+
+from ...core.base import ConstitutiveLaw
+
+
+class Elastic(ConstitutiveLaw):
+    """Class for elastic constitutive law."""
+
+    __materials__: t.Tuple[str] = (
+        'concrete',
+        'steel',
+        'rebars',
+    )
+
+    def __init__(self, E: float, name: t.Optional[str] = None) -> None:
+        """Initialize an Elastic Material.
+
+        Arguments:
+            E (float): The elastic modulus.
+
+        Keyword Arguments:
+            name (str): A descriptive name for the constitutive law.
+        """
+        name = name if name is not None else 'ElasticLaw'
+        super().__init__(name=name)
+        self._E = E
+        self._eps_su = None
+
+    def get_stress(
+        self, eps: t.Union[float, ArrayLike]
+    ) -> t.Union[float, ArrayLike]:
+        """Return stress given strain."""
+        eps = eps if np.isscalar(eps) else np.atleast_1d(eps)
+        return self._E * eps
+
+    def get_tangent(
+        self, eps: t.Union[float, ArrayLike]
+    ) -> t.Union[float, ArrayLike]:
+        """Return the tangent."""
+        if np.isscalar(eps):
+            return self._E
+        eps = np.atleast_1d(eps)
+        return np.ones_like(eps) * self._E
+
+    def __marin__(
+        self, strain: t.Tuple[float, float]
+    ) -> t.Tuple[t.List[t.Tuple], t.List[t.Tuple]]:
+        """Returns coefficients and strain limits for Marin integration in a
+        simply formatted way.
+
+        Arguments:
+            strain (float, float): Tuple defining the strain profile: eps =
+                strain[0] + strain[1]*y.
+
+        Example:
+            [(0, -0.002), (-0.002, -0.003)]
+            [(a0, a1, a2), (a0)]
+        """
+        strains = None
+        a0 = self._E * strain[0]
+        a1 = self._E * strain[1]
+        coeff = [(a0, a1)]
+        return strains, coeff
+
+    def __marin_tangent__(
+        self, strain: t.Tuple[float, float]
+    ) -> t.Tuple[t.List[t.Tuple], t.List[t.Tuple]]:
+        """Returns coefficients and strain limits for Marin integration of
+        tangent in a simply formatted way.
+
+        Arguments:
+            strain (float, float): Tuple defining the strain profile: eps =
+                strain[0] + strain[1]*y.
+
+        Example:
+            [(0, -0.002), (-0.002, -0.003)]
+            [(a0, a1, a2), (a0)]
+        """
+        strains = None
+        a0 = self._E
+        coeff = [(a0,)]
+        return strains, coeff
+
+    def get_ultimate_strain(self, **kwargs) -> t.Tuple[float, float]:
+        """Return the ultimate strain (negative and positive)."""
+        # There is no real strain limit, so set it to very large values
+        # unlesse specified by the user differently
+        del kwargs
+        return self._eps_su or (-100, 100)
+
+    def set_ultimate_strain(
+        self, eps_su=t.Union[float, t.Tuple[float, float]]
+    ) -> None:
+        """Set ultimate strains for Elastic Material if needed.
+
+        Arguments:
+            eps_su (float or (float, float)): Defining ultimate strain. If a
+                single value is provided the same is adopted for both negative
+                and positive strains. If a tuple is provided, it should be
+                given as (negative, positive).
+        """
+        if isinstance(eps_su, float):
+            self._eps_su = (-abs(eps_su), abs(eps_su))
+        elif isinstance(eps_su, tuple):
+            if len(eps_su) < 2:
+                raise ValueError(
+                    'Two values need to be provided when setting the tuple'
+                )
+            eps_su_n = eps_su[0]
+            eps_su_p = eps_su[1]
+            if eps_su_p < eps_su_n:
+                eps_su_p, eps_su_n = eps_su_n, eps_su_p
+            if eps_su_p < 0:
+                raise ValueError(
+                    'Positive ultimate strain should be non-negative'
+                )
+            if eps_su_n > 0:
+                raise ValueError(
+                    'Negative utimate strain should be non-positive'
+                )
+            self._eps_su = (eps_su_n, eps_su_p)
+        else:
+            raise ValueError(
+                'set_ultimate_strain requires a single value or a tuple \
+                with  two values'
+            )