git.hungrycats.org Git - xscreensaver/blob - hacks/webcollage-helper-cocoa.m

   1 /* webcollage-helper-cocoa --- scales and pastes one image into another
   2  * xscreensaver, Copyright (c) 2002-2009 Jamie Zawinski <jwz@jwz.org>
   3  *
   4  * Permission to use, copy, modify, distribute, and sell this software and its
   5  * documentation for any purpose is hereby granted without fee, provided that
   6  * the above copyright notice appear in all copies and that both that
   7  * copyright notice and this permission notice appear in supporting
   8  * documentation.  No representations are made about the suitability of this
   9  * software for any purpose.  It is provided "as is" without express or
  10  * implied warranty.
  11  */
  12
  13 /* This is the Cocoa implementation.  See webcollage-helper.c for the
  14    GDK + JPEGlib implementation.
  15  */
  16
  17 #import <Cocoa/Cocoa.h>
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <unistd.h>
  21 #include <sys/stat.h>
  22
  23
  24 #if MAC_OS_X_VERSION_MAX_ALLOWED <= MAC_OS_X_VERSION_10_4
  25  typedef int          NSInteger;
  26  typedef unsigned int NSUInteger;
  27 #endif
  28
  29
  30 char *progname;
  31 static int verbose_p = 0;
  32
  33 static void write_image (NSImage *img, const char *file);
  34
  35
  36 /* NSImage can't load PPMs by default...
  37  */
  38 static NSImage *
  39 load_ppm_image (const char *file)
  40 {
  41   FILE *in = fopen (file, "r");
  42   if (! in) return 0;
  43
  44   char buf[255];
  45
  46   char *s = fgets (buf, sizeof(buf)-1, in);             /* P6 */
  47   if (!s || !!strcmp (s, "P6\n"))
  48     return 0;
  49
  50   s = fgets (buf, sizeof(buf)-1, in);                   /* W H */
  51   if (!s)
  52     return 0;
  53
  54   int w = 0, h = 0, d = 0;
  55   if (2 != sscanf (buf, " %d %d \n", &w, &h))
  56     return 0;
  57   if (w <= 0 || h <= 0)
  58     return 0;
  59
  60   s = fgets (buf, sizeof(buf)-1, in);                   /* 255 */
  61   if (!s)
  62     return 0;
  63
  64   if (1 != sscanf (buf, " %d \n", &d))
  65     return 0;
  66   if (d != 255)
  67     return 0;
  68
  69   int size = (w * (h+1) * 3);
  70   unsigned char *bits = malloc (size);
  71   if (!bits) return 0;
  72
  73   int n = read (fileno (in), (void *) bits, size);      /* body */
  74   if (n < 20) return 0;
  75
  76   fclose (in);
  77
  78   NSBitmapImageRep *rep =
  79     [[NSBitmapImageRep alloc]
  80       initWithBitmapDataPlanes: &bits
  81                     pixelsWide: w
  82                     pixelsHigh: h
  83                  bitsPerSample: 8
  84                samplesPerPixel: 3
  85                       hasAlpha: NO
  86                       isPlanar: NO
  87                 colorSpaceName: NSDeviceRGBColorSpace
  88                   bitmapFormat: NSAlphaFirstBitmapFormat
  89                    bytesPerRow: w * 3
  90                   bitsPerPixel: 8 * 3];
  91
  92   NSImage *image = [[NSImage alloc] initWithSize: NSMakeSize (w, h)];
  93   [image addRepresentation: rep];
  94   [rep release];
  95
  96   // #### 'bits' is leaked... the NSImageRep doesn't free it when freed.
  97
  98   return image;
  99 }
 100
 101
 102 static NSImage *
 103 load_image (const char *file)
 104 {
 105   NSImage *image = [[NSImage alloc]
 106                      initWithContentsOfFile:
 107                        [NSString stringWithCString: file
 108                                           encoding: kCFStringEncodingUTF8]];
 109   if (! image)
 110     image = load_ppm_image (file);
 111
 112   if (! image) {
 113     fprintf (stderr, "%s: unable to load %s\n", progname, file);
 114     exit (1);
 115   }
 116
 117   return image;
 118 }
 119
 120
 121 static void
 122 bevel_image (NSImage *img, int bevel_pct,
 123              int x, int y, int w, int h, double scale)
 124 {
 125   int small_size = (w > h ? h : w);
 126
 127   int bevel_size = small_size * (bevel_pct / 100.0);
 128
 129   bevel_size /= scale;
 130
 131   /* Use a proportionally larger bevel size for especially small images. */
 132   if      (bevel_size < 20 && small_size > 40) bevel_size = 20;
 133   else if (bevel_size < 10 && small_size > 20) bevel_size = 10;
 134   else if (bevel_size < 5)    /* too small to bother bevelling */
 135     return;
 136
 137
 138   NSBitmapImageRep *rep =
 139     [[NSBitmapImageRep alloc]
 140       initWithBitmapDataPlanes: NULL
 141                     pixelsWide: w
 142                     pixelsHigh: h
 143                  bitsPerSample: 8
 144                samplesPerPixel: 4
 145                       hasAlpha: YES
 146                       isPlanar: NO
 147                 colorSpaceName: NSDeviceRGBColorSpace
 148                   bitmapFormat: NSAlphaFirstBitmapFormat
 149                    bytesPerRow: 0
 150                   bitsPerPixel: 0];
 151
 152   NSInteger xx, yy;
 153   double *ramp = (double *) malloc (sizeof(*ramp) * (bevel_size + 1));
 154
 155   if (!ramp)
 156     {
 157       fprintf (stderr, "%s: out of memory (%d)\n", progname, bevel_size);
 158       exit (1);
 159     }
 160
 161   for (xx = 0; xx <= bevel_size; xx++)
 162     {
 163 # if 0  /* linear */
 164       ramp[xx] = xx / (double) bevel_size;
 165
 166 # else /* sinusoidal */
 167       double p = (xx / (double) bevel_size);
 168       double s = sin (p * M_PI / 2);
 169       ramp[xx] = s;
 170 # endif
 171     }
 172
 173   memset ([rep bitmapData], 0xFFFFFFFF,
 174           [rep bytesPerRow] * h);
 175
 176   for (yy = 0; yy < h; yy++)
 177     {
 178       for (xx = 0; xx < w; xx++)
 179         {
 180           double rx, ry, r;
 181
 182           if (xx < bevel_size)           rx = ramp[xx];
 183           else if (xx >= w - bevel_size) rx = ramp[w - xx - 1];
 184           else rx = 1;
 185
 186           if (yy < bevel_size)           ry = ramp[yy];
 187           else if (yy >= h - bevel_size) ry = ramp[h - yy - 1];
 188           else ry = 1;
 189
 190           r = rx * ry;
 191           if (r != 1)
 192             {
 193               NSUInteger p[4];
 194               p[0] = 0xFF * r;
 195               p[1] = p[2] = p[3] = 0xFF;
 196               [rep setPixel:p atX:xx y:yy];
 197             }
 198         }
 199     }
 200
 201   free (ramp);
 202
 203   NSImage *bevel_img = [[NSImage alloc]
 204                          initWithData: [rep TIFFRepresentation]];
 205
 206   [img lockFocus];
 207   y = [img size].height - (y + h);
 208   [bevel_img drawAtPoint: NSMakePoint (x, y)
 209                 fromRect: NSMakeRect (0, 0, w, h)
 210                operation: NSCompositeDestinationIn /* Destination image
 211                                                       wherever both images are
 212                                                       opaque, transparent
 213                                                       elsewhere. */
 214                 fraction: 1.0];
 215   [img unlockFocus];
 216
 217   [rep release];
 218   [bevel_img release];
 219
 220   if (verbose_p)
 221     fprintf (stderr, "%s: added %d%% bevel (%d px)\n", progname,
 222              bevel_pct, bevel_size);
 223 }
 224
 225
 226 static void
 227 paste (const char *paste_file,
 228        const char *base_file,
 229        double from_scale,
 230        double opacity, int bevel_pct,
 231        int from_x, int from_y, int to_x, int to_y,
 232        int w, int h)
 233 {
 234   NSImage *paste_img = load_image (paste_file);
 235   NSImage *base_img  = load_image (base_file);
 236
 237   int paste_w = [paste_img size].width;
 238   int paste_h = [paste_img size].height;
 239
 240   int base_w  = [base_img size].width;
 241   int base_h  = [base_img size].height;
 242
 243   if (verbose_p)
 244     {
 245       fprintf (stderr, "%s: loaded %s: %dx%d\n",
 246                progname, base_file, base_w, base_h);
 247       fprintf (stderr, "%s: loaded %s: %dx%d\n",
 248                progname, paste_file, paste_w, paste_h);
 249     }
 250
 251   if (bevel_pct > 0 && paste_w > 5 && paste_h > 5)
 252     bevel_image (paste_img, bevel_pct,
 253                  from_x, from_y, w, h,
 254                  from_scale);
 255
 256   int scaled_w = w * from_scale;
 257   int scaled_h = h * from_scale;
 258
 259   from_y = paste_h - (from_y + h);  // Cocoa flipped coordinate system
 260   to_y   = base_h  - (to_y + scaled_h);
 261
 262   [base_img lockFocus];
 263   [paste_img drawInRect: NSMakeRect (to_x, to_y, scaled_w, scaled_h)
 264                fromRect: NSMakeRect (from_x, from_y, w, h)
 265               operation: NSCompositeSourceOver
 266                fraction: opacity];
 267   [base_img unlockFocus];
 268
 269   if (verbose_p)
 270     fprintf (stderr, "%s: pasted %dx%d (%dx%d) from %d,%d to %d,%d\n",
 271              progname, w, h, scaled_w, scaled_h, from_x, from_y, to_x, to_y);
 272
 273   [paste_img release];
 274   write_image (base_img, base_file);
 275   [base_img release];
 276 }
 277
 278
 279 static void
 280 write_image (NSImage *img, const char *file)
 281 {
 282   float jpeg_quality = .85;
 283
 284   // Load the NSImage's contents into an NSBitmapImageRep:
 285   NSBitmapImageRep *bit_rep = [NSBitmapImageRep
 286                                 imageRepWithData:[img TIFFRepresentation]];
 287
 288   // Write the bitmapImageRep to a JPEG file.
 289   if (bit_rep == nil)
 290     {
 291       fprintf (stderr, "%s: error converting image?\n", progname);
 292       exit (1);
 293     }
 294
 295   if (verbose_p)
 296     fprintf (stderr, "%s: writing %s (q=%d%%) ", progname, file,
 297              (int) (jpeg_quality * 100));
 298
 299   NSDictionary *props = [NSDictionary
 300                           dictionaryWithObject:
 301                             [NSNumber numberWithFloat:jpeg_quality]
 302                           forKey:NSImageCompressionFactor];
 303   NSData *jpeg_data = [bit_rep representationUsingType:NSJPEGFileType
 304                                properties:props];
 305
 306   [jpeg_data writeToFile:
 307                [NSString stringWithCString:file
 308                                   encoding:NSISOLatin1StringEncoding]
 309              atomically:YES];
 310
 311   if (verbose_p)
 312     {
 313       struct stat st;
 314       if (stat (file, &st))
 315         {
 316           char buf[255];
 317           sprintf (buf, "%.100s: %.100s", progname, file);
 318           perror (buf);
 319           exit (1);
 320         }
 321       fprintf (stderr, " %luK\n", ((unsigned long) st.st_size + 1023) / 1024);
 322     }
 323 }
 324
 325
 326 static void
 327 usage (void)
 328 {
 329   fprintf (stderr, "usage: %s [-v] paste-file base-file\n"
 330            "\t from-scale opacity\n"
 331            "\t from-x from-y to-x to-y w h\n"
 332            "\n"
 333            "\t Pastes paste-file into base-file.\n"
 334            "\t base-file will be overwritten (with JPEG data.)\n"
 335            "\t scaling is applied first: coordinates apply to scaled image.\n",
 336            progname);
 337   exit (1);
 338 }
 339
 340
 341 int
 342 main (int argc, char **argv)
 343 {
 344   int i;
 345   char *paste_file, *base_file, *s, dummy;
 346   double from_scale, opacity;
 347   int from_x, from_y, to_x, to_y, w, h, bevel_pct;
 348
 349   i = 0;
 350   progname = argv[i++];
 351   s = strrchr (progname, '/');
 352   if (s) progname = s+1;
 353
 354   if (argc != 11 && argc != 12) usage();
 355
 356   if (!strcmp(argv[i], "-v"))
 357     verbose_p++, i++;
 358
 359   paste_file = argv[i++];
 360   base_file = argv[i++];
 361
 362   if (*paste_file == '-') usage();
 363   if (*base_file == '-') usage();
 364
 365   s = argv[i++];
 366   if (1 != sscanf (s, " %lf %c", &from_scale, &dummy)) usage();
 367   if (from_scale <= 0 || from_scale > 100) usage();
 368
 369   s = argv[i++];
 370   if (1 != sscanf (s, " %lf %c", &opacity, &dummy)) usage();
 371   if (opacity <= 0 || opacity > 1) usage();
 372
 373   s = argv[i++]; if (1 != sscanf (s, " %d %c", &from_x, &dummy)) usage();
 374   s = argv[i++]; if (1 != sscanf (s, " %d %c", &from_y, &dummy)) usage();
 375   s = argv[i++]; if (1 != sscanf (s, " %d %c", &to_x, &dummy)) usage();
 376   s = argv[i++]; if (1 != sscanf (s, " %d %c", &to_y, &dummy)) usage();
 377   s = argv[i++]; if (1 != sscanf (s, " %d %c", &w, &dummy)) usage();
 378   s = argv[i];   if (1 != sscanf (s, " %d %c", &h, &dummy)) usage();
 379
 380   bevel_pct = 10; /* #### */
 381
 382   if (w < 0) usage();
 383   if (h < 0) usage();
 384
 385   if (w == 0 || h == 0 ||
 386       w > 10240 || h > 10240) {
 387     fprintf (stderr, "%s: absurd size: %d x %d\n", progname, w, h);
 388     exit (1);
 389   }
 390
 391
 392   // Much of Cocoa needs one of these to be available.
 393   NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
 394
 395   //Need an NSApp instance to make [NSImage TIFFRepresentation] work
 396   NSApp = [NSApplication sharedApplication];
 397   [NSApp autorelease];
 398
 399   paste (paste_file, base_file,
 400          from_scale, opacity, bevel_pct,
 401          from_x, from_y, to_x, to_y,
 402          w, h);
 403
 404   [pool release];
 405
 406   exit (0);
 407 }