robot_controller.py

from defines import *
from utils import *
import random
class RobotController(object):
    """ You have to implement this class """
    def init(starting_position, steering_noise, distance_noise, sonar_noise,
                     measurement_noise, speed, turning_speed, gps_delay, execution_cpu_time_limit):
        """ @param starting_position - (x,y) tuple representing current_position """
        raise NotImplementedError()

    def act(self):
        """ Return next action """
        raise NotImplementedError()

    def on_sense_sonar(self, dist):
        """ React to sensory data """
        raise NotImplementedError()

    def on_sense_field(self, field_type, field_parameter):
        """ React to sensory data """
        raise NotImplementedError()

    def on_sense_gps(self, x, y):
        """ React to sensory data """
        raise NotImplementedError()



#class OmitCollisionsCheckAccuracy(RobotController):
#    """ Exemplary robot controller omitting collisions """
#    STATE_FORWARD = 0
#    STATE_LOOK_FOR_SPACE = 1
#
#    def init(self, starting_position, steering_noise, distance_noise, measurement_noise, speed, turning_speed):
#        self.phase = OmitCollisions.STATE_LOOK_FOR_SPACE
#        self.speed = speed
#        self.turn_speed = turning_speed
#        self.command_queue = []
#        self.last_distance = 0.0
#
#    def act(self):
#        if len(self.command_queue) == 0:
#            if self.phase == OmitCollisions.STATE_LOOK_FOR_SPACE:
#                self.command_queue.append([TURN, random.randint(-1, 1)* 50])
#                self.command_queue.append([SENSE_SONAR])
#            else:
#                self.command_queue.append([MOVE, int(self.last_distance/0.01)])
#                self.command_queue.append([SENSE_SONAR])
#                self.command_queue.append([SENSE_FIELD])
#
#        return self.command_queue.pop(0)
#
#    def on_sense_sonar(self, distance):
#        self.last_distance = distance
#
#        if distance < 0.04:
#            self.phase = OmitCollisions.STATE_LOOK_FOR_SPACE
#        else:
#            self.phase = OmitCollisions.STATE_FORWARD
#

from collections import defaultdict



#TODO: fill in this function
def load_python_controller(file):
    pass

import datetime

class PythonTimedRobotController(RobotController):
    """ Wrapper class to manage time consumption (also for other language packages) """
    def __init__(self, rc):
        self.rc = rc
        self.time_consumed = datetime.timedelta(0)

    def init(self, *args, **dargs):
        x = datetime.datetime.now()
        self.rc.init(*args, **dargs)
        self.time_consumed += datetime.datetime.now() - x

    def act(self):
        """ Return next action """
        x = datetime.datetime.now()
        ret = self.rc.act()
        self.time_consumed += datetime.datetime.now() - x
        return ret

    def on_sense_sonar(self, dist):
        x = datetime.datetime.now()
        self.rc.on_sense_sonar(dist)
        self.time_consumed += datetime.datetime.now() - x

    def on_sense_field(self, field_type, field_parameter):
        x = datetime.datetime.now()
        self.rc.on_sense_field(field_type, field_parameter)
        self.time_consumed += datetime.datetime.now() - x

    def on_sense_gps(self, x, y):
        tmp = datetime.datetime.now()
        self.rc.on_sense_gps(x,y)
        self.time_consumed += datetime.datetime.now() - tmp




def importCode(file_name, name, add_to_sys_modules=0):
    import sys,imp
    return imp.load_source(name, file_name)



counter_module = 0
def compile_robot(file_name, module_name = "contestant_module"):
    """ Compiles robot from given file and returns class object """
    global counter_module
    module_name += str(counter_module)
    counter_module += 1
    mod =  importCode(file_name, module_name)
    compiled_class = None
    for symbol in dir(mod):
        if hasattr(getattr(mod, symbol), "act") and getattr(mod, symbol).__name__ != "RobotController":
            compiled_class = getattr(mod, symbol)
    print compiled_class
    globals()[compiled_class.__name__] = compiled_class
    if compiled_class is None:
        raise KrakrobotException("Not found class with act() function named different than RobotController in provided .py")
    return compiled_class, mod