BasaltMeter/BasaltTrainer/main.py

from cmd_server import CmdServer, HandshakePayload, PixelKind
from while_stopper import INSTANCE as STOPPER
import logging


def receive_data() -> None:
    logging.info("Data received")


def main():
    server = CmdServer()

    print("Please launch BasaltPresenter now.")
    print("Then press Enter to continue...")
    input()

    logging.info("Waiting BasaltPresenter...")
    server.wait_handshake(HandshakePayload(PixelKind.GRAY_U8, 600, 600))

    logging.info("Start to running.")
    STOPPER.register()
    while True:
        if server.tick(receive_data, STOPPER.is_stop_requested()):
            break
    STOPPER.unregister()

    logging.info("Program stop.")


if __name__ == "__main__":
    logging.basicConfig(
        level=logging.DEBUG,
        format="[%(asctime)s] [%(levelname)s] %(message)s",
        datefmt="%Y-%m-%d %H:%M:%S",
    )
    main()