From http://www.jwz.org/xscreensaver/xscreensaver-5.30.tar.gz
[xscreensaver] / 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 }