1
1
import numpy as np
2
2
3
3
from .bin_data import bin_data
4
- from .calculate_advection import calculate_advection
4
+ from .calculate_advection_2d import calculate_advection_2d
5
5
from .calculate_separation_distances import calculate_separation_distances
6
- from .calculate_structure_function import (
7
- calculate_structure_function ,
6
+ from .calculate_structure_function_2d import (
7
+ calculate_structure_function_2d ,
8
8
)
9
9
from .shift_array_1d import shift_array_1d
10
10
11
11
12
- def generate_structure_functions ( # noqa: C901, D417
12
+ def generate_structure_functions_2d ( # noqa: C901, D417
13
13
u ,
14
14
v ,
15
15
x = None ,
@@ -195,7 +195,7 @@ def generate_structure_functions( # noqa: C901, D417
195
195
if "ASF_V" in sf_type :
196
196
SF_adv_x = np .zeros (len (sep_x ) + 1 )
197
197
SF_adv_y = np .zeros (len (sep_y ) + 1 )
198
- adv_x , adv_y = calculate_advection (
198
+ adv_x , adv_y = calculate_advection_2d (
199
199
u ,
200
200
v ,
201
201
x ,
@@ -211,7 +211,7 @@ def generate_structure_functions( # noqa: C901, D417
211
211
if "ASF_S" in sf_type :
212
212
SF_x_scalar = np .zeros (len (sep_x ) + 1 )
213
213
SF_y_scalar = np .zeros (len (sep_y ) + 1 )
214
- adv_scalar = calculate_advection (
214
+ adv_scalar = calculate_advection_2d (
215
215
u , v , x , y , lats , lons , dx , dy , grid_type , scalar , sphere_circumference
216
216
)
217
217
if "LL" in sf_type :
@@ -243,7 +243,7 @@ def generate_structure_functions( # noqa: C901, D417
243
243
else :
244
244
xroll = shift_array_1d (x , shift_by = x_shift , boundary = None )
245
245
246
- SF_dicts = calculate_structure_function (
246
+ SF_dicts = calculate_structure_function_2d (
247
247
u ,
248
248
v ,
249
249
adv_x ,
@@ -301,7 +301,7 @@ def generate_structure_functions( # noqa: C901, D417
301
301
else :
302
302
yroll = shift_array_1d (y , shift_by = y_shift , boundary = None )
303
303
304
- SF_dicts = calculate_structure_function (
304
+ SF_dicts = calculate_structure_function_2d (
305
305
u ,
306
306
v ,
307
307
adv_x ,
0 commit comments