pig/pig.c

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