Ejemplo n.º 1
0
def simplegrid():

    nzones = 3
    gr = gp.FVGrid2d(nzones, nzones, ng=0)

    # plot a domain without ghostcells
    gr.draw_grid()

    #------------------------------------------------------------------------
    # label
    gr.label_cell_center(nzones // 2,
                         nzones // 2,
                         r"$a_{i,j}$",
                         fontsize="large")
    gr.label_cell_center(nzones // 2 + 1,
                         nzones // 2,
                         r"$a_{i+1,j}$",
                         fontsize="large")
    gr.label_cell_center(nzones // 2,
                         nzones // 2 + 1,
                         r"$a_{i,j+1}$",
                         fontsize="large")

    # i+1/2,j interface
    gr.mark_cell_left_state_x(nzones // 2,
                              nzones // 2,
                              r"$a^{n+\myhalf}_{i+\myhalf,j,L}$",
                              color="b")
    gr.mark_cell_right_state_x(nzones // 2 + 1,
                               nzones // 2,
                               r"$a^{n+\myhalf}_{i+\myhalf,j,R}$",
                               color="b")

    # i,j+1/2 interface
    gr.mark_cell_left_state_y(nzones // 2,
                              nzones // 2,
                              r"$a^{n+\myhalf}_{i,j+\myhalf,L}$",
                              color="b")
    gr.mark_cell_right_state_y(nzones // 2,
                               nzones // 2 + 1,
                               r"$a^{n+\myhalf}_{i,j+\myhalf,R}$",
                               color="b")

    # grid labels
    gr.label_center_x(nzones // 2 - 1, r"$i-1$")
    gr.label_center_x(nzones // 2, r"$i$")
    gr.label_center_x(nzones // 2 + 1, r"$i+1$")

    gr.label_center_y(nzones // 2 - 1, r"$j-1$")
    gr.label_center_y(nzones // 2, r"$j$")
    gr.label_center_y(nzones // 2 + 1, r"$j+1$")

    # axes
    gr.clean_axes()
    plt.subplots_adjust(left=0.02, right=0.98, bottom=0.02, top=0.98)

    f = plt.gcf()
    f.set_size_inches(7.0, 7.0)

    plt.savefig("2dgrid.pdf")
Ejemplo n.º 2
0
def simplegrid():

    nzones = 5
    gr = gp.FVGrid2d(nzones, nzones, ng=0)

    # plot a domain without ghostcells
    gr.draw_grid()

    #------------------------------------------------------------------------
    # label
    gr.label_cell_center(nzones/2, nzones/2, r"$a_{i,j}$", 
                         fontsize="x-large", color="r")
    gr.shade_cell(nzones/2, nzones/2)

    # grid labels
    gr.label_center_x(nzones/2-2, r"$i-2$")
    gr.label_center_x(nzones/2-1, r"$i-1$")
    gr.label_center_x(nzones/2, r"$i$")
    gr.label_center_x(nzones/2+1, r"$i+1$")
    gr.label_center_x(nzones/2+2, r"$i+2$")

    gr.label_center_y(nzones/2-2, r"$j-2$")
    gr.label_center_y(nzones/2-1, r"$j-1$")
    gr.label_center_y(nzones/2, r"$j$")
    gr.label_center_y(nzones/2+1, r"$j+1$")
    gr.label_center_y(nzones/2+2, r"$j+2$")

    # axes
    gr.clean_axes()
    plt.subplots_adjust(left=0.025,right=0.98,bottom=0.1,top=0.98)

    f = plt.gcf()
    f.set_size_inches(7.0,7.0)

    plt.savefig("2dFV.png")
def simplegrid():

    nzones = 3
    gr = gp.FVGrid2d(nzones, nzones, ng=0)

    # plot a domain without ghostcells
    gr.draw_grid()

    #------------------------------------------------------------------------
    # label
    gr.label_cell_center(nzones // 2,
                         nzones // 2,
                         r"$a_{i,j}$",
                         fontsize="large")
    gr.label_cell_center(nzones // 2 + 1,
                         nzones // 2,
                         r"$a_{i+1,j}$",
                         fontsize="large")
    gr.label_cell_center(nzones // 2,
                         nzones // 2 + 1,
                         r"$a_{i,j+1}$",
                         fontsize="large")
    gr.label_cell_center(nzones // 2,
                         nzones // 2 - 1,
                         r"$a_{i,j-1}$",
                         fontsize="large")

    # i+1/2,j interface
    gr.mark_cell_left_state_x(nzones // 2,
                              nzones // 2,
                              r"$\hat{a}^{n+\myhalf}_{i+\myhalf,j,L}$",
                              color="b")

    # i,j+1/2 interface
    gr.mark_cell_state_y(nzones // 2, nzones // 2, r"${a}^T_{i,j+\myhalf}$")

    # i,j-1/2 interface
    gr.mark_cell_state_y(nzones // 2,
                         nzones // 2 - 1,
                         r"${a}^T_{i,j-\myhalf}$",
                         off_sign=-1)

    # helpful line showing the transverse bits
    plt.plot([gr.xc[nzones // 2], gr.xc[nzones // 2]], [
        gr.yl[nzones // 2] + 0.025 * gr.dy, gr.yr[nzones // 2] - 0.025 * gr.dy
    ],
             linestyle=":",
             color="0.5",
             zorder=1000)

    plt.plot([gr.xc[nzones // 2], gr.xr[nzones // 2] - 0.26 * gr.dx],
             [gr.yc[nzones // 2], gr.yc[nzones // 2]],
             linestyle=":",
             color="0.5",
             zorder=1000)

    plt.plot(
        [gr.xr[nzones // 2] - 0.34 * gr.dx, gr.xr[nzones // 2] - 0.26 * gr.dx],
        [gr.yc[nzones // 2] + 0.04 * gr.dy, gr.yc[nzones // 2]],
        linestyle=":",
        color="0.5",
        zorder=1000)

    plt.plot(
        [gr.xr[nzones // 2] - 0.34 * gr.dx, gr.xr[nzones // 2] - 0.26 * gr.dx],
        [gr.yc[nzones // 2] - 0.04 * gr.dy, gr.yc[nzones // 2]],
        linestyle=":",
        color="0.5",
        zorder=1000)

    # grid labels
    gr.label_center_x(nzones // 2 - 1, r"$i-1$")
    gr.label_center_x(nzones // 2, r"$i$")
    gr.label_center_x(nzones // 2 + 1, r"$i+1$")

    gr.label_center_y(nzones // 2 - 1, r"$j-1$")
    gr.label_center_y(nzones // 2, r"$j$")
    gr.label_center_y(nzones // 2 + 1, r"$j+1$")

    # axes
    gr.clean_axes()

    plt.subplots_adjust(left=0.02, right=0.98, bottom=0.02, top=0.98)

    f = plt.gcf()
    f.set_size_inches(7.0, 7.0)

    plt.savefig("2dgrid-transverse.pdf")