| | | 1 | | using UnityEngine; |
| | | 2 | | |
| | | 3 | | // Launch angle input. |
| | | 4 | | public struct LaunchAngleInput { |
| | | 5 | | // Horizontal distance in meters. |
| | 43 | 6 | | public float Distance { get; } |
| | | 7 | | |
| | | 8 | | // Altitude in meters. |
| | 43 | 9 | | public float Altitude { get; } |
| | | 10 | | |
| | 43 | 11 | | public LaunchAngleInput(float distance, float altitude) { |
| | 43 | 12 | | Distance = distance; |
| | 43 | 13 | | Altitude = altitude; |
| | 43 | 14 | | } |
| | | 15 | | } |
| | | 16 | | |
| | | 17 | | // Launch angle output. |
| | | 18 | | public struct LaunchAngleOutput { |
| | | 19 | | // Launch angle in degrees. |
| | | 20 | | public float LaunchAngle { get; } |
| | | 21 | | |
| | | 22 | | // Time to reach the target position in seconds. |
| | | 23 | | public float TimeToPosition { get; } |
| | | 24 | | |
| | | 25 | | public LaunchAngleOutput(float launchAngle, float timeToPosition) { |
| | | 26 | | LaunchAngle = launchAngle; |
| | | 27 | | TimeToPosition = timeToPosition; |
| | | 28 | | } |
| | | 29 | | } |
| | | 30 | | |
| | | 31 | | // Launch angle data point. |
| | | 32 | | public struct LaunchAngleDataPoint { |
| | | 33 | | // Launch angle input. |
| | | 34 | | public LaunchAngleInput Input { get; } |
| | | 35 | | |
| | | 36 | | // Launch angle output. |
| | | 37 | | public LaunchAngleOutput Output { get; } |
| | | 38 | | |
| | | 39 | | public LaunchAngleDataPoint(in LaunchAngleInput input, in LaunchAngleOutput output) { |
| | | 40 | | Input = input; |
| | | 41 | | Output = output; |
| | | 42 | | } |
| | | 43 | | } |
| | | 44 | | |
| | | 45 | | // The launch angle planner class is an interface for a planner that outputs the optimal launch |
| | | 46 | | // angle and the time-to-target. |
| | | 47 | | public interface ILaunchAnglePlanner { |
| | | 48 | | // Calculate the optimal launch angle in degrees and the time-to-target in seconds. |
| | | 49 | | public LaunchAngleOutput Plan(in LaunchAngleInput input); |
| | | 50 | | public LaunchAngleOutput Plan(float distance, float altitude) { |
| | | 51 | | return Plan(new LaunchAngleInput(distance, altitude)); |
| | | 52 | | } |
| | | 53 | | public LaunchAngleOutput Plan(Vector3 position) { |
| | | 54 | | Vector2 direction = ConvertToDirection(position); |
| | | 55 | | return Plan(distance: direction[0], altitude: direction[1]); |
| | | 56 | | } |
| | | 57 | | |
| | | 58 | | // Get the intercept position. |
| | | 59 | | public LaunchAngleInput GetInterceptPosition(in LaunchAngleInput input); |
| | | 60 | | public LaunchAngleInput GetInterceptPosition(float distance, float altitude) { |
| | | 61 | | return GetInterceptPosition(new LaunchAngleInput(distance, altitude)); |
| | | 62 | | } |
| | | 63 | | public LaunchAngleInput GetInterceptPosition(Vector3 position) { |
| | | 64 | | Vector2 direction = ConvertToDirection(position); |
| | | 65 | | return GetInterceptPosition(distance: direction[0], altitude: direction[1]); |
| | | 66 | | } |
| | | 67 | | |
| | | 68 | | // Convert from a 3D vector to a 2D direction that ignores the azimuth. |
| | | 69 | | public static Vector2 ConvertToDirection(Vector3 position) { |
| | | 70 | | return new Vector2(Vector3.ProjectOnPlane(position, Vector3.up).magnitude, |
| | | 71 | | Vector3.Project(position, Vector3.up).magnitude); |
| | | 72 | | } |
| | | 73 | | } |