Merge commit '638ecc4cde79b0296f2e04f683ca4cb881c28f36' as 'rtengine/libraw'
This commit is contained in:
Lawrence Lee
212
rtengine/libraw/samples/half_mt_win32.c
Normal file
212
rtengine/libraw/samples/half_mt_win32.c
Normal file
@@ -0,0 +1,212 @@
|
||||
/* -*- C++ -*-
|
||||
* File: halt_mt_win32.c
|
||||
* Copyright 2008-2021 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 LIBRAW_WIN32_CALLS
|
||||
#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 parallel job count (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;
|
||||
}
|
||||
Reference in New Issue
Block a user