private double Tasklet(AscentSimulationControl currentBestGuess, double currentBestScore, double perturbationScale, bool print,
out AscentSimulationControl control,
out LookAheadState lookAheadState,
out State state)
{
control = new AscentSimulationControl()
{
InitialVerticalTime = Math.Max(5, currentBestGuess.InitialVerticalTime + (perturbationScale * _random.NextDouble(1))),
KickPitchTime = Math.Max(0, currentBestGuess.KickPitchTime + (perturbationScale * _random.NextDouble(4))),
KickPitchAngle = currentBestGuess.KickPitchAngle, //Math.Max(0, currentBestGuess.KickPitchAngle + (perturbationScale * _random.NextDouble(1))),
StagingAngle = Math.Max(0, Math.Min(90, currentBestGuess.StagingAngle + (perturbationScale * _random.NextDouble(2)))),
MaxAcceleration = Math.Max(0, currentBestGuess.MaxAcceleration + (perturbationScale * _random.NextDouble(0.1))),
Rocket = currentBestGuess.Rocket
//Stage1Duration = currentBestGuess.Stage1Duration, // + (perturbationScale * _random.NextDouble(3)),
//Stage2Duration = currentBestGuess.Stage2Duration, // + (perturbationScale * _random.NextDouble(3)),
//Stage1InitialAcceleration = currentBestGuess.Stage1InitialAcceleration,
//Stage1MaxAcceleration = currentBestGuess.Stage1MaxAcceleration,
//Stage2InitialAcceleration = currentBestGuess.Stage2InitialAcceleration,
//Stage2MaxAcceleration = currentBestGuess.Stage2MaxAcceleration
//InitialTurnDuration = currentBestGuess.InitialTurnDuration + (perturbationScale * _random.NextDouble(0.2)),
//MaxThrust1 = currentBestGuess.MaxThrust1,
//MinThrust1 = currentBestGuess.MinThrust1,
//MaxThrust2 = currentBestGuess.MaxThrust2,
//MinThrust2 = currentBestGuess.MinThrust2,
//ThrustCutoff = 1000, // Math.Max(0, currentBestGuess.ThrustCutoff + (perturbationScale * _random.NextDouble(5))),
////ThrustCurve = Math.Max(0, currentBestGuess.ThrustCurve + (perturbationScale * _random.NextDouble(5))),
//ThrustDuration1 = Math.Max(0, currentBestGuess.ThrustDuration1 + (perturbationScale * _random.NextDouble(5))),
//ThrustDuration2 = Math.Max(0, currentBestGuess.ThrustDuration2 + (perturbationScale * _random.NextDouble(5))),
//TurnDelay = currentBestGuess.TurnDelay, //Math.Max(0, currentBestGuess.TurnDelay + (perturbationScale * _random.NextDouble(1))),
//TurnDuration = Math.Max(0, currentBestGuess.TurnDuration + (perturbationScale * _random.NextDouble(1))),
};
var simulation = new AscentSimulation(Goal, control, InitialState, SurfaceVelocity);
double totalFuelMass = control.Rocket.TotalFuelMass();
while (!simulation.CurrentState.IsDone)
{
//Console.WriteLine("{0} {1}", simulation.CurrentState.ExpendedMass - totalFuelMass, simulation.CurrentState.Time);
simulation.FastTick(TimeStep, MicroSteps);
if (simulation.LookAheadState.Apoapsis / 1000.0 > (Goal.Apoapsis + 15) || simulation.LookAheadState.Periapsis / 1000.0 > (Goal.Periapsis + 15))
{
//control.Stage2Duration = Math.Max(0, simulation.CurrentState.Time - control.Stage1Duration);
break;
}
}
if (print)
{
Print(Goal, simulation.LookAheadState, simulation.CurrentState, control, currentBestScore);
}
lookAheadState = simulation.LookAheadState;
state = simulation.CurrentState;
return(ComputeScore(Goal, simulation.LookAheadState, state, control));
}
public AscentOptimization(double surfaceVelocity, AscentSimulationGoal goal, AscentSimulationControl initialGuess, State initialState, double timeStep, int microSteps, double bestScore = -1)
{
Goal = goal;
InitialGuess = initialGuess;
BestGuess = initialGuess;
InitialState = initialState;
TimeStep = timeStep;
MicroSteps = microSteps;
SurfaceVelocity = surfaceVelocity;
LookAheadState lookAheadState;
State state;
BestScore = bestScore >= 0 ? bestScore : Tasklet(BestGuess, 0, 0.0, false, out BestGuess, out lookAheadState, out state);
}
public AscentSimulation(
AscentSimulationGoal goal,
AscentSimulationControl controlInput,
State initialState,
double windSurfaceVelocity = 0.0
)
{
Goal = goal;
ControlInput = controlInput;
States.Add(initialState);
_lastState = initialState;
_currentState = initialState;
InitialState = initialState;
WindSurfaceVelocity = windSurfaceVelocity;
for (var i = 0; i < LookAheadState.FuturePositions.Length; i++)
{
LookAheadState.FuturePositions[i] = initialState.Position;
}
}
private static void Print(AscentSimulationGoal goal, LookAheadState lookAheadState, State state, AscentSimulationControl guess, double bestScore)
{
Console.WriteLine("{0,1:F} {1,1:F}, {2,2:F}/{3,2:F}: g {4,1:F}, d {5,1:F}, mass left {6,2:F} kg - score {7,3:F}/{8,3:F}",
(lookAheadState.Periapsis) / 1000.0,
(lookAheadState.Apoapsis) / 1000.0,
(state.Position.Length - Constants.EarthRadius) / 1000.0,
state.ReachedAltitude / 1000.0,
state.LossesToGravity,
state.LossesToDrag,
guess.Rocket.TotalFuelMass() - state.ExpendedMass,
ComputeScore(goal, lookAheadState, state, guess),
bestScore
);
}
public static double ComputeScore(AscentSimulationGoal goal, LookAheadState lookAheadState, State state, AscentSimulationControl guess)
{
const double distanceScaling = 10.0;
double distanceFromApoapsisGoal = lookAheadState.Apoapsis / 1000.0 - goal.Apoapsis;
double distanceFromPeriapsisGoal = lookAheadState.Periapsis / 1000.0 - goal.Periapsis;
double distanceFromReachedAltitudeGoal = state.ReachedAltitude / 1000.0 - goal.Periapsis;
double distanceFromAltitudeGoal = (state.Position.Length - Constants.EarthRadius) / 1000.0 - goal.Periapsis;
double totalFuel = guess.Rocket.TotalFuelMass();
double fuelLeft = (totalFuel - state.ExpendedMass) / totalFuel;
return(ErrorFunction(distanceFromApoapsisGoal, distanceScaling)
+ ErrorFunction(distanceFromPeriapsisGoal, distanceScaling)
+ ErrorFunction(distanceFromReachedAltitudeGoal, distanceScaling)
+ ErrorFunction(distanceFromAltitudeGoal, distanceScaling)
+ fuelLeft * 5);
//+ ErrorFunction((guess.ThrustDuration1 + guess.ThrustDuration2) - guess.ThrustCutoff, 1);
}
