/* -*- C++ -*-
 * File: halt_mt_win32.c
 * Copyright 2008-2016 LibRaw LLC (info@libraw.org)
 * Created: Sat Mar  8, 2008
 *
 * LibRaw  C API mutithreaded sample:  emulates call to "dcraw  -h [-w] [-a] [-v]"
 * Win32 version

LibRaw is free software; you can redistribute it and/or modify
it under the terms of the one of two licenses as you choose:

1. GNU LESSER GENERAL PUBLIC LICENSE version 2.1
   (See file LICENSE.LGPL provided in LibRaw distribution archive for details).

2. COMMON DEVELOPMENT AND DISTRIBUTION LICENSE (CDDL) Version 1.0
   (See file LICENSE.CDDL provided in LibRaw distribution archive for details).


 */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <windows.h>
#include "libraw/libraw.h"

#ifdef WIN32
#define snprintf _snprintf
#endif

#define HANDLE_ERRORS(ret)                                                                                             \
  do                                                                                                                   \
  {                                                                                                                    \
    if (ret)                                                                                                           \
    {                                                                                                                  \
      fprintf(stderr, "%s: %s\n", fn, libraw_strerror(ret));                                                           \
      if (LIBRAW_FATAL_ERROR(ret))                                                                                     \
      {                                                                                                                \
        libraw_close(iprc);                                                                                            \
        return -1;                                                                                                     \
      }                                                                                                                \
    }                                                                                                                  \
  } while (0)

// global settings
int verbose = 0, use_camera_wb = 0, use_auto_wb = 0, tiff_mode = 0;

// global file queue
HANDLE qmutex;
char **queue = NULL;
size_t qsize = 0, qptr = 0;

char *get_next_file()
{
  char *ret;
  DWORD dwWaitResult;
  if (!queue)
    return NULL;
  if (qptr >= qsize)
    return NULL;

  dwWaitResult = WaitForSingleObject(qmutex,    // handle to mutex
                                     INFINITE); // no time-out interval
  switch (dwWaitResult)
  {
  // The thread got ownership of the mutex
  case WAIT_OBJECT_0:
    ret = queue[qptr++];
    ReleaseMutex(qmutex);
    break;
  case WAIT_ABANDONED:
    return NULL; // cannot obtain the lock
  };
  return ret;
}

// thread routine
int process_files(void *q)
{
  int ret;
  int count = 0;
  char outfn[1024], *fn;
  libraw_data_t *iprc = libraw_init(0);

  if (!iprc)
  {
    fprintf(stderr, "Cannot create libraw handle\n");
    return -1;
  }

  while ((fn = get_next_file()))
  {

    iprc->params.half_size = 1; /* dcraw -h */
    iprc->params.use_camera_wb = use_camera_wb;
    iprc->params.use_auto_wb = use_auto_wb;
    iprc->params.output_tiff = tiff_mode;

    ret = libraw_open_file(iprc, fn);
    if (verbose)
      fprintf(stderr, "%s: %s/%s\n", fn, iprc->idata.make, iprc->idata.model);
    HANDLE_ERRORS(ret);

    ret = libraw_unpack(iprc);
    HANDLE_ERRORS(ret);

    ret = libraw_dcraw_process(iprc);
    HANDLE_ERRORS(ret);

    snprintf(outfn, 1023, "%s.%s", fn, tiff_mode ? "tif" : "ppm");

    if (verbose)
      fprintf(stderr, "Writing file %s\n", outfn);
    ret = libraw_dcraw_ppm_tiff_writer(iprc, outfn);
    HANDLE_ERRORS(ret);
    count++;
  }
  libraw_close(iprc);
  printf("Processed %d files\n", count);
  return 0;
}

void usage(const char *p)
{
  printf("Options:\n"
         "-J n  - set parrallel job coun (default 2)\n"
         "-v    - verbose\n"
         "-w    - use camera white balance\n"
         "-T    - output TIFF instead of PPM\n"
         "-a    - average image for white balance\n");
  exit(1);
}

int show_files(void *q)
{
  char *p;
  int cnt = 0;
  while (p = get_next_file())
  {
    printf("%s\n", p);
    cnt++;
  }
  return cnt;
}

int main(int ac, char *av[])
{
  int i, max_threads = 2;
  HANDLE *threads;
  DWORD ThreadID;

  if (ac < 2)
    usage(av[0]);

  queue = calloc(ac - 1, sizeof(queue[0]));

  for (i = 1; i < ac; i++)
  {
    if (av[i][0] == '-')
    {
      if (av[i][1] == 'w')
        use_camera_wb = 1;
      if (av[i][1] == 'a')
        use_auto_wb = 1;
      if (av[i][1] == 'v')
        verbose = 1;
      if (av[i][1] == 'T')
        tiff_mode = 1;
      if (av[i][1] == 'J')
      {
        max_threads = atoi(av[++i]);
        if (max_threads < 1)
        {
          fprintf(stderr, "Job count should be at least 1\n");
          exit(1);
        }
      }
    }
    else
      queue[qsize++] = av[i];
  }
  qmutex = CreateMutex(NULL, FALSE, NULL);
  threads = calloc(max_threads, sizeof(threads[0]));
  for (i = 0; i < max_threads; i++)
  {

    if (NULL == (threads[i] = CreateThread(NULL, // default security attributes
                                           0,    // default stack size
                                           (LPTHREAD_START_ROUTINE)process_files,
                                           NULL,      // no thread function arguments
                                           0,         // default creation flags
                                           &ThreadID) // receive thread identifier
                 ))
    {
      printf("CreateThread error: %d\n", GetLastError());
      return 1;
    }
  }

  WaitForMultipleObjects(max_threads, threads, TRUE, INFINITE);

  // Close thread and mutex handles

  for (i = 0; i < max_threads; i++)
    CloseHandle(threads[i]);

  CloseHandle(qmutex);

  return 0;
}