LookOut.py

from __future__ import print_function

import random
from display import *

""" BASELINES """
# Baseline 1
def best_greedy( graph, budget ): # TopK
	return graph.get_plot_ranks()[:budget]

# Baseline 2
def random_plots( graph, budget ): # Random
	plots = graph.get_plot_ranks()
	return random.sample( plots, budget )

""" LookOut Algorithm """
# Select the optimised plot cover
def best_plots( graph, budget ):
	best_plot_list = []
	# Continue choosing plots till the budget is exhausted
	while budget > 0:
		# Choose the next best plot which mazimizes score
		plot = graph.get_best_plot()
		best_plot_list.append( int( plot ) )
		# Update the graph after removing chosen plot
		graph.update_graph( plot )
		budget -= 1
	# Return a list of the chosen plots
	return best_plot_list

# Optimized Algorithm
def LookOut( graph, budget, algo="LookOut" ):
	print( "\t-> Choosing best plots" )
	if algo == "LookOut":
		plots = best_plots( graph, budget )
	elif algo == "TopK":
		plots = best_greedy( graph, budget )
	elif algo == "Random":
		plots = random_plots( graph, budget )
	else:
		print_fail( "The Algorithm specified doesn't lie in { LookOut, TopK, Random }. Skipping ..." )
		return
	print( "\t-> Choosen Plots are ", end='' ) 
	cprint( str(plots), OKBLUE )
	return plots