pig/pig.c

   1 /*      $NetBSD: pig.c,v 1.8 1999/09/18 19:38:53 jsm Exp $      */
   2
   3 /*-
   4  * Copyright (c) 1992, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *      This product includes software developed by the University of
  18  *      California, Berkeley and its contributors.
  19  * 4. Neither the name of the University nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35
  36 #include <sys/cdefs.h>
  37 #ifndef lint
  38 __COPYRIGHT("@(#) Copyright (c) 1992, 1993\n\
  39         The Regents of the University of California.  All rights reserved.\n");
  40 #endif /* not lint */
  41
  42 #ifndef lint
  43 #if 0
  44 static char sccsid[] = "@(#)pig.c       8.2 (Berkeley) 5/4/95";
  45 #else
  46 __RCSID("$NetBSD: pig.c,v 1.8 1999/09/18 19:38:53 jsm Exp $");
  47 #endif
  48 #endif /* not lint */
  49
  50 #include <sys/types.h>
  51
  52 #include <ctype.h>
  53 #include <err.h>
  54 #include <stdio.h>
  55 #include <stdlib.h>
  56 #include <string.h>
  57 #include <unistd.h>
  58
  59 int main __P((int, char *[]));
  60 void pigout __P((char *, int));
  61 void usage __P((void)) __attribute__((__noreturn__));
  62
  63 int
  64 main(argc, argv)
  65         int argc;
  66         char *argv[];
  67 {
  68         int len;
  69         int ch;
  70         char buf[1024];
  71
  72         while ((ch = getopt(argc, argv, "")) != -1)
  73                 switch(ch) {
  74                 case '?':
  75                 default:
  76                         usage();
  77                 }
  78         argc -= optind;
  79         argv += optind;
  80
  81         for (len = 0; (ch = getchar()) != EOF;) {
  82                 if (isalpha(ch)) {
  83                         if ((size_t)len >= sizeof(buf))
  84                                 errx(1, "ate too much!");
  85                         buf[len++] = ch;
  86                         continue;
  87                 }
  88                 if (len != 0) {
  89                         pigout(buf, len);
  90                         len = 0;
  91                 }
  92                 (void)putchar(ch);
  93         }
  94         exit(0);
  95 }
  96
  97 void
  98 pigout(buf, len)
  99         char *buf;
 100         int len;
 101 {
 102         int ch, start, i;
 103         int olen, allupper, firstupper;
 104
 105         /* See if the word is all upper case */
 106         allupper = firstupper = isupper(buf[0]);
 107         for (i = 1; i < len && allupper; i++)
 108                 allupper = allupper && isupper(buf[i]);
 109
 110         /*
 111          * If the word starts with a vowel, append "way".  Don't treat 'y'
 112          * as a vowel if it appears first.
 113          */
 114         if (strchr("aeiouAEIOU", buf[0]) != NULL) {
 115                 (void)printf("%.*s%s", len, buf,
 116                     allupper ? "WAY" : "way");
 117                 return;
 118         }
 119
 120         /*
 121          * Copy leading consonants to the end of the word.  The unit "qu"
 122          * isn't treated as a vowel.
 123          */
 124         if (!allupper)
 125                 buf[0] = tolower(buf[0]);
 126         for (start = 0, olen = len;
 127             !strchr("aeiouyAEIOUY", buf[start]) && start < olen;) {
 128                 ch = buf[len++] = buf[start++];
 129                 if ((ch == 'q' || ch == 'Q') && start < olen &&
 130                     (buf[start] == 'u' || buf[start] == 'U'))
 131                         buf[len++] = buf[start++];
 132         }
 133         if (firstupper)
 134                 buf[start] = toupper(buf[start]);
 135         (void)printf("%.*s%s", olen, buf + start, allupper ? "AY" : "ay");
 136 }
 137
 138 void
 139 usage()
 140 {
 141         (void)fprintf(stderr, "usage: pig\n");
 142         exit(1);
 143 }