Beispiel #1
0
def start_test_signal_gen(lag, SNR, K1, K2, col1, col2):
    data = signal_gen.signal_gen(SNR, K1, K2)
    Xnew = data[col1]
    Ynew = data[col2]
    d = np.vstack((Xnew, Ynew)).T
    res = gtest.grangercausalitytests(d, lag, verbose=True)
    return res
Beispiel #2
0
def start_test_signal_gen(lag, SNR, K1, K2, col1, col2):
    data = signal_gen.signal_gen(SNR, K1, K2)
    Xnew = data[col1]
    Ynew = data[col2]
    d = np.vstack((Xnew, Ynew)).T
    res = gtest.grangercausalitytests(d, lag, verbose=True)
    return res
Beispiel #3
0
    def __init__(self):
        super(DDTF, self).__init__()

        self.connect("delete_event", self.delete_event)
        self.connect("destroy", self.destroy)
        e1, e2, e3 = signal_gen.signal_gen(.2, .01, .001)

        (f, step, NS11, NS12, NS13, NS21, NS22, NS23, NS31, NS32,
         NS33) = self.ddtf(e1, e2, e3)

        # gtk.Window.__init__(self)
        self.fig = Figure(figsize=(20, 15), dpi=72)

        self.canvas = FigureCanvas(self.fig)
        self.canvas.set_size_request(1800, 640)

        t = np.arange(0.0, 50.0, 0.01)
        xlim = np.array([0, 10])

        self.axes = self.fig.add_axes([0.075, 0.25, 0.9, 0.725],
                                      axisbg='#FFFFCC')

        self.axes.plot(
            t,
            np.sin(2 * 0.32 * np.pi * t) * np.sin(2 * 2.44 * np.pi * t))
        self.axes.set_xlim([0.0, 10.0])
        self.axes.set_xticklabels([])

        self.axesSpec = self.fig.add_axes([0.075, 0.05, 0.9, 0.2])
        t = self.axesSpec.text(
            0.5,
            0.5,
            'Click on EEG channel for spectrogram (scroll mouse to expand)',
            verticalalignment='center',
            horizontalalignment='center',
        )
        t.set_transform(self.axes.transAxes)
        self.axesSpec.set_xlim([0.0, 10.0])
        self.axesSpec.set_xticklabels([])
        self.axesSpec.set_yticklabels([])
        self.canvas.show()
        self.show()
Beispiel #4
0
    def __init__(self):
        super(DDTF,self).__init__()
        
        self.connect("delete_event", self.delete_event)
        self.connect("destroy", self.destroy)
        e1,e2,e3 = signal_gen.signal_gen(.2,.01,.001)

        (f ,step,NS11, NS12, NS13, NS21, NS22, NS23, NS31, NS32, NS33) = self.ddtf(e1,e2,e3)
        
        # gtk.Window.__init__(self)
        self.fig = Figure(figsize = (20,15), dpi=72)
        
        self.canvas = FigureCanvas(self.fig)
        self.canvas.set_size_request(1800, 640)
        
        t = np.arange(0.0,50.0, 0.01)
        xlim = np.array([0,10])

        
        self.axes = self.fig.add_axes([0.075, 0.25, 0.9, 0.725], axisbg='#FFFFCC')

        self.axes.plot(t, np.sin(2*0.32*np.pi*t) * np.sin(2*2.44*np.pi*t) )
        self.axes.set_xlim([0.0,10.0])
        self.axes.set_xticklabels([])

        self.axesSpec = self.fig.add_axes([0.075, 0.05, 0.9, 0.2])
        t = self.axesSpec.text(
            0.5, 0.5,
            'Click on EEG channel for spectrogram (scroll mouse to expand)',
            verticalalignment='center',
            horizontalalignment='center',
            )
        t.set_transform(self.axes.transAxes)
        self.axesSpec.set_xlim([0.0,10.0])
        self.axesSpec.set_xticklabels([])
        self.axesSpec.set_yticklabels([])
        self.canvas.show()
        self.show()
Beispiel #5
0
def test_do():
    e1,e2,e3 = signal_gen.signal_gen(3,.1,.001)
    do_ddtf_loop(e1,e2)
Beispiel #6
0
    def __init__(self):

        ttle1='Spectrum el1'
        ttle2=' el2 - . el1'
        ttle3=' el3 - . el1'
        ttle4=' el1 - . el2'
        ttle5=' Spectrum el2'
        ttle6=' el3 - . el2'
        ttle7=' el1 - . el3'
        ttle8='el2 - . el3'
        ttle9='Spectrum el3'


        self.win = gtk.Window()
        self.win.set_border_width(5)
        self.win.resize(800,400)
        vbox = gtk.VBox(spacing=3)
        self.win.add(vbox)
        vbox.show()
        self.fig = Figure(figsize=(7,5), dpi=72)

        self.canvas = FigureCanvas(self.fig)  # a gtk.DrawingArea
        self.canvas.show()
        vbox.pack_start(self.canvas, True, True)
        (self.e1,self.e2,self.e3) = signal_gen.signal_gen(3,.1,.001)

        self.ax1 = self.fig.add_subplot(431, title=ttle1)
        self.ax1.set_xlim(0,60)
        self.ax1.set_ylim(0,1)

        self.ax2 = self.fig.add_subplot(432, title=ttle2)
        self.ax2.set_xlim(0,60)
        self.ax2.set_ylim(0,1)

        self.ax3 = self.fig.add_subplot(433, title=ttle3)
        self.ax3.set_xlim(0,60)
        self.ax3.set_ylim(0,1)

        self.ax4 = self.fig.add_subplot(434, title=ttle4)
        self.ax4.set_xlim(0,60)
        self.ax4.set_ylim(0,1)


        self.ax5 = self.fig.add_subplot(435, title=ttle5)
        self.ax5.set_xlim(0,60)
        self.ax5.set_ylim(0,1)


        self.ax6 = self.fig.add_subplot(436, title=ttle6)
        self.ax6.set_xlim(0,60)
        self.ax6.set_ylim(0,1)

        self.ax7 = self.fig.add_subplot(437, title=ttle7)
        self.ax7.set_xlim(0,60)
        self.ax7.set_ylim(0,1)

        self.ax8 = self.fig.add_subplot(438, title=ttle8)
        self.ax8.set_xlim(0,60)
        self.ax8.set_ylim(0,1)

        self.ax9 = self.fig.add_subplot(439, title=ttle9)
        self.ax9.set_xlim(0,60)
        self.ax9.set_ylim(0,1)

        self.ax10 = self.fig.add_subplot(4,3,10, title="el1")
        self.ax11 = self.fig.add_subplot(4,3,11, title="el2")
Beispiel #7
0
    def __init__(self):

        ttle1 = 'Spectrum el1'
        ttle2 = ' el2 - . el1'
        ttle3 = ' el3 - . el1'
        ttle4 = ' el1 - . el2'
        ttle5 = ' Spectrum el2'
        ttle6 = ' el3 - . el2'
        ttle7 = ' el1 - . el3'
        ttle8 = 'el2 - . el3'
        ttle9 = 'Spectrum el3'

        self.win = gtk.Window()
        self.win.set_border_width(5)
        self.win.resize(800, 400)
        vbox = gtk.VBox(spacing=3)
        self.win.add(vbox)
        vbox.show()
        self.fig = Figure(figsize=(7, 5), dpi=72)

        self.canvas = FigureCanvas(self.fig)  # a gtk.DrawingArea
        self.canvas.show()
        vbox.pack_start(self.canvas, True, True)
        (self.e1, self.e2, self.e3) = signal_gen.signal_gen(3, .1, .001)

        c = 0
        for i in self.e1:
            print i, self.e2[c]
            c += 1

        self.ax1 = self.fig.add_subplot(431, title=ttle1)
        # self.ax1.set_xlim(0,60)
        # self.ax1.set_ylim(0,1)

        self.ax2 = self.fig.add_subplot(432, title=ttle2)
        self.ax2.set_xlim(0, 60)
        self.ax2.set_ylim(0, 1)

        self.ax3 = self.fig.add_subplot(433, title=ttle3)
        self.ax3.set_xlim(0, 60)
        self.ax3.set_ylim(0, 1)

        self.ax4 = self.fig.add_subplot(434, title=ttle4)
        self.ax4.set_xlim(0, 60)
        self.ax4.set_ylim(0, 1)

        self.ax5 = self.fig.add_subplot(435, title=ttle5)
        self.ax5.set_xlim(0, 60)
        self.ax5.set_ylim(0, 1)

        self.ax6 = self.fig.add_subplot(436, title=ttle6)
        self.ax6.set_xlim(0, 60)
        self.ax6.set_ylim(0, 1)

        self.ax7 = self.fig.add_subplot(437, title=ttle7)
        self.ax7.set_xlim(0, 60)
        self.ax7.set_ylim(0, 1)

        self.ax8 = self.fig.add_subplot(438, title=ttle8)
        self.ax8.set_xlim(0, 60)
        self.ax8.set_ylim(0, 1)

        self.ax9 = self.fig.add_subplot(439, title=ttle9)
        self.ax9.set_xlim(0, 60)
        self.ax9.set_ylim(0, 1)

        self.ax10 = self.fig.add_subplot(4, 3, 10, title="el1")
        self.ax11 = self.fig.add_subplot(4, 3, 11, title="el2")
        self.ax12 = self.fig.add_subplot(4, 3, 12, title="el3")
Beispiel #8
0
def test_do():
    e1, e2, e3 = signal_gen.signal_gen(3, .1, .001)
    do_ddtf_loop(e1, e2)