import time
from multiprocessing import Process, Pipe
import numpy as np
import matplotlib.pyplot as plt


class DataStreamProcess(Process):
    def __init__(self, connec, *args, **kwargs):
        self.connec = connec
        Process.__init__(self, *args, **kwargs)

    def run(self):
        random_gen = np.random.mtrand.RandomState(seed=127260)
        for _ in range(30000):
            time.sleep(0.01)
            new_pt = random_gen.uniform(-1., 1., size=2)
            self.connec.send(new_pt)

class ComediStreamProcess(Process):
    def __init__(self, connec, comediConfig, *args, **kwargs):
        self._connect = connec
        self._comediConfig = comediConfig
        super(ComediStreamProcess, self).__init__(*args, **kwargs)


def main():
    conn1, conn2 = Pipe()
    data_stream = DataStreamProcess(conn1)
    data_stream.start()

    plt.gca().set_xlim([-1, 1.])
    plt.gca().set_ylim([-1, 1.])
    plt.gca().set_title("Running...")
    plt.ion()

    pt = None
    while True:
        if not (conn2.poll(0.1)):
            if not (data_stream.is_alive()):
                break
            else:
                continue
        new_pt = conn2.recv()
        if pt is not None:
            plt.plot([pt[0], new_pt[0]], [pt[1], new_pt[1]], "bs:")
            plt.pause(0.001)
        pt = new_pt

    plt.gca().set_title("Terminated.")
    plt.draw()
    plt.show(block=True)


if __name__ == '__main__':
    main()