RSS

(root)/iphone/common : 61 : Jigs/source/JigsawGenerator.m

« back to all changes in this revision

Viewing changes to Jigs/source/JigsawGenerator.m

Dömötör Gulyás
2010-01-18 09:03:51
Revision ID: dognotdog@gmail.com-20100118080351-ib2knxvk4w8ssw3h
made common a nested tree

Show diffs side-by-side

added added

removed removed

1
 
//
2
 
//  JigsawGenerator.m
3
 
//  JigsawGenerator
4
 
//
5
 
//  Created by DoG on 19.02.07.
6
 
//  Copyright 2007 Doemoetoer Gulyas. All rights reserved.
7
 
//
8
 
 
9
 
#import "JigsawGenerator.h"
10
 
//#import "JigsawView.h"
11
 
#import "SimplexNoise.h"
12
 
#import "MersenneTwister.h"
13
 
#import "geometry.h"
14
 
 
15
 
#define SEG_LENGTH 10.0f
16
 
 
17
 
void FreeJigPieces(JigPieces pieces)
18
 
{
19
 
        for (int i = 0; i < pieces.hcells; ++i)
20
 
                for (int j = 0; j < pieces.vcells; ++j)
21
 
                {
22
 
                        JigCell cell = pieces.cells[i][j];
23
 
                        CGPathRelease(cell.outline);
24
 
                        for (int k = 0; k < 4; ++k)
25
 
                        {
26
 
                                CGPathRelease(cell.segments[k].bpath);
27
 
                                free(cell.segments[k].points);
28
 
                        }
29
 
                }
30
 
 
31
 
        free(pieces.cells[0]);
32
 
        free(pieces.cells);
33
 
}
34
 
 
35
 
static void ReverseOrderOfPoints(CGPoint* points, int count)
36
 
{
37
 
        for (int i = 0; i < count/2; ++i)
38
 
        {
39
 
                CGPoint buf = points[i];
40
 
                points[i] = points[count - 1 - i];
41
 
                points[count - 1 - i] = buf;
42
 
        }
43
 
};
44
 
 
45
 
static void FlipYOfPoints(CGPoint* points, unsigned count)
46
 
{
47
 
        for (size_t i = 0; i < count; ++i)
48
 
        {
49
 
                points[i].y = -points[i].y;
50
 
        }
51
 
};
52
 
 
53
 
static void FlipXOfSegment(JigSegment* seg)
54
 
{
55
 
        for (size_t i = 0; i < seg->count; ++i)
56
 
        {
57
 
                seg->points[i].x = seg->xlen - seg->points[i].x;
58
 
        }
59
 
};
60
 
 
61
 
 
62
 
 
63
 
void transformPointArray(CGAffineTransform transform, CGPoint* points, size_t count)
64
 
{
65
 
        for (size_t i = 0; i < count; ++i)
66
 
                points[i] = CGPointApplyAffineTransform(points[i], transform);
67
 
};
68
 
 
69
 
 
70
 
 
71
 
float diminishingWeight(float p, float size)
72
 
{
73
 
        float w = 1.0f - fclampf(fabsf(p)/(0.5*size), 0.0f, 1.0f);
74
 
        return w;
75
 
}
76
 
 
77
 
CGPoint weightedNoiseAtPoint(CGPoint p, SimplexNoise* noise, float distortion, float freq, CGSize size)
78
 
{
79
 
        float nx = [noise noise3dWithX: p.x*freq Y: p.y*freq Z: 0.0f]*distortion;
80
 
        float ny = [noise noise3dWithX: p.x*freq Y: p.y*freq Z: 1.0f]*distortion;
81
 
        
82
 
        //NSLog(@"%.2f, %.2f", nx, ny);
83
 
        
84
 
        float wx = diminishingWeight(p.x, size.width);
85
 
        float wy = diminishingWeight(p.y, size.height);
86
 
        
87
 
//      wx = wy = 0.0f;
88
 
 
89
 
        return CGPointMake(p.x + nx*wx, p.y + ny*wy);
90
 
}
91
 
 
92
 
@implementation JigsawGenerator
93
 
 
94
 
- (JigPieces) generateJigsaw
95
 
{       
96
 
//      MersenneTwister* rgen = [[[MersenneTwister alloc] initWithSeed: seed] autorelease];
97
 
 
98
 
        // determine enabled edges
99
 
        int* enabledIndices = calloc([edgeFiles count], sizeof(int)); // freed
100
 
        int enabledEdgeCount = 0;
101
 
        for (size_t i = 0; i < [edgeFiles count]; ++i)
102
 
        {
103
 
                //id edge = [[[edgeFiles objectAtIndex: i] lastPathComponent] stringByDeletingPathExtension];
104
 
                id edge = [edgeFiles objectAtIndex: i];
105
 
                if ([[enabledEdges objectForKey: edge] intValue] != 1)
106
 
                        continue;
107
 
                enabledIndices[enabledEdgeCount++] = i;
108
 
        }
109
 
 
110
 
        // make sure we got at least one edge
111
 
        if (enabledEdgeCount == 0)
112
 
                enabledIndices[enabledEdgeCount++] = 0;
113
 
 
114
 
 
115
 
        
116
 
        
117
 
        SimplexNoise* noise = [[[SimplexNoise alloc] init] autorelease];
118
 
        [noise setSeed: seed];
119
 
        
120
 
        // create cells
121
 
 
122
 
        JigCell**       cells = calloc(hcells, sizeof(JigCell*));
123
 
        cells[0] = calloc(hcells*vcells, sizeof(JigCell));
124
 
        for (int i = 1; i < hcells; ++i)
125
 
                cells[i] = cells[i-1] + vcells;
126
 
 
127
 
        // load template edge
128
 
//      NSArray* edges = [NSBundle pathsForResourcesOfType: @"jigedge" inDirectory: [[NSBundle mainBundle] bundlePath]];
129
 
 
130
 
        CGSize jigsawSize = CGSizeMake(width, height);
131
 
 
132
 
        float cellWidth = width/hcells;
133
 
        float cellHeight = height/vcells;
134
 
        float cellAspect = cellWidth/cellHeight;
135
 
        float xAdjust = MIN(1.0f, 1.0f/cellAspect);
136
 
        float yAdjust = MIN(1.0f, cellAspect);
137
 
 
138
 
        CGPoint segmentStarts[4] = {
139
 
                { 0.5*cellWidth, -0.5*cellHeight},
140
 
                { 0.5*cellWidth,  0.5*cellHeight},
141
 
                {-0.5*cellWidth,  0.5*cellHeight},
142
 
                {-0.5*cellWidth, -0.5*cellHeight},
143
 
        };
144
 
        
145
 
        // init cells
146
 
        for (int i = 0; i < hcells; ++i)
147
 
        {
148
 
                float cellx = -0.5*width + (0.5f+(float)i)*cellWidth;
149
 
                for (int j = 0; j < vcells; ++j)
150
 
                {
151
 
                        float celly = -0.5*height + (0.5f+(float)j)*cellHeight;
152
 
                        JigCell* cell = cells[i]+j;
153
 
 
154
 
                        cell->pos = CGPointMake(cellx,celly);
155
 
                        cell->attachmentPoints[0] = CGPointMake(cellx + 0.5*cellWidth, celly);
156
 
                        cell->attachmentPoints[1] = CGPointMake(cellx, celly + 0.5*cellHeight);
157
 
                        cell->attachmentPoints[2] = CGPointMake(cellx - 0.5*cellWidth, celly);
158
 
                        cell->attachmentPoints[3] = CGPointMake(cellx, celly - 0.5*cellHeight);
159
 
 
160
 
                        for (int k = 0; k < 4; ++k)
161
 
                        {
162
 
                                float snoise = 0.0f;
163
 
                                snoise = [noise noise3dWithX: 10*(i+(k==0?1:0)) Y: 10*(j+(k==1?1:0)) Z: 2.0f withOctaves: 1 ofType: 0];
164
 
 
165
 
                                cell->segments[k].count = 2;
166
 
                                cell->segments[k].xlen = SEG_LENGTH;
167
 
                                cell->segments[k].format = 0;
168
 
                                if ((j == 0) && (k == 3))
169
 
                                {
170
 
                                        cell->segments[k].points = calloc(2, sizeof(CGPoint));
171
 
                                        cell->segments[k].points[0] = (CGPointMake(0.0f, 0.0f));
172
 
                                        cell->segments[k].points[1] = (CGPointMake(SEG_LENGTH, 0.0f));
173
 
                                }
174
 
                                else if ((j == vcells-1) && (k == 1))
175
 
                                {
176
 
                                        cell->segments[k].points = calloc(2, sizeof(CGPoint));
177
 
                                        cell->segments[k].points[0] = (CGPointMake(0.0f, 0.0f));
178
 
                                        cell->segments[k].points[1] = (CGPointMake(SEG_LENGTH, 0.0f));
179
 
                                }
180
 
                                else if ((i == 0) && (k == 2))
181
 
                                {
182
 
                                        cell->segments[k].points = calloc(2, sizeof(CGPoint));
183
 
                                        cell->segments[k].points[0] = (CGPointMake(0.0f, 0.0f));
184
 
                                        cell->segments[k].points[1] = (CGPointMake(SEG_LENGTH, 0.0f));
185
 
                                }
186
 
                                else if ((i == hcells-1) && (k == 0))
187
 
                                {
188
 
                                        cell->segments[k].points = calloc(2, sizeof(CGPoint));
189
 
                                        cell->segments[k].points[0] = (CGPointMake(0.0f, 0.0f));
190
 
                                        cell->segments[k].points[1] = (CGPointMake(SEG_LENGTH, 0.0f));
191
 
                                }
192
 
                                else
193
 
                                {
194
 
                                        // randomly select edge
195
 
                                        JigSegment templateSegment;
196
 
                                        
197
 
                                        {
198
 
                                                
199
 
                                                //NSLog(@"selection noise %f %f %f", nval, 0.5f*(p.x+q.x), 0.5f*(p.y+q.y));
200
 
 
201
 
                                                float nval = snoise + 1.0f;
202
 
                                                nval *= 0.5f; // now in [0,1];
203
 
                                                
204
 
                                                // multiply by large prime
205
 
                                                nval *= 3943.0f;
206
 
                                                
207
 
                                                nval = fmodf(nval,enabledEdgeCount);
208
 
                                                
209
 
                                                //nval *= enabledEdgeCount*(1.0f - 2.0f*FLT_EPSILON);
210
 
                                                assert((int)(nval) < enabledEdgeCount);
211
 
                                                
212
 
                                                int edgeIndex = enabledIndices[(int)(nval)];
213
 
                                                
214
 
                                                //id edgeName = [[[edgeFiles objectAtIndex: edgeIndex] lastPathComponent] stringByDeletingPathExtension];
215
 
                                                id edgeName = [edgeFiles objectAtIndex: edgeIndex];
216
 
                                                
217
 
                                                templateSegment = GetJigSegmentFromFile([edgeFiles objectAtIndex: edgeIndex]);
218
 
                                                
219
 
                                                if (((templateSegment.format == 0) || (templateSegment.format == 2)) && ([[smoothedEdges objectForKey: edgeName] intValue] == 1))
220
 
                                                {
221
 
                                                        templateSegment.format = 1;
222
 
                                                }
223
 
                                        }
224
 
 
225
 
 
226
 
                                
227
 
                                        cell->segments[k].count = templateSegment.count;
228
 
                                        cell->segments[k].xlen = templateSegment.xlen;
229
 
                                        cell->segments[k].format = templateSegment.format;
230
 
                                        cell->segments[k].points = calloc(cell->segments[k].count, sizeof(CGPoint));
231
 
                                        memcpy(cell->segments[k].points, templateSegment.points, templateSegment.count*sizeof(CGPoint));
232
 
                                        
233
 
                                        // flip based on noise
234
 
                                        if (snoise > 0.0f)
235
 
                                        {
236
 
                                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
237
 
                                                        cell->segments[k].points[ii].y = -cell->segments[k].points[ii].y;
238
 
                                        }
239
 
                                        
240
 
                                        free(templateSegment.points);
241
 
                                        
242
 
                                        
243
 
                                }
244
 
                                // rotate and scale segment to cell size
245
 
                                switch (k)
246
 
                                {
247
 
                                        case 0:
248
 
                                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
249
 
                                                {
250
 
                                                        CGPoint p = cell->segments[k].points[ii];
251
 
                                                        p = CGPointMake(-p.y, p.x);
252
 
                                                        p.x *= cellWidth/SEG_LENGTH*xAdjust;
253
 
                                                        p.y *= cellHeight/SEG_LENGTH;
254
 
                                                        cell->segments[k].points[ii] = p;
255
 
                                                }
256
 
                                                break;
257
 
                                        case 1:
258
 
                                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
259
 
                                                {
260
 
                                                        CGPoint p = cell->segments[k].points[ii];
261
 
                                                        p = CGPointMake(-p.x, p.y);
262
 
                                                        p.x *= cellWidth/SEG_LENGTH;
263
 
                                                        p.y *= cellHeight/SEG_LENGTH*yAdjust;
264
 
                                                        cell->segments[k].points[ii] = p;
265
 
                                                }
266
 
                                                //ReverseOrderOfPoints(cell->segments[k].points, cell->segments[k].count);
267
 
                                                break;
268
 
                                        case 2:
269
 
                                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
270
 
                                                {
271
 
                                                        CGPoint p = cell->segments[k].points[ii];
272
 
                                                        p = CGPointMake(-p.y, -p.x);
273
 
                                                        p.x *= cellWidth/SEG_LENGTH*xAdjust;
274
 
                                                        p.y *= cellHeight/SEG_LENGTH;
275
 
                                                        cell->segments[k].points[ii] = p;
276
 
                                                }
277
 
                                                //ReverseOrderOfPoints(cell->segments[k].points, cell->segments[k].count);
278
 
                                                break;
279
 
                                        case 3:
280
 
                                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
281
 
                                                {
282
 
                                                        CGPoint p = cell->segments[k].points[ii];
283
 
                                                        p.x *= cellWidth/SEG_LENGTH;
284
 
                                                        p.y *= cellHeight/SEG_LENGTH*yAdjust;
285
 
                                                        cell->segments[k].points[ii] = p;
286
 
                                                }
287
 
                                                break;
288
 
                                }
289
 
                                // transform local to global coords
290
 
                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
291
 
                                {
292
 
                                        CGPoint p = cell->segments[k].points[ii];
293
 
                                        p = CGPointAdd(p, segmentStarts[k]);
294
 
                                        p = CGPointAdd(p, CGPointMake(cellx, celly));
295
 
                                        cell->segments[k].points[ii] = p;
296
 
                                }
297
 
                        }
298
 
                        
299
 
                        // and now add some noise
300
 
                        
301
 
                        cell->pos = weightedNoiseAtPoint(cell->pos, noise, distortion, frequency, jigsawSize);
302
 
                        for (size_t k = 0; k < 4; ++k)
303
 
                        {
304
 
                                cell->attachmentPoints[k] = weightedNoiseAtPoint(cell->attachmentPoints[k], noise, distortion, frequency, jigsawSize);
305
 
                                for (size_t ii = 0; ii < cell->segments[k].count; ++ii)
306
 
                                {
307
 
                                        CGPoint p = cell->segments[k].points[ii];
308
 
                                        p = weightedNoiseAtPoint(p, noise, distortion, frequency, jigsawSize);
309
 
                                        cell->segments[k].points[ii] = p;
310
 
                                }
311
 
                        }
312
 
                        
313
 
                }
314
 
        }
315
 
 
316
 
        for (int i = 0; i < hcells; ++i)
317
 
        {
318
 
                for (int j = 0; j < vcells; ++j)
319
 
                {
320
 
                        JigCell* cell = (cells[i]+j);
321
 
 
322
 
                        for (int k = 0; k < 4; ++k)
323
 
                        {
324
 
                                // apply smoothing
325
 
                                if (cell->segments[k].format == 1)
326
 
                                {
327
 
                                        SplineSegmentToPiecewiseBezier( &(cell->segments[k]) );
328
 
                                }
329
 
                                
330
 
                                // generate NSBeziers
331
 
 
332
 
                                CGMutablePathRef bpath = CGPathCreateMutable();
333
 
                                CGPathMoveToPoint(bpath, NULL, cell->segments[k].points[0].x, cell->segments[k].points[0].y);
334
 
                                
335
 
                                switch(cell->segments[k].format)
336
 
                                {
337
 
                                        case 0:
338
 
                                        {
339
 
                                                for (size_t l = 1; l < cell->segments[k].count; ++l)
340
 
                                                        CGPathAddLineToPoint(bpath, NULL, cell->segments[k].points[l].x, cell->segments[k].points[l].y);
341
 
                                                break;
342
 
                                        }
343
 
                                        case 2:
344
 
                                        {
345
 
                                                for (size_t l = 1; l < cell->segments[k].count; l+=3)
346
 
                                                        CGPathAddCurveToPoint(bpath, NULL, cell->segments[k].points[l].x, cell->segments[k].points[l].y, cell->segments[k].points[l+1].x, cell->segments[k].points[l+1].y, cell->segments[k].points[l+2].x, cell->segments[k].points[l+2].y);
347
 
                                                break;
348
 
                                        }
349
 
                                        default:
350
 
                                                NSLog(@"dammit, unkown point format: %d", cell->segments[k].format);
351
 
                                };
352
 
                                
353
 
                                cell->segments[k].bpath = bpath;
354
 
 
355
 
                        };
356
 
                        
357
 
                        // generate outline path
358
 
                        CGMutablePathRef bpath = CGPathCreateMutable();
359
 
                        for (int k = 0; k < 4; ++k)
360
 
                        {
361
 
                                if (k == 0)
362
 
                                        CGPathMoveToPoint(bpath, NULL, cell->segments[k].points[0].x, cell->segments[k].points[0].y);
363
 
                                        
364
 
                                int pcount = cell->segments[k].count;
365
 
                                //if (k == 3)
366
 
                                //      pcount--;
367
 
                                
368
 
                                switch(cell->segments[k].format)
369
 
                                {
370
 
                                        case 0:
371
 
                                        {
372
 
                                                for (int l = 1; l < pcount; ++l)
373
 
                                                        CGPathAddLineToPoint(bpath, NULL, cell->segments[k].points[l].x, cell->segments[k].points[l].y);
374
 
                                                break;
375
 
                                        }
376
 
                                        case 2:
377
 
                                        {
378
 
                                                for (int l = 1; l < pcount; l+=3)
379
 
                                                        CGPathAddCurveToPoint(bpath, NULL, cell->segments[k].points[l].x, cell->segments[k].points[l].y, cell->segments[k].points[l+1].x, cell->segments[k].points[l+1].y, cell->segments[k].points[l+2].x, cell->segments[k].points[l+2].y);
380
 
                                                break;
381
 
                                        }
382
 
                                        default:
383
 
                                                NSLog(@"dammit, unkown point format: %d", cell->segments[k].format);
384
 
                                };
385
 
                                
386
 
                        };
387
 
                        CGPathCloseSubpath(bpath);
388
 
                        cell->outline = bpath;
389
 
                };
390
 
        };
391
 
 
392
 
// cleanup
393
 
 
394
 
        free(enabledIndices);
395
 
 
396
 
 
397
 
 
398
 
        JigPieces pieces;
399
 
        pieces.hcells = hcells;
400
 
        pieces.vcells = vcells;
401
 
        pieces.cells = cells;
402
 
 
403
 
 
404
 
 
405
 
        return pieces;
406
 
};
407
 
 
408
 
JigSegment      GetJigSegmentFromFile(NSString* path)
409
 
{
410
 
        FILE* file = fopen([path fileSystemRepresentation], "r");
411
 
        float x = 0.0, y = 0.0;
412
 
        int i = 0;
413
 
        
414
 
        
415
 
        CGPoint* buf = calloc(100, sizeof(CGPoint)); // retained
416
 
        while(2 == fscanf(file, "%f %f", &x, &y))
417
 
        {
418
 
                if (i == 100)
419
 
                {
420
 
                        fprintf(stderr, __FILE__ ", %d: Buffer size exceeded\n", __LINE__);
421
 
                        exit(-1);
422
 
                };
423
 
                buf[i++] = CGPointMake(x,y);
424
 
        };
425
 
        char ch = 0;
426
 
        int format = 0;
427
 
        if (!feof(file))
428
 
                if (1 == fscanf(file, "%c", &ch))
429
 
                {
430
 
                        if (ch == 'b')
431
 
                                format = 2;
432
 
                        else if (ch == 's')
433
 
                                format = 1;
434
 
                }
435
 
        
436
 
        fclose(file);
437
 
        
438
 
        return (JigSegment){buf, i, x, format, NULL};
439
 
};
440
 
 
441
 
 
442
 
static CGPoint interpolate(CGPoint p, CGPoint q, float t)
443
 
{
444
 
        return CGPointMake(p.x*(1.0f-t) + q.x*(t), p.y*(1.0f-t) + q.y*(t)); 
445
 
};
446
 
 
447
 
void            SplineSegmentToPiecewiseBezier(JigSegment* seg)
448
 
{
449
 
        int npieces = seg->count - 1;
450
 
        int newpointcount = npieces*3 - 2;
451
 
        CGPoint* newPoints = calloc(newpointcount, sizeof(CGPoint)); // retained
452
 
        
453
 
        if (npieces < 2)
454
 
                return;
455
 
        int j = 0;
456
 
        for (int i = 0; i < npieces; ++i)
457
 
        {
458
 
                CGPoint p = seg->points[i];
459
 
                CGPoint q = seg->points[i+1];
460
 
                if (i == 0)
461
 
                {
462
 
                        newPoints[j++] = p;
463
 
                        newPoints[j++] = interpolate(p,q,0.5f);
464
 
                        //newPoints[j++] = q;
465
 
                }
466
 
                else if (i < npieces-1)
467
 
                {
468
 
                        newPoints[j++] = interpolate(p,q,0.25f);
469
 
                        newPoints[j++] = interpolate(p,q,0.5f);
470
 
                        newPoints[j++] = interpolate(p,q,0.75f);
471
 
                }
472
 
                else
473
 
                {
474
 
                        //newPoints[j++] = p;
475
 
                        newPoints[j++] = interpolate(p,q,0.5f);
476
 
                        newPoints[j++] = q;
477
 
                }
478
 
 
479
 
        };
480
 
        
481
 
        free(seg->points);
482
 
        seg->points = newPoints;
483
 
        seg->count = newpointcount;
484
 
        seg->format = 2;
485
 
};
486
 
 
487
 
- (CGSize) puzzleSize
488
 
{
489
 
        return CGSizeMake(width, height);
490
 
};
491
 
 
492
 
@synthesize width, height, frequency, distortion, hcells, vcells, seed, octaves, noiseType, roundEdgeCornerCoordsToInt, edgeFiles, enabledEdges, smoothedEdges;
493
 
 
494
 
@end
495
 
 
496
 
 
497
 
 
498
 
 
499
 
 

Loggerhead 1.17 is a web-based interface for Bazaar branches