dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

renaming

Browse Source
This commit is contained in:
Benoît Sierro
2021-11-11 16:28:00 +01:00
parent 472cd37814
commit 4b9425b519
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/scgenerator/solver.py
View File

@@ -87,7 +87,7 @@ class ConstantStepIntegrator(Integrator):
lin = self.linear_operator(self.state) lin = self.linear_operator(self.state)
nonlin = self.nonlinear_operator(self.state) nonlin = self.nonlinear_operator(self.state)
self.record_tracked_values() self.record_tracked_values()
new_spec = RK4IP_step( new_spec = rk4ip_step(
self.nonlinear_operator, self.nonlinear_operator,
self.state, self.state,
self.state.solution.spectrum, self.state.solution.spectrum,
@@ -129,7 +129,7 @@ class ConservedQuantityIntegrator(Integrator):
while True: while True:
h = h_next_step h = h_next_step
new_state = self.state.replace( new_state = self.state.replace(
RK4IP_step( rk4ip_step(
self.nonlinear_operator, self.nonlinear_operator,
self.state, self.state,
self.state.solution.spectrum, self.state.solution.spectrum,
@@ -215,7 +215,7 @@ class RK4IPSD(Integrator):
def take_step( def take_step(
self, h: float, spec: np.ndarray, lin: np.ndarray, nonlin: np.ndarray self, h: float, spec: np.ndarray, lin: np.ndarray, nonlin: np.ndarray
) -> np.ndarray: ) -> np.ndarray:
return RK4IP_step(self.nonlinear_operator, self.state, spec, h, lin, nonlin) return rk4ip_step(self.nonlinear_operator, self.state, spec, h, lin, nonlin)
def compute_diff(self, coarse_spec: np.ndarray, fine_spec: np.ndarray) -> float: def compute_diff(self, coarse_spec: np.ndarray, fine_spec: np.ndarray) -> float:
return np.sqrt(math.abs2(coarse_spec - fine_spec).sum() / math.abs2(fine_spec).sum()) return np.sqrt(math.abs2(coarse_spec - fine_spec).sum() / math.abs2(fine_spec).sum())
@@ -351,7 +351,7 @@ class ERK54(ERK43):
yield self.state yield self.state
def RK4IP_step( def rk4ip_step(
nonlinear_operator: NonLinearOperator, nonlinear_operator: NonLinearOperator,
init_state: CurrentState, init_state: CurrentState,
spectrum: np.ndarray, spectrum: np.ndarray,