Source code for sfepy.solvers.ts

import numpy as nm

from sfepy.base.base import output, get_default, assert_, Struct

[docs] def get_print_info(n_step): if n_step > 1: n_digit = int(nm.log10(n_step - 1) + 1) else: n_digit = 1 format = '%%%dd of %%%dd' % (n_digit, n_digit) suffix = '%%0%dd' % n_digit return n_digit, format, suffix
[docs] class TimeStepper(Struct): """ Time stepper class. """
[docs] @staticmethod def from_conf(conf): return TimeStepper(conf.t0, conf.t1, dt=conf.dt, n_step=conf.n_step, is_quasistatic=conf.quasistatic)
def __init__(self, t0, t1, dt=None, n_step=None, step=None, is_quasistatic=False): self.set_from_data(t0, t1, dt=dt, n_step=n_step, step=step) self.is_quasistatic = is_quasistatic self.step_start_time = None def _get_n_step(self, t0, t1, dt): n_step = int(round(nm.floor(((t1 - t0) / dt) + 0.5) + 1.0)) return n_step
[docs] def set_from_data(self, t0, t1, dt=None, n_step=None, step=None): self.t0, self.t1 = t0, t1 dt = get_default(dt, t1 - t0) self.n_step = get_default(n_step, self._get_n_step(self.t0, self.t1, dt)) if self.n_step > 1: self.times, self.dt = nm.linspace(self.t0, self.t1, self.n_step, endpoint=True, retstep=True) else: self.times = nm.array((self.t0,), dtype=nm.float64) self.dt = self.t1 - self.t0 self.n_digit, self.format, self.suffix = get_print_info(self.n_step) self.set_step(step)
[docs] def set_from_ts(self, ts, step=None): step = get_default(step, ts.step) self.set_from_data(ts.t0, ts.t1, ts.dt, ts.n_step, step=step)
[docs] def get_state(self): return {'step' : self.step}
[docs] def set_state(self, step=0, **kwargs): self.set_step(step=step)
[docs] def set_substep_time(self, sub_dt): self.step_start_time = self.time self.time += sub_dt
[docs] def restore_step_time(self): if self.step_start_time is not None: self.time = self.step_start_time self.step_start_time = None
[docs] def advance(self): if self.step < (self.n_step - 1): self.step += 1 self.time = self.times[self.step] self.normalize_time()
def __iter__(self): """ts.step, ts.time is consistent with step, time returned here ts.nt is normalized time in [0, 1]""" return self.iter_from(0)
[docs] def iter_from(self, step): self.set_step(step=step) for time in self.times[step:]: yield self.step, self.time self.advance()
[docs] def normalize_time(self): self.nt = (self.time - self.t0) / (self.t1 - self.t0)
[docs] def set_step(self, step=0, nt=0.0): nm1 = self.n_step - 1 if step is None: step = int(round(nt * nm1)) if step < 0: step = self.n_step + step if (step >= self.n_step) or (step < 0): output('time step must be in [%d, %d]' % (-nm1, nm1) ) raise ValueError self.step = step self.time = self.times[step] self.normalize_time()
def __eq__(self, other): if type(other) == type(self): return (abs(self.t0 == other.t0) < 1e-15) and \ (abs(self.t1 == other.t1) < 1e-15) and \ (self.n_step == other.n_step) else: raise ValueError
[docs] class VariableTimeStepper(TimeStepper): """ Time stepper class with a variable time step. """
[docs] @staticmethod def from_conf(conf): return VariableTimeStepper(conf.t0, conf.t1, dt=conf.dt, n_step=conf.n_step, is_quasistatic=conf.quasistatic)
[docs] def set_from_data(self, t0, t1, dt=None, n_step=None, step=None): self.t0, self.t1 = t0, t1 self.dtime = self.t1 - self.t0 dt = get_default(dt, self.dtime) self.n_step0 = get_default(n_step, self._get_n_step(self.t0, self.t1, dt)) if self.n_step0 > 1: self.dt = self.dtime / (self.n_step0 - 1) else: self.dt = self.dtime self.dt0 = self.dt self.n_digit, self.format, self.suffix = get_print_info(self.n_step0) self.set_step(step)
[docs] def set_from_ts(self, ts, step=None): self.set_from_data(ts.t0, ts.t1, ts.dt, ts.n_step0, step=0)
[docs] def get_state(self): return {'step' : self.step, 'dt' : self.dt, 'times' : self.times}
[docs] def set_state(self, step=0, dt=None, times=None, **kwargs): assert_(len(times) == (step + 1)) self.step = step self.dt = dt self.times = times self.time = self.times[-1] self.normalize_time()
[docs] def set_n_digit_from_min_dt(self, dt): n_step = self._get_n_step(self.t0, self.t1, dt) self.n_digit, self.format, self.suffix = get_print_info(n_step)
[docs] def set_step(self, step=0, nt=0.0): if step is None: step = 0 if (step > 0) and (step != self.step): msg = 'cannot set step != self.step or 0 in VariableTimeStepper!' raise ValueError(msg) if step == 0: self.step = 0 self.time = self.t0 self.nt = 0.0 self.times = [self.time] self.n_step = 1
[docs] def get_default_time_step(self): return self.dt0
[docs] def set_time_step(self, dt, update_time=False): self.dt = dt if update_time: self.time = self.times[self.step - 1] + self.dt self.times[self.step] = self.time self.normalize_time()
[docs] def advance(self): self.step += 1 self.time += self.dt self.normalize_time() self.times.append(self.time) self.n_step = self.step + 1
[docs] def iter_from(self, step): self.set_step(step=step) return self.iter_from_current()
[docs] def iter_from_current(self): """ ts.step, ts.time is consistent with step, time returned here ts.nt is normalized time in [0, 1]. """ while 1: yield self.step, self.time if self.nt >= 1.0: break self.advance()
def __iter__(self): self.set_step(0) return self.iter_from_current()