Commit 72935072 authored by Gurecky, William's avatar Gurecky, William
Browse files

add checks on outlet velocity and flow rate in rom cfd preprocessing

parent a1e7656e

dthydro_cfd/rom_cfd/preprocess_cfd.py

+42 −14
Original line number Diff line number Diff line
@@ -53,20 +53,20 @@ run_csv_files = [ 
        "alpha-9_runner.csv",

        "alpha-10_runner.csv",

        ]

# run_csv_files = [

# #         "alpha-01_spiral-casing-and-draft-tube.csv",

# #         "alpha-02_spiral-casing-and-draft-tube.csv",

# #         "alpha-03_spiral-casing-and-draft-tube.csv",

# #         "alpha-04_spiral-casing-and-draft-tube.csv",

# #         "alpha-05_spiral-casing-and-draft-tube.csv",

# #         "alpha-06_spiral-casing-and-draft-tube.csv",

# #         "alpha-07_spiral-casing-and-draft-tube.csv",

# #         "alpha-08_spiral-casing-and-draft-tube.csv",

# #         "alpha-09_spiral-casing-and-draft-tube.csv",

# #         "alpha-10_spiral-casing-and-draft-tube.csv",

#           "alpha-9_casing-draft_tube.csv",

#           "alpha-10_casing-draft_tube.csv",

#          ]

run_csv_files = [

#         "alpha-01_spiral-casing-and-draft-tube.csv",

#         "alpha-02_spiral-casing-and-draft-tube.csv",

#         "alpha-03_spiral-casing-and-draft-tube.csv",

#         "alpha-04_spiral-casing-and-draft-tube.csv",

#         "alpha-05_spiral-casing-and-draft-tube.csv",

#         "alpha-06_spiral-casing-and-draft-tube.csv",

#         "alpha-07_spiral-casing-and-draft-tube.csv",

#         "alpha-08_spiral-casing-and-draft-tube.csv",

#         "alpha-09_spiral-casing-and-draft-tube.csv",

#         "alpha-10_spiral-casing-and-draft-tube.csv",

          "alpha-9_casing-draft_tube.csv",

          "alpha-10_casing-draft_tube.csv",

         ]

# run_csv_files = [

#         # "alpha-01_distributor.csv",

#         # "alpha-02_distributor.csv",
@@ -107,6 +107,23 @@ def plot_cfd_vel_field(mesh_x, mesh_y, mesh_z, v_i, v_j, v_k, cbar_label=""): 
    v_mag = np.sqrt(v_i ** 2 + v_j ** 2 + v_k ** 2)

    print("min vmag:", np.min(v_mag))

    print("max vmag:", np.max(v_mag))



    outlet_x_bounds = [-4, 4]

    outlet_y_bounds = [-2.5, -2.4]

    outlet_z_bounds = [-4, 4]



    outlet_idxs = np.argwhere(

            (outlet_x_bounds[0] < mesh_x) & (mesh_x < outlet_x_bounds[1]) & \

            (outlet_y_bounds[0] < mesh_y) & (mesh_y < outlet_y_bounds[1]) & \

            (outlet_z_bounds[0] < mesh_z) & (mesh_z < outlet_z_bounds[1]) \

            )

    v_j_outlet = v_j[outlet_idxs]

    mean_v_j_outlet = np.mean(v_j_outlet)

    print("mean y vel outlet (m/s):", mean_v_j_outlet)

    # vel * area:  (m/s) * m^2 = m^3/s

    outlet_area = np.pi * (3.0 / 2.) ** 2.

    print("mean y vol flow outlet (m^3/s):", mean_v_j_outlet * outlet_area)



    fig_v = go.Figure(data=go.Cone(

        x=mesh_x[subsample_idxs].flatten(),

        y=mesh_y[subsample_idxs].flatten(),
@@ -130,6 +147,17 @@ def plot_cfd_vel_field(mesh_x, mesh_y, mesh_z, v_i, v_j, v_k, cbar_label=""): 
        hovertemplate=('%{customdata}%'),

        ))

    fig_p.show()

    fig_ph = go.Figure(data=go.Scatter3d(

        x=mesh_x[outlet_idxs].flatten(),

        y=mesh_y[outlet_idxs].flatten(),

        z=mesh_z[outlet_idxs].flatten(),

        mode='markers',

        marker=dict(size=4, color=v_j[outlet_idxs], opacity=0.95,

                    colorbar=dict(thickness=5, title=cbar_label+"CFD Vel[j] Magnitude (m/s)")),

        customdata=v_j[outlet_idxs],

        hovertemplate=('%{customdata}%'),

        ))

    fig_ph.show()





def plot_cfd_pres_field(mesh_x, mesh_y, mesh_z, p_s):