def solveVectorized(self, timer = None): if timer is not None: print("Solving... vectorized") start = clock() for counter in range(self.timeSteps): waves, s = vec_rp(self.mwaves,self.q[:,0:self.mx-1],self.q[:,1:self.mx],self.aux[:,0:self.mx-1],self.aux[:,1:self.mx]) if timer is not None: end = clock() self.elapsedTime = end-start print("elapsed time: ", self.elapsedTime, " seconds") return waves, s
from __future__ import absolute_import from __future__ import print_function from numpy import * from FRS import vec_rp q = random.random((2,800)) aux = random.random((2,800)) waves, s = vec_rp(2,q[:,0:799],q[:,1:800],aux[:,0:799],aux[:,1:800]) print(waves) print(s)