human/render/base_render.py

#encoding = utf8
import logging
import time
from abc import ABC, abstractmethod
from queue import Queue
from threading import Event, Thread

from utils import SyncQueue

logger = logging.getLogger(__name__)


class BaseRender(ABC):
    def __init__(self, play_clock, context, type_, delay=0.02):
        self._play_clock = play_clock
        self._context = context
        self._type = type_
        self._delay = delay
        self._queue = SyncQueue(context.batch_size)
        self._exit_event = Event()
        self._thread = Thread(target=self._on_run)
        self._exit_event.set()
        self._thread.start()

    def _on_run(self):
        logging.info(f'{self._type} render run')
        while self._exit_event.is_set():
            self._run_step()
            time.sleep(self._delay)

        logging.info(f'{self._type} render exit')

    def put(self, frame, ps):
        self._queue.put((frame, ps))

    def size(self):
        return self._queue.size()

    def pause_talk(self):
        self._queue.clear()

    def stop(self):
        self._queue.clear()
        self._exit_event.clear()
        self._thread.join()

    @abstractmethod
    def _run_step(self):
        pass