def run(visualize): grid = GridGenerator.GenGrid(m) r2d2 = np.argwhere(grid == 2) global initialState initialState = SearchNode.State(r2d2[0][0], r2d2[0][1], grid, grid.max()) operators = ['Up', 'Down', 'Left', 'Right'] def goalTestFunction(state): return (state.max == 0) def pathCostFunction(node): if h == 1: return heuristics.HeuristicOne(node.state.grid) elif h == 2: return heuristics.HeuristicTwo(node.state.grid) problem = SearchProblem.createSearchProblem(initialState, operators, goalTestFunction, pathCostFunction) print(colored("Thinking", 'red')) for strategy in ["BFS", "DFS", "UC", "G", "A*"]: result_node = GenericSearch.Search(problem, strategy) output(result_node, strategy) for depth in range(99999999): GenericSearch.max_depth = depth result_node = GenericSearch.Search(problem, "ID") output(result_node, "ID") if result_node != None: print('the depth is ', depth) break if visualize == True: runGUI(GUIResults)
# Parse command line arguments: fileList = [] for arg in sys.argv[1:]: if 1: fileList.append(arg) # Get the primary star vsini values prim_vsini = [None for _ in fileList] # Use this one for the real data search Tvalues = range(7000, 10000, 200) + range(10000, 30000, 400) Tvalues = range(7000, 10000, 250) + range(10000, 30000, 500) GenericSearch.slow_companion_search(fileList, prim_vsini, hdf5_file=hdf5_filename, extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, metal_values=(0.0), logg_values=(3.5, 4.0, 4.5,), vsini_values=range(50, 300, 25), Tvalues=Tvalues, observatory='McDonald', debug=False, reject_outliers=False, vbary_correct=False, addmode='all', output_mode='hdf5', output_file='CCF_primary_20151129.hdf5')
modeldir = modeldir + "/" if __name__ == "__main__": # Parse command line arguments: fileList = [] for arg in sys.argv[1:]: if 1: fileList.append(arg) # Get the primary star vsini values prim_vsini = StarData.get_vsini(fileList) GenericSearch.slow_companion_search( fileList, prim_vsini, hdf5_file=hdf5_filename, extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, metal_values=(0.0, -0.5, 0.5), vsini_values=(1, 5.0, 10.0, 20.0, 30.0), Tvalues=range(3000, 9000, 100), observatory="McDonald", debug=False, vbary_correct=True, addmode="all", output_mode="hdf5", )
# Remove anything without a vsini new_file_list = [] new_prim_vsini = [] for vsini, fname in zip(prim_vsini, fileList): if vsini is not None: new_file_list.append(fname) new_prim_vsini.append(vsini) else: print('No vsini found for file {}'.format(fname)) GenericSearch.slow_companion_search(new_file_list, new_prim_vsini, hdf5_file=hdf5_filename, extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, interp_regions=interp_regions, #metal_values=(0, -0.5, 0.5), metal_values=(0.5,), # vsini_values=(1, 5.0, 10.0, 20.0, 30.0), logg_values=(4.5), Tvalues=range(8800, 9000, 100), vsini_values=(1, 5, 10, 20, 30), observatory='McDonald', debug=False, vbary_correct=True, addmode='all', output_mode='hdf5')
modeldir = modeldir + "/" if __name__ == "__main__": # Parse command line arguments: fileList = [] for arg in sys.argv[1:]: if 1: fileList.append(arg) vsini_list = [None for f in fileList] GenericSearch.slow_companion_search( fileList, vsini_list, hdf5_file="/media/ExtraSpace/PhoenixGrid/TS23_Grid.hdf5", extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, metal_values=(0.0, -0.5, 0.5), vsini_values=(1, 5), Tvalues=range(3000, 7000, 100), observatory="McDonald", debug=False, vbary_correct=False, addmode="simple", obstype="real", output_mode="hdf5", )
modeldir = raw_input("sys.platform not recognized. Please enter model directory below: ") if not modeldir.endswith("/"): modeldir = modeldir + "/" if __name__ == '__main__': # Parse command line arguments: fileList = [] for arg in sys.argv[1:]: if 1: fileList.append(arg) vsini_list = [None for f in fileList] GenericSearch.slow_companion_search(fileList, vsini_list, hdf5_file='/media/ExtraSpace/PhoenixGrid/TS23_Grid.hdf5', extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, metal_values=(0.0, ), vsini_values=(1, 5,), Tvalues=range(3000, 7000, 200), observatory='McDonald', debug=False, vbary_correct=False, addmode='simple', obstype='real', output_mode='hdf5')
if __name__ == '__main__': # Parse command line arguments: fileList = [] for arg in sys.argv[1:]: if 1: fileList.append(arg) vsini_list = [None for f in fileList] GenericSearch.slow_companion_search( fileList, vsini_list, hdf5_file='/media/ExtraSpace/PhoenixGrid/TS23_Grid.hdf5', extensions=True, resolution=None, trimsize=trimsize, modeldir=modeldir, badregions=badregions, metal_values=(0.0, ), vsini_values=( 1, 5, ), Tvalues=range(3000, 7000, 200), observatory='McDonald', debug=False, vbary_correct=False, addmode='simple', obstype='real', output_mode='hdf5')