Shepherd/recipemanager

shepherd/Alliance.py

@@ -12,6 +12,7 @@ class Alliance:
             team_1_number - Integer representing team number of first team
             team_2_number - Integer representing team number of second team
             score - Integer tracking the score of the Alliance
+            cooked - The number of ingredients that are cooked
     """
 
     def __init__(self, name, team_1_name, team_1_number, team_2_name,
@@ -21,6 +22,7 @@ def __init__(self, name, team_1_name, team_1_number, team_2_name,
         self.team_2_name = team_2_name
         self.team_1_number = team_1_number
         self.team_2_number = team_2_number
+        self.cooked = 0
         self.score = 0
         self.team_1_connection = False
         self.team_2_connection = False
@@ -37,6 +39,7 @@ def change_score(self, amount):
 
     def reset(self):
         self.score = 0
+        self.cooked = 0
         self.team_1_connection = False
         self.team_2_connection = False
         lcm_send(LCM_TARGETS.SCOREBOARD, SCOREBOARD_HEADER.SCORE,

shepherd/RecipeManager.py

@@ -0,0 +1,77 @@
+from Utils import *
+
+
+class RecipeManager():
+
+    master_recipes = [RECIPE_PROBABILITIES.EASY_RECIPE, RECIPE_PROBABILITIES.MEDIUM_RECIPE, \
+                      RECIPE_PROBABILITIES.HARD_RECIPE, RECIPE_PROBABILITIES.VERY_HARD_RECIPE]
+    gold_recipes = []
+    blue_recipes = []
+    gold_recipes_completed = 0
+    blue_recipes_completed = 0
+
+    def generate_recipes(self):
+        '''
+        reset recipe list with 4 newly selected recipes
+        both teams should have the same recipes
+        '''
+        self.gold_recipes.clear()
+        self.blue_recipes.clear()
+        for recipe in self.master_recipes:
+            self.gold_recipes += list(recipe.values)
+            self.blue_recipes += list(recipe.values)
+
+    def next_recipe(self, team):
+        '''
+        pop the recipe from the front of the list
+
+        The recipe being popped is done
+        '''
+        if team == ALLIANCE_COLOR.BLUE:
+            if self.blue_recipes:
+                self.blue_recipes.pop(0)
+                self.blue_recipes_completed += 1
+        else:
+            if self.gold_recipes:
+                self.gold_recipes.pop(0)
+                self.gold_recipes_completed += 1
+
+    def get_recipe(self, team):
+        '''
+        return first element of recipes
+        '''
+        if team == ALLIANCE_COLOR.BLUE:
+            if not self.blue_recipes:
+                return None
+            return self.blue_recipes[0]
+        else:
+            if not self.gold_recipes:
+                return None
+            return self.gold_recipes[0]
+
+    def check_recipe(self, team, ingredients, num_cooked=0):
+        '''
+        checks correctness of recipe
+        '''
+        recipes = []
+        if team == ALLIANCE_COLOR.BLUE:
+            recipes = self.blue_recipes
+        else:
+            recipes = self.gold_recipes
+        if len(recipes) != len(ingredients):
+            return False
+        else:
+            for item in ingredients:
+                if item not in recipes:
+                    return False
+        return True
+
+    def set_recipes(self, gold_recipes, blue_recipes):
+        '''
+        overwrite recipes
+        '''
+        if gold_recipes != None:
+            self.gold_recipes = list(gold_recipes)
+
+        if blue_recipes != None:
+            self.blue_recipes = list(blue_recipes)

shepherd/Utils.py

@@ -139,3 +139,54 @@ class STATE():
     WAIT = "wait"
     TELEOP = "teleop"
     END = "end"
+
+class INGREDIENT():
+    BREAD = "Bread"
+    VEGETABLES = "Vegetables"
+    MEAT = "Meat"
+    CHEESE = "Cheese"
+    # COOKED_MEAT = "COOKED " + MEAT
+    # COOKED_BREAD = "COOKED " + BREAD
+    # COOKED_CHEESE = "COOKED " + CHEESE
+
+class RECIPIES():
+    BREADSTICKS = "Breadsticks"
+    SALAD = "Salad"
+    COLD_CUTS = "Cold Cuts"
+    TURKEY_SANDWICH = "Turkey Sandwich"
+    CAESAR_SALAD = "Caesar Salad"
+    MEAT_AND_POTATOES = "Meat and Potatoes"
+    HAMBURGER = "Hamburgers"
+    GRILLED_CHEESE = "Grilled Cheese"
+    JALAPENO_POPPERZ = "Jalapeno Popperz"
+    CHEESEBURGER = "Cheeseburger"
+    SHEPHERDS_PIE = "Sherpherd's Pie"
+    CHEESE_AND_CRACKERS = "Cheese and Crackers"
+
+class RECIPE_PROBABILITIES():
+    Ingredients = "Ingredients"
+    Cooked = "Cooked"
+    RECIPE_PROBABILITIES = {0: [], 1: [], 2:[], 3:[]}
+    EASY_RECIPE = {RECIPIES.BREADSTICKS:[INGREDIENT.BREAD], \
+                   RECIPIES.SALAD:[INGREDIENT.VEGETABLES], \
+                   RECIPIES.COLD_CUTS:[INGREDIENT.MEAT]}
+    MEDIUM_RECIPE = {RECIPIES.TURKEY_SANDWICH: [INGREDIENT.BREAD, INGREDIENT.MEAT], \
+                     RECIPIES.CAESAR_SALAD: [INGREDIENT.VEGETABLES, INGREDIENT.BREAD], \
+                     RECIPIES.MEAT_AND_POTATOES: [INGREDIENT.MEAT, INGREDIENT.VEGETABLES], \
+                     RECIPIES.CHEESE_AND_CRACKERS: [INGREDIENT.BREAD, INGREDIENT.CHEESE]}
+    HARD_RECIPE = {RECIPIES.HAMBURGER: {Ingredients:[INGREDIENT.BREAD, INGREDIENT.MEAT, \
+                                        INGREDIENT.VEGETABLES], Cooked:1}, \
+                    RECIPIES.GRILLED_CHEESE: {Ingredients:[INGREDIENT.BREAD, \
+                                              INGREDIENT.CHEESE], Cooked:1}, \
+                    RECIPIES.JALAPENO_POPPERZ: {Ingredients:[INGREDIENT.CHEESE, \
+                                                INGREDIENT.VEGETABLES], Cooked:1}, \
+                    RECIPIES.CHEESEBURGER: {Ingredients:[INGREDIENT.BREAD, \
+                                            INGREDIENT.MEAT, INGREDIENT.CHEESE], Cooked:1}
+                  }
+    VERY_HARD_RECIPE = {RECIPIES.SHEPHERDS_PIE: {Ingredients:[INGREDIENT.MEAT,\
+                                                INGREDIENT.VEGETABLES,\
+                                                INGREDIENT.CHEESE], Cooked:2}}
+
+class TELEOP_RECIPE_MAP():
+    MAPPING = {0:[1, 0, 0, 0], 1:[0.1, 0.8, 0.1, 0], 2:[0.1, 0.1, 0.8, 0], 3:[0.1, 0.2, 0.3, 0.4]}
+    # 0: Easy, 1: Medium, 2: Hard, 3: Very Hard