Skip to content

Commit d9fd597

Browse files
committed
Use correct coordinate names
1 parent ba487bb commit d9fd597

6 files changed

Lines changed: 36 additions & 36 deletions

File tree

src/mdio/builder/templates/seismic_3d_obn.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@ def __init__(self, data_domain: SeismicDataDomain = "time"):
2727
self._spatial_dim_names = ("component", "receiver", "shot_line", "gun", "shot_point")
2828
self._dim_names = (*self._spatial_dim_names, self._data_domain)
2929
self._physical_coord_names = (
30-
"receiver_coord_x",
31-
"receiver_coord_y",
30+
"group_coord_x",
31+
"group_coord_y",
3232
"source_coord_x",
3333
"source_coord_y",
3434
)
@@ -78,16 +78,16 @@ def _add_coordinates(self) -> None:
7878

7979
# Add non-dimension coordinates
8080
self._builder.add_coordinate(
81-
"receiver_coord_x",
81+
"group_coord_x",
8282
dimensions=("receiver",),
8383
data_type=ScalarType.FLOAT64,
84-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_x")),
84+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_x")),
8585
)
8686
self._builder.add_coordinate(
87-
"receiver_coord_y",
87+
"group_coord_y",
8888
dimensions=("receiver",),
8989
data_type=ScalarType.FLOAT64,
90-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_y")),
90+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_y")),
9191
)
9292
self._builder.add_coordinate(
9393
"source_coord_x",

src/mdio/builder/templates/seismic_3d_obn_single_component.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@ def __init__(self, data_domain: SeismicDataDomain = "time"):
2727
self._spatial_dim_names = ("receiver", "shot_line", "gun", "shot_point")
2828
self._dim_names = (*self._spatial_dim_names, self._data_domain)
2929
self._physical_coord_names = (
30-
"receiver_coord_x",
31-
"receiver_coord_y",
30+
"group_coord_x",
31+
"group_coord_y",
3232
"source_coord_x",
3333
"source_coord_y",
3434
)
@@ -73,16 +73,16 @@ def _add_coordinates(self) -> None:
7373

7474
# Add non-dimension coordinates
7575
self._builder.add_coordinate(
76-
"receiver_coord_x",
76+
"group_coord_x",
7777
dimensions=("receiver",),
7878
data_type=ScalarType.FLOAT64,
79-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_x")),
79+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_x")),
8080
)
8181
self._builder.add_coordinate(
82-
"receiver_coord_y",
82+
"group_coord_y",
8383
dimensions=("receiver",),
8484
data_type=ScalarType.FLOAT64,
85-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_y")),
85+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_y")),
8686
)
8787
self._builder.add_coordinate(
8888
"source_coord_x",

src/mdio/builder/templates/seismic_3d_shot_receiver_line.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ def __init__(self, data_domain: SeismicDataDomain = "time"):
2929
self._physical_coord_names = (
3030
"source_coord_x",
3131
"source_coord_y",
32-
"receiver_coord_x",
33-
"receiver_coord_y",
32+
"group_coord_x",
33+
"group_coord_y",
3434
)
3535
self._logical_coord_names = ("orig_field_record_num",)
3636
self._var_chunk_shape = (1, 32, 1, 32, 2048)
@@ -85,16 +85,16 @@ def _add_coordinates(self) -> None:
8585
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("source_coord_y")),
8686
)
8787
self._builder.add_coordinate(
88-
"receiver_coord_x",
88+
"group_coord_x",
8989
dimensions=("receiver_line", "receiver"),
9090
data_type=ScalarType.FLOAT64,
91-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_x")),
91+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_x")),
9292
)
9393
self._builder.add_coordinate(
94-
"receiver_coord_y",
94+
"group_coord_y",
9595
dimensions=("receiver_line", "receiver"),
9696
data_type=ScalarType.FLOAT64,
97-
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("receiver_coord_y")),
97+
metadata=CoordinateMetadata(units_v1=self.get_unit_by_key("group_coord_y")),
9898
)
9999
self._builder.add_coordinate(
100100
"orig_field_record_num",

tests/unit/v1/templates/test_seismic_3d_obn.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@
2828
"time": "int32",
2929
}
3030
EXPECTED_COORDINATES = [
31-
"receiver_coord_x",
32-
"receiver_coord_y",
31+
"group_coord_x",
32+
"group_coord_y",
3333
"source_coord_x",
3434
"source_coord_y",
3535
"orig_field_record_num",
@@ -70,7 +70,7 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
7070
)
7171

7272
# Verify receiver coordinate variables (indexed by receiver only)
73-
for coord_name in ["receiver_coord_x", "receiver_coord_y"]:
73+
for coord_name in ["group_coord_x", "group_coord_y"]:
7474
coord = validate_variable(
7575
dataset,
7676
name=coord_name,
@@ -113,8 +113,8 @@ def test_configuration(self) -> None:
113113
assert t.name == "ObnReceiverGathers3D"
114114
assert t._dim_names == ("component", "receiver", "shot_line", "gun", "shot_point", "time")
115115
assert t._physical_coord_names == (
116-
"receiver_coord_x",
117-
"receiver_coord_y",
116+
"group_coord_x",
117+
"group_coord_y",
118118
"source_coord_x",
119119
"source_coord_y",
120120
)
@@ -170,7 +170,7 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
170170
t = Seismic3DObnReceiverGathersTemplate(data_domain="time")
171171

172172
# Add units
173-
t.add_units({"receiver_coord_x": UNITS_METER, "receiver_coord_y": UNITS_METER})
173+
t.add_units({"group_coord_x": UNITS_METER, "group_coord_y": UNITS_METER})
174174
t.add_units({"source_coord_x": UNITS_METER, "source_coord_y": UNITS_METER})
175175
t.add_units({"time": UNITS_SECOND})
176176

tests/unit/v1/templates/test_seismic_3d_obn_single_component.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
"time": "int32",
2727
}
2828
EXPECTED_COORDINATES = [
29-
"receiver_coord_x",
30-
"receiver_coord_y",
29+
"group_coord_x",
30+
"group_coord_y",
3131
"source_coord_x",
3232
"source_coord_y",
3333
"orig_field_record_num",
@@ -68,7 +68,7 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
6868
)
6969

7070
# Verify receiver coordinate variables
71-
for coord_name in ["receiver_coord_x", "receiver_coord_y"]:
71+
for coord_name in ["group_coord_x", "group_coord_y"]:
7272
coord = validate_variable(
7373
dataset,
7474
name=coord_name,
@@ -101,8 +101,8 @@ def test_configuration(self) -> None:
101101
assert t.name == "ObnSingleComponentGathers3D"
102102
assert t._dim_names == ("receiver", "shot_line", "gun", "shot_point", "time")
103103
assert t._physical_coord_names == (
104-
"receiver_coord_x",
105-
"receiver_coord_y",
104+
"group_coord_x",
105+
"group_coord_y",
106106
"source_coord_x",
107107
"source_coord_y",
108108
)
@@ -149,7 +149,7 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
149149
"""Test building a complete dataset with the template."""
150150
t = Seismic3DObnSingleComponentGathersTemplate(data_domain="time")
151151

152-
t.add_units({"receiver_coord_x": UNITS_METER, "receiver_coord_y": UNITS_METER})
152+
t.add_units({"group_coord_x": UNITS_METER, "group_coord_y": UNITS_METER})
153153
t.add_units({"source_coord_x": UNITS_METER, "source_coord_y": UNITS_METER})
154154
t.add_units({"time": UNITS_SECOND})
155155

tests/unit/v1/templates/test_seismic_3d_shot_receiver_line.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@
3434
EXPECTED_COORDINATES = [
3535
"source_coord_x",
3636
"source_coord_y",
37-
"receiver_coord_x",
38-
"receiver_coord_y",
37+
"group_coord_x",
38+
"group_coord_y",
3939
"orig_field_record_num",
4040
]
4141

@@ -87,7 +87,7 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
8787

8888
# Verify receiver coordinate variables
8989
receiver_dims = [(k, v) for k, v in DATASET_SIZE_MAP.items() if k in ["receiver_line", "receiver"]]
90-
for coord_name in ["receiver_coord_x", "receiver_coord_y"]:
90+
for coord_name in ["group_coord_x", "group_coord_y"]:
9191
coord = validate_variable(
9292
dataset,
9393
name=coord_name,
@@ -110,8 +110,8 @@ def test_configuration(self) -> None:
110110
assert t._physical_coord_names == (
111111
"source_coord_x",
112112
"source_coord_y",
113-
"receiver_coord_x",
114-
"receiver_coord_y",
113+
"group_coord_x",
114+
"group_coord_y",
115115
)
116116
assert t._logical_coord_names == ("orig_field_record_num",)
117117
assert t._var_chunk_shape == (1, 32, 1, 32, 2048)
@@ -157,7 +157,7 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
157157
t = Seismic3DShotReceiverLineGathersTemplate(data_domain="time")
158158

159159
t.add_units({"source_coord_x": UNITS_METER, "source_coord_y": UNITS_METER})
160-
t.add_units({"receiver_coord_x": UNITS_METER, "receiver_coord_y": UNITS_METER})
160+
t.add_units({"group_coord_x": UNITS_METER, "group_coord_y": UNITS_METER})
161161
t.add_units({"time": UNITS_SECOND})
162162

163163
sizes = tuple(DATASET_SIZE_MAP.values())

0 commit comments

Comments
 (0)