megapixels/src/main.c

#include "main.h"

#include <errno.h>
#include <fcntl.h>
#include <linux/videodev2.h>
#include <linux/media.h>
#include <linux/v4l2-subdev.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <time.h>
#include <assert.h>
#include <limits.h>
#include <linux/kdev_t.h>
#include <sys/sysmacros.h>
#include <asm/errno.h>
#include <wordexp.h>
#include <gtk/gtk.h>
#include <locale.h>
#include <zbar.h>
#include "gl_util.h"
#include "camera_config.h"
#include "io_pipeline.h"
#include "process_pipeline.h"

#define RENDERDOC

#ifdef RENDERDOC
#include <dlfcn.h>
#include <renderdoc/app.h>
RENDERDOC_API_1_1_2 *rdoc_api = NULL;
#endif

enum user_control { USER_CONTROL_ISO, USER_CONTROL_SHUTTER };

static bool camera_is_initialized = false;
static const struct mp_camera_config *camera = NULL;
static MPCameraMode mode;

static int preview_width = -1;
static int preview_height = -1;

static int device_rotation = 0;

static bool gain_is_manual = false;
static int gain;
static int gain_max;

static bool exposure_is_manual = false;
static int exposure;

static bool has_auto_focus_continuous;
static bool has_auto_focus_start;

static MPProcessPipelineBuffer *current_preview_buffer = NULL;
static int preview_buffer_width = -1;
static int preview_buffer_height = -1;

static char last_path[260] = "";

static MPZBarScanResult *zbar_result = NULL;

static int burst_length = 3;

// Widgets
GtkWidget *preview;
GtkWidget *main_stack;
GtkWidget *open_last_stack;
GtkWidget *thumb_last;
GtkWidget *process_spinner;
GtkWidget *scanned_codes;
GtkWidget *preview_top_box;
GtkWidget *preview_bottom_box;

int
remap(int value, int input_min, int input_max, int output_min, int output_max)
{
	const long long factor = 1000000000;
	long long output_spread = output_max - output_min;
	long long input_spread = input_max - input_min;

	long long zero_value = value - input_min;
	zero_value *= factor;
	long long percentage = zero_value / input_spread;

	long long zero_output = percentage * output_spread / factor;

	long long result = output_min + zero_output;
	return (int)result;
}

static void
update_io_pipeline()
{
	struct mp_io_pipeline_state io_state = {
		.camera = camera,
		.burst_length = burst_length,
		.preview_width = preview_width,
		.preview_height = preview_height,
		.device_rotation = device_rotation,
		.gain_is_manual = gain_is_manual,
		.gain = gain,
		.exposure_is_manual = exposure_is_manual,
		.exposure = exposure,
	};
	mp_io_pipeline_update_state(&io_state);
}

static bool
update_state(const struct mp_main_state *state)
{
	if (!camera_is_initialized) {
		camera_is_initialized = true;
	}

	if (camera == state->camera) {
		mode = state->mode;

		if (!gain_is_manual) {
			gain = state->gain;
		}
		gain_max = state->gain_max;

		if (!exposure_is_manual) {
			exposure = state->exposure;
		}

		has_auto_focus_continuous = state->has_auto_focus_continuous;
		has_auto_focus_start = state->has_auto_focus_start;
	}

	preview_buffer_width = state->image_width;
	preview_buffer_height = state->image_height;

	return false;
}

void
mp_main_update_state(const struct mp_main_state *state)
{
	struct mp_main_state *state_copy = malloc(sizeof(struct mp_main_state));
	*state_copy = *state;

	g_main_context_invoke_full(g_main_context_default(), G_PRIORITY_DEFAULT_IDLE,
				   (GSourceFunc)update_state, state_copy, free);
}

static bool set_zbar_result(MPZBarScanResult *result)
{
	if (zbar_result) {
		for (uint8_t i = 0; i < zbar_result->size; ++i) {
			free(zbar_result->codes[i].data);
		}

		free(zbar_result);
	}

	zbar_result = result;
	gtk_widget_queue_draw(preview);

	return false;
}

void mp_main_set_zbar_result(MPZBarScanResult *result)
{
	g_main_context_invoke_full(g_main_context_default(), G_PRIORITY_DEFAULT_IDLE,
				   (GSourceFunc)set_zbar_result, result, NULL);
}

static bool
set_preview(MPProcessPipelineBuffer *buffer)
{
	if (current_preview_buffer) {
		mp_process_pipeline_buffer_unref(current_preview_buffer);
	}
	current_preview_buffer = buffer;
	gtk_widget_queue_draw(preview);
	return false;
}

void
mp_main_set_preview(MPProcessPipelineBuffer *buffer)
{
	g_main_context_invoke_full(g_main_context_default(), G_PRIORITY_DEFAULT_IDLE,
				   (GSourceFunc)set_preview, buffer, NULL);
}

struct capture_completed_args {
	GdkTexture *thumb;
	char *fname;
};

static bool
capture_completed(struct capture_completed_args *args)
{
	strncpy(last_path, args->fname, 259);

	gtk_image_set_from_paintable(GTK_IMAGE(thumb_last),
				     GDK_PAINTABLE(args->thumb));

	gtk_spinner_stop(GTK_SPINNER(process_spinner));
	gtk_stack_set_visible_child(GTK_STACK(open_last_stack), thumb_last);

	g_object_unref(args->thumb);
	g_free(args->fname);

	return false;
}

void
mp_main_capture_completed(GdkTexture *thumb, const char *fname)
{
	struct capture_completed_args *args = malloc(sizeof(struct capture_completed_args));
	args->thumb = thumb;
	args->fname = g_strdup(fname);
	g_main_context_invoke_full(g_main_context_default(), G_PRIORITY_DEFAULT_IDLE,
				   (GSourceFunc)capture_completed, args, free);
}

static GLuint blit_program;
static GLuint blit_uniform_transform;
static GLuint blit_uniform_texture;
static GLuint solid_program;
static GLuint solid_uniform_color;
static GLuint quad;

static void
preview_realize(GtkGLArea *area)
{
	gtk_gl_area_make_current(area);

	if (gtk_gl_area_get_error(area) != NULL) {
		return;
	}

	// Make a VAO for OpenGL
	if (!gtk_gl_area_get_use_es(area)) {
		GLuint vao;
		glGenVertexArrays(1, &vao);
		glBindVertexArray(vao);
		check_gl();
	}

	GLuint blit_shaders[] = {
		gl_util_load_shader("/org/postmarketos/Megapixels/blit.vert", GL_VERTEX_SHADER, NULL, 0),
		gl_util_load_shader("/org/postmarketos/Megapixels/blit.frag", GL_FRAGMENT_SHADER, NULL, 0),
	};

	blit_program = gl_util_link_program(blit_shaders, 2);
	glBindAttribLocation(blit_program, GL_UTIL_VERTEX_ATTRIBUTE, "vert");
	glBindAttribLocation(blit_program, GL_UTIL_TEX_COORD_ATTRIBUTE, "tex_coord");
	check_gl();

	blit_uniform_transform = glGetUniformLocation(blit_program, "transform");
	blit_uniform_texture = glGetUniformLocation(blit_program, "texture");

	GLuint solid_shaders[] = {
		gl_util_load_shader("/org/postmarketos/Megapixels/solid.vert", GL_VERTEX_SHADER, NULL, 0),
		gl_util_load_shader("/org/postmarketos/Megapixels/solid.frag", GL_FRAGMENT_SHADER, NULL, 0),
	};

	solid_program = gl_util_link_program(solid_shaders, 2);
	glBindAttribLocation(solid_program, GL_UTIL_VERTEX_ATTRIBUTE, "vert");
	check_gl();

	solid_uniform_color = glGetUniformLocation(solid_program, "color");

	quad = gl_util_new_quad();
}

static void
position_preview(float *offset_x, float *offset_y, float *size_x, float *size_y)
{
	int buffer_width, buffer_height;
	if (device_rotation == 0 || device_rotation == 180) {
		buffer_width = preview_buffer_width;
		buffer_height = preview_buffer_height;
	} else {
		buffer_width = preview_buffer_height;
		buffer_height = preview_buffer_width;
	}

	int scale_factor = gtk_widget_get_scale_factor(preview);
	int top_height = gtk_widget_get_allocated_height(preview_top_box) * scale_factor;
	int bottom_height = gtk_widget_get_allocated_height(preview_bottom_box) * scale_factor;
	int inner_height = preview_height - top_height - bottom_height;

	double scale = MIN(preview_width / (float) buffer_width, preview_height / (float) buffer_height);

	*size_x = scale * buffer_width;
	*size_y = scale * buffer_height;

	*offset_x = (preview_width - *size_x) / 2.0;

	if (*size_y > inner_height) {
		*offset_y = (preview_height - *size_y) / 2.0;
	} else {
		*offset_y = top_height + (inner_height - *size_y) / 2.0;
	}

}

static gboolean
preview_draw(GtkGLArea *area, GdkGLContext *ctx, gpointer data)
{
	if (gtk_gl_area_get_error(area) != NULL) {
		return FALSE;
	}

	if (!camera_is_initialized) {
		return FALSE;
	}

#ifdef RENDERDOC
	if (rdoc_api) rdoc_api->StartFrameCapture(NULL, NULL);
#endif

	glClearColor(0, 0, 0, 1);
	glClear(GL_COLOR_BUFFER_BIT);

	float offset_x, offset_y, size_x, size_y;
	position_preview(&offset_x, &offset_y, &size_x, &size_y);
	glViewport(offset_x,
		   preview_height - size_y - offset_y,
		   size_x,
		   size_y);

	if (current_preview_buffer) {
		glUseProgram(blit_program);

		GLfloat rotation_list[4] = { 0, -1, 0, 1 };
		int rotation_index = device_rotation / 90;

		GLfloat sin_rot = rotation_list[rotation_index];
		GLfloat cos_rot = rotation_list[(4 + rotation_index - 1) % 4];
		GLfloat matrix[9] = {
			cos_rot, sin_rot, 0,
			-sin_rot, cos_rot, 0,
			0, 0, 1,
		};
		glUniformMatrix3fv(blit_uniform_transform, 1, GL_FALSE, matrix);
		check_gl();

		glActiveTexture(GL_TEXTURE0);
		glBindTexture(GL_TEXTURE_2D, mp_process_pipeline_buffer_get_texture_id(current_preview_buffer));
		glUniform1i(blit_uniform_texture, 0);
		check_gl();

		gl_util_bind_quad(quad);
		gl_util_draw_quad(quad);
	}

	if (zbar_result) {
		GLuint buffer;
		if (!gtk_gl_area_get_use_es(area)) {
			glGenBuffers(1, &buffer);
			glBindBuffer(GL_ARRAY_BUFFER, buffer);
			check_gl();
		}

		glUseProgram(solid_program);
		glEnable(GL_BLEND);
		glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);

		glUniform4f(solid_uniform_color, 1, 0, 0, 0.5);

		for (uint8_t i = 0; i < zbar_result->size; ++i) {
			MPZBarCode *code = &zbar_result->codes[i];

			GLfloat vertices[] = {
				code->bounds_x[0], code->bounds_y[0],
				code->bounds_x[1], code->bounds_y[1],
				code->bounds_x[3], code->bounds_y[3],
				code->bounds_x[2], code->bounds_y[2],
			};

			for (int i = 0; i < 4; ++i) {
				vertices[i * 2] = 2 * vertices[i * 2] / preview_buffer_width - 1.0;
				vertices[i * 2 + 1] = 1.0 - 2 * vertices[i * 2 + 1] / preview_buffer_height;
			}

			if (gtk_gl_area_get_use_es(area)) {
				glVertexAttribPointer(GL_UTIL_VERTEX_ATTRIBUTE, 2, GL_FLOAT, 0, 0, vertices);
				check_gl();
				glEnableVertexAttribArray(GL_UTIL_VERTEX_ATTRIBUTE);
				check_gl();
			} else {
				glBufferData(GL_ARRAY_BUFFER, sizeof(vertices), vertices, GL_STREAM_DRAW);
				check_gl();

				glVertexAttribPointer(GL_UTIL_VERTEX_ATTRIBUTE, 2, GL_FLOAT, GL_FALSE, 0, 0);
				glEnableVertexAttribArray(GL_UTIL_VERTEX_ATTRIBUTE);
				check_gl();
			}

			glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
			check_gl();
		}

		glDisable(GL_BLEND);
		glBindBuffer(GL_ARRAY_BUFFER, 0);
	}

	glFlush();

#ifdef RENDERDOC
	if(rdoc_api) rdoc_api->EndFrameCapture(NULL, NULL);
#endif

	return FALSE;
}

static gboolean
preview_resize(GtkWidget *widget, int width, int height, gpointer data)
{
	if (preview_width != width || preview_height != height) {
		preview_width = width;
		preview_height = height;
		update_io_pipeline();
	}

	return TRUE;
}

void
run_open_last_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	char uri[275];
	GError *error = NULL;

	if (strlen(last_path) == 0) {
		return;
	}
	sprintf(uri, "file://%s", last_path);
	if (!g_app_info_launch_default_for_uri(uri, NULL, &error)) {
		g_printerr("Could not launch image viewer for '%s': %s\n", uri, error->message);
	}
}

void
run_open_photos_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	char uri[270];
	GError *error = NULL;
	sprintf(uri, "file://%s", g_get_user_special_dir(G_USER_DIRECTORY_PICTURES));
	if (!g_app_info_launch_default_for_uri(uri, NULL, &error)) {
		g_printerr("Could not launch image viewer: %s\n", error->message);
	}
}

void
run_capture_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	gtk_spinner_start(GTK_SPINNER(process_spinner));
	gtk_stack_set_visible_child(GTK_STACK(open_last_stack), process_spinner);
	mp_io_pipeline_capture();
}

void
run_quit_action(GSimpleAction *action, GVariant *param, GApplication *app)
{
	g_application_quit(app);
}

static bool
check_point_inside_bounds(int x, int y, int *bounds_x, int *bounds_y)
{
	bool right = false, left = false, top = false, bottom = false;

	for (int i = 0; i < 4; ++i) {
		if (x <= bounds_x[i])
			left = true;
		if (x >= bounds_x[i])
			right = true;
		if (y <= bounds_y[i])
			top = true;
		if (y >= bounds_y[i])
			bottom = true;
	}

	return right && left && top && bottom;
}

static void
on_zbar_dialog_response(GtkDialog *dialog, int response, char *data)
{
	GError *error = NULL;
	switch (response) {
		case GTK_RESPONSE_YES:
			if (!g_app_info_launch_default_for_uri(data,
							       NULL, &error)) {
				g_printerr("Could not launch application: %s\n",
					   error->message);
			}
		case GTK_RESPONSE_ACCEPT:
		{
			GdkDisplay *display = gtk_widget_get_display(GTK_WIDGET(dialog));
			gdk_clipboard_set_text(
				gdk_display_get_clipboard(display),
				data);
		}
		case GTK_RESPONSE_CANCEL:
			break;
		default:
			g_printerr("Wrong dialog response: %d\n", response);
	}

	g_free(data);
	gtk_window_destroy(GTK_WINDOW(dialog));
}

static void
on_zbar_code_tapped(GtkWidget *widget, const MPZBarCode *code)
{
	GtkWidget *dialog;
	GtkDialogFlags flags = GTK_DIALOG_MODAL | GTK_DIALOG_DESTROY_WITH_PARENT;
	bool data_is_url = g_uri_is_valid(
		code->data, G_URI_FLAGS_PARSE_RELAXED, NULL);

	char* data = strdup(code->data);

	if (data_is_url) {
		dialog = gtk_message_dialog_new(
			GTK_WINDOW(gtk_widget_get_root(widget)),
			flags,
			GTK_MESSAGE_QUESTION,
			GTK_BUTTONS_NONE,
			"Found a URL '%s' encoded in a %s code.",
			code->data,
			code->type);
		gtk_dialog_add_buttons(
			GTK_DIALOG(dialog),
			"_Open URL",
			GTK_RESPONSE_YES,
			NULL);
	} else {
		dialog = gtk_message_dialog_new(
			GTK_WINDOW(gtk_widget_get_root(widget)),
			flags,
			GTK_MESSAGE_QUESTION,
			GTK_BUTTONS_NONE,
			"Found '%s' encoded in a %s code.",
			code->data,
			code->type);
	}
	gtk_dialog_add_buttons(
		GTK_DIALOG(dialog),
		"_Copy",
		GTK_RESPONSE_ACCEPT,
		"_Cancel",
		GTK_RESPONSE_CANCEL,
		NULL);

	g_signal_connect(dialog, "response", G_CALLBACK(on_zbar_dialog_response), data);

	gtk_widget_show(GTK_WIDGET(dialog));
}

static void
preview_pressed(GtkGestureClick *gesture, int n_press, double x, double y)
{
	GtkWidget *widget = gtk_event_controller_get_widget(GTK_EVENT_CONTROLLER(gesture));
	int scale_factor = gtk_widget_get_scale_factor(widget);

	// Tapped zbar result
	if (zbar_result) {
		// Transform the event coordinates to the image
		float offset_x, offset_y, size_x, size_y;
		position_preview(&offset_x, &offset_y, &size_x, &size_y);

		int zbar_x = (x - offset_x) * scale_factor / size_x * preview_buffer_width;
		int zbar_y = (y - offset_y) * scale_factor / size_y * preview_buffer_height;

		for (uint8_t i = 0; i < zbar_result->size; ++i) {
			MPZBarCode *code = &zbar_result->codes[i];

			if (check_point_inside_bounds(zbar_x, zbar_y, code->bounds_x, code->bounds_y)) {
				on_zbar_code_tapped(widget, code);
				return;
			}
		}
	}

	// Tapped preview image itself, try focussing
	if (has_auto_focus_start) {
		mp_io_pipeline_focus();
	}
}

static void
run_camera_switch_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	size_t next_index = camera->index + 1;
	const struct mp_camera_config *next_camera =
		mp_get_camera_config(next_index);

	if (!next_camera) {
		next_index = 0;
		next_camera = mp_get_camera_config(next_index);
	}

	camera = next_camera;
	update_io_pipeline();
}

static void
run_open_settings_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	gtk_stack_set_visible_child_name(GTK_STACK(main_stack), "settings");
}

static void
run_close_settings_action(GSimpleAction *action, GVariant *param, gpointer user_data)
{
	gtk_stack_set_visible_child_name(GTK_STACK(main_stack), "main");
}

static void
on_controls_scale_changed(GtkAdjustment *adjustment, void (*set_fn)(double))
{
	set_fn(gtk_adjustment_get_value(adjustment));
}

static void
update_value(GtkAdjustment *adjustment, GtkLabel *label)
{
	char buf[12];
	snprintf(buf, 12, "%.0f", gtk_adjustment_get_value(adjustment));
	gtk_label_set_label(label, buf);
}

static void
on_auto_controls_toggled(GtkToggleButton *button, void (*set_auto_fn)(bool))
{
	set_auto_fn(gtk_toggle_button_get_active(button));
}

static void
update_scale(GtkToggleButton *button, GtkScale *scale)
{
	gtk_widget_set_sensitive(GTK_WIDGET(scale), !gtk_toggle_button_get_active(button));
}

static void
open_controls(GtkWidget *parent, const char *title_name,
	      double min_value, double max_value, double current,
	      bool auto_enabled,
	      void (*set_fn)(double),
	      void (*set_auto_fn)(bool))
{
	GtkBuilder *builder = gtk_builder_new_from_resource(
		"/org/postmarketos/Megapixels/controls-popover.ui");
	GtkPopover *popover = GTK_POPOVER(gtk_builder_get_object(builder, "controls"));
	GtkScale *scale = GTK_SCALE(gtk_builder_get_object(builder, "scale"));
	GtkLabel *title = GTK_LABEL(gtk_builder_get_object(builder, "title"));
	GtkLabel *value_label = GTK_LABEL(gtk_builder_get_object(builder, "value-label"));
	GtkToggleButton *auto_button = GTK_TOGGLE_BUTTON(gtk_builder_get_object(builder, "auto-button"));

	gtk_label_set_label(title, title_name);

	GtkAdjustment *adjustment = gtk_range_get_adjustment(GTK_RANGE(scale));
	gtk_adjustment_set_lower(adjustment, min_value);
	gtk_adjustment_set_upper(adjustment, max_value);
	gtk_adjustment_set_value(adjustment, current);
	update_value(adjustment, value_label);

	gtk_toggle_button_set_active(auto_button, auto_enabled);
	update_scale(auto_button, scale);

	g_signal_connect(adjustment, "value-changed", G_CALLBACK(on_controls_scale_changed), set_fn);
	g_signal_connect(adjustment, "value-changed", G_CALLBACK(update_value), value_label);
	g_signal_connect(auto_button, "toggled", G_CALLBACK(on_auto_controls_toggled), set_auto_fn);
	g_signal_connect(auto_button, "toggled", G_CALLBACK(update_scale), scale);

	gtk_widget_set_parent(GTK_WIDGET(popover), parent);
	gtk_popover_popup(popover);
	// g_object_unref(popover);
}

static void
set_gain(double value)
{
	if (gain != (int)value) {
		gain = value;
		update_io_pipeline();
	}
}

static void
set_gain_auto(bool is_auto)
{
	if (gain_is_manual != !is_auto) {
		gain_is_manual = !is_auto;
		update_io_pipeline();
	}
}

static void
open_iso_controls(GtkWidget *button, gpointer user_data)
{
	open_controls(button, "ISO", 0, gain_max, gain, !gain_is_manual, set_gain, set_gain_auto);
}

static void
set_shutter(double value)
{
	int new_exposure =
		(int)(value / 360.0 * camera->capture_mode.height);
	if (new_exposure != exposure) {
		exposure = new_exposure;
		update_io_pipeline();
	}
}

static void
set_shutter_auto(bool is_auto)
{
	if (exposure_is_manual != !is_auto) {
		exposure_is_manual = !is_auto;
		update_io_pipeline();
	}
}

static void
open_shutter_controls(GtkWidget *button, gpointer user_data)
{
	open_controls(button, "Shutter", 1.0, 360.0, exposure, !exposure_is_manual, set_shutter, set_shutter_auto);
}

static void
on_realize(GtkWidget *window, gpointer *data)
{
	GtkNative *native = gtk_widget_get_native(window);
	mp_process_pipeline_init_gl(gtk_native_get_surface(native));

	camera = mp_get_camera_config(0);
	update_io_pipeline();
}

static GSimpleAction *
create_simple_action(GtkApplication *app, const char *name, GCallback callback)
{
	GSimpleAction *action = g_simple_action_new(name, NULL);
	g_signal_connect(action, "activate", callback, app);
	g_action_map_add_action(G_ACTION_MAP(app), G_ACTION(action));
	return action;
}

static void update_ui_rotation()
{
	if (device_rotation == 0 || device_rotation == 180) {
		// Portrait
		gtk_widget_set_halign(preview_top_box, GTK_ALIGN_FILL);
		gtk_orientable_set_orientation(GTK_ORIENTABLE(preview_top_box), GTK_ORIENTATION_VERTICAL);

		gtk_widget_set_halign(preview_bottom_box, GTK_ALIGN_FILL);
		gtk_orientable_set_orientation(GTK_ORIENTABLE(preview_bottom_box), GTK_ORIENTATION_HORIZONTAL);

		if (device_rotation == 0) {
			gtk_widget_set_valign(preview_top_box, GTK_ALIGN_START);
			gtk_widget_set_valign(preview_bottom_box, GTK_ALIGN_END);
		} else {
			gtk_widget_set_valign(preview_top_box, GTK_ALIGN_END);
			gtk_widget_set_valign(preview_bottom_box, GTK_ALIGN_START);
		}
	} else {
		// Landscape
		gtk_widget_set_valign(preview_top_box, GTK_ALIGN_FILL);
		gtk_orientable_set_orientation(GTK_ORIENTABLE(preview_top_box), GTK_ORIENTATION_HORIZONTAL);

		gtk_widget_set_valign(preview_bottom_box, GTK_ALIGN_FILL);
		gtk_orientable_set_orientation(GTK_ORIENTABLE(preview_bottom_box), GTK_ORIENTATION_VERTICAL);

		if (device_rotation == 90) {
			gtk_widget_set_halign(preview_top_box, GTK_ALIGN_END);
			gtk_widget_set_halign(preview_bottom_box, GTK_ALIGN_START);
		} else {
			gtk_widget_set_halign(preview_top_box, GTK_ALIGN_START);
			gtk_widget_set_halign(preview_bottom_box, GTK_ALIGN_END);
		}
	}
}

static void display_config_received(GDBusConnection *conn, GAsyncResult *res, gpointer user_data)
{
	GError *error = NULL;
	GVariant *result = g_dbus_connection_call_finish(conn, res, &error);

	if (!result) {
		printf("Failed to get display configuration: %s\n", error->message);
		return;
	}

	GVariant *configs = g_variant_get_child_value(result, 1);
	if (g_variant_n_children(configs) == 0) {
		return;
	}

	GVariant *config = g_variant_get_child_value(configs, 0);
	GVariant *rot_config = g_variant_get_child_value(config, 7);
	uint32_t rotation_index = g_variant_get_uint32(rot_config);

	assert(rotation_index < 4);
	int new_rotation = rotation_index * 90;

	if (new_rotation != device_rotation) {
		device_rotation = new_rotation;
		update_io_pipeline();
		update_ui_rotation();
	}
}

static void update_screen_rotation(GDBusConnection *conn)
{
	g_dbus_connection_call(conn,
		"org.gnome.Mutter.DisplayConfig",
		"/org/gnome/Mutter/DisplayConfig",
		"org.gnome.Mutter.DisplayConfig",
		"GetResources",
		NULL,
		NULL,
		G_DBUS_CALL_FLAGS_NO_AUTO_START,
		-1,
		NULL,
		(GAsyncReadyCallback)display_config_received,
		NULL);
}

static void on_screen_rotate(
	GDBusConnection *conn,
        const gchar *sender_name,
        const gchar *object_path,
        const gchar *interface_name,
        const gchar *signal_name,
        GVariant *parameters,
        gpointer user_data)
{
	update_screen_rotation(conn);
}

static void
activate(GtkApplication *app, gpointer data)
{
	g_object_set(gtk_settings_get_default(), "gtk-application-prefer-dark-theme",
		     TRUE, NULL);

	GdkDisplay *display = gdk_display_get_default();
	GtkIconTheme *icon_theme = gtk_icon_theme_get_for_display(display);
	gtk_icon_theme_add_resource_path(icon_theme, "/org/postmarketos/Megapixels");

	GtkCssProvider *provider = gtk_css_provider_new();
	gtk_css_provider_load_from_resource(
		provider, "/org/postmarketos/Megapixels/camera.css");
	gtk_style_context_add_provider_for_display(
		display, GTK_STYLE_PROVIDER(provider),
		GTK_STYLE_PROVIDER_PRIORITY_APPLICATION);

	GtkBuilder *builder = gtk_builder_new_from_resource(
		"/org/postmarketos/Megapixels/camera.ui");

	GtkWidget *window = GTK_WIDGET(gtk_builder_get_object(builder, "window"));
	GtkWidget *iso_button = GTK_WIDGET(gtk_builder_get_object(builder, "iso-controls-button"));
	GtkWidget *shutter_button = GTK_WIDGET(gtk_builder_get_object(builder, "shutter-controls-button"));
	preview = GTK_WIDGET(gtk_builder_get_object(builder, "preview"));
	main_stack = GTK_WIDGET(gtk_builder_get_object(builder, "main_stack"));
	open_last_stack = GTK_WIDGET(gtk_builder_get_object(builder, "open_last_stack"));
	thumb_last = GTK_WIDGET(gtk_builder_get_object(builder, "thumb_last"));
	process_spinner = GTK_WIDGET(gtk_builder_get_object(builder, "process_spinner"));
	scanned_codes = GTK_WIDGET(gtk_builder_get_object(builder, "scanned-codes"));
	preview_top_box = GTK_WIDGET(gtk_builder_get_object(builder, "top-box"));
	preview_bottom_box = GTK_WIDGET(gtk_builder_get_object(builder, "bottom-box"));

	g_signal_connect(window, "realize", G_CALLBACK(on_realize), NULL);

	g_signal_connect(preview, "realize", G_CALLBACK(preview_realize), NULL);
	g_signal_connect(preview, "render", G_CALLBACK(preview_draw), NULL);
	g_signal_connect(preview, "resize", G_CALLBACK(preview_resize), NULL);
	GtkGesture *click = gtk_gesture_click_new();
	g_signal_connect(click, "pressed", G_CALLBACK(preview_pressed), NULL);
	gtk_widget_add_controller(preview, GTK_EVENT_CONTROLLER(click));

	g_signal_connect(iso_button, "clicked", G_CALLBACK(open_iso_controls), NULL);
	g_signal_connect(shutter_button, "clicked", G_CALLBACK(open_shutter_controls), NULL);

	// Setup actions
	create_simple_action(app, "capture", G_CALLBACK(run_capture_action));
	create_simple_action(app, "switch-camera", G_CALLBACK(run_camera_switch_action));
	create_simple_action(app, "open-settings", G_CALLBACK(run_open_settings_action));
	create_simple_action(app, "close-settings", G_CALLBACK(run_close_settings_action));
	create_simple_action(app, "open-last", G_CALLBACK(run_open_last_action));
	create_simple_action(app, "open-photos", G_CALLBACK(run_open_photos_action));
	create_simple_action(app, "quit", G_CALLBACK(run_quit_action));

	// Setup shortcuts
	const char *capture_accels[] = { "space", NULL };
	gtk_application_set_accels_for_action(app, "app.capture", capture_accels);

	const char *quit_accels[] = { "<Ctrl>q", "<Ctrl>w", NULL };
	gtk_application_set_accels_for_action(app, "app.quit", quit_accels);

	// Listen for phosh rotation
	GDBusConnection *conn = g_application_get_dbus_connection(G_APPLICATION(app));
	g_dbus_connection_signal_subscribe(
		conn,
		NULL,
		"org.gnome.Mutter.DisplayConfig",
		"MonitorsChanged",
		"/org/gnome/Mutter/DisplayConfig",
		NULL,
		G_DBUS_SIGNAL_FLAGS_NONE,
		&on_screen_rotate,
		NULL,
		NULL);
	update_screen_rotation(conn);

	mp_io_pipeline_start();

	gtk_application_add_window(app, GTK_WINDOW(window));
	gtk_widget_show(window);
}

static void
shutdown(GApplication *app, gpointer data)
{
	// Only do cleanup in development, let the OS clean up otherwise
#ifdef DEBUG
	mp_io_pipeline_stop();
#endif
}

int
main(int argc, char *argv[])
{
#ifdef RENDERDOC
	{
		void *mod = dlopen("librenderdoc.so", RTLD_NOW | RTLD_NOLOAD);
		if (mod)
		{
		    pRENDERDOC_GetAPI RENDERDOC_GetAPI = (pRENDERDOC_GetAPI)dlsym(mod, "RENDERDOC_GetAPI");
		    int ret = RENDERDOC_GetAPI(eRENDERDOC_API_Version_1_1_2, (void **)&rdoc_api);
		    assert(ret == 1);
		}
		else
		{
			printf("Renderdoc not found\n");
		}
	}
#endif

	if (!mp_load_config())
		return 1;

	setenv("LC_NUMERIC", "C", 1);

	GtkApplication *app = gtk_application_new("org.postmarketos.Megapixels", 0);

	g_signal_connect(app, "activate", G_CALLBACK(activate), NULL);
	g_signal_connect(app, "shutdown", G_CALLBACK(shutdown), NULL);

	g_application_run(G_APPLICATION(app), argc, argv);

	return 0;
}