megapixels/src/process_pipeline.c

761 lines
20 KiB
C
Raw Normal View History

#include "process_pipeline.h"
#include "pipeline.h"
#include "zbar_pipeline.h"
#include "io_pipeline.h"
#include "main.h"
#include "config.h"
#include "gles2_debayer.h"
#include <tiffio.h>
#include <assert.h>
#include <math.h>
#include <wordexp.h>
#include <gtk/gtk.h>
#include "gl_util.h"
2021-04-10 14:39:55 +02:00
#include <sys/mman.h>
#include <drm/drm_fourcc.h>
2021-04-09 14:57:14 +02:00
#define TIFFTAG_FORWARDMATRIX1 50964
2020-12-06 15:44:31 +01:00
static const float colormatrix_srgb[] = { 3.2409, -1.5373, -0.4986, -0.9692, 1.8759,
0.0415, 0.0556, -0.2039, 1.0569 };
static MPPipeline *pipeline;
static char burst_dir[23];
static char processing_script[512];
static volatile bool is_capturing = false;
static volatile int frames_processed = 0;
static volatile int frames_received = 0;
static const struct mp_camera_config *camera;
static MPCameraMode mode;
static int burst_length;
static int captures_remaining = 0;
static int preview_width;
static int preview_height;
// static bool gain_is_manual;
static int gain;
static int gain_max;
static bool exposure_is_manual;
static int exposure;
static char capture_fname[255];
static void
register_custom_tiff_tags(TIFF *tif)
{
2020-12-06 15:44:31 +01:00
static const TIFFFieldInfo custom_fields[] = {
{ TIFFTAG_FORWARDMATRIX1, -1, -1, TIFF_SRATIONAL, FIELD_CUSTOM, 1, 1,
"ForwardMatrix1" },
};
// Add missing dng fields
TIFFMergeFieldInfo(tif, custom_fields,
sizeof(custom_fields) / sizeof(custom_fields[0]));
}
static bool
find_processor(char *script)
{
2020-12-06 15:44:31 +01:00
char *xdg_config_home;
char filename[] = "postprocess.sh";
wordexp_t exp_result;
// Resolve XDG stuff
if ((xdg_config_home = getenv("XDG_CONFIG_HOME")) == NULL) {
xdg_config_home = "~/.config";
}
wordexp(xdg_config_home, &exp_result, 0);
xdg_config_home = strdup(exp_result.we_wordv[0]);
wordfree(&exp_result);
// Check postprocess.h in the current working directory
sprintf(script, "data/%s", filename);
2020-12-06 15:44:31 +01:00
if (access(script, F_OK) != -1) {
sprintf(script, "./data/%s", filename);
2020-12-06 15:44:31 +01:00
printf("Found postprocessor script at %s\n", script);
return true;
}
// Check for a script in XDG_CONFIG_HOME
sprintf(script, "%s/megapixels/%s", xdg_config_home, filename);
if (access(script, F_OK) != -1) {
printf("Found postprocessor script at %s\n", script);
return true;
}
// Check user overridden /etc/megapixels/postprocessor.sh
sprintf(script, "%s/megapixels/%s", SYSCONFDIR, filename);
if (access(script, F_OK) != -1) {
printf("Found postprocessor script at %s\n", script);
return true;
}
// Check packaged /usr/share/megapixels/postprocessor.sh
sprintf(script, "%s/megapixels/%s", DATADIR, filename);
if (access(script, F_OK) != -1) {
printf("Found postprocessor script at %s\n", script);
return true;
}
return false;
}
2020-12-06 15:44:31 +01:00
static void
setup(MPPipeline *pipeline, const void *data)
{
2020-12-06 15:44:31 +01:00
TIFFSetTagExtender(register_custom_tiff_tags);
2020-12-06 15:44:31 +01:00
if (!find_processor(processing_script)) {
g_printerr("Could not find any post-process script\n");
exit(1);
}
}
2020-12-06 15:44:31 +01:00
void
mp_process_pipeline_start()
{
2020-12-06 15:44:31 +01:00
pipeline = mp_pipeline_new();
2020-12-06 15:44:31 +01:00
mp_pipeline_invoke(pipeline, setup, NULL, 0);
mp_zbar_pipeline_start();
}
2020-12-06 15:44:31 +01:00
void
mp_process_pipeline_stop()
{
2020-12-06 15:44:31 +01:00
mp_pipeline_free(pipeline);
mp_zbar_pipeline_stop();
}
void
mp_process_pipeline_sync()
{
mp_pipeline_sync(pipeline);
}
2021-04-09 14:57:14 +02:00
#define NUM_BUFFERS 4
2021-04-15 15:14:20 +02:00
struct _MPProcessPipelineBuffer {
GLuint texture_id;
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
_Atomic(int) refcount;
};
static MPProcessPipelineBuffer output_buffers[NUM_BUFFERS];
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
void
mp_process_pipeline_buffer_ref(MPProcessPipelineBuffer *buf)
2021-04-09 14:57:14 +02:00
{
2021-04-15 15:14:20 +02:00
++buf->refcount;
2021-04-09 14:57:14 +02:00
}
2021-04-15 15:14:20 +02:00
void
mp_process_pipeline_buffer_unref(MPProcessPipelineBuffer *buf)
{
--buf->refcount;
}
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
uint32_t
mp_process_pipeline_buffer_get_texture_id(MPProcessPipelineBuffer *buf)
{
2021-04-15 15:14:20 +02:00
return buf->texture_id;
}
2021-04-09 14:57:14 +02:00
static GLES2Debayer *gles2_debayer = NULL;
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
static GdkGLContext *context;
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
#define RENDERDOC
2020-12-06 15:44:31 +01:00
2021-04-15 15:14:20 +02:00
#ifdef RENDERDOC
#include <renderdoc/app.h>
extern RENDERDOC_API_1_1_2 *rdoc_api;
#endif
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
static void
init_gl(MPPipeline *pipeline, GdkSurface **surface)
{
GError *error = NULL;
context = gdk_surface_create_gl_context(*surface, &error);
if (context == NULL) {
printf("Failed to initialize OpenGL context: %s\n", error->message);
g_clear_error(&error);
2021-04-09 14:57:14 +02:00
return;
}
2021-04-15 15:14:20 +02:00
gdk_gl_context_set_use_es(context, true);
gdk_gl_context_set_required_version(context, 2, 0);
gdk_gl_context_set_forward_compatible(context, false);
2021-04-15 15:14:20 +02:00
#ifdef DEBUG
gdk_gl_context_set_debug_enabled(context, true);
#else
gdk_gl_context_set_debug_enabled(context, false);
#endif
2021-04-09 14:57:14 +02:00
2021-04-15 15:14:20 +02:00
gdk_gl_context_realize(context, &error);
if (error != NULL) {
printf("Failed to create OpenGL context: %s\n", error->message);
g_clear_object(&context);
g_clear_error(&error);
2021-04-09 14:57:14 +02:00
return;
}
2021-04-15 15:14:20 +02:00
gdk_gl_context_make_current(context);
2021-04-09 14:57:14 +02:00
check_gl();
// Make a VAO for OpenGL
if (!gdk_gl_context_get_use_es(context)) {
GLuint vao;
glGenVertexArrays(1, &vao);
glBindVertexArray(vao);
check_gl();
}
2020-12-06 15:44:31 +01:00
gles2_debayer = gles2_debayer_new(MP_PIXEL_FMT_BGGR8);
check_gl();
2021-04-09 14:57:14 +02:00
glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
2021-04-09 14:57:14 +02:00
check_gl();
gles2_debayer_use(gles2_debayer);
2021-04-15 15:14:20 +02:00
for (size_t i = 0; i < NUM_BUFFERS; ++i) {
glGenTextures(1, &output_buffers[i].texture_id);
glBindTexture(GL_TEXTURE_2D, output_buffers[i].texture_id);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
}
glBindTexture(GL_TEXTURE_2D, 0);
gboolean is_es = gdk_gl_context_get_use_es(context);
int major, minor;
gdk_gl_context_get_version(context, &major, &minor);
printf("Initialized %s %d.%d\n", is_es ? "OpenGL ES" : "OpenGL", major, minor);
2021-04-09 14:57:14 +02:00
}
void
2021-04-15 15:14:20 +02:00
mp_process_pipeline_init_gl(GdkSurface *surface)
2021-04-09 14:57:14 +02:00
{
2021-04-15 15:14:20 +02:00
mp_pipeline_invoke(pipeline, (MPPipelineCallback) init_gl, &surface, sizeof(GdkSurface *));
2021-04-09 14:57:14 +02:00
}
static cairo_surface_t *
process_image_for_preview(const uint8_t *image)
{
2021-04-10 14:39:55 +02:00
clock_t t1 = clock();
2021-04-15 15:14:20 +02:00
// Pick an available buffer
MPProcessPipelineBuffer *output_buffer = NULL;
for (size_t i = 0; i < NUM_BUFFERS; ++i) {
if (output_buffers[i].refcount == 0) {
output_buffer = &output_buffers[i];
}
}
if (output_buffer == NULL) {
return NULL;
}
assert(output_buffer != NULL);
2021-04-09 14:57:14 +02:00
#ifdef RENDERDOC
2021-04-15 15:14:20 +02:00
if (rdoc_api) rdoc_api->StartFrameCapture(NULL, NULL);
2021-04-09 14:57:14 +02:00
#endif
2021-04-15 15:14:20 +02:00
// Copy image to a GL texture. TODO: This can be avoided
GLuint input_texture;
glGenTextures(1, &input_texture);
glBindTexture(GL_TEXTURE_2D, input_texture);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE, mode.width, mode.height, 0, GL_LUMINANCE, GL_UNSIGNED_BYTE, image);
check_gl();
2021-04-09 14:57:14 +02:00
gles2_debayer_process(
gles2_debayer, output_buffer->texture_id, input_texture);
2021-04-15 15:14:20 +02:00
check_gl();
glFinish();
glDeleteTextures(1, &input_texture);
2021-04-15 15:14:20 +02:00
clock_t t2 = clock();
printf("%fms\n", (float)(t2 - t1) / CLOCKS_PER_SEC * 1000);
// {
// glBindTexture(GL_TEXTURE_2D, textures[1]);
// EGLImage egl_image = eglCreateImage(egl_display, egl_context, EGL_GL_TEXTURE_2D, (EGLClientBuffer)(size_t)textures[1], NULL);
// // Make sure it's in the expected format
// int fourcc;
// eglExportDMABUFImageQueryMESA(egl_display, egl_image, &fourcc, NULL, NULL);
// assert(fourcc == DRM_FORMAT_ABGR8888);
// int dmabuf_fd;
// int stride, offset;
// eglExportDMABUFImageMESA(egl_display, egl_image, &dmabuf_fd, &stride, &offset);
// int fsize = lseek(dmabuf_fd, 0, SEEK_END);
// printf("SIZE %d STRIDE %d OFFSET %d SIZE %d:%d\n", fsize, stride, offset, preview_width, preview_height);
// size_t size = stride * preview_height;
// uint32_t *data = mmap(NULL, fsize, PROT_READ, MAP_SHARED, dmabuf_fd, 0);
// assert(data != MAP_FAILED);
// int pixel_stride = stride / 4;
// for (size_t y = 0; y < preview_height; ++y) {
// for (size_t x = 0; x < preview_width; ++x) {
// uint32_t p = data[x + y * pixel_stride];
// pixels[x + y * preview_width] = p;
// // uint16_t p = data[x + y * stride];
// // uint32_t r = (p & 0b11111);
// // uint32_t g = ((p >> 5) & 0b11111);
// // uint32_t b = ((p >> 10) & 0b11111);
// // pixels[x + y * preview_width] = (r << 16) | (g << 8) | b;
// }
// // memcpy(pixels + preview_width * y, data + stride * y, preview_width * sizeof(uint32_t));
// }
// {
// FILE *f = fopen("test.raw", "w");
// fwrite(data, fsize, 1, f);
// fclose(f);
// }
// // memcpy(pixels, data, size);
// munmap(data, size);
// close(dmabuf_fd);
// }
// glReadPixels(0, 0, preview_width, preview_height, GL_RGBA, GL_UNSIGNED_BYTE, pixels);
// check_gl();
2021-04-09 14:57:14 +02:00
#ifdef RENDERDOC
2021-04-15 15:14:20 +02:00
if(rdoc_api) rdoc_api->EndFrameCapture(NULL, NULL);
2021-04-09 14:57:14 +02:00
#endif
2021-04-15 15:14:20 +02:00
mp_process_pipeline_buffer_ref(output_buffer);
mp_main_set_preview(output_buffer);
2020-12-06 15:44:31 +01:00
// Create a thumbnail from the preview for the last capture
cairo_surface_t *thumb = NULL;
2021-04-15 15:14:20 +02:00
// if (captures_remaining == 1) {
// printf("Making thumbnail\n");
// thumb = cairo_image_surface_create(
// CAIRO_FORMAT_ARGB32, MP_MAIN_THUMB_SIZE, MP_MAIN_THUMB_SIZE);
// cairo_t *cr = cairo_create(thumb);
// draw_surface_scaled_centered(
// cr, MP_MAIN_THUMB_SIZE, MP_MAIN_THUMB_SIZE, surface);
// cairo_destroy(cr);
// }
// Pass processed preview to main and zbar
2021-04-15 15:14:20 +02:00
// mp_zbar_pipeline_process_image(cairo_surface_reference(surface));
return thumb;
}
static void
process_image_for_capture(const uint8_t *image, int count)
{
2020-12-06 15:44:31 +01:00
time_t rawtime;
time(&rawtime);
struct tm tim = *(localtime(&rawtime));
char datetime[20] = { 0 };
strftime(datetime, 20, "%Y:%m:%d %H:%M:%S", &tim);
char fname[255];
sprintf(fname, "%s/%d.dng", burst_dir, count);
TIFF *tif = TIFFOpen(fname, "w");
if (!tif) {
printf("Could not open tiff\n");
}
// Define TIFF thumbnail
TIFFSetField(tif, TIFFTAG_SUBFILETYPE, 1);
TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, mode.width >> 4);
TIFFSetField(tif, TIFFTAG_IMAGELENGTH, mode.height >> 4);
2020-12-06 15:44:31 +01:00
TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, 8);
TIFFSetField(tif, TIFFTAG_COMPRESSION, COMPRESSION_NONE);
TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_RGB);
TIFFSetField(tif, TIFFTAG_MAKE, mp_get_device_make());
TIFFSetField(tif, TIFFTAG_MODEL, mp_get_device_model());
uint16_t orientation;
if (camera->rotate == 0) {
orientation = camera->mirrored ? ORIENTATION_TOPRIGHT :
ORIENTATION_TOPLEFT;
} else if (camera->rotate == 90) {
orientation = camera->mirrored ? ORIENTATION_RIGHTBOT :
ORIENTATION_LEFTBOT;
} else if (camera->rotate == 180) {
orientation = camera->mirrored ? ORIENTATION_BOTLEFT :
ORIENTATION_BOTRIGHT;
} else {
orientation = camera->mirrored ? ORIENTATION_LEFTTOP :
ORIENTATION_RIGHTTOP;
}
TIFFSetField(tif, TIFFTAG_ORIENTATION, orientation);
TIFFSetField(tif, TIFFTAG_DATETIME, datetime);
TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 3);
TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
TIFFSetField(tif, TIFFTAG_SOFTWARE, "Megapixels");
long sub_offset = 0;
TIFFSetField(tif, TIFFTAG_SUBIFD, 1, &sub_offset);
TIFFSetField(tif, TIFFTAG_DNGVERSION, "\001\001\0\0");
TIFFSetField(tif, TIFFTAG_DNGBACKWARDVERSION, "\001\0\0\0");
char uniquecameramodel[255];
sprintf(uniquecameramodel, "%s %s", mp_get_device_make(),
mp_get_device_model());
TIFFSetField(tif, TIFFTAG_UNIQUECAMERAMODEL, uniquecameramodel);
if (camera->colormatrix[0]) {
TIFFSetField(tif, TIFFTAG_COLORMATRIX1, 9, camera->colormatrix);
} else {
TIFFSetField(tif, TIFFTAG_COLORMATRIX1, 9, colormatrix_srgb);
}
if (camera->forwardmatrix[0]) {
TIFFSetField(tif, TIFFTAG_FORWARDMATRIX1, 9, camera->forwardmatrix);
}
static const float neutral[] = { 1.0, 1.0, 1.0 };
TIFFSetField(tif, TIFFTAG_ASSHOTNEUTRAL, 3, neutral);
TIFFSetField(tif, TIFFTAG_CALIBRATIONILLUMINANT1, 21);
// Write black thumbnail, only windows uses this
{
unsigned char *buf =
(unsigned char *)calloc(1, (int)mode.width >> 4);
for (int row = 0; row < (mode.height >> 4); row++) {
2020-12-06 15:44:31 +01:00
TIFFWriteScanline(tif, buf, row, 0);
}
free(buf);
}
TIFFWriteDirectory(tif);
// Define main photo
TIFFSetField(tif, TIFFTAG_SUBFILETYPE, 0);
TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, mode.width);
TIFFSetField(tif, TIFFTAG_IMAGELENGTH, mode.height);
2020-12-06 15:44:31 +01:00
TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, 8);
TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_CFA);
TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 1);
TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
static const short cfapatterndim[] = { 2, 2 };
TIFFSetField(tif, TIFFTAG_CFAREPEATPATTERNDIM, cfapatterndim);
#if (TIFFLIB_VERSION < 20201219) && !LIBTIFF_CFA_PATTERN
2020-12-06 15:44:31 +01:00
TIFFSetField(tif, TIFFTAG_CFAPATTERN, "\002\001\001\000"); // BGGR
2020-12-22 16:05:06 +01:00
#else
TIFFSetField(tif, TIFFTAG_CFAPATTERN, 4, "\002\001\001\000"); // BGGR
#endif
printf("TIFF version %d\n", TIFFLIB_VERSION);
2020-12-06 15:44:31 +01:00
if (camera->whitelevel) {
TIFFSetField(tif, TIFFTAG_WHITELEVEL, 1, &camera->whitelevel);
}
if (camera->blacklevel) {
TIFFSetField(tif, TIFFTAG_BLACKLEVEL, 1, &camera->blacklevel);
}
TIFFCheckpointDirectory(tif);
printf("Writing frame to %s\n", fname);
unsigned char *pLine = (unsigned char *)malloc(mode.width);
for (int row = 0; row < mode.height; row++) {
TIFFWriteScanline(tif, (void *) image + (row * mode.width), row, 0);
2020-12-06 15:44:31 +01:00
}
free(pLine);
TIFFWriteDirectory(tif);
// Add an EXIF block to the tiff
TIFFCreateEXIFDirectory(tif);
// 1 = manual, 2 = full auto, 3 = aperture priority, 4 = shutter priority
if (!exposure_is_manual) {
TIFFSetField(tif, EXIFTAG_EXPOSUREPROGRAM, 2);
} else {
TIFFSetField(tif, EXIFTAG_EXPOSUREPROGRAM, 1);
}
TIFFSetField(tif, EXIFTAG_EXPOSURETIME,
(mode.frame_interval.numerator /
(float)mode.frame_interval.denominator) /
((float)mode.height / (float)exposure));
2020-12-06 15:44:31 +01:00
uint16_t isospeed[1];
isospeed[0] = (uint16_t)remap(gain - 1, 0, gain_max, camera->iso_min,
camera->iso_max);
TIFFSetField(tif, EXIFTAG_ISOSPEEDRATINGS, 1, isospeed);
TIFFSetField(tif, EXIFTAG_FLASH, 0);
TIFFSetField(tif, EXIFTAG_DATETIMEORIGINAL, datetime);
TIFFSetField(tif, EXIFTAG_DATETIMEDIGITIZED, datetime);
if (camera->fnumber) {
TIFFSetField(tif, EXIFTAG_FNUMBER, camera->fnumber);
}
if (camera->focallength) {
TIFFSetField(tif, EXIFTAG_FOCALLENGTH, camera->focallength);
}
if (camera->focallength && camera->cropfactor) {
TIFFSetField(tif, EXIFTAG_FOCALLENGTHIN35MMFILM,
(short)(camera->focallength * camera->cropfactor));
}
uint64_t exif_offset = 0;
TIFFWriteCustomDirectory(tif, &exif_offset);
TIFFFreeDirectory(tif);
// Update exif pointer
TIFFSetDirectory(tif, 0);
TIFFSetField(tif, TIFFTAG_EXIFIFD, exif_offset);
TIFFRewriteDirectory(tif);
TIFFClose(tif);
}
static void
post_process_finished(GSubprocess *proc, GAsyncResult *res, cairo_surface_t *thumb)
{
char *stdout;
g_subprocess_communicate_utf8_finish(proc, res, &stdout, NULL, NULL);
// The last line contains the file name
int end = strlen(stdout);
// Skip the newline at the end
stdout[--end] = '\0';
char *path = path = stdout + end - 1;
do {
if (*path == '\n') {
path++;
break;
}
--path;
} while (path > stdout);
mp_main_capture_completed(thumb, path);
}
static void
process_capture_burst(cairo_surface_t *thumb)
{
2020-12-06 15:44:31 +01:00
time_t rawtime;
time(&rawtime);
struct tm tim = *(localtime(&rawtime));
2020-12-06 15:44:31 +01:00
char timestamp[30];
strftime(timestamp, 30, "%Y%m%d%H%M%S", &tim);
if (g_get_user_special_dir(G_USER_DIRECTORY_PICTURES) != NULL) {
sprintf(capture_fname,
"%s/IMG%s",
g_get_user_special_dir(G_USER_DIRECTORY_PICTURES),
timestamp);
} else if (getenv("XDG_PICTURES_DIR") != NULL) {
sprintf(capture_fname,
"%s/IMG%s",
getenv("XDG_PICTURES_DIR"),
timestamp);
} else {
sprintf(capture_fname,
"%s/Pictures/IMG%s",
getenv("HOME"),
timestamp);
}
2020-12-06 15:44:31 +01:00
// Start post-processing the captured burst
g_print("Post process %s to %s.ext\n", burst_dir, capture_fname);
GError *error = NULL;
GSubprocess *proc = g_subprocess_new(
G_SUBPROCESS_FLAGS_STDOUT_PIPE,
&error,
processing_script,
burst_dir,
capture_fname,
NULL);
if (!proc) {
g_printerr("Failed to spawn postprocess process: %s\n",
error->message);
return;
}
g_subprocess_communicate_utf8_async(
proc,
NULL,
NULL,
(GAsyncReadyCallback)post_process_finished,
thumb);
}
static void
process_image(MPPipeline *pipeline, const MPBuffer *buffer)
{
size_t size =
mp_pixel_format_width_to_bytes(mode.pixel_format, mode.width) *
mode.height;
uint8_t *image = malloc(size);
memcpy(image, buffer->data, size);
mp_io_pipeline_release_buffer(buffer->index);
2021-04-26 10:09:04 +02:00
MPZBarImage *zbar_image = mp_zbar_image_new(image, mode.pixel_format, mode.width, mode.height, camera->rotate, camera->mirrored);
mp_zbar_pipeline_process_image(mp_zbar_image_ref(zbar_image));
cairo_surface_t *thumb = process_image_for_preview(image);
2020-12-06 15:44:31 +01:00
if (captures_remaining > 0) {
int count = burst_length - captures_remaining;
--captures_remaining;
2020-12-06 15:44:31 +01:00
process_image_for_capture(image, count);
2020-12-06 15:44:31 +01:00
if (captures_remaining == 0) {
assert(thumb);
process_capture_burst(thumb);
} else {
assert(!thumb);
2020-12-06 15:44:31 +01:00
}
} else {
assert(!thumb);
2020-12-06 15:44:31 +01:00
}
2021-04-26 10:09:04 +02:00
mp_zbar_image_unref(zbar_image);
2020-12-06 15:44:31 +01:00
++frames_processed;
if (captures_remaining == 0) {
is_capturing = false;
}
}
2020-12-06 15:44:31 +01:00
void
mp_process_pipeline_process_image(MPBuffer buffer)
{
2020-12-06 15:44:31 +01:00
// If we haven't processed the previous frame yet, drop this one
if (frames_received != frames_processed && !is_capturing) {
printf("Dropped frame at capture\n");
mp_io_pipeline_release_buffer(buffer.index);
2020-12-06 15:44:31 +01:00
return;
}
2020-12-06 15:44:31 +01:00
++frames_received;
mp_pipeline_invoke(pipeline, (MPPipelineCallback)process_image, &buffer,
sizeof(MPBuffer));
}
2020-12-06 15:44:31 +01:00
static void
capture()
{
2020-12-06 15:44:31 +01:00
char template[] = "/tmp/megapixels.XXXXXX";
char *tempdir;
tempdir = mkdtemp(template);
2020-12-06 15:44:31 +01:00
if (tempdir == NULL) {
g_printerr("Could not make capture directory %s\n", template);
exit(EXIT_FAILURE);
}
2020-12-06 15:44:31 +01:00
strcpy(burst_dir, tempdir);
2020-12-06 15:44:31 +01:00
captures_remaining = burst_length;
}
2020-12-06 15:44:31 +01:00
void
mp_process_pipeline_capture()
{
2020-12-06 15:44:31 +01:00
is_capturing = true;
2020-12-06 15:44:31 +01:00
mp_pipeline_invoke(pipeline, capture, NULL, 0);
}
2021-04-19 16:04:53 +02:00
static int output_buffer_width = -1;
static int output_buffer_height = -1;
2021-04-15 15:14:20 +02:00
static void
on_output_changed()
2021-04-15 15:14:20 +02:00
{
2021-04-19 16:04:53 +02:00
output_buffer_width = mode.width / 2;
output_buffer_height = mode.height / 2;
2021-04-15 15:14:20 +02:00
if (camera->rotate != 0 || camera->rotate != 180) {
int tmp = output_buffer_width;
output_buffer_width = output_buffer_height;
output_buffer_height = tmp;
}
for (size_t i = 0; i < NUM_BUFFERS; ++i) {
glBindTexture(GL_TEXTURE_2D, output_buffers[i].texture_id);
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, output_buffer_width, output_buffer_height, 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
}
glBindTexture(GL_TEXTURE_2D, 0);
gles2_debayer_configure(
gles2_debayer,
output_buffer_width, output_buffer_height,
mode.width, mode.height,
camera->rotate, camera->mirrored,
camera->previewmatrix[0] == 0 ? NULL : camera->previewmatrix,
camera->blacklevel);
2021-04-15 15:14:20 +02:00
}
static void
update_state(MPPipeline *pipeline, const struct mp_process_pipeline_state *state)
{
const bool output_changed = (!mp_camera_mode_is_equivalent(&mode, &state->mode) || preview_width != state->preview_width || preview_height != state->preview_height);
2021-04-15 15:14:20 +02:00
2020-12-06 15:44:31 +01:00
camera = state->camera;
mode = state->mode;
preview_width = state->preview_width;
preview_height = state->preview_height;
2021-04-15 15:14:20 +02:00
burst_length = state->burst_length;
2020-12-06 15:44:31 +01:00
// gain_is_manual = state->gain_is_manual;
gain = state->gain;
gain_max = state->gain_max;
exposure_is_manual = state->exposure_is_manual;
exposure = state->exposure;
if (output_changed) {
on_output_changed();
2021-04-15 15:14:20 +02:00
}
2020-12-06 15:44:31 +01:00
struct mp_main_state main_state = {
.camera = camera,
.mode = mode,
2021-04-19 16:04:53 +02:00
.image_width = output_buffer_width,
.image_height = output_buffer_height,
2020-12-06 15:44:31 +01:00
.gain_is_manual = state->gain_is_manual,
.gain = gain,
.gain_max = gain_max,
.exposure_is_manual = exposure_is_manual,
.exposure = exposure,
.has_auto_focus_continuous = state->has_auto_focus_continuous,
.has_auto_focus_start = state->has_auto_focus_start,
};
mp_main_update_state(&main_state);
}
2020-12-06 15:44:31 +01:00
void
mp_process_pipeline_update_state(const struct mp_process_pipeline_state *new_state)
{
2020-12-06 15:44:31 +01:00
mp_pipeline_invoke(pipeline, (MPPipelineCallback)update_state, new_state,
sizeof(struct mp_process_pipeline_state));
}
2021-04-15 15:14:20 +02:00
// FUCK YOU GTK
void pango_fc_font_get_languages() {}