[sigrok-cli.git] / options.c

/*
 * This file is part of the sigrok-cli project.
 *
 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <glib.h>
#include "sigrok-cli.h"

gboolean opt_version = FALSE;
gint opt_loglevel = SR_LOG_WARN; /* Show errors+warnings by default. */
gboolean opt_scan_devs = FALSE;
gboolean opt_wait_trigger = FALSE;
gchar *opt_input_file = NULL;
gchar *opt_output_file = NULL;
gchar *opt_drv = NULL;
gchar *opt_config = NULL;
gchar *opt_channels = NULL;
gchar *opt_channel_group = NULL;
gchar *opt_triggers = NULL;
gchar *opt_pds = NULL;
#ifdef HAVE_SRD
gchar *opt_pd_stack = NULL;
gchar *opt_pd_annotations = NULL;
gchar *opt_pd_meta = NULL;
gchar *opt_pd_binary = NULL;
#endif
gchar *opt_input_format = NULL;
gchar *opt_output_format = NULL;
gchar *opt_transform_module = NULL;
gchar *opt_show = NULL;
gchar *opt_time = NULL;
gchar *opt_samples = NULL;
gchar *opt_frames = NULL;
gchar *opt_continuous = NULL;
gchar *opt_get = NULL;
gchar *opt_set = NULL;

/*
 * Defines a callback function that generates an error if an
 * option occurs twice.
 */
#define CHECK_ONCE(option) \
static gboolean check_ ## option                                          \
	(const gchar *option_name, const gchar *value,                    \
	gpointer data, GError **error)                                    \
{                                                                         \
	(void)data;                                                       \
                                                                          \
	static gboolean seen = FALSE;                                     \
	if (seen) {                                                       \
		g_set_error(error, G_OPTION_ERROR, G_OPTION_ERROR_FAILED, \
		            "superfluous option \"%s\"", option_name);    \
		return FALSE;                                             \
	}                                                                 \
                                                                          \
	option = g_strdup(value);                                         \
	seen = TRUE;                                                      \
	return TRUE;                                                      \
}

CHECK_ONCE(opt_drv)
CHECK_ONCE(opt_config)
CHECK_ONCE(opt_input_format)
CHECK_ONCE(opt_output_format)
CHECK_ONCE(opt_transform_module)
CHECK_ONCE(opt_channels)
CHECK_ONCE(opt_channel_group)
CHECK_ONCE(opt_triggers)
#ifdef HAVE_SRD
CHECK_ONCE(opt_pds)
CHECK_ONCE(opt_pd_stack)
CHECK_ONCE(opt_pd_annotations)
CHECK_ONCE(opt_pd_meta)
CHECK_ONCE(opt_pd_binary)
#endif
CHECK_ONCE(opt_time)
CHECK_ONCE(opt_samples)
CHECK_ONCE(opt_frames)
CHECK_ONCE(opt_get)

#undef CHECK_STR_ONCE

static gchar **input_file_array = NULL;
static gchar **output_file_array = NULL;

static const GOptionEntry optargs[] = {
	{"version", 'V', 0, G_OPTION_ARG_NONE, &opt_version,
			"Show version and support list", NULL},
	{"loglevel", 'l', 0, G_OPTION_ARG_INT, &opt_loglevel,
			"Set loglevel (5 is most verbose)", NULL},
	{"driver", 'd', 0, G_OPTION_ARG_CALLBACK, &check_opt_drv,
			"The driver to use", NULL},
	{"config", 'c', 0, G_OPTION_ARG_CALLBACK, &check_opt_config,
			"Specify device configuration options", NULL},
	{"input-file", 'i', 0, G_OPTION_ARG_FILENAME_ARRAY, &input_file_array,
			"Load input from file", NULL},
	{"input-format", 'I', 0, G_OPTION_ARG_CALLBACK, &check_opt_input_format,
			"Input format", NULL},
	{"output-file", 'o', 0, G_OPTION_ARG_FILENAME_ARRAY, &output_file_array,
			"Save output to file", NULL},
	{"output-format", 'O', 0, G_OPTION_ARG_CALLBACK, &check_opt_output_format,
			"Output format", NULL},
	{"transform-module", 'T', 0, G_OPTION_ARG_CALLBACK, &check_opt_transform_module,
			"Transform module", NULL},
	{"channels", 'C', 0, G_OPTION_ARG_CALLBACK, &check_opt_channels,
			"Channels to use", NULL},
	{"channel-group", 'g', 0, G_OPTION_ARG_CALLBACK, &check_opt_channel_group,
			"Channel groups", NULL},
	{"triggers", 't', 0, G_OPTION_ARG_CALLBACK, &check_opt_triggers,
			"Trigger configuration", NULL},
	{"wait-trigger", 'w', 0, G_OPTION_ARG_NONE, &opt_wait_trigger,
			"Wait for trigger", NULL},
#ifdef HAVE_SRD
	{"protocol-decoders", 'P', 0, G_OPTION_ARG_CALLBACK, &check_opt_pds,
			"Protocol decoders to run", NULL},
	{"protocol-decoder-stack", 'S', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_stack,
			"Protocol decoder stack", NULL},
	{"protocol-decoder-annotations", 'A', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_annotations,
			"Protocol decoder annotation(s) to show", NULL},
	{"protocol-decoder-meta", 'M', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_meta,
			"Protocol decoder meta output to show", NULL},
	{"protocol-decoder-binary", 'B', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_binary,
			"Protocol decoder binary output to show", NULL},
#endif
	{"scan", 0, 0, G_OPTION_ARG_NONE, &opt_scan_devs,
			"Scan for devices", NULL},
	{"show", 0, 0, G_OPTION_ARG_NONE, &opt_show,
			"Show device detail", NULL},
	{"time", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_time,
			"How long to sample (ms)", NULL},
	{"samples", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_samples,
			"Number of samples to acquire", NULL},
	{"frames", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_frames,
			"Number of frames to acquire", NULL},
	{"continuous", 0, 0, G_OPTION_ARG_NONE, &opt_continuous,
			"Sample continuously", NULL},
	{"get", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_get, "Get device option only", NULL},
	{"set", 0, 0, G_OPTION_ARG_NONE, &opt_set, "Set device options only", NULL},
	{NULL, 0, 0, 0, NULL, NULL, NULL}
};

/*
 * Parses the command line and sets all the 'opt_...' variables.
 * Returns zero on success, non-zero otherwise.
 */
int parse_options(int argc, char **argv)
{
	GError *error = NULL;
	GOptionContext *context = g_option_context_new(NULL);
	int ret = 1;

	g_option_context_add_main_entries(context, optargs, NULL);

	if (!g_option_context_parse(context, &argc, &argv, &error)) {
		g_critical("%s", error->message);
		goto done;
	}

	/*
	 * Because of encoding issues with filenames (mentioned in the glib
	 * documentation), we don't check them with a callback function, but
	 * collect them into arrays and then check if the arrays contain at
	 * most one element.
	 */
	if (NULL != input_file_array) {
		if (NULL != input_file_array[0] && NULL != input_file_array[1]) {
			g_critical("option \"--input-file/-i\" only allowed once");
			goto done;
		}
		opt_input_file = g_strdup(input_file_array[0]);
	}

	if (NULL != output_file_array) {
		if (NULL != output_file_array[0] && NULL != output_file_array[1]) {
			g_critical("option \"--output-file/-o\" only allowed once");
			goto done;
		}
		opt_output_file = g_strdup(output_file_array[0]);
	}

	if (1 != argc) {
		g_critical("superfluous command line argument \"%s\"", argv[1]);
		goto done;
	}

	ret = 0;

done:
	g_option_context_free(context);
	g_strfreev(input_file_array);
	g_strfreev(output_file_array);
	input_file_array = NULL;
	output_file_array = NULL;

	return ret;
}

void show_help(void)
{
	GOptionContext *context = g_option_context_new(NULL);
	g_option_context_add_main_entries(context, optargs, NULL);

	char *help = g_option_context_get_help(context, TRUE, NULL);
	printf("%s", help);
	g_free(help);

	g_option_context_free(context);
}
Commit	Line	Data
cd62e027 JS	1	/*
	2	* This file is part of the sigrok-cli project.
	3	*
	4	* Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
	5	*
	6	* This program is free software: you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License as published by
	8	* the Free Software Foundation, either version 3 of the License, or
	9	* (at your option) any later version.
	10	*
	11	* This program is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	* GNU General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU General Public License
	17	* along with this program. If not, see <http://www.gnu.org/licenses/>.
	18	*/
	19
d486cbdd	20	#include <config.h>
cd62e027	21	#include <glib.h>
662a1e27	22	#include "sigrok-cli.h"
cd62e027 JS	23
	24	gboolean opt_version = FALSE;
	25	gint opt_loglevel = SR_LOG_WARN; /* Show errors+warnings by default. */
	26	gboolean opt_scan_devs = FALSE;
	27	gboolean opt_wait_trigger = FALSE;
	28	gchar *opt_input_file = NULL;
	29	gchar *opt_output_file = NULL;
	30	gchar *opt_drv = NULL;
	31	gchar *opt_config = NULL;
	32	gchar *opt_channels = NULL;
	33	gchar *opt_channel_group = NULL;
	34	gchar *opt_triggers = NULL;
	35	gchar *opt_pds = NULL;
	36	#ifdef HAVE_SRD
	37	gchar *opt_pd_stack = NULL;
	38	gchar *opt_pd_annotations = NULL;
	39	gchar *opt_pd_meta = NULL;
	40	gchar *opt_pd_binary = NULL;
	41	#endif
	42	gchar *opt_input_format = NULL;
	43	gchar *opt_output_format = NULL;
6f7b4c5d	44	gchar *opt_transform_module = NULL;
cd62e027 JS	45	gchar *opt_show = NULL;
	46	gchar *opt_time = NULL;
	47	gchar *opt_samples = NULL;
	48	gchar *opt_frames = NULL;
	49	gchar *opt_continuous = NULL;
62a64762	50	gchar *opt_get = NULL;
cd62e027 JS	51	gchar *opt_set = NULL;
cd62e027 JS	52
f0f54487 UH	53	/*
	54	* Defines a callback function that generates an error if an
	55	* option occurs twice.
	56	*/
0894b972 JS	57	#define CHECK_ONCE(option) \
	58	static gboolean check_ ## option \
	59	(const gchar option_name, const gchar value, \
	60	gpointer data, GError **error) \
	61	{ \
	62	(void)data; \
	63	\
	64	static gboolean seen = FALSE; \
	65	if (seen) { \
	66	g_set_error(error, G_OPTION_ERROR, G_OPTION_ERROR_FAILED, \
	67	"superfluous option \"%s\"", option_name); \
	68	return FALSE; \
	69	} \
	70	\
	71	option = g_strdup(value); \
	72	seen = TRUE; \
	73	return TRUE; \
	74	}
	75
	76	CHECK_ONCE(opt_drv)
	77	CHECK_ONCE(opt_config)
	78	CHECK_ONCE(opt_input_format)
	79	CHECK_ONCE(opt_output_format)
6f7b4c5d	80	CHECK_ONCE(opt_transform_module)
0894b972 JS	81	CHECK_ONCE(opt_channels)
	82	CHECK_ONCE(opt_channel_group)
	83	CHECK_ONCE(opt_triggers)
	84	#ifdef HAVE_SRD
	85	CHECK_ONCE(opt_pds)
	86	CHECK_ONCE(opt_pd_stack)
	87	CHECK_ONCE(opt_pd_annotations)
	88	CHECK_ONCE(opt_pd_meta)
	89	CHECK_ONCE(opt_pd_binary)
	90	#endif
	91	CHECK_ONCE(opt_time)
	92	CHECK_ONCE(opt_samples)
	93	CHECK_ONCE(opt_frames)
62a64762	94	CHECK_ONCE(opt_get)
0894b972 JS	95
	96	#undef CHECK_STR_ONCE
	97
44ac7614 JS	98	static gchar **input_file_array = NULL;
	99	static gchar **output_file_array = NULL;
	100
cd62e027 JS	101	static const GOptionEntry optargs[] = {
	102	{"version", 'V', 0, G_OPTION_ARG_NONE, &opt_version,
	103	"Show version and support list", NULL},
	104	{"loglevel", 'l', 0, G_OPTION_ARG_INT, &opt_loglevel,
	105	"Set loglevel (5 is most verbose)", NULL},
0894b972	106	{"driver", 'd', 0, G_OPTION_ARG_CALLBACK, &check_opt_drv,
cd62e027	107	"The driver to use", NULL},
0894b972	108	{"config", 'c', 0, G_OPTION_ARG_CALLBACK, &check_opt_config,
cd62e027	109	"Specify device configuration options", NULL},
44ac7614	110	{"input-file", 'i', 0, G_OPTION_ARG_FILENAME_ARRAY, &input_file_array,
cd62e027	111	"Load input from file", NULL},
0894b972	112	{"input-format", 'I', 0, G_OPTION_ARG_CALLBACK, &check_opt_input_format,
cd62e027	113	"Input format", NULL},
44ac7614	114	{"output-file", 'o', 0, G_OPTION_ARG_FILENAME_ARRAY, &output_file_array,
cd62e027	115	"Save output to file", NULL},
0894b972	116	{"output-format", 'O', 0, G_OPTION_ARG_CALLBACK, &check_opt_output_format,
cd62e027	117	"Output format", NULL},
6f7b4c5d UH	118	{"transform-module", 'T', 0, G_OPTION_ARG_CALLBACK, &check_opt_transform_module,
6f7b4c5d UH	119	"Transform module", NULL},
0894b972	120	{"channels", 'C', 0, G_OPTION_ARG_CALLBACK, &check_opt_channels,
cd62e027	121	"Channels to use", NULL},
0894b972	122	{"channel-group", 'g', 0, G_OPTION_ARG_CALLBACK, &check_opt_channel_group,
cd62e027	123	"Channel groups", NULL},
0894b972	124	{"triggers", 't', 0, G_OPTION_ARG_CALLBACK, &check_opt_triggers,
cd62e027 JS	125	"Trigger configuration", NULL},
	126	{"wait-trigger", 'w', 0, G_OPTION_ARG_NONE, &opt_wait_trigger,
	127	"Wait for trigger", NULL},
	128	#ifdef HAVE_SRD
0894b972	129	{"protocol-decoders", 'P', 0, G_OPTION_ARG_CALLBACK, &check_opt_pds,
cd62e027	130	"Protocol decoders to run", NULL},
0894b972	131	{"protocol-decoder-stack", 'S', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_stack,
cd62e027	132	"Protocol decoder stack", NULL},
0894b972	133	{"protocol-decoder-annotations", 'A', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_annotations,
cd62e027	134	"Protocol decoder annotation(s) to show", NULL},
0894b972	135	{"protocol-decoder-meta", 'M', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_meta,
cd62e027	136	"Protocol decoder meta output to show", NULL},
0894b972	137	{"protocol-decoder-binary", 'B', 0, G_OPTION_ARG_CALLBACK, &check_opt_pd_binary,
cd62e027 JS	138	"Protocol decoder binary output to show", NULL},
	139	#endif
	140	{"scan", 0, 0, G_OPTION_ARG_NONE, &opt_scan_devs,
	141	"Scan for devices", NULL},
	142	{"show", 0, 0, G_OPTION_ARG_NONE, &opt_show,
	143	"Show device detail", NULL},
0894b972	144	{"time", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_time,
cd62e027	145	"How long to sample (ms)", NULL},
0894b972	146	{"samples", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_samples,
cd62e027	147	"Number of samples to acquire", NULL},
0894b972	148	{"frames", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_frames,
cd62e027 JS	149	"Number of frames to acquire", NULL},
	150	{"continuous", 0, 0, G_OPTION_ARG_NONE, &opt_continuous,
	151	"Sample continuously", NULL},
62a64762	152	{"get", 0, 0, G_OPTION_ARG_CALLBACK, &check_opt_get, "Get device option only", NULL},
cd62e027 JS	153	{"set", 0, 0, G_OPTION_ARG_NONE, &opt_set, "Set device options only", NULL},
	154	{NULL, 0, 0, 0, NULL, NULL, NULL}
	155	};
	156
f0f54487 UH	157	/*
	158	* Parses the command line and sets all the 'opt_...' variables.
	159	* Returns zero on success, non-zero otherwise.
	160	*/
cd62e027 JS	161	int parse_options(int argc, char **argv)
	162	{
	163	GError *error = NULL;
	164	GOptionContext *context = g_option_context_new(NULL);
	165	int ret = 1;
	166
	167	g_option_context_add_main_entries(context, optargs, NULL);
	168
	169	if (!g_option_context_parse(context, &argc, &argv, &error)) {
	170	g_critical("%s", error->message);
	171	goto done;
	172	}
	173
44ac7614 JS	174	/*
	175	* Because of encoding issues with filenames (mentioned in the glib
	176	* documentation), we don't check them with a callback function, but
	177	* collect them into arrays and then check if the arrays contain at
	178	* most one element.
	179	*/
	180	if (NULL != input_file_array) {
	181	if (NULL != input_file_array[0] && NULL != input_file_array[1]) {
	182	g_critical("option \"--input-file/-i\" only allowed once");
	183	goto done;
	184	}
	185	opt_input_file = g_strdup(input_file_array[0]);
	186	}
	187
	188	if (NULL != output_file_array) {
	189	if (NULL != output_file_array[0] && NULL != output_file_array[1]) {
	190	g_critical("option \"--output-file/-o\" only allowed once");
	191	goto done;
	192	}
	193	opt_output_file = g_strdup(output_file_array[0]);
	194	}
	195
41ce2cbb JS	196	if (1 != argc) {
	197	g_critical("superfluous command line argument \"%s\"", argv[1]);
	198	goto done;
	199	}
	200
cd62e027 JS	201	ret = 0;
	202
	203	done:
	204	g_option_context_free(context);
44ac7614 JS	205	g_strfreev(input_file_array);
	206	g_strfreev(output_file_array);
	207	input_file_array = NULL;
	208	output_file_array = NULL;
cd62e027 JS	209
	210	return ret;
	211	}
	212
	213	void show_help(void)
	214	{
	215	GOptionContext *context = g_option_context_new(NULL);
	216	g_option_context_add_main_entries(context, optargs, NULL);
	217
	218	char *help = g_option_context_get_help(context, TRUE, NULL);
	219	printf("%s", help);
	220	g_free(help);
	221
	222	g_option_context_free(context);
	223	}