Flow encoding

In [ ]:

from timeit import default_timer as timer

import matplotlib.pyplot as plt
import numpy as np

import gropt

from timeit import default_timer as timer import matplotlib.pyplot as plt import numpy as np import gropt

In [ ]:

gparams = gropt.GroptParams()
gparams.N = 133
gparams.dt = 10e-6

gparams.add_gmax(.08)
gparams.add_smax(200)
gparams.add_moment(0,0.0)
gparams.add_moment(1,14.0)

start_t = timer()
result = gropt.solve(gparams)
out = result.X
stop_t = timer()

print(f'Converged: {result.converged}  Solve time: {1000*(stop_t-start_t):.1f} ms')

plt.figure(figsize=(6,3))
tt_ms = np.arange(out.size)*gparams.dt*1e3
plt.plot(tt_ms, out)
plt.show()

gparams = gropt.GroptParams() gparams.N = 133 gparams.dt = 10e-6 gparams.add_gmax(.08) gparams.add_smax(200) gparams.add_moment(0,0.0) gparams.add_moment(1,14.0) start_t = timer() result = gropt.solve(gparams) out = result.X stop_t = timer() print(f'Converged: {result.converged} Solve time: {1000*(stop_t-start_t):.1f} ms') plt.figure(figsize=(6,3)) tt_ms = np.arange(out.size)*gparams.dt*1e3 plt.plot(tt_ms, out) plt.show()