Skip to content

Commit 53fbae5

Browse files
committed
Update logic
1 parent 5840289 commit 53fbae5

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

src/mdio/segy/_disaster_recovery_wrapper.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,17 +13,25 @@
1313
def _reverse_single_transform(data: NDArray, transform: Transform) -> NDArray:
1414
"""Reverse a single transform operation."""
1515
from segy.schema import Endianness
16+
from segy.transforms import ByteSwapTransform
17+
from segy.transforms import IbmFloatTransform
1618

1719
if isinstance(transform, ByteSwapTransform):
1820
# Reverse the endianness conversion
1921
# TODO: I don't think this is correct
20-
if transform.target_order == Endianness.LITTLE:
21-
reverse_target = Endianness.BIG
22-
else:
22+
if transform.target_order == Endianness.BIG:
2323
reverse_target = Endianness.LITTLE
24+
reverse_transform = ByteSwapTransform(reverse_target)
25+
return reverse_transform.apply(data)
26+
return data
2427

25-
reverse_transform = ByteSwapTransform(reverse_target)
26-
return reverse_transform.apply(data)
28+
# if transform.target_order == Endianness.LITTLE:
29+
# reverse_target = Endianness.BIG
30+
# else:
31+
# reverse_target = Endianness.LITTLE
32+
33+
# reverse_transform = ByteSwapTransform(reverse_target)
34+
# return reverse_transform.apply(data)
2735

2836
elif isinstance(transform, IbmFloatTransform):
2937
# Reverse IBM float conversion

0 commit comments

Comments
 (0)