diff --git a/src/common/csvreader.py b/src/common/csvreader.py index f702e84..a5437f4 100644 --- a/src/common/csvreader.py +++ b/src/common/csvreader.py @@ -3,6 +3,7 @@ from common.architecture import Architecture from common.budget import Budget +from common.task import Task from common.scheduler import Scheduler from common.utils import get_project_root @@ -50,6 +51,24 @@ def read_budgets(csv:str)-> list[Budget]: return budgets +def read_tasks(csv:str)-> list[Task]: + csv = _get_csv_path(csv) + + df = pd.read_csv(csv) + + tasks = [] + for _,row in df.iterrows(): + task = Task( + task_name=row['task_name'], + wcet=row['wcet'], + period=row['period'], + component_id=row['component_id'], + priority=row['priority'], + ) + tasks.append(task) + + return tasks + def _get_csv_path(csv:str) -> str: if os.path.exists(csv): return csv