chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Revert "Limited shm frame count (#12346)" (#12362)

Browse Source 
This reverts commit 34812b7439.
This commit is contained in:
Nicolas Mowen
2024-07-09 10:49:08 -06:00
committed by GitHub
parent 34812b7439
commit f44df9fe61
4 changed files with 37 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
frigate/util/image.py
View File

@@ -687,31 +687,27 @@ class DictFrameManager(FrameManager):
class SharedMemoryFrameManager(FrameManager):
def __init__(self):
self.shm_store: dict[str, shared_memory.SharedMemory] = {}
self.shm_store = {}
def create(self, name: str, size) -> AnyStr:
def create(self, name, size) -> AnyStr:
shm = shared_memory.SharedMemory(name=name, create=True, size=size)
self.shm_store[name] = shm
return shm.buf
def get(self, name: str, shape) -> Optional[np.ndarray]:
try:
if name in self.shm_store:
shm = self.shm_store[name]
else:
shm = shared_memory.SharedMemory(name=name)
self.shm_store[name] = shm
return np.ndarray(shape, dtype=np.uint8, buffer=shm.buf)
except FileNotFoundError:
logger.error(f"Failed to get {name} from SHM")
return None
def get(self, name, shape):
if name in self.shm_store:
shm = self.shm_store[name]
else:
shm = shared_memory.SharedMemory(name=name)
self.shm_store[name] = shm
return np.ndarray(shape, dtype=np.uint8, buffer=shm.buf)
def close(self, name: str):
def close(self, name):
if name in self.shm_store:
self.shm_store[name].close()
del self.shm_store[name]
def delete(self, name: str):
def delete(self, name):
if name in self.shm_store:
self.shm_store[name].close()
self.shm_store[name].unlink()