refactor: create multipurpose RecurrencePlot

Author: hasanisaeed

main.py

@@ -1,35 +1,58 @@
 from __future__ import division, print_function
 import numpy as np
-import pylab
+import pylab as plt
 from scipy.spatial.distance import pdist, squareform
 
+from utils import calculate_convolve
 
-def calculate_convolve(signal, n=5):
-    return np.convolve(signal, np.ones((n,)) / n, mode='valid')
 
+# ----------------- Plot config -------------------------------------------
+SMALL = 8
+MEDIUM = 10
+BIGGER = 11
 
-def recurrence_plot(signal, eps=0.10, steps=3):
-    _2d_array = signal[:, None]
-    # Pairwise distances
-    distance = pdist(_2d_array)
-    print(distance)
-    distance = np.floor(distance / eps)
-    print(distance)
-    distance[distance > steps] = steps
-    print(distance.shape)
-    return squareform(distance)
+plt.rc('font', size=SMALL)          # controls default text sizes
+plt.rc('axes', titlesize=SMALL)     # font size of the axes title
+plt.rc('axes', labelsize=MEDIUM)    # font size of the x and y labels
+plt.rc('xtick', labelsize=SMALL)    # font size of the tick labels
+plt.rc('ytick', labelsize=SMALL)    # font size of the tick labels
+plt.rc('legend', fontsize=SMALL)    # legend font size
+plt.rc('figure', titlesize=BIGGER)  # font size of the figure title
+# -------------------------------------------------------------------------
 
 
-if __name__ == "__main__":
-    raw_signal = np.random.uniform(-1, 1, 1000)
-    convolved_signal = calculate_convolve(raw_signal)
+class RecurrencePlot(object):
+    def __init__(self, signal, row=2, col=2):
+        self.signal = signal
+        self.size = '%d%d' % (row, col)
+
+    def recurrence_plot(self, eps=0.10, steps=3):
+        _2d_array = self.signal[:, None]
+
+        # Pairwise distances
+        distance = pdist(_2d_array)
+        distance = np.floor(distance / eps)
+        distance[distance > steps] = steps
+        return squareform(distance)
 
-    pylab.title("Raw Signal")
-    pylab.subplot(211)
-    pylab.plot(convolved_signal)
+    def subplot(self, x, is_signal=True, index=1, title=None, grid=True):
+        plt.subplot(int('%s%d' % (self.size, index)))
+        plt.plot(x) if is_signal else plt.imshow(x)
+        plt.title(title)
+        plt.grid(grid)
 
-    # pylab.title("2D Image")
-    pylab.subplot(212)
+    def save(self):
+        # plot with various axes scales
+        plt.figure()
+        self.subplot(self.signal, index=1, title='Raw Signal')
+        self.subplot(self.recurrence_plot(), is_signal=False, index=2, title='2D Image')
+        self.subplot(self.signal, index=3, title='Raw Signal')
+        self.subplot(self.recurrence_plot(), is_signal=False, index=4, title='2D Image')
+        plt.savefig('result.jpg')
 
-    pylab.imshow(recurrence_plot(convolved_signal))
-    pylab.savefig('result.jpg')
+
+if __name__ == "__main__":
+    raw_signal = np.random.uniform(-1, 1, 500)
+    convolved_signal = calculate_convolve(raw_signal)
+    rp = RecurrencePlot(convolved_signal)
+    rp.save()