Source code for bmtk.simulator.filternet.io_tools

# from bmtk.simulator.core.io_tools import iof
from bmtk.simulator.core.io_tools import IOUtils
from bmtk.utils.io.ioutils import bmtk_world_comm


[docs]class FilterNetIOUtils(IOUtils):
    def __init__(self):
        super(FilterNetIOUtils, self).__init__()
        self.mpi_rank = bmtk_world_comm.MPI_rank
        self.mpi_size = bmtk_world_comm.MPI_size

[docs]    def log_info(self, message, all_ranks=False):
        if all_ranks is False and self.mpi_rank != 0:
            return

        self.logger.info(message)

[docs]    def log_debug(self, message, all_ranks=False):
        if all_ranks is False and self.mpi_rank != 0:
            return

        self.logger.debug(message)

[docs]    def barrier(self):
        bmtk_world_comm.barrier()


io = FilterNetIOUtils()