54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
import random
|
|
|
|
from data.Item import Item
|
|
from data.Order import Order
|
|
from data.enum.ItemType import ItemType
|
|
from util.ClientParamsFactory import ClientParamsFactory
|
|
|
|
|
|
class InitialStateFactory:
|
|
|
|
@staticmethod
|
|
def generate_item_list(output_list_size: int):
|
|
item_list : [Item] = []
|
|
for i in range(output_list_size):
|
|
item_list.append(InitialStateFactory.__generate_item())
|
|
|
|
return item_list
|
|
|
|
@staticmethod
|
|
def generate_order_list(output_order_list_size: int):
|
|
order_list: [Order] = []
|
|
for i in range(output_order_list_size):
|
|
order_list.append(InitialStateFactory.__generate_order())
|
|
|
|
return order_list
|
|
|
|
@staticmethod
|
|
def __generate_order() -> Order:
|
|
order_size = random.randint(1, 4)
|
|
|
|
items: [Item] = []
|
|
for i in range(order_size):
|
|
items.append(InitialStateFactory.__generate_item())
|
|
|
|
time_base = random.randint(8, 20)
|
|
final_time = time_base * order_size
|
|
|
|
client_params = ClientParamsFactory.get_client_params()
|
|
|
|
return Order(final_time, items, None, client_params)
|
|
|
|
@staticmethod
|
|
def generate_input_sequence(self, input_sequence_size):
|
|
sequence : [Item] = []
|
|
for i in range(0, input_sequence_size):
|
|
sequence.append(self.__generate_item())
|
|
|
|
return sequence
|
|
|
|
@staticmethod
|
|
def __generate_item() -> Item:
|
|
randomly_picked_type = random.choice(list(ItemType))
|
|
return Item(randomly_picked_type)
|