mathjs/examples/browser/rocket_trajectory_optimization.html
Jos de Jong 6f00715754
Specify import require paths (continuation of #1941) (#1962)
* Add `.js` extension to source file imports

* Specify package `exports` in `package.json`

Specify package type as `commonjs` (It's good to be specific)

* Move all compiled scripts into `lib` directory

Remove ./number.js (You can use the compiled ones in `./lib/*`)

Tell node that the `esm` directory is type `module` and enable tree shaking.

Remove unused files from packages `files` property

* Allow importing of package.json

* Make library ESM first

* - Fix merge conflicts
- Refactor `bundleAny` into `defaultInstance.js` and `browserBundle.cjs`
- Refactor unit tests to be able to run with plain nodejs (no transpiling)
- Fix browser examples

* Fix browser and browserstack tests

* Fix running unit tests on Node 10 (which has no support for modules)

* Fix node.js examples (those are still commonjs)

* Remove the need for `browserBundle.cjs`

* Generate minified bundle only

* [Security] Bump node-fetch from 2.6.0 to 2.6.1 (#1963)

Bumps [node-fetch](https://github.com/bitinn/node-fetch) from 2.6.0 to 2.6.1. **This update includes a security fix.**
- [Release notes](https://github.com/bitinn/node-fetch/releases)
- [Changelog](https://github.com/node-fetch/node-fetch/blob/master/docs/CHANGELOG.md)
- [Commits](https://github.com/bitinn/node-fetch/compare/v2.6.0...v2.6.1)

Signed-off-by: dependabot-preview[bot] <support@dependabot.com>

Co-authored-by: dependabot-preview[bot] <27856297+dependabot-preview[bot]@users.noreply.github.com>

* Cleanup console.log

* Add integration tests to test the entry points (commonjs/esm, full/number only)

* Create backward compatibility error messages in the files moved/removed since v8

* Describe breaking changes in HISTORY.md

* Bump karma from 5.2.1 to 5.2.2 (#1965)

Bumps [karma](https://github.com/karma-runner/karma) from 5.2.1 to 5.2.2.
- [Release notes](https://github.com/karma-runner/karma/releases)
- [Changelog](https://github.com/karma-runner/karma/blob/master/CHANGELOG.md)
- [Commits](https://github.com/karma-runner/karma/compare/v5.2.1...v5.2.2)

Signed-off-by: dependabot-preview[bot] <support@dependabot.com>

Co-authored-by: dependabot-preview[bot] <27856297+dependabot-preview[bot]@users.noreply.github.com>

Co-authored-by: Lee Langley-Rees <lee@greenimp.co.uk>
Co-authored-by: dependabot-preview[bot] <27856297+dependabot-preview[bot]@users.noreply.github.com>
2020-09-20 18:01:29 +02:00

153 lines
6.3 KiB
HTML

<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
<title>math.js | rocket trajectory optimization</title>
<script src="../../lib/browser/math.js"></script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/Chart.js/2.5.0/Chart.min.js"></script>
<style>
body {
font-family: sans-serif;
}
</style>
</head>
<body>
<h1>Rocket trajectory optimization</h1>
<p>
This example simulates the ascent stage of the Apollo Lunar Module modeled using a system of ordinary differential equations.
</p>
<canvas id=canvas1 width=1600 height=400></canvas>
<canvas id=canvas2 width=800 height=400></canvas>
<script>
function ndsolve(f, x0, dt, tmax) {
const n = f.size()[0] // Number of variables
const x = x0.clone() // Current values of variables
const dxdt = [] // Temporary variable to hold time-derivatives
const result = [] // Contains entire solution
const nsteps = math.divide(tmax, dt) // Number of time steps
for(let i=0; i<nsteps; i++) {
// Compute derivatives
for(let j=0; j<n; j++) {
dxdt[j] = f.get([j]).apply(null, x.toArray())
}
// Euler method to compute next time step
for(let j=0; j<n; j++) {
x.set([j], math.add(x.get([j]), math.multiply(dxdt[j], dt)))
}
result.push(x.clone())
}
return math.matrix(result)
}
// Import the numerical ODE solver
math.import({ndsolve:ndsolve})
// Create a math.js context for our simulation. Everything else occurs in the context of the expression parser!
const sim = math.parser()
sim.evaluate("G = 6.67408e-11 m^3 kg^-1 s^-2") // Gravitational constant
sim.evaluate("mbody = 5.972e24 kg") // Mass of Earth
sim.evaluate("mu = G * mbody")
sim.evaluate("dt = 1.0 s") // Simulation timestep
sim.evaluate("tfinal = 162 s") // Simulation duration
sim.evaluate("T = 1710000 lbf * 0.9") // Engine thrust
sim.evaluate("g0 = 9.80665 m/s^2") // Standard gravity: used for calculating prop consumption (dmdt)
sim.evaluate("isp = 290 s") // Specific impulse
sim.evaluate("gamma0 = 89.99883 deg") // Initial pitch angle (90 deg is vertical)
sim.evaluate("r0 = 6378.1370 km") // Equatorial radius of Earth
sim.evaluate("v0 = 10 m/s") // Initial velocity (must be non-zero because ODE is ill-conditioned)
sim.evaluate("phi0 = 0 deg") // Initial orbital reference angle
sim.evaluate("m0 = 1207920 lbm + 30000 lbm") // Initial mass of rocket and fuel
// Define the equations of motion. It is important to maintain the same argument order for each of these functions.
sim.evaluate("drdt(r, v, m, phi, gamma) = v sin(gamma)")
sim.evaluate("dvdt(r, v, m, phi, gamma) = -mu / r^2 * sin(gamma) + T / m")
sim.evaluate("dmdt(r, v, m, phi, gamma) = -T/g0/isp")
sim.evaluate("dphidt(r, v, m, phi, gamma) = v/r * cos(gamma) * rad")
sim.evaluate("dgammadt(r, v, m, phi, gamma) = (1/r * (v - mu / (r v)) * cos(gamma)) * rad")
// Again, remember to maintain the same variable order in the call to ndsolve.
sim.evaluate("result_stage1 = ndsolve([drdt, dvdt, dmdt, dphidt, dgammadt], [r0, v0, m0, phi0, gamma0], dt, tfinal)")
// Reset initial conditions for interstage flight
sim.evaluate("T = 0 lbf")
sim.evaluate("tfinal = 12 s")
sim.evaluate("x = flatten(result_stage1[result_stage1.size()[1],:])")
sim.evaluate("result_interstage = ndsolve([drdt, dvdt, dmdt, dphidt, dgammadt], x, dt, tfinal)")
console.log(sim.evaluate("result_interstage[result_interstage.size()[1],3]").toString())
// Reset initial conditions for stage 2 flight
sim.evaluate("T = 210000 lbf")
sim.evaluate("isp = 348 s")
sim.evaluate("tfinal = 397 s")
sim.evaluate("x = flatten(result_interstage[result_interstage.size()[1],:])")
sim.evaluate("x[3] = 273600 lbm") // Lighten the rocket a bit since we discarded the first stage
sim.evaluate("result_stage2 = ndsolve([drdt, dvdt, dmdt, dphidt, dgammadt], x, dt, tfinal)")
// Reset initial conditions for unpowered flight
sim.evaluate("T = 0 lbf")
sim.evaluate("tfinal = 60 s")
sim.evaluate("x = flatten(result_stage2[result_stage2.size()[1],:])")
sim.evaluate("result_unpowered = ndsolve([drdt, dvdt, dmdt, dphidt, dgammadt], x, dt, tfinal)")
// Extract the useful information from the results so it can be plotted
const data_stage1 = sim.evaluate("transpose(concat( transpose( result_stage1[:,4] - phi0) * r0 / rad / km, ( transpose(result_stage1[:,1]) - r0) / km, 1 ))").toArray().map(function(e) { return {x: e[0], y: e[1]} })
const data_interstage = sim.evaluate("transpose(concat( transpose(result_interstage[:,4] - phi0) * r0 / rad / km, (transpose(result_interstage[:,1]) - r0) / km, 1 ))").toArray().map(function(e) { return {x: e[0], y: e[1]} })
const data_stage2 = sim.evaluate("transpose(concat( transpose( result_stage2[:,4] - phi0) * r0 / rad / km, ( transpose(result_stage2[:,1]) - r0) / km, 1 ))").toArray().map(function(e) { return {x: e[0], y: e[1]} })
const data_unpowered = sim.evaluate("transpose(concat( transpose( result_unpowered[:,4] - phi0) * r0 / rad / km, ( transpose(result_unpowered[:,1]) - r0) / km, 1 ))").toArray().map(function(e) { return {x: e[0], y: e[1]} })
window['chart'] = new Chart(document.getElementById('canvas1'), {
type: 'line',
data: {
datasets: [{
label: "Stage 1",
data: data_stage1,
fill: false,
borderColor: "red",
pointRadius: 0
}, {
label: "Interstage",
data: data_interstage,
fill: false,
borderColor: "green",
pointRadius: 0
}, {
label: "Stage 2",
data: data_stage2,
fill: false,
borderColor: "orange",
pointRadius: 0
}, {
label: "Unpowered",
data: data_unpowered,
fill: false,
borderColor: "blue",
pointRadius: 0
}]
},
options: {
scales: {
xAxes: [{
type: 'linear',
position: 'bottom'
}]
}
}
})
</script>
</body>
</html>