aboutsummaryrefslogtreecommitdiff
path: root/args.c
blob: e2248a69435307d025347dacb1725c29b273719c (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
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
129
130
131
132
133
134
135
136
137
/*
 * V4L2 Codec decoding example application
 * Kamil Debski <k.debski@samsung.com>
 *
 * Argument parser
 *
 * Copyright 2012 Samsung Electronics Co., Ltd.
 * Copyright (c) 2015 Linaro Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <linux/videodev2.h>

#include "common.h"

void print_usage(char *name)
{
	printf("Usage:\n");
	printf("\t%s\n", name);
	printf("\t-c <codec> - The codec to be used for encoding\n");
	printf("\t\t     Available codecs: mpeg4, h264, h263, xvid, mpeg2, mpeg1, vp8\n");
	printf("\t-d use dmabuf instead of mmap\n");
	printf("\t-m <device> - video encoder device (e.g. /dev/video33)\n");
	printf("\t-w video width\n");
	printf("\t-h video height\n");
	printf("\t-f save encoded frames on disk\n");
	printf("\t-g disable test pattern generator\n");
	printf("\t-v verbose mode\n");
	printf("\t-B bitrate\n");
	printf("\t-F framerate\n");
	printf("\n");
}

void init_to_defaults(struct instance *i)
{
	memset(i, 0, sizeof(*i));
}

int get_codec(char *str)
{
	if (strncasecmp("mpeg4", str, 5) == 0) {
		return V4L2_PIX_FMT_MPEG4;
	} else if (strncasecmp("h264", str, 5) == 0) {
		return V4L2_PIX_FMT_H264;
	} else if (strncasecmp("h263", str, 5) == 0) {
		return V4L2_PIX_FMT_H263;
	} else if (strncasecmp("xvid", str, 5) == 0) {
		return V4L2_PIX_FMT_XVID;
	} else if (strncasecmp("mpeg2", str, 5) == 0) {
		return V4L2_PIX_FMT_MPEG2;
	} else if (strncasecmp("mpeg1", str, 5) == 0) {
		return V4L2_PIX_FMT_MPEG1;
	} else if (strncasecmp("vp8", str, 5) == 0) {
		return V4L2_PIX_FMT_VP8;
	}
	return 0;
}

int parse_args(struct instance *i, int argc, char **argv)
{
	int c;

	init_to_defaults(i);

	while ((c = getopt(argc, argv, "w:h:c:d:m:f:n:gvB:F:")) != -1) {
		switch (c) {
		case 'c':
			i->codec = get_codec(optarg);
			i->codec_name = optarg;
			break;
		case 'd':
			i->use_dmabuf = 1;
			break;
		case 'm':
			i->video.name = optarg;
			break;
		case 'w':
			i->width = atoi(optarg);
			break;
		case 'h':
			i->height = atoi(optarg);
			break;
		case 'f':
			i->save_encoded = 1;
			i->save_path = optarg;
			break;
		case 'n':
			i->num_frames_to_save = atoi(optarg);
			break;
		case 'g':
			i->disable_gentest = true;
			break;
		case 'v':
			i->verbose = true;
			break;
		case 'B':
			i->bitrate = atoi(optarg);
			break;
		case 'F':
			i->framerate = atoi(optarg);
			break;
		default:
			err("Bad argument");
			return -1;
		}
	}

	if (!i->video.name) {
		err("The following arguments are required: -m -c");
		return -1;
	}

	if (!i->codec) {
		err("Unknown or not set codec (-c)");
		return -1;
	}

	return 0;
}