import logging
from multiprocessing import Process, Queue
from trajpred.config import parser
from trajpred.prediction_server import InferenceServer, run_inference_server
from trajpred.socket_forwarder import run_ws_forwarder


logger = logging.getLogger("trajpred.plumbing")

def start():
    args = parser.parse_args()
    loglevel = logging.NOTSET if args.verbose > 1 else logging.DEBUG if args.verbose > 0  else logging.INFO
    logging.basicConfig(
        level=loglevel,
    )
    # rootLogger = logging.getLogger()
    # rootLogger.setLevel(loglevel)

    movement_q = Queue()
    prediction_q = Queue()


    # instantiating process with arguments
    procs = [
        Process(target=run_ws_forwarder, args=(args,))
    ]
    if not args.bypass_prediction:
        procs.append(
            Process(target=run_inference_server, args=(args, movement_q, prediction_q)),
        )

    logger.info("start")
    for proc in procs:
        proc.start()

    for proc in procs:
        proc.join()