#include <sys/resource.h>
#include <limits.h>

#include <cstdio>
#include <cstdlib>

#include "cereal/visionipc/visionipc_client.h"
#include "common/params.h"
#include "common/swaglog.h"
#include "common/util.h"
#include "selfdrive/modeld/models/nav.h"

ExitHandler do_exit;

void run_model(NavModelState &model, VisionIpcClient &vipc_client) {
  SubMaster sm({"navInstruction"});
  PubMaster pm({"navModel"});

  //double last_ts = 0;
  //uint32_t last_frame_id = 0;
  VisionIpcBufExtra extra = {};

  while (!do_exit) {
    VisionBuf *buf = vipc_client.recv(&extra);
    if (buf == nullptr) continue;

    sm.update(0);

    double t1 = millis_since_boot();
    NavModelResult *model_res = navmodel_eval_frame(&model, buf);
    double t2 = millis_since_boot();

    // send navmodel packet
    navmodel_publish(pm, extra, *model_res, (t2 - t1) / 1000.0, sm["navInstruction"].getValid());

    //printf("navmodel process: %.2fms, from last %.2fms\n", t2 - t1, t1 - last_ts);
    //last_ts = t1;
    //last_frame_id = extra.frame_id;
  }
}

int main(int argc, char **argv) {
  setpriority(PRIO_PROCESS, 0, -15);

  // there exists a race condition when two processes try to create a
  // SNPE model runner at the same time, wait for dmonitoringmodeld to finish
  LOGW("waiting for dmonitoringmodeld to initialize");
  if (!Params().getBool("DmModelInitialized", true)) {
    return 0;
  }

  // init the models
  NavModelState model;
  navmodel_init(&model);
  LOGW("models loaded, navmodeld starting");

  VisionIpcClient vipc_client = VisionIpcClient("navd", VISION_STREAM_MAP, true);
  while (!do_exit && !vipc_client.connect(false)) {
    util::sleep_for(100);
  }

  // run the models
  if (vipc_client.connected) {
    LOGW("connected with buffer size: %zu", vipc_client.buffers[0].len);
    run_model(model, vipc_client);
  }

  navmodel_free(&model);
  return 0;
}