aniva
/
Cosplay
1
1
Fork 0
Code Issues 1 Pull Requests 2 Packages Projects Releases Wiki Activity

feat: Calculation of total mass

This commit is contained in:
Leni Aniva 2024-07-17 19:13:06 -07:00
parent 9de4159166
commit 014784be34
Signed by: aniva
GPG Key ID: 4D9B1C8D10EA4C50
2 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
nhf/materials.py
View File

@ -54,16 +54,18 @@ ROLE_COLOR_MAP = {
class Material(Enum): class Material(Enum):
""" """
A catalog of common material properties A catalog of common material properties
Density listed is in g/cm^3 or mg/mm^3
""" """
WOOD_BIRCH = 0.8, _color('bisque', 0.9) WOOD_BIRCH = 0.71, _color('bisque', 0.9)
PLASTIC_PLA = 0.5, _color('mistyrose', 0.8) PLASTIC_PLA = 1.2, _color('mistyrose', 0.8)
RESIN_TRANSPERENT = 1.1, _color('cadetblue2', 0.6) RESIN_TRANSPERENT = 1.17, _color('cadetblue2', 0.6)
RESIN_TOUGH_1500 = 1.1, _color('seashell3', 0.7) RESIN_TOUGH_1500 = 1.17, _color('seashell3', 0.7)
ACRYLIC_BLACK = 0.5, _color('gray5', 0.8) ACRYLIC_BLACK = 1.18, _color('gray5', 0.8)
ACRYLIC_TRANSLUSCENT = 0.5, _color('ivory2', 0.8) ACRYLIC_TRANSLUSCENT = 1.18, _color('ivory2', 0.8)
ACRYLIC_TRANSPARENT = 0.5, _color('ghostwhite', 0.5) ACRYLIC_TRANSPARENT = 1.18, _color('ghostwhite', 0.5)
STEEL_SPRING = 1.0, _color('gray', 0.8) STEEL_SPRING = 7.8, _color('gray', 0.8)
def __init__(self, density: float, color: Cq.Color): def __init__(self, density: float, color: Cq.Color):
self.density = density self.density = density
@ -110,3 +112,16 @@ def color_by_role(self: Cq.Assembly, avg: bool = True) -> Cq.Assembly:
a.color = role.color_avg() if avg else role.color_head() a.color = role.color_avg() if avg else role.color_head()
return self return self
Cq.Assembly.color_by_role = color_by_role Cq.Assembly.color_by_role = color_by_role
def total_mass(self: Cq.Assembly) -> float:
"""
Calculates the total mass in units of g
"""
total = 0.0
for _, a in self.traverse():
if 'material' not in a.metadata:
continue
vol = a.toCompound().Volume()
total += vol * a.metadata['material'].density
return total / 1000.0
Cq.Assembly.total_mass = total_mass

12
nhf/touhou/houjuu_nue/__init__.py
View File

@ -146,11 +146,19 @@ class Parameters(Model):
bbox = a.toCompound().BoundingBox() bbox = a.toCompound().BoundingBox()
return { return {
"wing-span": bbox.xlen, "wing-span": bbox.xlen,
"wing-thickness": bbox.ylen, "wing-depth": bbox.ylen,
"wing-height": bbox.zlen, "wing-height": bbox.zlen,
"wing-mass": a.total_mass(),
} }
if __name__ == '__main__': if __name__ == '__main__':
import sys
p = Parameters() p = Parameters()
p.build_all() if len(sys.argv) == 1:
p.build_all()
sys.exit(0)
if sys.argv[1] == 'stat':
print(p.stat())