def compute_traces(self, initial_condition=None): if initial_condition == None: initial_condition = self.initial_condition V_i = fh.integrate_four_rk4( initial_condition, self.network.get_coupling(), self.system.dt/float(self.system.stride), self.system.N_output(self.CYCLES), self.system.stride) t = self.system.dt*np.arange(V_i.shape[0]) ticks = np.asarray(t[::t.size/10], dtype=int) self.li_b.set_data(t, V_i[:, 0]) self.li_g.set_data(t, V_i[:, 1]-2.) self.li_r.set_data(t, V_i[:, 2]-4.) self.li_y.set_data(t, V_i[:, 3]-6.) self.ax.set_xticks(ticks) self.ax.set_xticklabels(ticks) self.ax.set_xlim(t[0], t[-1]) self.fig.canvas.draw() return t, V_i
def phase_trace_cpu(self, i, X): V_i = np.transpose( model.integrate_four_rk4( X[i, :], self.network.get_coupling(), self.dt/float(self.stride), self.N_output, self.stride)) ti, d = tl.phase_difference(V_i, V_trigger=torus.V_trigger) return d