busybox/coreutils/seq.c

/* vi: set sw=4 ts=4: */
/*
 * seq implementation for busybox
 *
 * Copyright (C) 2004, Glenn McGrath
 *
 * Licensed under the GPL v2, see the file LICENSE in this tarball.
 */
#include "libbb.h"

/* This is a NOFORK applet. Be very careful! */

int seq_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int seq_main(int argc, char **argv)
{
	enum {
		OPT_w = (1 << 0),
		OPT_s = (1 << 1),
	};
	double first, last, increment, v;
	unsigned n;
	unsigned width;
	unsigned frac_part;
	const char *sep, *opt_s = "\n";
	unsigned opt;

#if ENABLE_LOCALE_SUPPORT
	/* Undo busybox.c: on input, we want to use dot
	 * as fractional separator, regardless of current locale */
	setlocale(LC_NUMERIC, "C");
#endif

	opt = getopt32(argv, "+ws:", &opt_s);
	argc -= optind;
	argv += optind;
	first = increment = 1;
	errno = 0;
	switch (argc) {
			char *pp;
		case 3:
			increment = strtod(argv[1], &pp);
			errno |= *pp;
		case 2:
			first = strtod(argv[0], &pp);
			errno |= *pp;
		case 1:
			last = strtod(argv[argc-1], &pp);
			if (!errno && *pp == '\0')
				break;
		default:
			bb_show_usage();
	}

#if ENABLE_LOCALE_SUPPORT
	setlocale(LC_NUMERIC, "");
#endif

	/* Last checked to be compatible with: coreutils-6.10 */
	width = 0;
	frac_part = 0;
	while (1) {
		char *dot = strchrnul(*argv, '.');
		int w = (dot - *argv);
		int f = strlen(dot);
		if (width < w)
			width = w;
		argv++;
		if (!*argv)
			break;
		/* Why do the above _before_ frac check below?
		 * Try "seq 1 2.0" and "seq 1.0 2.0":
		 * coreutils never pay attention to the number
		 * of fractional digits in last arg. */
		if (frac_part < f)
			frac_part = f;
	}
	if (frac_part) {
		frac_part--;
		if (frac_part)
			width += frac_part + 1;
	}
	if (!(opt & OPT_w))
		width = 0;

	sep = "";
	v = first;
	n = 0;
	while (increment >= 0 ? v <= last : v >= last) {
		printf("%s%0*.*f", sep, width, frac_part, v);
		sep = opt_s;
		/* v += increment; - would accumulate floating point errors */
		n++;
		v = first + n * increment;
	}
	if (n) /* if while loop executed at least once */
		bb_putchar('\n');

	return fflush_all();
}
1	niro	532	/* vi: set sw=4 ts=4: */
2			/*
3			* seq implementation for busybox
4			*
5			* Copyright (C) 2004, Glenn McGrath
6			*
7			* Licensed under the GPL v2, see the file LICENSE in this tarball.
8			*/
9	niro	816	#include "libbb.h"
10	niro	532
11	niro	816	/* This is a NOFORK applet. Be very careful! */
12
13			int seq_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
14	niro	532	int seq_main(int argc, char **argv)
15			{
16	niro	984	enum {
17			OPT_w = (1 << 0),
18			OPT_s = (1 << 1),
19			};
20			double first, last, increment, v;
21			unsigned n;
22			unsigned width;
23			unsigned frac_part;
24			const char sep, opt_s = "\n";
25			unsigned opt;
26	niro	532
27	niro	984	#if ENABLE_LOCALE_SUPPORT
28			/* Undo busybox.c: on input, we want to use dot
29			* as fractional separator, regardless of current locale */
30			setlocale(LC_NUMERIC, "C");
31			#endif
32
33			opt = getopt32(argv, "+ws:", &opt_s);
34			argc -= optind;
35			argv += optind;
36			first = increment = 1;
37			errno = 0;
38	niro	532	switch (argc) {
39	niro	984	char *pp;
40	niro	532	case 3:
41	niro	984	increment = strtod(argv[1], &pp);
42			errno \|= *pp;
43	niro	532	case 2:
44	niro	984	first = strtod(argv[0], &pp);
45			errno \|= *pp;
46			case 1:
47			last = strtod(argv[argc-1], &pp);
48			if (!errno && *pp == '\0')
49			break;
50	niro	532	default:
51			bb_show_usage();
52			}
53
54	niro	984	#if ENABLE_LOCALE_SUPPORT
55			setlocale(LC_NUMERIC, "");
56			#endif
57
58			/* Last checked to be compatible with: coreutils-6.10 */
59			width = 0;
60			frac_part = 0;
61			while (1) {
62			char dot = strchrnul(argv, '.');
63			int w = (dot - *argv);
64			int f = strlen(dot);
65			if (width < w)
66			width = w;
67			argv++;
68			if (!*argv)
69			break;
70			/* Why do the above _before_ frac check below?
71			* Try "seq 1 2.0" and "seq 1.0 2.0":
72			* coreutils never pay attention to the number
73			* of fractional digits in last arg. */
74			if (frac_part < f)
75			frac_part = f;
76	niro	532	}
77	niro	984	if (frac_part) {
78			frac_part--;
79			if (frac_part)
80			width += frac_part + 1;
81			}
82			if (!(opt & OPT_w))
83			width = 0;
84	niro	532
85	niro	984	sep = "";
86			v = first;
87			n = 0;
88			while (increment >= 0 ? v <= last : v >= last) {
89			printf("%s%0.f", sep, width, frac_part, v);
90			sep = opt_s;
91			/* v += increment; - would accumulate floating point errors */
92			n++;
93			v = first + n * increment;
94			}
95			if (n) /* if while loop executed at least once */
96			bb_putchar('\n');
97
98			return fflush_all();
99	niro	532	}