diff --git a/nhf/touhou/shiki_eiki/rod.py b/nhf/touhou/shiki_eiki/rod.py index b53141d..fc2c541 100644 --- a/nhf/touhou/shiki_eiki/rod.py +++ b/nhf/touhou/shiki_eiki/rod.py @@ -7,16 +7,21 @@ import nhf.utils @dataclass class Rod(Model): + width: float = 120.0 + length: float = 550.0 + length_tip: float = 100.0 + width_tail: float = 60.0 + @target(name="surface") def top_profile(self) -> Cq.Sketch: - w, h = 10, 20 sketch = ( Cq.Sketch() .polygon([ - (w, h), - (w, -h), - (-w, -h), - (-w, h), + (self.length, 0), + (self.length - self.length_tip, self.width/2), + (0, self.width_tail / 2), + (0, -self.width_tail / 2), + (self.length - self.length_tip, -self.width/2), ]) ) return sketch