/*
 *
 * This file is part of the open-source SeetaFace engine, which includes three modules:
 * SeetaFace Detection, SeetaFace Alignment, and SeetaFace Identification.
 *
 * This file is an example of how to use SeetaFace engine for face alignment, the
 * face alignment method described in the following paper:
 *
 *
 *   Coarse-to-Fine Auto-Encoder Networks (CFAN) for Real-Time Face Alignment, 
 *   Jie Zhang, Shiguang Shan, Meina Kan, Xilin Chen. In Proceeding of the
 *   European Conference on Computer Vision (ECCV), 2014
 *
 *
 * Copyright (C) 2016, Visual Information Processing and Learning (VIPL) group,
 * Institute of Computing Technology, Chinese Academy of Sciences, Beijing, China.
 *
 * The codes are mainly developed by Jie Zhang (a Ph.D supervised by Prof. Shiguang Shan)
 *
 * As an open-source face recognition engine: you can redistribute SeetaFace source codes
 * and/or modify it under the terms of the BSD 2-Clause License.
 *
 * You should have received a copy of the BSD 2-Clause License along with the software.
 * If not, see < https://opensource.org/licenses/BSD-2-Clause>.
 *
 * Contact Info: you can send an email to SeetaFace@vipl.ict.ac.cn for any problems.
 *
 * Note: the above information must be kept whenever or wherever the codes are used.
 *
 */

#include <cstdint>
#include <fstream>
#include <iostream>
#include <string>

#include "cv.h"
#include "highgui.h"

#include "face_alignment.h"

#ifdef _WIN32
std::string DATA_DIR = "../../data/";
std::string MODEL_DIR = "../../model/";
#else
std::string DATA_DIR = "./data/";
std::string MODEL_DIR = "./model/";
#endif

int main2(int argc, char** argv)
{
 
  const char* model_path = argv[1];
  const char* img_path = argv[2];
  const char* img_out_path = argv[3];
  int pos_x = atoi(argv[4]);
  int pos_y = atoi(argv[5]);
  int pos_w = atoi(argv[6]);
  int pos_h = atoi(argv[7]);

  // Initialize face alignment model 
  seeta::FaceAlignment point_detector(model_path);

  //load image
  IplImage *img_grayscale = NULL;
  img_grayscale = cvLoadImage(img_path, 0);
  if (img_grayscale == NULL)
  {
    return 0;
  }

  IplImage *img_color = cvLoadImage(img_path, 1);
  int pts_num = 5;
  int im_width = img_grayscale->width;
  int im_height = img_grayscale->height;
  unsigned char* data = new unsigned char[im_width * im_height];
  unsigned char* data_ptr = data;
  unsigned char* image_data_ptr = (unsigned char*)img_grayscale->imageData;
  int h = 0;
  for (h = 0; h < im_height; h++) {
    memcpy(data_ptr, image_data_ptr, im_width);
    data_ptr += im_width;
    image_data_ptr += img_grayscale->widthStep;
  }

  seeta::ImageData image_data;
  image_data.data = data;
  image_data.width = im_width;
  image_data.height = im_height;
  image_data.num_channels = 1;

  // Detect faces
  // std::vector<seeta::FaceInfo> faces = detector.Detect(image_data);
  // int32_t face_num = static_cast<int32_t>(faces.size());

  // if (face_num == 0)
  // {
  //   delete[]data;
  //   cvReleaseImage(&img_grayscale);
  //   cvReleaseImage(&img_color);
  //   return 0;
  // }

  // Detect 5 facial landmarks
  seeta::FaceInfo info;
  info.bbox.x = pos_x;
  info.bbox.y = pos_y;
  info.bbox.width = pos_w;
  info.bbox.height = pos_h;
  seeta::FacialLandmark points[5];
  point_detector.PointDetectLandmarks(image_data, info, points);

  // Visualize the results
  cvRectangle(img_color, cvPoint(info.bbox.x, info.bbox.y), cvPoint(info.bbox.x + info.bbox.width - 1, info.bbox.y + info.bbox.height - 1), CV_RGB(255, 0, 0));
  for (int i = 0; i<pts_num; i++)
  {
    cvCircle(img_color, cvPoint(points[i].x, points[i].y), 2, CV_RGB(0, 255, 0), CV_FILLED);
  }
  cvSaveImage(img_out_path, img_color);

  // Release memory
  cvReleaseImage(&img_color);
  cvReleaseImage(&img_grayscale);
  delete[]data;
  return 0;
}