Skip to content

Commit f44971e

Browse files
refactor Lightness class to make sliding window length configurable
1 parent ab0d43a commit f44971e

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

pyeyesweb/mid_level/lightness.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,14 +8,22 @@
88

99

1010
class Lightness:
11-
sliding_window = SlidingWindow(50, 1) # Store velocity values
11+
# sliding_window = SlidingWindow(50, 1) # Store velocity values
1212
kinetic_energy = KineticEnergy()
1313
rarity = Rarity()
1414

15-
def __init__(self, rate_hz=50.0, use_filter=True, signal_type='velocity'):
15+
def __init__(self, sliding_window_max_length=50.0, rate_hz=50.0, use_filter=True, signal_type='velocity'):
16+
self.sliding_window_max_length = validate_numeric(
17+
sliding_window_max_length,
18+
'sliding_window_max_length',
19+
min_val=1,
20+
max_val=100000
21+
)
1622
self.rate_hz = validate_numeric(rate_hz, 'rate_hz', min_val=0.01, max_val=100000)
1723
self.use_filter = validate_boolean(use_filter, 'use_filter')
1824

25+
self.sliding_window = SlidingWindow(self.sliding_window_max_length, 1)
26+
1927
# Validate signal_type
2028
if signal_type not in ['velocity', 'position']:
2129
raise ValueError(f"signal_type must be 'velocity' or 'position', got '{signal_type}'")

0 commit comments

Comments
 (0)