store.py

MAX_CUBES_STORED = 5

ZONE_0_STORE_LOCATION_0 = {'x': 0.5, 'y': 7.5, 'max_d_theta': 10}
ZONE_0_STORE_LOCATION_1 = {'x': 1, 'y': 7.5, 'max_d_theta': 10}
ZONE_0_STORE_LOCATION_2 = {'x': 0.5, 'y': 7, 'max_d_theta': 10}
ZONE_0_STORE_LOCATION_3 = {'x': 0.75, 'y': 7.25, 'max_d_theta': 10}
ZONE_0_STORE_LOCATION_4 = {'x': 0.75, 'y': 7.25, 'max_d_theta': 10}
ZONE_0_STORE_LOCATIONS = [ZONE_0_STORE_LOCATION_0, ZONE_0_STORE_LOCATION_1, ZONE_0_STORE_LOCATION_2, ZONE_0_STORE_LOCATION_3, ZONE_0_STORE_LOCATION_4]
ZONE_0_RETURN_LOCATION_012 = {'x': 1, 'y': 7}
ZONE_0_RETURN_LOCATION_34 = {'x': 1.5, 'y': 6.5}
ZONE_0_RETURN_LOCATIONS = [ZONE_0_RETURN_LOCATION_012, ZONE_0_RETURN_LOCATION_012, ZONE_0_RETURN_LOCATION_012, ZONE_0_RETURN_LOCATION_34, ZONE_0_RETURN_LOCATION_34]
ZONE_0_NEXT_CUBE_LOCATION = {'x': 2.5, 'y': 5.5}

ZONE_1_STORE_LOCATION_0 = {'x': 7.5, 'y': 7.5, 'max_d_theta': 10}
ZONE_1_STORE_LOCATION_1 = {'x': 7.5, 'y': 7, 'max_d_theta': 10}
ZONE_1_STORE_LOCATION_2 = {'x': 7, 'y': 7.5, 'max_d_theta': 10}
ZONE_1_STORE_LOCATION_3 = {'x': 7.25, 'y': 7.25, 'max_d_theta': 10}
ZONE_1_STORE_LOCATION_4 = {'x': 7.25, 'y': 7.25, 'max_d_theta': 10}
ZONE_1_STORE_LOCATIONS = [ZONE_1_STORE_LOCATION_0, ZONE_1_STORE_LOCATION_1, ZONE_1_STORE_LOCATION_2, ZONE_1_STORE_LOCATION_3, ZONE_1_STORE_LOCATION_4]
ZONE_1_RETURN_LOCATION_012 = {'x': 7, 'y': 7}
ZONE_1_RETURN_LOCATION_34 = {'x': 6.5, 'y': 6.5}
ZONE_1_RETURN_LOCATIONS = [ZONE_1_RETURN_LOCATION_012, ZONE_1_RETURN_LOCATION_012, ZONE_1_RETURN_LOCATION_012, ZONE_1_RETURN_LOCATION_34, ZONE_1_RETURN_LOCATION_34]
ZONE_1_NEXT_CUBE_LOCATION = {'x': 5.5, 'y': 5.5}

ZONE_2_STORE_LOCATION_0 = {'x': 7.5, 'y': 0.5, 'max_d_theta': 10}
ZONE_2_STORE_LOCATION_1 = {'x': 7, 'y': 0.5, 'max_d_theta': 10}
ZONE_2_STORE_LOCATION_2 = {'x': 7.5, 'y': 1, 'max_d_theta': 10}
ZONE_2_STORE_LOCATION_3 = {'x': 7.25, 'y': 0.75, 'max_d_theta': 10}
ZONE_2_STORE_LOCATION_4 = {'x': 7.25, 'y': 0.75, 'max_d_theta': 10}
ZONE_2_STORE_LOCATIONS = [ZONE_2_STORE_LOCATION_0, ZONE_2_STORE_LOCATION_1, ZONE_2_STORE_LOCATION_2, ZONE_2_STORE_LOCATION_3, ZONE_2_STORE_LOCATION_4]
ZONE_2_RETURN_LOCATION_012 = {'x': 7, 'y': 1}
ZONE_2_RETURN_LOCATION_34 = {'x': 6.5, 'y': 1.5}
ZONE_2_RETURN_LOCATIONS = [ZONE_2_RETURN_LOCATION_012, ZONE_2_RETURN_LOCATION_012, ZONE_2_RETURN_LOCATION_012, ZONE_2_RETURN_LOCATION_34, ZONE_2_RETURN_LOCATION_34]
ZONE_2_NEXT_CUBE_LOCATION = {'x': 5.5, 'y': 2.5}

ZONE_3_STORE_LOCATION_0 = {'x': 0.5, 'y': 0.5, 'max_d_theta': 10}
ZONE_3_STORE_LOCATION_1 = {'x': 0.5, 'y': 1, 'max_d_theta': 10}
ZONE_3_STORE_LOCATION_2 = {'x': 1, 'y': 0.5, 'max_d_theta': 10}
ZONE_3_STORE_LOCATION_3 = {'x': 0.75, 'y': 0.75, 'max_d_theta': 10}
ZONE_3_STORE_LOCATION_4 = {'x': 0.75, 'y': 0.75, 'max_d_theta': 10}
ZONE_3_STORE_LOCATIONS = [ZONE_3_STORE_LOCATION_0, ZONE_3_STORE_LOCATION_1, ZONE_3_STORE_LOCATION_2, ZONE_3_STORE_LOCATION_3, ZONE_3_STORE_LOCATION_4]
ZONE_3_RETURN_LOCATION_012 = {'x': 1, 'y': 1}
ZONE_3_RETURN_LOCATION_34 = {'x': 1.5, 'y': 1.5}
ZONE_3_RETURN_LOCATIONS = [ZONE_3_RETURN_LOCATION_012, ZONE_3_RETURN_LOCATION_012, ZONE_3_RETURN_LOCATION_012, ZONE_3_RETURN_LOCATION_34, ZONE_3_RETURN_LOCATION_34]
ZONE_3_NEXT_CUBE_LOCATION = {'x': 2.5, 'y': 2.5}

STORE_LOCATIONS = [ZONE_0_STORE_LOCATIONS, ZONE_1_STORE_LOCATIONS, ZONE_2_STORE_LOCATIONS, ZONE_3_STORE_LOCATIONS]
RETURN_LOCATIONS = [ZONE_0_RETURN_LOCATIONS, ZONE_1_RETURN_LOCATIONS, ZONE_2_RETURN_LOCATIONS, ZONE_3_RETURN_LOCATIONS]
NEXT_CUBE_LOCATIONS = [ZONE_0_NEXT_CUBE_LOCATION, ZONE_1_NEXT_CUBE_LOCATION, ZONE_2_NEXT_CUBE_LOCATION, ZONE_3_NEXT_CUBE_LOCATION]

class StoreManager():
    
    def __init__(self, zone):
        self.store_locations = STORE_LOCATIONS[zone]
        self.return_locations = RETURN_LOCATIONS[zone]
        self.next_cube_location = NEXT_CUBE_LOCATIONS[zone]
        self.cubes_stored = 0
        
    def getReturnLocation(self):
        
        if (self.cubes_stored == MAX_CUBES_STORED):
            return None
        
        else:
            return_location = self.return_locations[self.cubes_stored]
            return return_location
        
    def getStoreLocation(self):
        
        if (self.cubes_stored == MAX_CUBES_STORED):
            return None
        
        else:
            store_location = self.store_locations[self.cubes_stored]
            self.cubes_stored += 1
            return store_location