Пример #1
0
def LaserStates(data_dir):
    data = read_data(join(data_dir, 'LaserStates'))
    fig = plt.figure(figsize=(6, 6))
    X = axis(data=data[0], magnitude=False, label='$x_1$')
    Y = axis(data=data[1], magnitude=False, label='$x_3$')
    ax = SubPlot(fig, (1, 1, 1), X, Y, color='b', label='Laser')
    fig.subplots_adjust(bottom=0.12)  # Make more space for label
    return fig
Пример #2
0
def main(argv=None):
    '''

    '''
    import numpy as np
    if sys.version_info >= (3,0):
       print('%s needs matplotlib.  However, no matplotlib for python %s'%(
           sys.argv[0],sys.version_info,))
       return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt
    from utilities import axis, SubPlot
    if argv is None:                    # Usual case
        argv = sys.argv[1:]
    name_fine, name_coarse, name_quantized, plot_file = argv

    read_data = lambda name: np.array([[float(x) for x in line.split()]
        for line in file(name, 'r').readlines()]).T

    fine = read_data(name_fine)
    coarse = read_data(name_coarse)
    quantized = read_data(name_quantized)

    params = {'axes.labelsize': 12,
                   'text.fontsize': 10,
                   'legend.fontsize': 10,
                   'text.usetex': True,
                   'xtick.labelsize': 11,
                   'ytick.labelsize': 11}
    mpl.rcParams.update(params)
    fig = plt.figure(figsize=(6,4))
    X = axis(data=fine[0], magnitude=False, label=r'$\tau$')
    Y = axis(data=fine[1], magnitude=False, ticks=np.arange(-10, 10.1, 10),
             label=r'$x_1(\tau)$')
    ax = SubPlot(fig,(2,1,1),X,Y, color='b')
    ax.plot(coarse[0], coarse[1],'ro')
    ax.set_ylim(-17, 17)
    ax.set_xlim(0,6)

    ax = fig.add_subplot(2,1,2)
    ax.plot(quantized[0],quantized[1],'kd')
    ax.set_xlabel(r'$t$')
    ax.set_ylabel(r'$y(t)$')
    ax.set_ylim(0.5, 4.5)
    ax.set_yticks(np.arange(1,4.1,1))
    ax.set_xticks(np.arange(0,40.1,10))
    fig.subplots_adjust(hspace=0.3)

    if DEBUG:
        plt.show()
    else:
        fig.savefig(plot_file)
    return 0
Пример #3
0
def main(argv=None):
    '''

    '''
    import numpy as np
    if sys.version_info >= (3,0):
       print('%s needs matplotlib.  However, no matplotlib for python %s'%(
           sys.argv[0],sys.version_info,))
       return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt
    from utilities import axis, SubPlot
    if argv is None:                    # Usual case
        argv = sys.argv[1:]
    data_file, plot_file = argv

    data = np.array([[float(x) for x in line.split()]
        for line in file(data_file, 'r').readlines()]).T

    params = {'axes.labelsize': 12,
                   'text.fontsize': 10,
                   'legend.fontsize': 10,
                   'text.usetex': True,
                   'xtick.labelsize': 11,
                   'ytick.labelsize': 11}
    mpl.rcParams.update(params)
    fig = plt.figure(figsize=(6,4))
    x_ticks = np.arange(100,201,50)
    X = axis(data=data[0], magnitude=False, label=r'$t$', ticks=x_ticks)
    Y = axis(data=data[1], magnitude=False, ticks=np.arange(-20, 20.1, 20),
             label=r'$y(t)$')
    ax = SubPlot(fig,(3,1,1),X,Y, color='b')

    ax = fig.add_subplot(3,1,2)
    ax.plot(data[0],data[2],'g', label=r'$(y(t)-\mu_\gamma(t))$')
    ax.plot(data[0],data[3],'k', label=r'$\sigma_\gamma(t)$')
    ax.set_xticks(x_ticks)
    y_ticks = np.arange(-0.06,0.121,0.06)
    ax.set_yticks(y_ticks)
    ax.set_yticklabels([r'$%4.2f$'%x for x in y_ticks])
    ax.legend()

    Y = axis(data=data[4], magnitude=False, ticks=np.arange(-6, 2.1, 2),
             label=r'$\log(P_\gamma (y(t))$')
    ax = SubPlot(fig,(3,1,3),X,Y, color='r')

    fig.subplots_adjust(hspace=0.3)

    if DEBUG:
        plt.show()
    else:
        fig.savefig(plot_file)
    return 0
Пример #4
0
def LaserForecast(data_dir):
    data = read_data(join(data_dir, 'LaserForecast'))
    fig = plt.figure(figsize=(7, 5))
    X = axis(data=data[0],
             magnitude=False,
             label='t',
             ticks=np.arange(300, 650, 100))
    Y = axis(data=data[1],
             magnitude=False,
             label='y(t)',
             ticks=np.arange(50, 256, 100))
    ax = SubPlot(fig, (1, 1, 1), X, Y, color='b', label='Laser')
    ax.plot(X.get_data(), data[2], 'r:', label='Forecast')
    ax.legend()
    fig.subplots_adjust(bottom=0.12)  # Make more space for label
    return fig
Пример #5
0
def main(argv=None):
    '''

    '''
    import numpy as np
    from utilities import read_data, SubPlot, axis
    from os.path import join

    if sys.version_info >= (3, 0):
        print('%s needs matplotlib.  However, no matplotlib for python %s' % (
            sys.argv[0],
            sys.version_info,
        ))
        return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt

    if argv is None:  # Usual case
        argv = sys.argv[1:]
    sim_file, fig_dir = argv

    params = {
        'axes.labelsize': 18,  # Plotting parameters for latex
        'text.fontsize': 15,
        'legend.fontsize': 15,
        'text.usetex': True,
        'font.family': 'serif',
        'font.serif': 'Computer Modern Roman',
        'xtick.labelsize': 15,
        'ytick.labelsize': 15
    }
    mpl.rcParams.update(params)

    data = read_data(sim_file)
    X = axis(data=data[0],
             magnitude=False,
             label=r'$t$',
             ticks=np.arange(0, 100.1, 25))

    def _plot(Y):
        fig = plt.figure(figsize=(3.5, 2.5))
        ax = SubPlot(fig, (1, 1, 1), X, Y, color='b')
        ax.set_ylim(-0.02, 1.02)
        fig.subplots_adjust(bottom=0.15)  # Make more space for label
        fig.subplots_adjust(left=.15, bottom=.18)
        return (ax, fig)

    ax, fig_b = _plot(
        axis(data=data[1],
             magnitude=False,
             label=r'$S(t)$',
             ticks=np.arange(0, 1.1, 1)))
    ax, fig_d = _plot(
        axis(data=data[3],
             magnitude=False,
             label=r'$S(t)$',
             ticks=np.arange(0, 1.1, 1)))

    ax, fig_c = _plot(
        axis(data=data[2],
             magnitude=False,
             label=r'$y(t)$',
             ticks=np.arange(-4, 4.1, 4)))
    ax.set_ylim(-5, 5)
    fig_c.subplots_adjust(left=.2)
    if DEBUG:
        plt.show()
    else:
        fig_b.savefig(join(fig_dir, 'SGO_b.pdf'))
        fig_c.savefig(join(fig_dir, 'SGO_c.pdf'))
        fig_d.savefig(join(fig_dir, 'SGO_d.pdf'))
    return 0
Пример #6
0
def main(argv=None):
    '''

    '''
    import numpy as np
    if sys.version_info >= (3, 0):
        print('%s needs matplotlib.  However, no matplotlib for python %s' % (
            sys.argv[0],
            sys.version_info,
        ))
        return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt
    from utilities import axis, SubPlot
    if argv is None:  # Usual case
        argv = sys.argv[1:]
    data_file, plot_file = argv

    data = np.array([[float(x) for x in line.split()]
                     for line in file(data_file, 'r').readlines()]).T

    params = {
        'axes.labelsize': 12,
        'text.fontsize': 10,
        'legend.fontsize': 10,
        'text.usetex': True,
        'xtick.labelsize': 11,
        'ytick.labelsize': 11
    }
    mpl.rcParams.update(params)
    fig = plt.figure(figsize=(6, 6))
    X = axis(data=data[0], magnitude=False)
    Y = axis(data=data[5],
             magnitude=False,
             ticks=np.arange(0.6, 1.41, 0.2),
             label=r'$\lambda(t)$')
    ax = SubPlot(fig, (2, 1, 1), X, Y, label=r'95\%', color='b')
    ax.plot(data[0], data[2], 'r', label=r'sample')
    ax.plot(data[0], data[3], 'r', label=r'sample')
    ax.plot(data[0], data[1], 'r', label=r'sample')
    ax.plot(data[0], data[4], 'b', label=r'5\%')
    ax.set_ylim(0.5, 1.5)
    ax.legend()
    ax.yaxis.tick_right()
    ax.yaxis.set_label_position("right")

    X = axis(data=data[0], magnitude=False, label=r'$t$')
    Y = axis(data=data[10],
             magnitude=False,
             ticks=np.arange(0.6, 1.41, 0.2),
             label=r'$\lambda(t)$')
    ax = SubPlot(fig, (2, 1, 2), X, Y, label=r'95\%', color='b')
    ax.plot(data[0], data[6], 'r', label=r'sample')
    ax.plot(data[0], data[7], 'r', label=r'sample')
    ax.plot(data[0], data[8], 'r', label=r'sample')
    ax.plot(data[0], data[9], 'b', label=r'5\%')
    ax.set_ylim(0.5, 1.5)
    ax.legend()
    ax.yaxis.tick_right()
    ax.yaxis.set_label_position("right")

    if DEBUG:
        plt.show()
    else:
        fig.savefig(plot_file)
    return 0
Пример #7
0
def main(argv=None):
    '''

    '''
    import numpy as np
    from utilities import read_data, SubPlot, axis
    from os.path import join

    if sys.version_info >= (3,0):
       print('%s needs matplotlib.  However, no matplotlib for python %s'%(
           sys.argv[0],sys.version_info,))
       return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt

    if argv is None:                    # Usual case
        argv = sys.argv[1:]
    sim_file, fig_dir = argv

    params = {'axes.labelsize': 18,     # Plotting parameters for latex
              'text.fontsize': 15,
              'legend.fontsize': 15,
              'text.usetex': True,
              'font.family':'serif',
              'font.serif':'Computer Modern Roman',
              'xtick.labelsize': 15,
              'ytick.labelsize': 15}
    mpl.rcParams.update(params)

    data = read_data(sim_file)
    X = axis(data=data[0], magnitude=False, label=r'$t$',
             ticks=np.arange(0, 100.1, 25))

    def _plot(Y):
        fig = plt.figure(figsize=(3.5,2.5))
        ax = SubPlot(fig,(1,1,1),X,Y, color='b')
        ax.set_ylim(-0.02, 1.02)
        fig.subplots_adjust(bottom=0.15) # Make more space for label
        fig.subplots_adjust(left=.15, bottom=.18)
        return (ax, fig)

    ax, fig_b = _plot(axis(data=data[1], magnitude=False, label=r'$S(t)$',
                         ticks=np.arange(0, 1.1, 1)))
    ax, fig_d = _plot(axis(data=data[3], magnitude=False, label=r'$S(t)$',
                           ticks=np.arange(0, 1.1, 1)))

    ax, fig_c = _plot(axis(data=data[2], magnitude=False, label=r'$y(t)$',
                         ticks=np.arange(-4, 4.1, 4)))
    ax.set_ylim(-5, 5)
    fig_c.subplots_adjust(left=.2)
    if DEBUG:
        plt.show()
    else:
        fig_b.savefig(join(fig_dir, 'SGO_b.pdf'))
        fig_c.savefig(join(fig_dir, 'SGO_c.pdf'))
        fig_d.savefig(join(fig_dir, 'SGO_d.pdf'))
    return 0
Пример #8
0
def main(argv=None):
    '''

    '''
    import numpy as np
    if sys.version_info >= (3,0):
       print('%s needs matplotlib.  However, no matplotlib for python %s'%(
           sys.argv[0],sys.version_info,))
       return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt
    from utilities import axis, SubPlot
    if argv is None:                    # Usual case
        argv = sys.argv[1:]
    data_file, plot_file = argv

    data = np.array([[float(x) for x in line.split()]
                     for line in file(data_file, 'r').readlines()]).T

    params = {'axes.labelsize': 12,
                   'text.fontsize': 10,
                   'legend.fontsize': 10,
                   'text.usetex': True,
                   'xtick.labelsize': 11,
                   'ytick.labelsize': 11}
    mpl.rcParams.update(params)
    fig = plt.figure(figsize=(6,6))
    X = axis(data=data[0], magnitude=False)
    Y = axis(data=data[5], magnitude=False, ticks=np.arange(0.6, 1.41, 0.2),
             label=r'$\lambda(t)$')
    ax = SubPlot(fig,(2,1,1),X,Y, label=r'95\%', color='b')
    ax.plot(data[0], data[2], 'r', label=r'sample')
    ax.plot(data[0], data[3], 'r', label=r'sample')
    ax.plot(data[0], data[1], 'r', label=r'sample')
    ax.plot(data[0], data[4], 'b', label=r'5\%')
    ax.set_ylim(0.5, 1.5)
    ax.legend()
    ax.yaxis.tick_right()
    ax.yaxis.set_label_position("right")

    X = axis(data=data[0], magnitude=False, label=r'$t$')
    Y = axis(data=data[10], magnitude=False, ticks=np.arange(0.6, 1.41, 0.2),
             label=r'$\lambda(t)$')
    ax = SubPlot(fig,(2,1,2),X,Y, label=r'95\%', color='b')
    ax.plot(data[0], data[6], 'r', label=r'sample')
    ax.plot(data[0], data[7], 'r', label=r'sample')
    ax.plot(data[0], data[8], 'r', label=r'sample')
    ax.plot(data[0], data[9],'b', label=r'5\%')
    ax.set_ylim(0.5, 1.5)
    ax.legend()
    ax.yaxis.tick_right()
    ax.yaxis.set_label_position("right")

    if DEBUG:
        plt.show()
    else:
        fig.savefig(plot_file)
    return 0
Пример #9
0
def main(argv=None):
    '''

    '''
    import numpy as np
    if sys.version_info >= (3, 0):
        print('%s needs matplotlib.  However, no matplotlib for python %s' % (
            sys.argv[0],
            sys.version_info,
        ))
        return -1
    import matplotlib as mpl
    global DEBUG
    if DEBUG:
        mpl.rcParams['text.usetex'] = False
    else:
        mpl.use('PDF')
    import matplotlib.pyplot as plt
    from utilities import axis, SubPlot
    if argv is None:  # Usual case
        argv = sys.argv[1:]
    data_file, plot_file = argv

    data = np.array([[float(x) for x in line.split()]
                     for line in file(data_file, 'r').readlines()]).T

    params = {
        'axes.labelsize': 12,
        'text.fontsize': 10,
        'legend.fontsize': 10,
        'text.usetex': True,
        'xtick.labelsize': 11,
        'ytick.labelsize': 11
    }
    mpl.rcParams.update(params)
    fig = plt.figure(figsize=(6, 4))
    x_ticks = np.arange(100, 201, 50)
    X = axis(data=data[0], magnitude=False, label=r'$t$', ticks=x_ticks)
    Y = axis(data=data[1],
             magnitude=False,
             ticks=np.arange(-20, 20.1, 20),
             label=r'$y(t)$')
    ax = SubPlot(fig, (3, 1, 1), X, Y, color='b')

    ax = fig.add_subplot(3, 1, 2)
    ax.plot(data[0], data[2], 'g', label=r'$(y(t)-\mu_\gamma(t))$')
    ax.plot(data[0], data[3], 'k', label=r'$\sigma_\gamma(t)$')
    ax.set_xticks(x_ticks)
    y_ticks = np.arange(-0.06, 0.121, 0.06)
    ax.set_yticks(y_ticks)
    ax.set_yticklabels([r'$%4.2f$' % x for x in y_ticks])
    ax.legend()

    Y = axis(data=data[4],
             magnitude=False,
             ticks=np.arange(-6, 2.1, 2),
             label=r'$\log(P_\gamma (y(t))$')
    ax = SubPlot(fig, (3, 1, 3), X, Y, color='r')

    fig.subplots_adjust(hspace=0.3)

    if DEBUG:
        plt.show()
    else:
        fig.savefig(plot_file)
    return 0