/*
       rtfwc - read rtf input, write word count (actually, char, word
               and paragraph counts).

       This installs callbacks for the ascii and control token classes.
       The control class is necessary so that special characters such as
       \par, \tab, \sect, etc.  can be counted.

       Counts paragraphs instead of lines, since the concept of "line"
       is relatively meaningless.

       It's problematic how to count text in headers and footers, and
       what to do about tables.

       04 Feb 91       Paul DuBois     [email protected]

       04 Feb 91 V1.0. Created.
       27 Feb 91 V1.01. Updated for distribution 1.05.
       16 Mar 91 V1.02. Updated for distribution 1.06.  Multiple files
               allowed.
*/

# include       <stdio.h>
# include       "rtf.h"


static long     chars, tchars = 0;
static long     words, twords = 0;
static long     paras, tparas = 0;
static long     wchars = 0;     /* chars in current word */

static void     Count ();
static void     Text ();
static void     Control ();

int main (argc, argv)
int     argc;
char    **argv;
{
int     i;

       --argc;
       ++argv;

       if (argc == 0)
       {
               Count ();
               printf ("\n");
       }
       else
       {
               for (i = 0; i < argc; i++)
               {
                       if (freopen (argv[i], "r", stdin) == NULL)
                       {
                               fprintf (stderr, "Can't open \"%s\"\n",
                                                               argv[i]);
                               exit (1);
                       }
                       Count ();
                       printf ("\t%s\n", argv[i]);
               }
               if (argc > 1)   /* multiple files, print totals */
                       printf ("%ld chars\t%ld words\t%ld paragraphs\ttotal\n",
                                               tchars, twords, tparas);
       }

       exit (0);
}


static void Count ()
{
       chars = words = paras = wchars = 0;

       RTFInit ();

       /* install counting hooks and process the input stream */

       RTFSetClassCallback (rtfText, Text);
       RTFSetClassCallback (rtfControl, Control);
       RTFRead ();
       printf ("%ld chars\t%ld words\t%ld paragraphs", chars, words, paras);
       tchars += chars;
       twords += words;
       tparas += paras;
}

static void Text ()
{
       ++chars;
       if (rtfMajor != ' ')
               ++wchars;
       else if (wchars > 0)
               ++words;
}


static void Control ()
{
       if (rtfMajor != rtfSpecialChar)
               return;
       switch (rtfMinor)
       {
       case rtfPage:
       case rtfSect:
       case rtfLine:
       case rtfPar:
               ++paras;
               ++chars;
               if (wchars > 0)
               {
                       ++words;
                       wchars = 0;
               }
               break;
       case rtfNoBrkSpace:
       case rtfTab:
               ++chars;
               if (wchars > 0)
               {
                       ++words;
                       wchars = 0;
               }
               break;
       case rtfNoBrkHyphen:
               ++chars;
               ++wchars;
               break;
       }
}