Subversion Repositories planix.SVN

/*
 * 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);
}