@@ -78,27 +78,27 @@ def unsplit_fluxes(my_data, rp, dt, scalar_name):
7878
7979 for index , vel in np .ndenumerate (u .v (buf = 1 )):
8080 if vel < 0 :
81- a_x .v (buf = 1 )[index ] = a .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ] - \
82- 0.5 * (1.0 + cx .v (buf = 1 )[index ]) * \
83- ldelta_ax .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]
81+ a_x .v (buf = 1 )[index ] = a .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ] \
82+ - 0.5 * (1.0 + cx .v (buf = 1 )[index ]) \
83+ * ldelta_ax .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]
8484 else :
85- a_x .v (buf = 1 )[index ] = a .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ] + \
86- 0.5 * (1.0 - cx .v (buf = 1 )[index ]) * \
87- ldelta_ax .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]
85+ a_x .v (buf = 1 )[index ] = a .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ] \
86+ + 0.5 * (1.0 - cx .v (buf = 1 )[index ]) \
87+ * ldelta_ax .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]
8888
8989 # upwind in y-direction
9090 a_y = myg .scratch_array ()
9191 shift_y = my_data .get_var ("y-shift" ).astype (int )
9292
9393 for index , vel in np .ndenumerate (v .v (buf = 1 )):
9494 if vel < 0 :
95- a_y .v (buf = 1 )[index ] = a .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ] - \
96- 0.5 * (1.0 + cy .v (buf = 1 )[index ]) * \
97- ldelta_ay .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]
95+ a_y .v (buf = 1 )[index ] = a .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ] \
96+ - 0.5 * (1.0 + cy .v (buf = 1 )[index ]) \
97+ * ldelta_ay .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]
9898 else :
99- a_y .v (buf = 1 )[index ] = a .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ] + \
100- 0.5 * (1.0 - cy .v (buf = 1 )[index ]) * \
101- ldelta_ay .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]
99+ a_y .v (buf = 1 )[index ] = a .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ] \
100+ + 0.5 * (1.0 - cy .v (buf = 1 )[index ]) \
101+ * ldelta_ay .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]
102102
103103 # compute the transverse flux differences. The flux is just (u a)
104104 # HOTF
@@ -114,13 +114,13 @@ def unsplit_fluxes(my_data, rp, dt, scalar_name):
114114 dtdy2 = 0.5 * dt / myg .dy
115115
116116 for index , vel in np .ndenumerate (u .v (buf = 1 )):
117- F_x .v (buf = 1 )[index ] = vel * (a_x .v (buf = 1 )[index ] -
118- dtdy2 * (F_yt .ip_jp (shift_x .v (buf = 1 )[index ], 1 , buf = 1 )[index ] -
119- F_yt .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]))
117+ F_x .v (buf = 1 )[index ] = vel * (a_x .v (buf = 1 )[index ] - dtdy2 *
118+ (F_yt .ip_jp (shift_x .v (buf = 1 )[index ], 1 , buf = 1 )[index ] -
119+ F_yt .ip (shift_x .v (buf = 1 )[index ], buf = 1 )[index ]))
120120
121121 for index , vel in np .ndenumerate (v .v (buf = 1 )):
122- F_y .v (buf = 1 )[index ] = vel * (a_y .v (buf = 1 )[index ] -
123- dtdx2 * (F_xt .ip_jp (1 , shift_y .v (buf = 1 )[index ], buf = 1 )[index ] -
124- F_xt .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]))
122+ F_y .v (buf = 1 )[index ] = vel * (a_y .v (buf = 1 )[index ] - dtdx2 *
123+ (F_xt .ip_jp (1 , shift_y .v (buf = 1 )[index ], buf = 1 )[index ] -
124+ F_xt .jp (shift_y .v (buf = 1 )[index ], buf = 1 )[index ]))
125125
126126 return F_x , F_y
0 commit comments