119 lines
2.2 KiB
C
119 lines
2.2 KiB
C
|
/*
|
||
|
|
* Count bytes within runes, if it fits in a uvlong, and other things.
|
||
|
|
*/
|
||
|
|
#include <u.h>
|
||
|
|
#include <libc.h>
|
||
|
|
#include <bio.h>
|
||
|
|
|
||
|
|
/* flags, per-file counts, and total counts */
|
||
|
|
static int pline, pword, prune, pbadr, pchar;
|
||
|
|
static uvlong nline, nword, nrune, nbadr, nchar;
|
||
|
|
static uvlong tnline, tnword, tnrune, tnbadr, tnchar;
|
||
|
|
|
||
|
|
enum{Space, Word};
|
||
|
|
|
||
|
|
static void
|
||
|
|
wc(Biobuf *bin)
|
||
|
|
{
|
||
|
|
int where;
|
||
|
|
long r;
|
||
|
|
|
||
|
|
nline = 0;
|
||
|
|
nword = 0;
|
||
|
|
nrune = 0;
|
||
|
|
nbadr = 0;
|
||
|
|
where = Space;
|
||
|
|
while ((long)(r = Bgetrune(bin)) >= 0) {
|
||
|
|
nrune++;
|
||
|
|
if(r == Runeerror) {
|
||
|
|
nbadr++;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
if(r == '\n')
|
||
|
|
nline++;
|
||
|
|
if(where == Word){
|
||
|
|
if(isspacerune(r))
|
||
|
|
where = Space;
|
||
|
|
}else
|
||
|
|
if(isspacerune(r) == 0){
|
||
|
|
where = Word;
|
||
|
|
nword++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
nchar = Boffset(bin);
|
||
|
|
tnline += nline;
|
||
|
|
tnword += nword;
|
||
|
|
tnrune += nrune;
|
||
|
|
tnbadr += nbadr;
|
||
|
|
tnchar += nchar;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void
|
||
|
|
report(uvlong nline, uvlong nword, uvlong nrune, uvlong nbadr, uvlong nchar, char *fname)
|
||
|
|
{
|
||
|
|
char line[1024], *s, *e;
|
||
|
|
|
||
|
|
s = line;
|
||
|
|
e = line + sizeof line;
|
||
|
|
line[0] = 0;
|
||
|
|
if(pline)
|
||
|
|
s = seprint(s, e, " %7llud", nline);
|
||
|
|
if(pword)
|
||
|
|
s = seprint(s, e, " %7llud", nword);
|
||
|
|
if(prune)
|
||
|
|
s = seprint(s, e, " %7llud", nrune);
|
||
|
|
if(pbadr)
|
||
|
|
s = seprint(s, e, " %7llud", nbadr);
|
||
|
|
if(pchar)
|
||
|
|
s = seprint(s, e, " %7llud", nchar);
|
||
|
|
if(fname != nil)
|
||
|
|
seprint(s, e, " %s", fname);
|
||
|
|
print("%s\n", line+1);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
char *sts;
|
||
|
|
Biobuf sin, *bin;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
sts = nil;
|
||
|
|
ARGBEGIN {
|
||
|
|
case 'l': pline++; break;
|
||
|
|
case 'w': pword++; break;
|
||
|
|
case 'r': prune++; break;
|
||
|
|
case 'b': pbadr++; break;
|
||
|
|
case 'c': pchar++; break;
|
||
|
|
default:
|
||
|
|
fprint(2, "Usage: %s [-lwrbc] [file ...]\n", argv0);
|
||
|
|
exits("usage");
|
||
|
|
} ARGEND
|
||
|
|
if(pline+pword+prune+pbadr+pchar == 0){
|
||
|
|
pline = 1;
|
||
|
|
pword = 1;
|
||
|
|
pchar = 1;
|
||
|
|
}
|
||
|
|
if(argc == 0){
|
||
|
|
Binit(&sin, 0, OREAD);
|
||
|
|
wc(&sin);
|
||
|
|
report(nline, nword, nrune, nbadr, nchar, nil);
|
||
|
|
Bterm(&sin);
|
||
|
|
}else{
|
||
|
|
for(i = 0; i < argc; i++){
|
||
|
|
bin = Bopen(argv[i], OREAD);
|
||
|
|
if(bin == nil){
|
||
|
|
perror(argv[i]);
|
||
|
|
sts = "can't open";
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
wc(bin);
|
||
|
|
report(nline, nword, nrune, nbadr, nchar, argv[i]);
|
||
|
|
Bterm(bin);
|
||
|
|
}
|
||
|
|
if(argc>1)
|
||
|
|
report(tnline, tnword, tnrune, tnbadr, tnchar, "total");
|
||
|
|
}
|
||
|
|
exits(sts);
|
||
|
|
}
|