/*
 * Copyright (C) 2019 Hertz Wang 1989wanghang@163.com
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see http://www.gnu.org/licenses
 *
 * Any non-GPL usage of this software or parts of this software is strictly
 * forbidden.
 *
 */

#define IN_RKNN_SSD_CC

#include <assert.h>

#include "npu_pp_output.h"

namespace NPU_UVC_SSD_DEMO {

#include "../../../../../../../external/rknpu/rknn/rknn_api/examples/rknn_ssd_demo/src/ssd.cc"

static SDLFont sdl_font(red, 16);
bool SSDDraw(SDL_Renderer *renderer, const SDL_Rect &render_rect,
             const SDL_Rect &coor_rect, int rotate,
             NPUPostProcessOutput *output, void *buffer, Uint32 sdl_fmt);
bool SSDDraw(SDL_Renderer *renderer, const SDL_Rect &render_rect,
             const SDL_Rect &coor_rect, int rotate,
             NPUPostProcessOutput *output, void *buffer, Uint32 sdl_fmt) {
  int npu_w = output->npuwh.width;
  int npu_h = output->npuwh.height;
  auto group = (NPU_UVC_SSD_DEMO::detect_result_group_t *)(output->pp_output);
  SDL_SetRenderDrawBlendMode(renderer, SDL_BLENDMODE_NONE);
  SDL_SetRenderDrawColor(renderer, 0xFF, 0x10, 0xEB, 0xFF);
  for (int i = 0; i < group->count; i++) {
    detect_result_t *det_result = &(group->results[i]);
    printf("%s @ (%d %d %d %d) %f\n", det_result->name, det_result->box.left,
           det_result->box.top, det_result->box.right, det_result->box.bottom,
           det_result->prop);
    int x1 = det_result->box.left;
    int y1 = det_result->box.top;
    int x2 = det_result->box.right;
    int y2 = det_result->box.bottom;
    SDL_Rect rect = {x1 * render_rect.w / npu_w + render_rect.x,
                     y1 * render_rect.h / npu_h + render_rect.y,
                     (x2 - x1) * render_rect.w / npu_w,
                     (y2 - y1) * render_rect.h / npu_h};
    SDL_Rect line_rect = transform(rect, coor_rect, rotate);
    int status = draw_rect(renderer, &line_rect, buffer, sdl_fmt);
    if (status)
      fprintf(stderr, "draw rect status: %d <%s>\n", status, SDL_GetError());
    if (!det_result->name)
      continue;
    int fontw = 0, fonth = 0;
    SDL_Surface *name = sdl_font.GetFontPicture(
        (char *)det_result->name, strlen(det_result->name), 32, &fontw, &fonth);
    if (name) {
      SDL_Rect texture_dimension;
      SDL_Texture *texture = load_texture(name, renderer, &texture_dimension);
      SDL_FreeSurface(name);
      SDL_Rect dst_dimension;
      dst_dimension.x = rect.x;
      dst_dimension.y = rect.y - 18;
      dst_dimension.w = texture_dimension.w;
      dst_dimension.h = texture_dimension.h;
      SDL_RenderCopyEx(renderer, texture, &texture_dimension, &dst_dimension,
                       rotate, NULL, SDL_FLIP_NONE);
      SDL_DestroyTexture(texture);
    }
  }
  return true;
}

} // namespace NPU_UVC_SSD_DEMO