def printfig(self,name): assert name == 'vtk' or name == 'png', 'Not a correct type file, choose vtk or png ' with config(verbose = 2, nprocs = 8, richoutput = True): for i in parallel.pariter(range(0,self.leni),8): for k in range(0,self.lenkoriginal): if name == 'vtk' : self.plotvtk(k,i) else: self.plotpng(k,i)
uerrs = ens.errors(tcase, rcase, tsol, rsol, tcase['u-h1s'](mu), scheme) res.append([rcase.size, rcase.meta['err-u'], *uerrs, rtime]) res = np.array(res) np.savetxt(f'beam-results-{length}.csv', res) @main.command() @click.option('--nred', default=5) @click.option('--length', default=5) @click.option('--num', default=7) @util.common_args def reduce(nred, length, num): get_reduced(nred, length, num) if __name__ == '__main__': with config(nprocs=multiprocessing.cpu_count()): main() # for i in range(1,16): # get_reduced(i, 20, 7) # get_reduced(i, 15, 7) # get_reduced(i, 10, 7) # get_reduced(i, 5, 7) # get_reduced(i, 3, 7) # get_reduced(i, 2, 7) # get_reduced(i, 1, 7)
def retval(verbose, nprocs, **kwargs): with config(verbose=verbose, nprocs=nprocs), log.TeeLog(log.DataLog('.'), log.RichOutputLog()): return func(**kwargs)