summaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2011-12-01 09:43:58 +0100
committerMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2011-12-01 09:43:58 +0100
commit6fbb97c1b908b45ebbc4313e87aa099519b10672 (patch)
treed7cf7f2240978262cabeca331686504abc44c811 /test
parent8ed172cab141f02390d7c77abec9eb18fe212182 (diff)
downloadufodecode-6fbb97c1b908b45ebbc4313e87aa099519b10672.tar.gz
ufodecode-6fbb97c1b908b45ebbc4313e87aa099519b10672.tar.bz2
ufodecode-6fbb97c1b908b45ebbc4313e87aa099519b10672.tar.xz
ufodecode-6fbb97c1b908b45ebbc4313e87aa099519b10672.zip
(Re-)move executable sources
Diffstat (limited to 'test')
-rw-r--r--test/deinterlace.c156
1 files changed, 156 insertions, 0 deletions
diff --git a/test/deinterlace.c b/test/deinterlace.c
new file mode 100644
index 0000000..07009dc
--- /dev/null
+++ b/test/deinterlace.c
@@ -0,0 +1,156 @@
+#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(const 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 weave(const uint16_t *frames_in, uint16_t *frame_out, int width, int height)
+{
+ const size_t row_size_bytes = width * sizeof(uint16_t);
+ const size_t frame_offset = width * height;
+ for (int row = 0; row < height; row++) {
+ memcpy(frame_out, frames_in + row*width, row_size_bytes);
+ frame_out += width;
+ memcpy(frame_out, frames_in + frame_offset + row*width, row_size_bytes);
+ frame_out += width;
+ }
+ 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);
+ }
+ */
+ for (int frame = 0; frame < num_frames-1; frame++) {
+ weave(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;
+}
+