eventlog.py

import pandas as pd
import processmodel as model

"""
Event log generator for our simulation model:
- generate an event log
- update an event log (adding new events)
"""

def add_start_event(process, event_id, case_id, activity, start_timestamp):
    process.event_log[event_id] = {
        'CaseID': case_id,
        'Activity': activity,
        'StartTimestamp': float(start_timestamp),
        'EndTimestamp': None
    }

def add_end_event(process, event_id, end_timestamp):
    # if event_id in process.event_log:
    event = process.event_log[event_id]
    event['EndTimestamp'] = end_timestamp
    # process.event_log.append(event)
    # del process.event_log[event_id]

# add functions for adding events with their attributes to the log


def export_to_csv(process, file_path):
    event_log_df = pd.DataFrame.from_dict(process.event_log)
    event_log_df.to_csv(file_path, index=False)

def export_to_xes(process, file_path):
    # Use appropriate code to export to XES format
    pass