def get_results(arc): if arc == 'gpu': try: import pycuda if sys.stdout != sys.__stdout__: atexit.register(report_g) except ImportError: if sys.stdout != sys.__stdout__: atexit.register(report_e) elif arc == 'cpu': if sys.stdout != sys.__stdout__: atexit.register(report_c) print('Generating results for lysozyme example...') if os.getcwd().rsplit('/', 1)[1] == 'tests': results = main(['', '../examples/lys'], log_output=False, return_results_dict=True) elif os.getcwd().rsplit('/', 1)[1] == 'pygbe': results = main(['', './examples/lys'], log_output=False, return_results_dict=True) else: print( "Run tests from either the main repo directory or the tests directory" ) return results
def get_results(): print('Generating results for lysozyme example...') if os.getcwd().rsplit('/', 1)[1] == 'tests': results = main(['','../examples/lys'], log_output=False, return_results_dict=True) elif os.getcwd().rsplit('/', 1)[1] == 'pygbe': results = main(['','./examples/lys'], log_output=False, return_results_dict=True) else: print("Run tests from either the main repo directory or the tests directory") return results
def get_results(): print('Generating results for two spheres example...') if os.getcwd().rsplit('/', 1)[1] == 'tests': results = main(['','../examples/two_spheres'], log_output=False, return_results_dict=True) elif os.getcwd().rsplit('/', 1)[1] == 'pygbe': results = main(['','./examples/two_spheres'], log_output=False, return_results_dict=True) else: print("Run tests from either the main repo directory or the tests directory") return results
folder_path = in_files_path + 'prism_reg_SE' full_path = os.path.abspath(folder_path) + '/' os.environ['PYGBE_PROBLEM_FOLDER'] = full_path #Creating dictionary field. We will modify the 'E' key in the for loop. field_dict_pillars = read_fields(full_path + 'prism_reg_38K.config') #Calculate Cext(lambda) for pillars' surface tic_ss = time.time() e_field = -1. wave, Cext_pillars = Cext_wave_scan(e_field, wave_s, diel_list, field_dict_pillars, full_path) toc_ss = time.time() numpy.savetxt('../results_data/prism_regular_SE_LE_res/' + 'prism_reg_38K_short_edge' + '10-20microns.txt', list(zip(wave, Cext_pillars)), fmt='%.9f %.9f', header='lambda [Ang], Cext [nm^2]') time_simulation = (toc_ss - tic_ss) with open( '../results_data/prism_regular_SE_LE_res/Time_' + 'prism_reg_38K_short_edge' + '.txt', 'w') as f: print('time_total: {} s'.format(time_simulation), file=f) if __name__ == "__main__": main(sys.argv)