summaryrefslogtreecommitdiff
path: root/src/util.c
blob: 7b7077d51b4e972943e98d94e7486fb8116f2dc3 (plain)
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
#define _XOPEN_SOURCE 700
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>

#include "video_decode.h"
#include "util.h"

int
pgm_read(char *filename, unsigned char **buf, int *xsize, int *ysize)
{
	FILE *f = NULL;
	unsigned char *b = NULL;
	int width, height;
	ssize_t n;

	f = fopen(filename, "r");
	if (f == NULL)
		return -1;

	if (fscanf(f, "P5\n%d %d\n255\n", &width, &height) == 0)
		return -1;

	if (width * height <= 0)
		goto fail;

	if ((b = malloc(width * height)) == NULL)
		goto fail;
	if ((n = fread(b, 1, width * height, f)) == 0 || n != width * height)
		goto fail;

	fclose(f);

	*buf = b;
	*xsize = width;
	*ysize = height;

	return 0;

fail:
	if (f) fclose(f);
	if (b) free(b);
	return -1;
}

int
pgm_save(unsigned char *buf, int wrap, int xsize, int ysize, char *filename)
{
	FILE *f;
	int i;

	f = fopen(filename, "w");
	if (f == NULL)
		return -1;
	fprintf(f, "P5\n%d %d\n%d\n", xsize, ysize, 255);
	for(i = 0; i < ysize; i++)
		fwrite(buf + i * wrap, 1, xsize, f);
	fclose(f);

	return 0;
}

void
frame_mix(struct video_frame *frame_a, struct video_frame *frame_b)
{
        uint8_t *row_a = frame_a->data, *col_a = NULL;
        uint8_t *row_b = frame_b->data, *col_b = NULL;

        for (int y = 0; y < frame_a->height; ++y) {
                col_a = row_a;
                col_b = row_b;
                for (int x = 0; x < frame_a->width; ++x) {
			if (x > frame_a->width / 2)
	                        *col_a = *col_b;
                        col_a++;
                        col_b++;
                }
                row_a += frame_a->stride;
                row_b += frame_b->stride;
        }
}

/* print time in format: "hh:mm:ss.xxx" */
void
print_time(int64_t msec) {

	int hh = msec / 1000 / 60 / 60;
	msec -= hh * 1000 * 60 * 60;
	int mm = msec / 1000 / 60;
	msec -= mm * 1000 * 60;
	int ss = msec / 1000;
	msec -= ss * 1000;

	printf("%02d:%02d:%02d.%03jd", hh, mm, ss, msec);
}