Skip to content

Commit 71dfe1f

Browse files
authored
Merge pull request #26 from pierGit7/feature/task_csv_reader
feat: Added function for reading contents of task csv
2 parents 33fbbfb + 93c2c9c commit 71dfe1f

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

src/common/csvreader.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33

44
from common.architecture import Architecture
55
from common.budget import Budget
6+
from common.task import Task
67
from common.scheduler import Scheduler
78
from common.utils import get_project_root
89

@@ -50,6 +51,24 @@ def read_budgets(csv:str)-> list[Budget]:
5051

5152
return budgets
5253

54+
def read_tasks(csv:str)-> list[Task]:
55+
csv = _get_csv_path(csv)
56+
57+
df = pd.read_csv(csv)
58+
59+
tasks = []
60+
for _,row in df.iterrows():
61+
task = Task(
62+
task_name=row['task_name'],
63+
wcet=row['wcet'],
64+
period=row['period'],
65+
component_id=row['component_id'],
66+
priority=row['priority'],
67+
)
68+
tasks.append(task)
69+
70+
return tasks
71+
5372
def _get_csv_path(csv:str) -> str:
5473
if os.path.exists(csv):
5574
return csv

0 commit comments

Comments
 (0)