-
Notifications
You must be signed in to change notification settings - Fork 1
/
numsum
executable file
·385 lines (292 loc) · 9.79 KB
/
numsum
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
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
#!/usr/bin/perl -w
# numsum: This program adds up all numbers it encounters and prints out
# the total at the end on STDOUT.
#
# Copyright (C) 2002-2004 Suso Banderas
# 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 2
# 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, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# You may contact the author at <suso@suso.org>.
#######################
# VARIABLES AND SETUP #
#######################
use bignum;
use Getopt::Std;
#use strict;
use vars qw/ %opts /;
my ($file, $finalsum, @number_array, @columns_to_print, @column_output, $output_line, $string_seperator, $verbose);
getopts('123456789cdhiIqrs:Vx:y:', \%opts);
if ($opts{'h'}) {
&help;
exit(0);
}
if ($opts{'d'}) {
$verbose = 3;
print STDERR "Debug mode\n";
} elsif ($opts{'V'}) {
$verbose = 2;
print STDERR "Verbose mode\n";
} elsif($opts{'q'}) {
$verbose = 0; # No output except the final answer.
} else {
$verbose = 1; # Normal output.
}
# I'm not sure whether this should stay in or not. It's mainly just for convenience.
# It's a bad hack because you can only specify 1 through 9. Maybe sometime I'll figure
# out how to bypass getopts to specify extra options.
for ($n = 1 ; $n < 10; $n++) {
if ($opts{$n}) {
$opts{'c'} = 1;
$opts{'x'} = $n;
}
}
if ($opts{'x'}) {
@columns_to_print = split(/,/, $opts{'x'});
}
if ($opts{'y'}) {
@rows_to_print = split(/,/, $opts{'y'});
}
if ($opts{'s'}) {
$string_seperator = $opts{'s'};
} else {
$string_seperator = "\\s+";
}
################
# MAIN PROGRAM #
################
if (@ARGV) {
foreach $file (@ARGV) {
print STDERR "Reading from file $file.\n" if ($verbose >= 2);
open (ARGFILE, "$file") && process_filehandle(\*ARGFILE, \@number_array)
|| $verbose && warn "Couldn't open file $file for reading: $!\n";
close(ARGFILE);
}
} else {
print STDERR "Reading from STDIN.\n" if ($verbose >= 2);
process_filehandle(\*STDIN, \@number_array);
}
my $number_array_length = @number_array;
if ($opts{'c'}) { # Handle column sum output.
my @column_output = ();
my $i;
for ($i = 0; $i < $number_array_length; $i++) {
my @add_this_array = split(/$string_seperator/, $number_array[$i]);
sum_columns(\@column_output, \@add_this_array);
}
# Now print out the columular output.
if ($opts{'x'}) {
my @final_output = ();
foreach $column_number (@columns_to_print) {
push (@final_output, $column_output[$column_number-1]);
}
$output_line = join(" ", @final_output);
} else {
$output_line = join(" ", @column_output);
}
print "$output_line\n";
} elsif ($opts{'r'}) { # Handle row sum output.
my $i;
if ($opts{'y'}) {
foreach $row_number (@rows_to_print) {
my @add_this_array = split(/ /, $number_array[$row_number-1]);
my $row_sum = 0;
foreach $row_value (@add_this_array) {
$row_sum += $row_value;
}
print "$row_sum\n";
}
} else {
for ($i = 0; $i < $number_array_length; $i++) {
my @add_this_array = split(/ /, $number_array[$i]);
my $row_sum = 0;
foreach $row_value (@add_this_array) {
next if ($row_value !~ m/^[0-9]+$/);
$row_sum += $row_value;
}
print "$row_sum\n";
}
}
} else {
$finalsum = add_array(\@number_array);
if ($opts{'i'}) {
$finalsum = int($finalsum);
} elsif ($opts{'I'}) {
if ($finalsum == int($finalsum)) {
$finalsum = 0;
} else {
$finalsum =~ s/^(\-?)[0-9]*\.([0-9]*)$/$1.$2/;
}
}
print "$finalsum\n";
}
exit(0);
###############
# SUBROUTINES #
###############
sub help {
print <<"EOF";
---------------------------------------------------------------------------
numsum : A program that adds up all numbers of input and returns the sum.
---------------------------------------------------------------------------
Usage:
numsum [options] <file> (Input from a file.)
| numsum [options] (Input from command pipeline.)
numsum [options] (Input on STDIN. Use Ctrl-D to stop.)
Options:
-i Only return the integer portion of the final sum.
-I Only return the decimal portion of the final sum
-c Print out the sum of each column.
-r Print out the sum of each row.
-x <n> Specify a comma separated list of columns to print.
-y <n> Specify a comma separated list of rows to print.
-s <string> Specify a separator string for splitting columns.
This defaults to consecutive whitespace.
-d Debug. For developers only.
-h Help: You're looking at it.
-V Increase verbosity.
-q Quiet mode, don't print any warnings.
EOF
}
# This function is used so that we can genericize what filehandle
# we are getting our information from.
sub process_filehandle {
my $filehandle = shift;
my $number_array_ref = shift;
while (<$filehandle>) {
my $line = $_;
chomp($line);
if ($opts{'c'} || $opts{'r'}) { # Process columns or rows
# Make each line column friendly by changing all non-numeric words into 0.
# Some ideas from the #perl channel on irc.freenode.org. Thanks to v, dkr, Khisanth and dudeman
#$input = join ' ', map { /\D/ ? 0 : $_ } split / +/, $input;
#s!(\S+)!$1=~/\D/?0:$1!ge;
# This one works best.
$line =~ s!(\S+)!$1=~/\D/?0:$1!ge;
push(@$number_array_ref, $line);
} else { # Normal processing.
if ($line =~ /^\s*(\-?[0-9]*\.?[0-9]+)/) {
print STDERR "number: $1\n" if ($verbose >= 3);
push(@$number_array_ref, $1);
}
}
}
return 1;
}
# Function for adding up numbers
sub add_array {
my $arrayref = shift;
my $runningtotal = 0;
my $number;
foreach $number (@$arrayref) {
print STDERR "$number\n" if ($verbose >= 2);
$runningtotal += $number;
}
return $runningtotal;
}
# Function for summing up the columns.
sub sum_columns {
my $sum_array_ref = shift;
my $input_array_ref = shift;
my $input_array_length = @$input_array_ref;
my $x;
for ($x = 0; $x < $input_array_length; $x++) {
if (${$sum_array_ref}[$x]) {
${$sum_array_ref}[$x] += ${$input_array_ref}[$x];
} else {
${$sum_array_ref}[$x] = ${$input_array_ref}[$x];
}
}
return 1;
}
# Lay down some of that perl pod action.
=pod
=head1 NAME
numsum - numsum program file
=head1 SYNOPSIS
B<numsum> [-iIcdhrsvxy] <FILE>
| B<numsum> [-iIcdhrsvxy] (Input on STDIN from pipeline.)
B<numsum> [-iIcdhrsvxy] (Input on STDIN. Use Ctrl-D to stop.)
=head1 DESCRIPTION
B<numsum> is a program that is part of the numeric utilities package. B<numsum>
will take all the numbers on STDIN and return the sum of those numbers. Currently
it only processes the first number on each line. Besides positive numbers, it also
handles negative numbers and numbers with decimals.
=head1 OPTIONS
-i Only return the integer portion of the final sum.
-I Only return the decimal portion of the final sum.
-c Print out the sum of each column.
-r Print out the sum of each row.
-x <n> Specify a comma separated list of columns to print.
-y <n> Specify a comma separated list of rows to print.
-s <string> Specify a string to use as a separator for columns.
This defaults to be consecutive whitespace (\s+).
-h Help: You're looking at it.
-V Increase verbosity.
-d Debug mode. For developers
-q Quiet mode, don't print any warnings.
=head1 EXAMPLES
Simply add up the numbers in a file.
$ numsum numbers.txt
4315
Enter your own numbers on STDIN. The last number is the answer.
$ numsum
4
21
98
100
223
Use it in a command pipeline.
$ ls -1s | grep .mp3 | numsum -c -x 5
72288
Add up the total byte count in a HTTP log file.
$ cat access_log | awk {'print $10'} numsum
or
numsum -c -x 10 access_log
Add up the columns of numbers of a file.
$ cat columns
1 6 11 16 21
2 7 12 17 22
3 8 13 18 23
4 9 14 19 24
5 10 15 20 25
$ numsum -c columns
15 40 65 90 115
Add up the 1st, 2nd and 5th columns only.
$ numsum -c -x 1,2,5 columns
15 40 115
Add up the rows of numbers of a file.
$ numsum -r columns
55
60
65
70
75
Add up the 2nd and 4th rows.
$ numsum -r -y 2,4 columns
60
70
=head1 SEE ALSO
average(1), bound(1), interval(1), normalize(1), numgrep(1), numprocess(1), random(1), range(1), round(1)
=head1 COPYRIGHT
numsum is part of the num-utils package, which is copyrighted by
Suso Banderas and released under the GPL license. Please read
the COPYING and LICENSE files that came with the num-utils package
Developers can read the GOALS file and contact me about providing
submissions or help for the project.
=head1 MORE INFO
More info on numsum can be found at:
=over 1
=item B<http://suso.suso.org/programs/num-utils/>
=back
=cut