face_cropper_dnd.cpp

#include "face_cropper.h"
#include "functions.h"
#include "read_srcs.h"

#include <omp.h>
#include <sys/stat.h>

#ifdef __CYGWIN__
#include <sys/cygwin.h>
#endif

int main(int argc, char *argv[]) {
  std::string date_suffix(get_date_suffix());
  struct src_set sst;

  std::string dir_str = "c:\\cygwin64\\";
  if (argc < 2) {
    // nop
    std::cout
        << "Give command line arguements or Drag and Drop files or directories."
        << std::endl;
    return 0;
  } else {
    for (int i = 1; i < argc; ++i) {
      dir_str = argv[i];
      std::cerr << dir_str;
#ifdef __CYGWIN__
      ssize_t size;
      char *posix;
      size = cygwin_conv_path(CCP_WIN_A_TO_POSIX | CCP_PROC_CYGDRIVE,
                              dir_str.c_str(), NULL, 0);
      if (size < 0) {
        perror("cygwin_conv_path");
      } else {
        posix = (char *)malloc(size);
        if (cygwin_conv_path(CCP_WIN_A_TO_POSIX | CCP_PROC_CYGDRIVE,
                             dir_str.c_str(), posix, size))
          perror("cygwin_conv_path");
      }
      dir_str = posix;
      std::cerr << "\t" << dir_str << std::endl;
      free(posix);
#endif
      sst.add(dir_str);
    }
  }

  create_all(date_suffix, sst);

  std::string error_file_name = "face_recognition_error" + date_suffix + ".txt";
  std::string result_file_name = "result_metric" + date_suffix + ".txt";

  std::string buf;
  // read file list
  std::ofstream ofs;
  ofs.open(result_file_name.c_str());
  //        file_list.push_back(buf);
  std::ofstream ofs_error;
  ofs_error.open(error_file_name.c_str());

// #pragma omp parallel for private(read_img_name, write_img_name, img_color,
// recog_ok, cropper_obj) private(read_img_name, write_img_name, img_color,
// recog_ok)
#pragma omp parallel  // num_threads(8)
  {
    // cropper cropper_obj;
    face_cropper cropper;
    // std::stringstream obuf;

    std::string filename, read_img_name, write_img_name, dest_dirname;
    std::ostringstream oss;
    int thread_num = omp_get_thread_num();
    bool read_recognize_error = false;

    // #pragma omp for
    for (auto src : sst.srcs) {
      if (src.type == 1) {
#pragma omp single
        {
          // TODO: treat single file
          oss.str("");
          read_img_name = src.filename;
          read_recognize_error = false;
#pragma omp critical
          {
            std::cerr << "th" << thread_num << ": " << read_img_name
                      << std::endl;
          }
          write_img_name = append_date_suffix(read_img_name, date_suffix);

          read_recognize_error =
              cropper.detect_and_output(read_img_name, write_img_name, oss);
#pragma omp critical
          {
            if (read_recognize_error) {
              ofs_error << "error: " << src.filename << std::endl;
              read_recognize_error = false;
            }
            std::cerr << oss.str() << std::endl;
            ofs << oss.str() << std::endl;
          }
        }
      } else if (src.type == 2) {
        dest_dirname =
            src.name + "/../" + get_dirname(src.name) + date_suffix + "/";
#pragma omp for
        for (int i = 0; i < src.dir.filelist.size(); ++i) {
          oss.str("");
#pragma omp critical
          {
            std::cerr << "th" << thread_num << ": " << src.dir.filelist[i]
                      << std::endl;
          }

          read_img_name = src.name + "/" + src.dir.filelist[i];
          write_img_name = dest_dirname + src.dir.filelist[i];

          read_recognize_error =
              cropper.detect_and_output(read_img_name, write_img_name, oss);
#pragma omp critical
          {
            if (read_recognize_error) {
              ofs_error << "error: " << src.filename << std::endl;
              read_recognize_error = false;
            }
            std::cerr << oss.str() << std::endl;
            // obuf << oss.str() << std::endl;
            ofs << oss.str() << std::endl << std::flush;
          }
        }
      }
    }
    // #pragma omp critical
    // {
    //                 ofs << obuf.str();
    // }
  }  // end omp parallel

  return 0;
}