git.hungrycats.org Git - xscreensaver/blob - hacks/vidwhacker

   1 #!/bin/sh
   2 #
   3 # vidwhacker, for xscreensaver.  Copyright (c) 1998, 1999 Jamie Zawinski.
   4 #
   5 # Permission to use, copy, modify, distribute, and sell this software and its
   6 # documentation for any purpose is hereby granted without fee, provided that
   7 # the above copyright notice appear in all copies and that both that
   8 # copyright notice and this permission notice appear in supporting
   9 # documentation.  No representations are made about the suitability of this
  10 # software for any purpose.  It is provided "as is" without express or
  11 # implied warranty.
  12 #
  13 #
  14 # This script grabs a frame of video, then uses various pbm filters to
  15 # munge the image in random nefarious ways, then uses xloadimage, xli, or xv
  16 # to put it on the root window.  This works out really nicely if you just
  17 # feed some random TV station into it...
  18 #
  19 # The video grabbing part is SGI-specific -- if you want to use this on
  20 # another system, add a new clause to the grab() procedure.
  21
  22
  23 # need perl to generate random numbers -- I don't know another way to do
  24 # that from a shell script.
  25 perl=perl
  26
  27
  28 onroot=false
  29 verbose=false
  30 delay=3
  31 use_stdin=false
  32 use_stdout=false
  33
  34 pid=""
  35 tmp=${TMPDIR:-/tmp}/vidwhacker.$$
  36 tmp_rgb=$tmp-00000.rgb
  37 tmp_ppm0=$tmp-0.ppm
  38 tmp_ppm1=$tmp-1.ppm
  39 tmp_ppm2=$tmp-2.ppm
  40 tmp_ppm3=$tmp-3.ppm
  41 tmp_ppm4=$tmp-4.ppm
  42 tmp_ppmS=$tmp-S.ppm
  43
  44
  45 # Figure out whether to use xloadimage, xli, or xv.
  46 pick_displayer() {
  47   displayer=
  48   for prog in xloadimage xli xv ; do
  49     IFS=:
  50     for p in $PATH; do
  51       IFS=
  52       if [ "$p" = "" ]; then p=. ; fi
  53       if [ -x $p/$prog ]; then
  54         displayer=$prog
  55         break 2
  56       fi
  57     done
  58   done
  59
  60   if [ "$displayer" = "xloadimage" ]; then
  61     displayer_args="-quiet"
  62     displayer_win_args=""
  63     displayer_root_args="-onroot"
  64   elif [ "$displayer" = "xli" ]; then
  65     displayer_args="-quiet"
  66     displayer_win_args=""
  67     displayer_root_args="-onroot -center -border black"
  68   elif [ "$displayer" = "xv" ]; then
  69     displayer_args="-quick24"
  70     displayer_win_args="-geom +0+0"
  71     displayer_root_args="-root -rmode 5 -noresetroot -rfg black -rbg black -viewonly"
  72   else
  73     echo "$0: neither xli nor xv found on \$PATH"
  74     exit -1
  75   fi
  76 }
  77
  78 # Process command-line args
  79 getargs() {
  80
  81   while [ $# != 0 ]; do
  82     case "$1" in
  83     -display | -disp | -dis | -dpy | -d )
  84       shift
  85       DISPLAY="$1"
  86       export DISPLAY
  87       ;;
  88     -root )
  89       onroot=true
  90       ;;
  91     -window )
  92       onroot=false
  93       ;;
  94     -verbose )
  95       verbose=true
  96       ;;
  97     -stdin )
  98       use_stdin=true
  99       ;;
 100     -stdout )
 101       use_stdout=true
 102       ;;
 103     -delay)
 104       shift
 105       delay="$1"
 106       ;;
 107     * )
 108       echo "VidWhacker, Copyright (c) 1999 Jamie Zawinski <jwz@jwz.org>" >&2
 109       echo "            http://www.jwz.org/xscreensaver/" >&2
 110       echo "" >&2
 111       echo "usage: $0 [-display dpy] [-verbose] [-root | -window]" >&2
 112       echo "                  [-stdin] [-stdout] [-delay secs]" >&2
 113       exit 1
 114       ;;
 115     esac
 116     shift
 117   done
 118
 119   pick_displayer
 120
 121   if [ "$onroot" = true ]; then
 122     displayer_args="$displayer_args $displayer_root_args"
 123   else
 124     displayer_args="$displayer_args $displayer_win_args"
 125   fi
 126
 127
 128   screen_width=''
 129   if [ "$use_stdout" = false ]; then
 130     screen_width=`xdpyinfo 2>/dev/null |
 131         sed -n 's/.* dimensions: *\([0-9]*\).*/\1/p'`
 132     if [ "$screen_width" = "" ]; then
 133       screen_width=800
 134     fi
 135   fi
 136 }
 137
 138
 139 clean() {
 140   rm -f $tmp_rgb $tmp_ppm1 $tmp_ppm2 $tmp_ppm3 $tmp_ppm4
 141 }
 142
 143 clean2() {
 144   clean
 145   rm -f $tmp_ppm0 $tmp_ppmS
 146 }
 147
 148
 149 # Grab a frame of video.  leaves it in $tmp_ppm1.
 150 #
 151 grab() {
 152   uname=`uname`
 153   if [ $uname = IRIX ]; then
 154     #
 155     # SGI's "vidtomem" returns an SGI RGB image of the default video input,
 156     # and has stupid non-overridable ouput-file-naming conventions.  So, let
 157     # it write its file; and then convert it to a pgm.
 158     #
 159
 160     vidtomem -f $tmp
 161     sgitopnm $tmp_rgb > $tmp_ppm1
 162
 163     # Cut off the close-captioning blips in the NTSC overscan region.  YMMV.
 164     #  | pnmcut 12 7 695 477
 165
 166   elif [ $uname = Linux ]; then
 167
 168     # Marcus Herbert says the following works with his Connectix Qcam.
 169     # Don't have qcam?  Well, do something else then...  and send me a patch.
 170
 171     qcam > $tmp_ppm1
 172
 173     # Friedrich Delgado Friedrichs says the following works if you have
 174     # XawTV installed:
 175     #
 176     #   streamer -o $tmp_ppm1
 177     #
 178
 179   else
 180     echo "$0: don't know how to grab video on this OS." >&2
 181     clean2
 182     exit 1
 183   fi
 184 }
 185
 186
 187 # Use perl to pick a random foreground/background color in pbm's syntax.
 188 #
 189 randcolor() {
 190   $perl -e 'srand(time ^ $$);
 191             printf("#%02x%02x%02x-#%02x%02x%02x",
 192                    int(rand()*60),
 193                    int(rand()*60),
 194                    int(rand()*60),
 195                    120+int(rand()*135),
 196                    120+int(rand()*135),
 197                    120+int(rand()*135))'
 198 }
 199
 200 rand() {
 201   $perl -e "srand(time ^ $$); print int(rand() * $1)"
 202 }
 203
 204
 205
 206 # Frobnicate the image in some random way.
 207 #
 208 frob() {
 209
 210   w_h=`head -2 $tmp_ppm1 | tail -1`
 211   width=`echo $w_h | awk '{print $1}'`
 212   height=`echo $w_h | awk '{print $2}'`
 213
 214   N=`rand 17`
 215
 216   if [ "$verbose" = true ]; then
 217     echo "mode $N..." >&2
 218   fi
 219
 220   if   [ $N = 0 ]; then
 221     ppmtopgm $tmp_ppm1 | pgmedge | pgmtoppm `randcolor` | ppmnorm
 222
 223   elif [ $N = 1 ]; then
 224     ppmtopgm $tmp_ppm1 |
 225     pgmenhance |
 226     pgmtoppm `randcolor`
 227
 228   elif [ $N = 2 ]; then
 229     ppmtopgm $tmp_ppm1 | pgmoil | pgmtoppm `randcolor`
 230
 231   elif [ $N = 3 ]; then
 232     ppmrelief $tmp_ppm1 | ppmtopgm | pgmedge | ppmrelief | ppmtopgm |
 233       pgmedge | pnminvert | pgmtoppm `randcolor`
 234
 235   elif [ $N = 4 ]; then
 236     ppmspread 71 $tmp_ppm1 > $tmp_ppm2
 237     pnmarith -add $tmp_ppm1 $tmp_ppm2
 238
 239   elif [ $N = 5 ]; then
 240     pnmflip -lr $tmp_ppm1 > $tmp_ppm2
 241     pnmarith -multiply $tmp_ppm1 $tmp_ppm2 > $tmp_ppm3
 242     pnmflip -tb $tmp_ppm3 | ppmnorm > $tmp_ppm2
 243     pnmarith -multiply $tmp_ppm1 $tmp_ppm2
 244
 245   elif [ $N = 6 ]; then
 246     N2=`rand 3`
 247     if [ $N2 = 0 ]; then
 248       pnmflip -lr $tmp_ppm1 > $tmp_ppm2
 249     elif [ $N2 = 1 ]; then
 250       pnmflip -tb $tmp_ppm1 > $tmp_ppm2
 251     else
 252       pnmflip -lr $tmp_ppm1 > $tmp_ppm2
 253       pnmflip -tb $tmp_ppm2 > $tmp_ppm3
 254       cp $tmp_ppm3 $tmp_ppm2
 255     fi
 256
 257     pnmarith -difference $tmp_ppm1 $tmp_ppm2
 258
 259   elif [ $N = 7 ]; then
 260
 261     for i in 1 2 3 ; do
 262       ppmtopgm $tmp_ppm1 | pgmedge > $tmp_ppm2
 263       pnmarith -difference $tmp_ppm1 $tmp_ppm2 > $tmp_ppm3
 264       cp $tmp_ppm3 $tmp_ppm1
 265     done
 266     ppmnorm < $tmp_ppm1
 267
 268   elif [ $N = 8 ]; then
 269     pnmflip -lr $tmp_ppm1 > $tmp_ppm2
 270     pnmarith -multiply $tmp_ppm1 $tmp_ppm2 | ppmrelief | ppmnorm | pnminvert
 271
 272   elif [ $N = 9 ]; then
 273     pnmflip -lr $tmp_ppm1 > $tmp_ppm2
 274     pnmarith -subtract $tmp_ppm1 $tmp_ppm2 | ppmrelief | ppmtopgm | pgmedge
 275
 276   elif [ $N = 10 ]; then
 277     ppmtopgm $tmp_ppm1 | pgmbentley | pgmtoppm `randcolor`
 278
 279   elif [ $N = 11 ]; then
 280     pgmcrater -number 20000 -height $height -width $width | pgmtoppm `randcolor` > $tmp_ppm2
 281     pnmarith -difference $tmp_ppm1 $tmp_ppm2 > $tmp_ppm3
 282     pnmflip -tb $tmp_ppm3 | ppmnorm > $tmp_ppm2
 283     pnmarith -multiply $tmp_ppm1 $tmp_ppm2
 284
 285   elif [ $N = 12 ]; then
 286     ppmshift 30 $tmp_ppm1 | ppmtopgm | pgmoil | pgmedge |  pgmtoppm `randcolor` > $tmp_ppm2
 287     pnmarith -difference $tmp_ppm1 $tmp_ppm2
 288
 289  elif [ $N = 13 ]; then
 290     ppmpat -madras $width $height | pnmdepth 255 > $tmp_ppm2
 291     pnmarith -difference $tmp_ppm1 $tmp_ppm2
 292
 293   elif [ $N = 14 ]; then
 294     ppmpat -tartan $width $height | pnmdepth 255 > $tmp_ppm2
 295     pnmarith -difference  $tmp_ppm1 $tmp_ppm2
 296
 297   elif [ $N = 15 ]; then
 298     ppmpat -camo $width $height | pnmdepth 255 | ppmshift 50 > $tmp_ppm2
 299     pnmarith -multiply $tmp_ppm1 $tmp_ppm2
 300
 301   elif [ $N = 16 ]; then
 302     pgmnoise $width $height | pgmedge | pgmtoppm `randcolor` > $tmp_ppm2
 303     pnmarith -difference $tmp_ppm1 $tmp_ppm2 | pnmdepth 255 | pnmsmooth
 304
 305   else cat $tmp_ppm1
 306   fi
 307 }
 308
 309
 310 # Grab a frame and frob it.  leave it in $tmp_ppm3.
 311 #
 312 whack() {
 313   clean
 314
 315   while [ ! -f $tmp_ppm1 ]; do
 316     if [ "$use_stdin" != true ]; then
 317       grab
 318     else
 319       cp $tmp_ppmS $tmp_ppm0
 320       cp $tmp_ppm0 $tmp_ppm1
 321     fi
 322   done
 323
 324   rm -f $tmp_rgb
 325
 326   if [ "$screen_width" != "" ]; then
 327     frob | pnmscale -width $screen_width > $tmp_ppm3
 328   else
 329     frob > $tmp_ppm3
 330   fi
 331
 332   rm -f $tmp_ppm1 $tmp_ppm2
 333 }
 334
 335
 336 # Kill off the xli or xv subprocess, if it's running
 337 #
 338 kill_pid() {
 339   if [ "$pid" != "" ]; then
 340
 341     if [ "$verbose" = true ]; then
 342       echo "killing pid $pid..." >&2
 343     fi
 344
 345     # need to do this to avoid "6898 Terminated" messages!
 346     # apparently one can't redirect the output of the builtin `kill' command.
 347 #    ( sh -c "kill $pid" ) >/dev/null 2>/dev/null </dev/null
 348
 349     # wtf?  that doesn't work either.  Is it writing to /dev/tty??
 350     kill $pid >/dev/null 2>&1
 351
 352     pid=""
 353   fi
 354 }
 355
 356 # called when this process is signalled (for cleanup)
 357 #
 358 my_trap() {
 359   if [ "$verbose" = true ]; then
 360     echo "trapped signal!" >&2
 361   fi
 362   kill_pid
 363   clean2
 364   exit 1
 365 }
 366
 367 main() {
 368
 369   getargs $@
 370
 371   trap my_trap 1 2 3 6 9 13 15
 372
 373   if [ "$use_stdin" = true ]; then
 374    cat > $tmp_ppmS
 375   fi
 376
 377   while true; do
 378
 379     # Loop grabbing and frobbing images.
 380     #
 381     # If we're running on the root, run xv or xli in the foreground
 382     # (with -exit, if xv) and then wait.
 383     #
 384     # If we're running in a window, spawn xv or xli in the background; then
 385     # when it's time to put up the new image, kill off the currently-running
 386     # xv or xli.
 387
 388     if [ "$verbose" = true ]; then
 389       whack
 390     else
 391       whack >/dev/null 2>&1
 392     fi
 393
 394     kill_pid
 395
 396     if [ ! -s $tmp_ppm3 ]; then
 397       echo "$0: no image grabbed" >&2
 398
 399     elif [ "$use_stdout" = true ]; then
 400
 401       cat $tmp_ppm3
 402       clean2
 403       exit 0
 404
 405     else
 406
 407 #      pnmtosgi < $tmp_ppm3 > $tmp_ppm2
 408 #      rm -f $tmp_ppm3
 409        mv $tmp_ppm3 $tmp_ppm2
 410
 411       if [ -s $tmp_ppm2 ]; then
 412         if [ "$verbose" = true ]; then
 413           echo "launching $displayer $displayer_args $tmp_ppm2" >&2
 414           ls -lF $tmp_ppm2
 415         fi
 416
 417         mv $tmp_ppm2 $tmp_ppm0
 418         eval "$displayer $displayer_args $tmp_ppm0 &"
 419
 420 # this doesn't work -- leaves xv processes around, instead of stray xset
 421 # data.  Sigh.
 422 #
 423 #       # cat the file so that we can nuke it without racing against xv.
 424 #        cat $tmp_ppm2 | $displayer $displayer_args - &
 425
 426         pid=$!
 427       fi
 428     fi
 429
 430     clean
 431     sleep $delay
 432
 433   done
 434   exit 1
 435 }
 436
 437 main $@
 438
 439 # to find stray xv data:
 440 # xwininfo -root -children|grep 'xv image comments' | awk '{print "xkill -id ", $1}'