An apparatus for treatment planning and/or treatment setup includes: a simulator 200 configured to obtain a first model representing a first component of a medical system, and virtually move the first model to simulate a movement of the first component of the medical system; an analyzer 204 configured to determine imaging waypoints during a treatment process based on the virtual movement of the first model; and a graphic generator 206configured to generate a graphic based on the determined imaging waypoints.