mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-09-03 10:23:43 +08:00
royjr
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import os
|
|
from openpilot.common.file_chunker import open_file_chunked, get_existing_chunks
|
|
from openpilot.common.params import Params
|
|
|
|
PARAM = "UsbGpuLoadProgress"
|
|
|
|
|
|
class ProgressReader:
|
|
def __init__(self, inner, total):
|
|
self._inner = inner
|
|
self._total = total
|
|
self._params = Params()
|
|
self._read = 0
|
|
self._pct = -1
|
|
self._step = max(64 * 1024, total // 100)
|
|
|
|
def _bump(self, n):
|
|
self._read += n
|
|
if self._total:
|
|
pct = min(100, self._read * 100 // self._total)
|
|
if pct != self._pct:
|
|
self._pct = pct
|
|
self._params.put(PARAM, pct)
|
|
|
|
def read(self, size=-1):
|
|
data = self._inner.read(size)
|
|
self._bump(len(data))
|
|
return data
|
|
|
|
def readinto(self, b):
|
|
view = memoryview(b)
|
|
done = 0
|
|
while done < len(view):
|
|
n = self._inner.readinto(view[done:done + self._step])
|
|
if not n:
|
|
break
|
|
done += n
|
|
self._bump(n)
|
|
return done
|
|
|
|
|
|
def open_with_progress(pkl_path):
|
|
chunks = [p for p in get_existing_chunks(pkl_path) if not p.endswith(".chunkmanifest")]
|
|
total = sum(os.path.getsize(p) for p in chunks)
|
|
return ProgressReader(open_file_chunked(pkl_path), total)
|