def run_rp2(self, rp_steps, caching, cache_dir):
'''Compute the spectrum using the new RP-Solver'''
print('\nRunning New RP-Solver method...')
t = time()
if caching:
cache_dir = CACHE2 if cache_dir is None else cache_dir
else:
cache_dir = None
spectrum = []
for i, (gam, ep) in enumerate(zip(self.gammas, self.eps)):
sys.stdout.write('\r{0:.2f}%'.format(i * 100. / self.nsteps))
sys.stdout.flush()
ep = max(ep, EPS_MIN)
solver = RP_Solver(ep * self.h,
ep * self.J,
gam,
cache_dir=cache_dir)
solver.solve()
e_vals, e_vecs = solver.node.evd()
spectrum.append(list(e_vals))
return spectrum
# split schedule into iso-field steps
rp_steps = max(1, rp_steps)
niso = int(np.ceil(len(self.gammas) * 1. / rp_steps))
isos = []
for i in range(rp_steps):
gams = self.gammas[i * niso:(i + 1) * niso]
eps = self.eps[i * niso:(i + 1) * niso]
isos.append((gams, eps))
# solve spectrum within each iso-step
spectrum = []
i = 0
for gammas, eps in isos:
i += 1
# solve first problem is iso-step
gam, ep = gammas[0], max(eps[0], EPS_MIN)
solver = RP_Solver(ep * self.h,
ep * self.J,
gam,
cache_dir=cache_dir)
solver.solve()
e_vals, e_vecs = solver.node.evd()
spectrum.append(list(e_vals))
nx, nz = solver.get_current_fields()
modes = solver.node.modes
# solve remaining problems is iso-step
for gam, ep in zip(gammas[1:], eps[1:]):
i += 1
sys.stdout.write('\r{0:.2f}%'.format(i * 100. / self.nsteps))
sys.stdout.flush()
ep = max(ep, EPS_MIN)
solver = RP_Solver(ep * self.h,
ep * self.J,
gam,
nx=nx,
nz=nz,
cache_dir=cache_dir)
solver.mode_solve(modes)
spectrum.append(list(solver.node.e_vals))
return spectrum
def run_rp2(self, rp_steps, caching, cache_dir):
'''Compute the spectrum using the new RP-Solver'''
print('\nRunning New RP-Solver method...')
t = time()
if caching:
cache_dir = CACHE2 if cache_dir is None else cache_dir
else:
cache_dir = None
spectrum = []
for i, (gam, ep) in enumerate(zip(self.gammas, self.eps)):
sys.stdout.write('\r{0:.2f}%'.format(i*100./self.nsteps))
sys.stdout.flush()
ep = max(ep, EPS_MIN)
solver = RP_Solver(ep*self.h, ep*self.J, gam,
cache_dir=cache_dir)
solver.solve()
e_vals, e_vecs = solver.node.evd()
spectrum.append(list(e_vals))
return spectrum
# split schedule into iso-field steps
rp_steps = max(1, rp_steps)
niso = int(np.ceil(len(self.gammas)*1./rp_steps))
isos = []
for i in range(rp_steps):
gams = self.gammas[i*niso:(i+1)*niso]
eps = self.eps[i*niso:(i+1)*niso]
isos.append((gams, eps))
# solve spectrum within each iso-step
spectrum = []
i = 0
for gammas, eps in isos:
i += 1
# solve first problem is iso-step
gam, ep = gammas[0], max(eps[0], EPS_MIN)
solver = RP_Solver(ep*self.h, ep*self.J, gam,
cache_dir=cache_dir)
solver.solve()
e_vals, e_vecs = solver.node.evd()
spectrum.append(list(e_vals))
nx, nz = solver.get_current_fields()
modes = solver.node.modes
# solve remaining problems is iso-step
for gam, ep in zip(gammas[1:], eps[1:]):
i += 1
sys.stdout.write('\r{0:.2f}%'.format(i*100./self.nsteps))
sys.stdout.flush()
ep = max(ep, EPS_MIN)
solver = RP_Solver(ep*self.h, ep*self.J, gam,
nx=nx, nz=nz, cache_dir=cache_dir)
solver.mode_solve(modes)
spectrum.append(list(solver.node.e_vals))
return spectrum
