"""Approximate long multi-collar shaft inferred from reference 001957.png."""

from shaft_common import cut_end_bore, cut_side_flat, export_horizontal, stepped_body


SEGMENTS = [
    (10.0, 34.0),   # reduced drive tip
    (14.0, 218.0),  # long shaft
    (19.0, 18.0),   # transition shoulder
    (30.0, 4.0),
    (25.0, 7.0),
    (32.0, 5.0),
    (25.0, 8.0),
    (31.0, 5.0),
    (23.0, 8.0),
    (14.0, 30.0),   # hollow output nose
]


def gen_step():
    body, total = stepped_body(SEGMENTS)
    body = cut_side_flat(body, diameter=10.0, z_start=27.0, length=7.0, flat_depth=1.4)
    body = cut_end_bore(body, total, diameter=8.0, depth=18.0)
    return export_horizontal(body, "shaft_001957_like")