from typing import Union
# The element class Tag in OpenSees, which is used to determine the element type
ELE_TAG_Truss = (12, 13, 14, 15, 16, 17, 18, 169) # 169 is CatenaryCable
ELE_TAG_Link = (19, 20, 21, 22, 23, 24, 25, 26, 260, 27, # zeroLength
86, # 86-twoNodeLink
84, 85, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104,
105, 106, 107, 108, 109, 130, 131, 132, 147, 148, 149, 150, 151, 152, 153,
158, 159, 160, 161, 165, 166, ) # Bearing
ELE_TAG_Beam = (3, 4, 5, 5001, 6, 7, 8, 9, 10, 11, 28, 29, 30, 34, 35, 62, 621, 63, 64, 640,
641, 642, 65, 66, 67, 68, 69, 70, 71, 72, 73, 731, 74, 75, 751, 76, 77, 78,
30766, 30765, 30767, 79, 128)
ELE_TAG_Plane = (31, 32, 33, 40, 47, 50, 52, 53, 54, 55, 59, 61, 119, 120, 126, 134, 156, 157,
167, 168, 173, 174, 175, 203, 204, 207, 208, 209)
ELE_TAG_Joint = (71, 72, 81, 8181, 82, 83)
ELE_TAG_Tetrahedron = (179,)
ELE_TAG_Brick = (36, 37, 38, 39, 41, 42, 43, 44, 45, 46,
48, 49, 51, 56, 57, 58, 121, 122, 127)
# shape dict used to subplots
shape_dict = {1: (1, 1), 2: (1, 2), 3: (1, 3), 4: (2, 2), 5: (2, 3),
6: (2, 3), 7: (3, 3), 8: (3, 3), 9: (3, 3), 10: (3, 4),
11: (3, 4), 12: (3, 4), 13: (4, 4), 14: (4, 4), 15: (4, 4),
16: (4, 4), 17: (4, 5), 18: (4, 5), 19: (4, 5), 20: (4, 5),
21: (5, 5), 22: (5, 5), 23: (5, 5), 24: (5, 5), 25: (5, 5),
26: (5, 6), 27: (5, 6), 28: (5, 6), 29: (5, 6), 30: (5, 6),
31: (6, 6), 32: (6, 6), 33: (6, 6), 34: (6, 6), 35: (6, 6),
36: (6, 6), 37: (6, 7), 38: (6, 7), 39: (6, 7), 40: (6, 7),
41: (6, 7), 42: (6, 7), 43: (7, 7), 44: (7, 7), 45: (7, 7),
46: (7, 7), 47: (7, 7), 48: (7, 7), 49: (7, 7)}
def check_file(file_name: str, file_type: Union[str, list[str], tuple[str]]):
"""Check file type.
Parameters
----------
file_name: str
The file to be checked.
file_type: str
The target file type.
Returns
-------
None
"""
if file_name:
if isinstance(file_type, str):
if not file_name.endswith(file_type):
raise ValueError(f"file must be endswith {file_type}!")
elif isinstance(file_type, list) or isinstance(file_type, tuple):
check = False
for type_ in file_type:
if file_name.endswith(type_):
check = True
if not check:
raise ValueError(f"file must be endswith in {file_type}!")
else:
raise ValueError("file_type must be str or list or tuple!")
[docs]def load_ops_examples(name: str):
"""load the examples.
Parameters:
-----------
name: str,
Optional, "ArchBridge", "CableStayedBridge", "Dam", "Frame3D", "Igloo", "Pier", "SuspensionBridge", "SDOF".
Returns:
--------
None
"""
if name.lower() == "archbridge":
exec("from opstool.examples.ArchBridge import *")
elif name.lower() == "cablestayedbridge":
exec("from opstool.examples.CableStayedBridge import *")
elif name.lower() == "dam":
exec("from opstool.examples.Dam import *")
elif name.lower() == "frame3d":
exec("from opstool.examples.Frame3D import *")
elif name.lower() == "igloo":
exec("from opstool.examples.Igloo import *")
elif name.lower() == "pier":
exec("from opstool.examples.Pier import *")
elif name.lower() == "suspensionbridge":
exec("from opstool.examples.SuspensionBridge import *")
elif name.lower() == "sdof":
exec("from opstool.examples.SDOF import *")
else:
raise ValueError("not supported example {name}!")