67 lines
2 KiB
Python
67 lines
2 KiB
Python
"""
|
|
tdma.py: EMANE TDMA model bindings for CORE
|
|
"""
|
|
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Set
|
|
|
|
from core import constants, utils
|
|
from core.config import Configuration
|
|
from core.emane import emanemodel
|
|
from core.emulator.enumerations import ConfigDataTypes
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class EmaneTdmaModel(emanemodel.EmaneModel):
|
|
# model name
|
|
name: str = "emane_tdma"
|
|
|
|
# mac configuration
|
|
mac_library: str = "tdmaeventschedulerradiomodel"
|
|
mac_xml: str = "tdmaeventschedulerradiomodel.xml"
|
|
|
|
# add custom schedule options and ignore it when writing emane xml
|
|
schedule_name: str = "schedule"
|
|
default_schedule: Path = (
|
|
constants.CORE_DATA_DIR / "examples" / "tdma" / "schedule.xml"
|
|
)
|
|
config_ignore: Set[str] = {schedule_name}
|
|
|
|
@classmethod
|
|
def load(cls, emane_prefix: Path) -> None:
|
|
cls.mac_defaults["pcrcurveuri"] = str(
|
|
emane_prefix
|
|
/ "share/emane/xml/models/mac/tdmaeventscheduler/tdmabasemodelpcr.xml"
|
|
)
|
|
super().load(emane_prefix)
|
|
config_item = Configuration(
|
|
id=cls.schedule_name,
|
|
type=ConfigDataTypes.STRING,
|
|
default=str(cls.default_schedule),
|
|
label="TDMA schedule file (core)",
|
|
)
|
|
cls.mac_config.insert(0, config_item)
|
|
|
|
def post_startup(self) -> None:
|
|
"""
|
|
Logic to execute after the emane manager is finished with startup.
|
|
|
|
:return: nothing
|
|
"""
|
|
# get configured schedule
|
|
config = self.session.emane.get_configs(node_id=self.id, config_type=self.name)
|
|
if not config:
|
|
return
|
|
schedule = config[self.schedule_name]
|
|
|
|
# get the set event device
|
|
event_device = self.session.emane.event_device
|
|
|
|
# initiate tdma schedule
|
|
logger.info(
|
|
"setting up tdma schedule: schedule(%s) device(%s)", schedule, event_device
|
|
)
|
|
args = f"emaneevent-tdmaschedule -i {event_device} {schedule}"
|
|
utils.cmd(args)
|