alee/stegasoo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix decode progress getting stuck at 25%

Browse Source 
- Reduce PROGRESS_INTERVAL from 2000 to 500 for responsive updates
- Scale extraction progress to 5-70% range
- Add progress updates before/after RS decode (75% and 95%)
- RS decode is the slow part, now visible in progress

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Aaron D. Lee
2026-01-09 22:48:53 -05:00
parent 4e819b80cc
commit 5c685cba67
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/stegasoo/dct_steganography.py
View File

@@ -1232,8 +1232,8 @@ def _extract_scipy_dct_safe(
embed_rows = np.array([pos[0] for pos in DEFAULT_EMBED_POSITIONS]) embed_rows = np.array([pos[0] for pos in DEFAULT_EMBED_POSITIONS])
embed_cols = np.array([pos[1] for pos in DEFAULT_EMBED_POSITIONS]) embed_cols = np.array([pos[1] for pos in DEFAULT_EMBED_POSITIONS])
# Progress reporting interval # Progress reporting interval - report frequently for responsive UI
PROGRESS_INTERVAL = 2000 # Report every N blocks PROGRESS_INTERVAL = 500 # Report every N blocks (matches BATCH_SIZE)
block_idx = 0 block_idx = 0
while block_idx < len(block_order): while block_idx < len(block_order):
@@ -1264,9 +1264,10 @@ def _extract_scipy_dct_safe(
del blocks, dct_blocks, coeffs, quantized del blocks, dct_blocks, coeffs, quantized
block_idx = batch_end block_idx = batch_end
# Report progress # Report progress (scale to 5-70% range, RS decode gets 70-100%)
if progress_file and block_idx % PROGRESS_INTERVAL < BATCH_SIZE: if progress_file and block_idx % PROGRESS_INTERVAL < BATCH_SIZE:
_write_progress(progress_file, block_idx, num_blocks, "extracting") extract_pct = 5 + int(65 * block_idx / num_blocks)
_write_progress(progress_file, extract_pct, 100, "extracting")
# Check if we have enough bits (early exit) # Check if we have enough bits (early exit)
if len(all_bits) >= HEADER_SIZE * 8: if len(all_bits) >= HEADER_SIZE * 8:
@@ -1281,7 +1282,8 @@ def _extract_scipy_dct_safe(
del padded del padded
gc.collect() gc.collect()
_write_progress(progress_file, 80, 100, "decoding") # Extraction done, RS decode starts at 70%
_write_progress(progress_file, 70, 100, "decoding")
# Try RS-protected format first (has 24-byte length prefix: 3 copies of 8-byte header) # Try RS-protected format first (has 24-byte length prefix: 3 copies of 8-byte header)
if HAS_REEDSOLO and len(all_bits) >= RS_LENGTH_PREFIX_SIZE * 8: if HAS_REEDSOLO and len(all_bits) >= RS_LENGTH_PREFIX_SIZE * 8:
@@ -1335,10 +1337,16 @@ def _extract_scipy_dct_safe(
] ]
) )
# 75% - bits converted, starting RS decode (slow part)
_write_progress(progress_file, 75, 100, "decoding")
try: try:
# RS decode to get header + data # RS decode to get header + data
raw_payload = _rs_decode(rs_encoded) raw_payload = _rs_decode(rs_encoded)
# 95% - RS decode done
_write_progress(progress_file, 95, 100, "decoding")
# Parse header from decoded payload # Parse header from decoded payload
_, flags, data_length = _parse_header( _, flags, data_length = _parse_header(
[((raw_payload[i // 8] >> (7 - i % 8)) & 1) for i in range(HEADER_SIZE * 8)] [((raw_payload[i // 8] >> (7 - i % 8)) & 1) for i in range(HEADER_SIZE * 8)]
@@ -1454,8 +1462,9 @@ def _extract_jpegio(
) )
try: try:
_write_progress(progress_file, 70, 100, "decoding") _write_progress(progress_file, 75, 100, "decoding")
raw_payload = _rs_decode(rs_encoded) raw_payload = _rs_decode(rs_encoded)
_write_progress(progress_file, 95, 100, "decoding")
_, flags, data_length = _jpegio_parse_header(raw_payload[:HEADER_SIZE]) _, flags, data_length = _jpegio_parse_header(raw_payload[:HEADER_SIZE])
data = raw_payload[HEADER_SIZE : HEADER_SIZE + data_length] data = raw_payload[HEADER_SIZE : HEADER_SIZE + data_length]
_write_progress(progress_file, 100, 100, "complete") _write_progress(progress_file, 100, 100, "complete")