Skip to content

Commit 18d9ed9

Browse files
Renaming get_fn_needs_length to the more accurate read_fn_needs_length.
1 parent a5b8ac9 commit 18d9ed9

File tree

4 files changed

+19
-19
lines changed

4 files changed

+19
-19
lines changed

bitstring/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -199,13 +199,13 @@ def bool_bits2chars(bitlength: Literal[1]):
199199
multiplier=8, description="a bytes object"),
200200
# Unknown length types
201201
DtypeDefinition('se', Bits._setse, Bits._getse, int, True, None,
202-
get_fn_needs_length=False, description="a signed exponential-Golomb code"),
202+
read_fn_needs_length=False, description="a signed exponential-Golomb code"),
203203
DtypeDefinition('ue', Bits._setue, Bits._getue, int, False, None,
204-
get_fn_needs_length=False, description="an unsigned exponential-Golomb code"),
204+
read_fn_needs_length=False, description="an unsigned exponential-Golomb code"),
205205
DtypeDefinition('sie', Bits._setsie, Bits._getsie, int, True, None,
206-
get_fn_needs_length=False, description="a signed interleaved exponential-Golomb code"),
206+
read_fn_needs_length=False, description="a signed interleaved exponential-Golomb code"),
207207
DtypeDefinition('uie', Bits._setuie, Bits._getuie, int, False, None,
208-
get_fn_needs_length=False, description="an unsigned interleaved exponential-Golomb code"),
208+
read_fn_needs_length=False, description="an unsigned interleaved exponential-Golomb code"),
209209
# Special case pad type
210210
DtypeDefinition('pad', Bits._setpad, Bits._getpad, None, False, None,
211211
description="a skipped section of padding"),

bitstring/bits.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1191,20 +1191,20 @@ def _read_dtype_list(self, dtypes: List[Dtype], pos: int) -> Tuple[List[Union[in
11911191
has_stretchy_token = False
11921192
bits_after_stretchy_token = 0
11931193
for dtype in dtypes:
1194-
stretchy = dtype.bitlength is None and dtype.get_fn_needs_length
1194+
stretchy = dtype.bitlength is None and dtype.read_fn_needs_length
11951195
if stretchy:
11961196
if has_stretchy_token:
11971197
raise bitstring.Error("It's not possible to have more than one 'filler' token.")
11981198
has_stretchy_token = True
11991199
elif has_stretchy_token:
1200-
if not dtype.get_fn_needs_length:
1200+
if not dtype.read_fn_needs_length:
12011201
raise bitstring.Error(f"It's not possible to parse a variable length token '{dtype}' after a 'filler' token.")
12021202
bits_after_stretchy_token += dtype.bitlength
12031203

12041204
# We should have precisely zero or one stretchy token
12051205
vals = []
12061206
for dtype in dtypes:
1207-
stretchy = dtype.bitlength is None and dtype.get_fn_needs_length
1207+
stretchy = dtype.bitlength is None and dtype.read_fn_needs_length
12081208
if stretchy:
12091209
bits_remaining = len(self) - pos
12101210
# Set length to the remaining bits
@@ -1660,9 +1660,9 @@ def _pp(self, dtype1: Dtype, dtype2: Optional[Dtype], bits_per_group: int, width
16601660
colour = Colour(bitstring.options.colourful_prettyprinting)
16611661
name1 = dtype1.name
16621662
name2 = dtype2.name if dtype2 is not None else None
1663-
if not dtype1.get_fn_needs_length:
1663+
if not dtype1.read_fn_needs_length:
16641664
raise ValueError(f"Can't use Dtype '{dtype1}' in pp() as it has an unknown length.")
1665-
if dtype2 is not None and not dtype2.get_fn_needs_length:
1665+
if dtype2 is not None and not dtype2.read_fn_needs_length:
16661666
raise ValueError(f"Can't use Dtype '{dtype1}' in pp() as it has an unknown length.")
16671667
offset_width = 0
16681668
offset_sep = ' :' if lsb0 else ': '

bitstring/bitstream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def read(self, fmt: Union[int, str, Dtype]) -> Union[int, float, str, Bits, bool
356356
self._pos += fmt
357357
return bs
358358
dtype = bitstring.dtypes.Dtype(fmt)
359-
if dtype.bitlength is None and dtype.get_fn_needs_length:
359+
if dtype.bitlength is None and dtype.read_fn_needs_length:
360360
# No length specified? Try again, but read to end.
361361
bitlength = len(self) - self._pos
362362
items, remainder = divmod(bitlength, dtype.bits_per_item)

bitstring/dtypes.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
class Dtype:
1313

14-
__slots__ = ('read_fn', 'name', 'set_fn', 'get_fn', 'return_type', 'is_signed', 'set_fn_needs_length', 'get_fn_needs_length', 'bitlength', 'bits_per_item', 'length')
14+
__slots__ = ('read_fn', 'name', 'set_fn', 'get_fn', 'return_type', 'is_signed', 'set_fn_needs_length', 'read_fn_needs_length', 'bitlength', 'bits_per_item', 'length')
1515

1616
def __new__(cls, token: Union[str, Dtype, None] = None, /, length: Optional[int] = None) -> Dtype:
1717
if isinstance(token, cls):
@@ -42,8 +42,8 @@ def create(cls, definition: DtypeDefinition, length: Optional[int]) -> Dtype:
4242
if x.bitlength is not None:
4343
x.bitlength *= x.bits_per_item
4444
x.set_fn_needs_length = definition.set_fn_needs_length
45-
x.get_fn_needs_length = definition.get_fn_needs_length
46-
if not x.get_fn_needs_length or len(dtype_register.names[x.name].allowed_lengths) == 1:
45+
x.read_fn_needs_length = definition.read_fn_needs_length
46+
if not x.read_fn_needs_length or len(dtype_register.names[x.name].allowed_lengths) == 1:
4747
x.read_fn = definition.read_fn
4848
else:
4949
x.read_fn = functools.partial(definition.read_fn, length=x.bitlength)
@@ -60,12 +60,12 @@ def create(cls, definition: DtypeDefinition, length: Optional[int]) -> Dtype:
6060
return x
6161

6262
def __str__(self) -> str:
63-
hide_length = not self.get_fn_needs_length or len(dtype_register.names[self.name].allowed_lengths) == 1 or self.length is None
63+
hide_length = not self.read_fn_needs_length or len(dtype_register.names[self.name].allowed_lengths) == 1 or self.length is None
6464
length_str = '' if hide_length else str(self.length)
6565
return f"{self.name}{length_str}"
6666

6767
def __repr__(self) -> str:
68-
hide_length = not self.get_fn_needs_length or len(dtype_register.names[self.name].allowed_lengths) == 1 or self.length is None
68+
hide_length = not self.read_fn_needs_length or len(dtype_register.names[self.name].allowed_lengths) == 1 or self.length is None
6969
length_str = '' if hide_length else ', ' + str(self.length)
7070
return f"{self.__class__.__name__}('{self.name}{length_str}')"
7171

@@ -103,7 +103,7 @@ class DtypeDefinition:
103103
# Represents a class of dtypes, such as uint or float, rather than a concrete dtype such as uint8.
104104

105105
def __init__(self, name: str, set_fn, get_fn, return_type: Any = Any, is_signed: bool = False, bitlength2chars_fn = None,
106-
get_fn_needs_length: bool = True, allowed_lengths: Tuple[int, ...] = tuple(), multiplier: int = 1, description: str = ''):
106+
read_fn_needs_length: bool = True, allowed_lengths: Tuple[int, ...] = tuple(), multiplier: int = 1, description: str = ''):
107107

108108
# Consistency checks
109109
# if not set_fn_needs_length and allowed_lengths:
@@ -115,7 +115,7 @@ def __init__(self, name: str, set_fn, get_fn, return_type: Any = Any, is_signed:
115115
self.description = description
116116
self.return_type = return_type
117117
self.is_signed = is_signed
118-
self.get_fn_needs_length = get_fn_needs_length
118+
self.read_fn_needs_length = read_fn_needs_length
119119
self.allowed_lengths = AllowedLengths(allowed_lengths)
120120

121121
self.multiplier = multiplier
@@ -134,7 +134,7 @@ def length_checked_get_fn(bs):
134134
self.get_fn = get_fn # Interpret everything
135135

136136
# Create a reading function from the get_fn.
137-
if self.get_fn_needs_length:
137+
if self.read_fn_needs_length:
138138
if len(self.allowed_lengths) == 1:
139139
def read_fn(bs, start):
140140
return self.get_fn(bs[start:start + self.allowed_lengths.value[0]])
@@ -173,7 +173,7 @@ def getDtype(self, length: Optional[int] = None) -> Dtype:
173173
if length is None:
174174
d = Dtype.create(self, None)
175175
return d
176-
if not self.get_fn_needs_length:
176+
if not self.read_fn_needs_length:
177177
raise ValueError(f"A length ({length}) shouldn't be supplied for the dtype '{self.name}'.")
178178
d = Dtype.create(self, length)
179179
return d

0 commit comments

Comments
 (0)