# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
"""
The protocol class
"""

class Protocol(object):
    """
    Protocol class
    """

    def __init__(self, _name):
        """
        Init

        Args:
            _name: The protocol name
        """

        self.name = _name
        self.load = 0       # bps

    def get_name(self):
        """
        Gets the protocol name

        Returns:
             The protocol name
        """

        return self.name

    def get_load(self):
        """
        Gets the protocol load

        Returns:
             The protocol load (bps)
        """

        return self.load

    def add_load(self, _load):
        """
        Adds load to the protocol

        Args:
            _load: The load to add
        """

        self.load += _load

    def clear_load(self):
        """
        Clears the load on this protocol
        """

        self.load = 0