summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2011-08-15 11:20:51 +0200
committerMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2011-08-15 11:20:51 +0200
commit6e1c9e9db78464ad49e2afdf7b81afa5c4396e74 (patch)
tree8c0d431df91130a867622cf3dd6b1ba1641bca9f
parent2d5e90ecabacd21652f7ab14eb84eb828fc7e093 (diff)
downloadufodecode-6e1c9e9db78464ad49e2afdf7b81afa5c4396e74.tar.gz
ufodecode-6e1c9e9db78464ad49e2afdf7b81afa5c4396e74.tar.bz2
ufodecode-6e1c9e9db78464ad49e2afdf7b81afa5c4396e74.tar.xz
ufodecode-6e1c9e9db78464ad49e2afdf7b81afa5c4396e74.zip
Simple (spatial) de-interlacer
-rw-r--r--Makefile12
-rw-r--r--deinterlace.c135
2 files changed, 144 insertions, 3 deletions
diff --git a/Makefile b/Makefile
index 0edd281..b7bf0e5 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,7 @@
CC = gcc
-CFLAGS = -g -std=c99
+CFLAGS = -g -ggdb -std=c99 -Wall -Werror
-all: ipedec
+all: ipedec deinterlace
ipedec: ipedec.o
$(CC) $(LDFLAGS) -o $@ $^
@@ -9,8 +9,14 @@ ipedec: ipedec.o
ipedec.o: ipedec.c
$(CC) $(CFLAGS) -c -o $@ $<
+deinterlace: deinterlace.o
+ $(CC) $(LDFLAGS) -o $@ $^
+
+deinterlace.o: deinterlace.c
+ $(CC) $(CFLAGS) -c -o $@ $<
+
clean:
- rm -f ipedec ipedec.o
+ rm -f ipedec ipedec.o deinterlace deinterlace.o
.PHONY: clean
diff --git a/deinterlace.c b/deinterlace.c
new file mode 100644
index 0000000..a7a81aa
--- /dev/null
+++ b/deinterlace.c
@@ -0,0 +1,135 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdint.h>
+#include <errno.h>
+#include <assert.h>
+#include <getopt.h>
+
+#define IPE_DEFAULT_WIDTH 2048
+
+char *strdup(const char *str)
+{
+ int n = strlen(str) + 1;
+ char *dup = malloc(n);
+ if (dup)
+ strcpy(dup, str);
+ return dup;
+}
+
+int process_simple_frame(uint16_t *frame_in, uint16_t *frame_out, int width, int height)
+{
+ const size_t row_size_bytes = width * sizeof(uint16_t);
+
+ for (int row = 0; row < height; row++) {
+ /* Copy one line */
+ memcpy(frame_out, frame_in + row*width, row_size_bytes);
+ frame_out += width;
+
+ /* Interpolate between source row and row+1 */
+ for (int x = 0; x < width; x++) {
+ frame_out[x] = (int) (0.5 * frame_in[row*width + x] + 0.5 * frame_in[(row+1)*width + x]);
+ }
+ frame_out += width;
+ }
+
+ /* Copy last row */
+ memcpy(frame_out, frame_in + width * (height - 1), row_size_bytes);
+ return 0;
+}
+
+int process_file(const char *filename, int width, int height, int num_lines_skipped)
+{
+ FILE *fp = fopen(filename, "rb");
+ if (fp == NULL) {
+ return EIO;
+ }
+ fseek(fp, 0, SEEK_END);
+ const size_t file_size = ftell(fp);
+ rewind(fp);
+
+ uint16_t *frame_in_buffer = (uint16_t *) malloc(file_size);
+ if (frame_in_buffer == NULL) {
+ fclose(fp);
+ return ENOMEM;
+ }
+
+ size_t buffer_length = fread(frame_in_buffer, 1, file_size, fp);
+ fclose(fp);
+ if (buffer_length != file_size) {
+ free(frame_in_buffer);
+ return EIO;
+ }
+
+ uint16_t *frame_out_buffer = (uint16_t *) malloc(file_size * 2);
+ if (frame_out_buffer == NULL) {
+ fclose(fp);
+ return ENOMEM;
+ }
+
+ const int num_frames = file_size / (width * height * sizeof(uint16_t));
+ printf("de-interlacing %i frames...\n", num_frames);
+ for (int frame = 0; frame < num_frames; frame++) {
+ process_simple_frame(
+ frame_in_buffer + frame * (width * height),
+ frame_out_buffer + frame * (width * height * 2),
+ width, height);
+ }
+
+ fp = fopen("result.raw", "wb");
+ fwrite(frame_out_buffer, 1, file_size * 2, fp);
+ fclose(fp);
+
+ free(frame_in_buffer);
+ free(frame_out_buffer);
+ return 0;
+}
+
+int main(int argc, char const* argv[])
+{
+ static struct option long_options[] = {
+ { "width", 1, 0, 'w' },
+ { "interlaced-height", 1, 0, 'i' },
+ { "target-height", 1, 0, 't' },
+ { "file", 1, 0, 'f' },
+ { NULL, 0, NULL, 0 }
+ };
+
+ int c, option_index = 0, errnum;
+ int width = IPE_DEFAULT_WIDTH;
+ int interlaced_height = -1;
+ int target_height = -1;
+ char *file_name = NULL;
+ char *end;
+
+ while ((c = getopt_long(argc, (char *const *) argv, "w:i:t:f:", long_options, &option_index)) != -1) {
+ switch (c) {
+ case 'w':
+ width = (int) strtol(optarg, &end, 10);
+ break;
+ case 'i':
+ interlaced_height = (int) strtol(optarg, &end, 10);
+ break;
+ case 't':
+ target_height = (int) strtol(optarg, &end, 10);
+ break;
+ case 'f':
+ file_name = strdup( (char *) optarg);
+ break;
+ }
+ }
+
+ if (interlaced_height == -1 || file_name == NULL) {
+ printf("Usage: deinterlace --interlaced-height=[number] --target-height=[number] --file=[name]\n");
+ exit(EXIT_FAILURE);
+ }
+
+ if (target_height == -1)
+ target_height = interlaced_height * 2;
+
+ if ((errnum = process_file(file_name, width, interlaced_height, (target_height / interlaced_height) - 1)))
+ printf("Error occured: %s\n", strerror(errnum));
+
+ free(file_name);
+ return 0;
+}