/* * 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 voidwc(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 voidreport(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);} voidmain(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);}