import matplotlib.pyplot as plt
from threading import Thread, Lock, Condition
import random, time, socket
from mpl_toolkits.mplot3d import Axes3D
import numpy as np


class Plotter:
    def __init__(self):

        # temp list of data received by server
        self.data = [[] for i in range(3)]

        # threading variables
        self.mutex = Lock()
        self.points_received = 0
        self.cv = Condition()

        # initialize figure
        self.fig = plt.figure()
        self.run_plot = False
        self.ax = self.fig.add_subplot(111, projection='3d')
        self.ax.set_xlim(0, 1)
        self.ax.set_ylim(0, 1)
        self.ax.set_zlim(0, 1)
        plt.ion()
        plt.show()

    # This just simulates reading from a socket.
    def data_listener(self):
        while True:
            time.sleep(1)
            with self.mutex:
                new_data = [random.random() for i in range(3)]

                # append new data to be plotted and check max/min to update axes limits
                for axis in range(3):
                    self.data[axis].append(new_data[axis])

    def sub_plot(self):
        # wait until ready, <Enter> is pressed
        self.cv.acquire()
        while not self.run_plot:
            self.cv.wait()  # I think this causes the plot to not respond before hitting <Enter>
        self.cv.release()

        while self.run_plot:
            time.sleep(0.01)
            with self.mutex:
                try:
                    # add data and clear graph
                    self.ax.scatter(self.data[0], self.data[1], self.data[2])
                    self.data = [[] for i in range(3)]

                    plt.pause(0.000001)
                    # self.fig.canvas.flush_events()
                except:
                    print ("error")

    def cmdline_input(self):
        # Start then End Plot
        for status in ["start", "end"]:
            text_start = raw_input("Hit <Enter> to " + status + " plotting")
            while text_start != "":
                print "Wrong input"
                text_start = raw_input("Hit <Enter> to " + status + " plotting")

            self.cv.acquire()
            if status == "start":
                self.run_plot = True
            elif status == "end":
                self.run_plot = False
            self.cv.notify()
            self.cv.release()

    def run(self):
        d = Thread(target=self.data_listener)
        d.start()

        # start cmdline input
        c_start = Thread(target=self.cmdline_input)
        c_start.start()

        # subplot
        self.sub_plot()


if __name__ == '__main__':
    p = Plotter()
    p.run()