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')
