1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/time.h>
#include <getopt.h>
#include <ufodecode.h>
static int read_raw_file(const char *filename, char **buffer, size_t *length)
{
FILE *fp = fopen(filename, "rb");
if (fp == NULL)
return ENOENT;
fseek(fp, 0, SEEK_END);
*length = ftell(fp);
rewind(fp);
*buffer = (char *) malloc(*length);
if (*buffer == NULL) {
fclose(fp);
return ENOMEM;
}
size_t buffer_length = fread(*buffer, 1, *length, fp);
fclose(fp);
if (buffer_length != *length) {
free(buffer);
return ERANGE;
}
return 0;
}
static void process_file(const char *filename, int rows, int clear_frame)
{
char *buffer = NULL;
size_t num_bytes = 0;
int err = 0;
uint16_t *pixels = (uint16_t *) malloc(2048 * 1088 * sizeof(uint16_t));
uint32_t num_rows, frame_number, time_stamp;
int num_frames = 0;
struct timeval start, end;
long seconds = 0L, useconds = 0L;
int error = read_raw_file(filename, &buffer, &num_bytes);
if (error) {
fprintf(stderr, "Error processing %s: %s\n", filename, strerror(error));
return;
}
ufo_decoder decoder = ufo_decoder_new(rows, 2048, (uint32_t *) buffer, num_bytes);
if (!decoder) {
fprintf(stderr, "Failed to initialize decoder\n");
return;
}
char output_name[256];
snprintf(output_name, 256, "%s.raw", filename);
FILE *fp = fopen(output_name, "wb");
if (!fp) {
fprintf(stderr, "Failed to open file for writing\n");
return;
}
while (!err) {
if (clear_frame)
memset(pixels, 0, 2048 * 1088 * sizeof(uint16_t));
gettimeofday(&start, NULL);
err = ufo_decoder_get_next_frame(decoder, &pixels, &num_rows, &frame_number, &time_stamp, NULL);
gettimeofday(&end, NULL);
if (!err) {
num_frames++;
seconds += end.tv_sec - start.tv_sec;
useconds += end.tv_usec - start.tv_usec;
fwrite(pixels, sizeof(uint16_t), 2048 * 1088, fp);
}
}
fclose(fp);
float mtime = seconds * 1000.0 + useconds / 1000.0;
printf("Decoded %i frames in %.5fms\n", num_frames, mtime);
free(pixels);
ufo_decoder_free(decoder);
free(buffer);
}
int main(int argc, char const* argv[])
{
int getopt_ret, index;
static struct option long_options[] = {
{ "num-rows", required_argument, 0, 'r' },
{ "clear-frame", no_argument, 0, 'c' },
{ "help", no_argument, 0, '?' },
{ 0, 0, 0, 0 }
};
int clear_frame = 0;
int rows = -1;
while ((getopt_ret = getopt_long(argc, (char *const *) argv, "r:c:?", long_options, &index)) != -1) {
switch (getopt_ret) {
case 'r':
rows = atoi(optarg);
break;
case 'c':
clear_frame = 1;
break;
default:
break;
}
}
while (optind < argc)
process_file(argv[optind++], rows, clear_frame);
return 0;
}
|
