mine_layer.py

import numpy as np
import zmq
import time


class Minelayer:

    def __init__(self, mean = [72, -53], cov=[[300, 0], [0, 300]], total_mines=50):
        # for Zmq
        context = zmq.Context()
        self.publisher = context.socket(zmq.PUB)
        self.publisher_mine = context.socket(zmq.PUB)
        self.publisher.connect("tcp://127.0.0.1:5505")
        self.publisher_mine.bind("tcp://127.0.0.1:5522")

        # Mean and the covariance
        self.mean = mean
        self.cov = cov  # diagonal covariance
        self.total_mines = total_mines

    def send_message(self):
        # distribution of the x and the y
        x, y = np.random.multivariate_normal(self.mean, self.cov, self.total_mines).T
        for i, element in enumerate(x):
            content = [b"M", b"x=" + str(x[i]) + ",y=" + str(y[i]) + ",label= " + str(i) + ", type=hazard"]
            self.publisher.send_multipart(content)
            time.sleep(0.1)
            content_add = [b"M", b"hazard = x=" + str(x[i]) + ",y=" + str(y[i]) + ",label= " + str(i) + ", type=hazard"]
            self.publisher_mine.send_multipart(content_add)
            time.sleep(0.1)