def main():
global DT, Theta, dTheta, ddTheta, F
ddTheta_in = ddTheta # Backup for the predictor-corrector
# Predictor
Theta, dTheta = predictor(DT, Theta, dTheta, ddTheta)
# Get the new time and time step
t = aqua.get("t")
dt = aqua.get("dt")
# Get new mass position (and experimental resulting angle)
xi = np.interp(t, T, XI)
dXi = np.interp(t, T, DXI)
exp_theta = np.interp(t, T, THETA)
# Compute ddTheta
M = aqua.get("forces_M")[0]
ddTheta = angularForce(M, xi, dXi, Theta, dTheta)
# Corrector
dTheta = corrector(dt, dTheta, ddTheta, ddTheta_in)
DT = dt # Store it for the predictor in the following time step
# Send the data to SPH
a = np.zeros(4, dtype=np.float32)
a[0] = Theta
aqua.set("motion_a", a)
dadt = np.zeros(4, dtype=np.float32)
dadt[0] = dTheta
aqua.set("motion_dadt", dadt)
# Write output
F.write('{}\t{}\t{}\t{}\t{}\t{}\n'.format(
t, xi, exp_theta,
np.degrees(a[0]), np.degrees(dadt[0]),
M))
F.flush()
return True
def main():
global DT, Theta, dTheta, ddTheta, F
ddTheta_in = ddTheta # Backup for the predictor-corrector
# Predictor
Theta, dTheta = predictor(DT, Theta, dTheta, ddTheta)
# Get the new time and time step
t = aqua.get("t")
dt = aqua.get("dt")
# Get new mass position (and experimental resulting angle)
xi = np.interp(t, T, XI)
dXi = np.interp(t, T, DXI)
exp_theta = np.interp(t, T, THETA)
# Compute ddTheta
M = aqua.get("forces_M")[0]
ddTheta = angularForce(M, xi, dXi, Theta, dTheta)
# Corrector
dTheta = corrector(dt, dTheta, ddTheta, ddTheta_in)
DT = dt # Store it for the predictor in the following time step
# Send the data to SPH
a = np.zeros(4, dtype=np.float32)
a[0] = Theta
aqua.set("motion_a", a)
dadt = np.zeros(4, dtype=np.float32)
dadt[0] = dTheta
aqua.set("motion_dadt", dadt)
# Write output
F.write('{}\t{}\t{}\t{}\t{}\t{}\n'.format(t, xi, exp_theta,
np.degrees(a[0]),
np.degrees(dadt[0]), M))
F.flush()
return True
def main():
dt = aqua.get("dt")
for e_name in E_NAMES:
# Get the data
e = aqua.get('energy_' + e_name)
dedt = aqua.get('energy_d' + e_name + 'dt')
dedt_in = aqua.get('energy_d' + e_name + 'dt_in')
# Perform the corrector
e += 0.5 * dt * (dedt - dedt_in)
aqua.set('energy_' + e_name, e)
return True
def main():
dt = aqua.get("dt")
for e_name in E_NAMES:
# Get the data
e = aqua.get('energy_' + e_name)
dedt = aqua.get('energy_d' + e_name + 'dt')
dedt_in = aqua.get('energy_d' + e_name + 'dt_in')
# Perform the corrector
e += 0.5 * dt * (dedt - dedt_in)
aqua.set('energy_' + e_name, e)
return True
def main():
dt = aqua.get("dt")
for e_name in E_NAMES:
# Get the data
e = aqua.get('energy_' + e_name)
dedt = aqua.get('energy_d' + e_name + 'dt')
# Perform the predictor
e += dt * dedt
aqua.set('energy_' + e_name, e)
# Store the energy variation for the Corrector
aqua.set('energy_d' + e_name + 'dt_in', dedt)
return True
def main():
dt = aqua.get("dt")
for e_name in E_NAMES:
# Get the data
e = aqua.get('energy_' + e_name)
dedt = aqua.get('energy_d' + e_name + 'dt')
# Perform the predictor
e += dt * dedt
aqua.set('energy_' + e_name, e)
# Store the energy variation for the Corrector
aqua.set('energy_d' + e_name + 'dt_in', dedt)
return True
def main():
global motion_r, motion_a
# Get the affected set of particles (it is also an identifier on the motion)
motion_iset = aqua.get("motion_iset")
# Check if they were not already set, eventually copying them from the
# current motion state.
if motion_r is None or motion_a is None:
n_sets = aqua.get("n_sets")
motion_r = [None] * n_sets
motion_a = [None] * n_sets
if motion_r[motion_iset] is None:
motion_r[motion_iset] = aqua.get("motion_r")
if motion_a[motion_iset] is None:
motion_a[motion_iset] = aqua.get("motion_a")
# Set the backuped state
aqua.set("motion_r_in", motion_r[motion_iset])
aqua.set("motion_a_in", motion_a[motion_iset])
# Backuo the new state variables
motion_r[motion_iset] = np.copy(aqua.get("motion_r"))
motion_a[motion_iset] = np.copy(aqua.get("motion_a"))
return True
def main():
global motion_r, motion_a
# Get the affected set of particles (it is also an identifier on the motion)
motion_iset = aqua.get("motion_iset")
# Check if they were not already set, eventually copying them from the
# current motion state.
if motion_r is None or motion_a is None:
n_sets = aqua.get("n_sets")
motion_r = [None] * n_sets
motion_a = [None] * n_sets
if motion_r[motion_iset] is None:
motion_r[motion_iset] = aqua.get("motion_r")
if motion_a[motion_iset] is None:
motion_a[motion_iset] = aqua.get("motion_a")
# Set the backuped state
aqua.set("motion_r_in", motion_r[motion_iset])
aqua.set("motion_a_in", motion_a[motion_iset])
# Backuo the new state variables
motion_r[motion_iset] = np.copy(aqua.get("motion_r"))
motion_a[motion_iset] = np.copy(aqua.get("motion_a"))
return True
def main():
# Get the time instant
t = aqua.get("t")
# Interpolate the data
a = np.zeros(4, dtype=np.float32)
a[0] = np.radians(np.interp(t, T, A))
dadt = np.zeros(4, dtype=np.float32)
dadt[0] = np.radians(np.interp(t, T, DADT))
# Send it to AQUAgpusph
aqua.set("motion_a", a)
aqua.set("motion_dadt", dadt)
return True
def main():
# Get the time instant
t = aqua.get("t")
# Interpolate the data
a = np.zeros(4, dtype=np.float32)
a[0] = np.radians(np.interp(t, T, A))
dadt = np.zeros(4, dtype=np.float32)
dadt[0] = np.radians(np.interp(t, T, DADT))
# Send it to AQUAgpusph
aqua.set("motion_a", a)
aqua.set("motion_dadt", dadt)
return True
def main():
# Get the time instant
t = aqua.get("t")
# Interpolate the data
r = np.zeros(2, dtype=np.float32)
r[0] = np.interp(t, T, X)
u = np.zeros(2, dtype=np.float32)
u[0] = np.interp(t, T, U)
dudt = np.zeros(2, dtype=np.float32)
dudt[0] = np.interp(t, T, DUDT)
# Send it to AQUAgpusph
aqua.set("motion_r", r)
aqua.set("motion_drdt", u)
aqua.set("motion_ddrddt", dudt)
# Write output
F.write('{}\t{}\t{}\t{}\n'.format(t, r[0], u[0], dudt[0]))
F.flush()
return True
def main():
# Get the time instant
t = aqua.get("t")
# Interpolate the data
r = np.zeros(2, dtype=np.float32)
r[0] = np.interp(t, T, X)
u = np.zeros(2, dtype=np.float32)
u[0] = np.interp(t, T, U)
dudt = np.zeros(2, dtype=np.float32)
dudt[0] = np.interp(t, T, DUDT)
# Send it to AQUAgpusph
aqua.set("motion_r", r)
aqua.set("motion_drdt", u)
aqua.set("motion_ddrddt", dudt)
# Write output
F.write('{}\t{}\t{}\t{}\n'.format(t, r[0], u[0], dudt[0]))
F.flush()
return True