Source code for examples.banksim.model.model_result

"""
Author: Changbeom Choi (@cbchoi)
Copyright (c) 2014-2020 Handong Global University
Copyright (c) 2021-2024 Hanbat National University
License: MIT.  The full license text is available at:
https://github.com/eventsim/pyjevsim/blob/main/LICENSE

This module contains Banksim User Generator Model 
"""
import os
from pyjevsim.behavior_model import BehaviorModel
from pyjevsim.definition import *



class BankResult(BehaviorModel):
    """A Model representing a bank user generator."""

    def __init__(self, name, max_user):
        """
        Args:
            name (str): Name of Model
            cycle (float): Generation cycle time
            max_user (int): Maximum number of users to generate
            proc_time (float): Processing time for each user
        """
        BehaviorModel.__init__(self, name)
        self.init_state("WAIT")  # Initialize initial state
        self.insert_state("WAIT", Infinite)  # Add "WAIT" state
        
        self.insert_input_port("process")  # Add input port "process"
        self.insert_input_port("drop")  # Add input port "drop"

        self.max_user = max_user  # Maximum number of users to generate
        #self.proc_time = proc_time  # Processing time for each user
        self.user_count = 0
        self.user = []
        self.drop_user = []
        self.drop_user_count = 0
        


    def ext_trans(self, port, msg):
        """
        Handles external transitions based on the input port.

        Args:
            port (str): The port that received the message
            msg (SysMessage): The received message
        """
        if port == "process":
            proc = msg.retrieve()[0]
            _user = msg.retrieve()[1]
                
            self.user.append((proc, _user))
            self.user_count += 1
            if self.user_count >= self.max_user :
                self.get_result()
                os._exit(0)
                
        if port == "drop" :
            drop_user_list = msg.retrieve()[0]
            for _user in drop_user_list : 
                self.drop_user.append(_user)
                self.drop_user_count += 1




    def output(self, msg_deliver):
        pass




    def int_trans(self):
        pass

        


    def get_result(self) : 
        #print("sim time", self.global_time, flush = True)
        print("[BANKSIM RESULT]")
        print("- Accountant user : ", self.user_count,  flush = True)
        print("- Dropped user : ", self.drop_user_count,  flush = True)
        
        print("\n [Accountant user list]")
        for p, u in self.user :
            print(p, u.__str__(), flush=True)
            
        print("\n [Dropped user list]", flush=True)
        for d in self.drop_user :
            print(d.__str__(), flush=True)