from data.enum.Direction import Direction
from data.Item import Item
from data.Order import Order
from data.enum.Priority import Priority
from decision.ActionType import ActionType
from util.PathDefinitions import GridLocation


class State:
    def __init__(self,
                 action_taken: ActionType,
                 forklift_position: GridLocation,
                 forklift_rotation: Direction,
                 pending_orders: [Priority, [Order]],
                 filled_orders: [Order],
                 input_items: [Item]
                 ):
        self.action_taken = action_taken
        self.forklift_position = forklift_position
        self.forklift_rotation = forklift_rotation
        self.pending_orders = pending_orders
        self.filled_orders = filled_orders
        self.input_items = input_items