Ejemplo n.º 1
0
def make_disp_sq_plots(comp_key,conn):
    """Takes in a comp_key for a track_stat computation and plots the
    square displacement histograms"""

    (fin,) = conn.execute("select fout from comps where function = 'track_stats' and comp_key = ?",
                          (comp_key,)).fetchone()

    F = h5py.File(fin,'r')
    
    g = F[_fd('disp_sq_hist',comp_key)]
    cmap = color_mapper(0,len(g))

    (fig,ax) = plots.set_up_plot()
    istatus = plots.non_i_plot_start();
    for s in g:
        step = int(s[-7:])
        val = g[s]['bin_value'][:]
        
        ax.semilogy(g[s]['bin_edges'],val,color = cmap.get_color(step))
    F.close()

    (iden_fun,dset_key ) = conn.execute("select function,dset_key from comps where " +
                                "comp_key in " +
                                "(select iden_key from trk_stat_prams where "+
                                "comp_key = ?)",(comp_key,)).fetchone()

    ax.set_title("dset: " + str(dset_key) + " " + iden_fun)

    plots.non_i_plot_stop(istatus)
Ejemplo n.º 2
0
def planet_dynamics(horizon, step, make_plot=True):
    delta_in_sec = step * days_to_sec

    # define your planets here
    planets = [static_sun("1"),
               planet("2", R_sun, M_earth, D_earth_sun, 0.0, 0.0, 0.0, V_earth, 0.0)]

    x, y, z = [[]], [[]], [[]]
    set_up_positions(x, y, z, len(planets))
    lines = []
    if make_plot:
        set_up_plot(lines, x, y, z, planets)

    total_time = 0  # in years
    while total_time <= horizon:

        print("Time (in y): ", total_time, " --- Number of planets: ", len(planets))

        for p in planets:
            p.report()
        append_positions(x, y, z, planets)
        if make_plot:
            update_plot(lines, x, y, z, planets, total_time)

        compute_accelerations(planets)
        for p in planets:
            p.update_velocity(delta_in_sec)
            p.update_position(delta_in_sec)

        # planets = check_for_colliding_planets(planets)

        total_time += step * days_to_years  # in years

    if make_plot:
        plt.show()
    return x, y, z
Ejemplo n.º 3
0
def plot_tracks(tracks):
    """
    Takes in a 

    """
    print len(tracks)
    # set up figure
    (fig,ax) = plots.set_up_plot()

    init_frame = np.min([t.start_frame for t in tracks])
    
    ax.set_title(' frame: ' + str(init_frame)
                 + ' dtime: ' + str(tracks.dtime) + 'ms')

    def trk_len_hash(trk):
        return len(trk)
    def trk_disp_hash(trk):
        return np.sqrt((np.sum(np.array(trk[-1]) - np.array(trk[0]))**2))

    trk_hash = trk_len_hash

    t_len = [trk_hash(trk) for trk in tracks]
    cm = plots.color_mapper(min(t_len),max(t_len))
    print (min(t_len),max(t_len))

    # loop over tracks and plot
    [ax.plot(np.array([t[0] for t in trk.positions])*tracks.sp_scale,
             np.array([t[1] for t in trk.positions])*tracks.sp_scale,
             '-',
             color=cm.get_color(trk_hash(trk)))
     for trk in tracks]

    x,y,I = zip(*[trk.positions[0] for trk in tracks])
    # plot the starting points
    ax.plot(np.array(x)*tracks.sp_scale,
            np.array(y)*tracks.sp_scale,'xk')

    ax.set_aspect('equal')
    plt.draw()