ShapeKitGeometry.mm

//
//  ShapeKitGeometry.m
//  ShapeKit
//
//  Created by Michael Weisman on 10-08-21.

// * This is free software; you can redistribute and/or modify it under
// the terms of the GNU Lesser General Public Licence as published
// by the Free Software Foundation. 
// See the COPYING file for more information.
//

#import "ShapeKitGeometry.h"
#import <geos_c.h>


@implementation ShapeKitGeometry
@synthesize wktGeom,geomType,geosGeom;

#pragma mark ShapeKitPoint init and dealloc methods
- (id) init
{
    self = [super init];
    if (self != nil) {
        // initialize GEOS library
        initGEOS(notice, log_and_exit);
    }
    return self;
}


-(id)initWithWKT:(NSString *) wkt {
    [self init];
    
    GEOSWKTReader *WKTReader = GEOSWKTReader_create();
    self.geosGeom = GEOSWKTReader_read(WKTReader, [wkt UTF8String]);
    GEOSWKTReader_destroy(WKTReader);
    
    self.geomType = [NSString stringWithUTF8String:GEOSGeomType(geosGeom)];
    
    GEOSWKTWriter *WKTWriter = GEOSWKTWriter_create();
    self.wktGeom = [NSString stringWithUTF8String:GEOSWKTWriter_write(WKTWriter,geosGeom)];
    GEOSWKTWriter_destroy(WKTWriter);
    
    return self;
}


-(id)initWithGeosGeometry:(GEOSGeometry *)geom {
    geosGeom = geom;
    self.geomType = [NSString stringWithUTF8String:GEOSGeomType(geosGeom)];
    GEOSWKTWriter *WKTWriter = GEOSWKTWriter_create();
    self.wktGeom = [NSString stringWithUTF8String:GEOSWKTWriter_write(WKTWriter,geosGeom)];
    GEOSWKTWriter_destroy(WKTWriter);
    
    return self;    
}


- (void) dealloc
{
    [geomType release];
    [wktGeom release];
    GEOSGeom_destroy(geosGeom);
    finishGEOS();
    [super dealloc];
}

#pragma mark GEOS init functions
void notice(const char *fmt,...) {
	va_list ap;
    
    fprintf( stdout, "NOTICE: ");
    
	va_start (ap, fmt);
    vfprintf( stdout, fmt, ap);
    va_end(ap);
    fprintf( stdout, "\n" );
}

void log_and_exit(const char *fmt,...) {
	va_list ap;
    
    fprintf( stdout, "ERROR: ");
    
	va_start (ap, fmt);
    vfprintf( stdout, fmt, ap);
    va_end(ap);
    fprintf( stdout, "\n" );
//	exit(1);
}

@end

#pragma mark -

@implementation ShapeKitPoint
@synthesize geometry,numberOfCoords;

-(id)initWithWKT:(NSString *) wkt {
    [super initWithWKT:wkt];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(geosGeom));
    geometry = [[MKPointAnnotation alloc] init];
    double xCoord;
    GEOSCoordSeq_getX(sequence, 0, &xCoord);
    
    double yCoord;
    GEOSCoordSeq_getY(sequence, 0, &yCoord);
    geometry.coordinate = CLLocationCoordinate2DMake(yCoord, xCoord);
    
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    GEOSCoordSeq_destroy(sequence);
        
    return self;
}

-(id)initWithGeosGeometry:(GEOSGeometry *)geom {
    [super initWithGeosGeometry:geom];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(geosGeom));
    geometry = [[MKPointAnnotation alloc] init];
    double xCoord;
    GEOSCoordSeq_getX(sequence, 0, &xCoord);
    
    double yCoord;
    GEOSCoordSeq_getY(sequence, 0, &yCoord);
    geometry.coordinate = CLLocationCoordinate2DMake(yCoord, xCoord);
    
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    GEOSCoordSeq_destroy(sequence);
    
    return self;
}


- (void) dealloc
{
    [geometry release];
    [super dealloc];
}

@end

#pragma mark -

@implementation ShapeKitPolyline
@synthesize geometry,numberOfCoords;

-(id)initWithWKT:(NSString *) wkt {
    [super initWithWKT:wkt];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(geosGeom));
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    CLLocationCoordinate2D coords[numberOfCoords];
    
    for (int coord = 0; coord < numberOfCoords; coord++) {
        double xCoord = NULL;
        GEOSCoordSeq_getX(sequence, coord, &xCoord);
        
        double yCoord = NULL;
        GEOSCoordSeq_getY(sequence, coord, &yCoord);
        coords[coord] = CLLocationCoordinate2DMake(yCoord, xCoord);
    }
    geometry = [MKPolyline polylineWithCoordinates:coords count:numberOfCoords];

    GEOSCoordSeq_destroy(sequence);
    
    return self;
}

-(id)initWithGeosGeometry:(GEOSGeometry *)geom {
    [super initWithGeosGeometry:geom];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(geosGeom));
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    CLLocationCoordinate2D coords[numberOfCoords];
    
    for (int coord = 0; coord < numberOfCoords; coord++) {
        double xCoord = NULL;
        GEOSCoordSeq_getX(sequence, coord, &xCoord);
        
        double yCoord = NULL;
        GEOSCoordSeq_getY(sequence, coord, &yCoord);
        coords[coord] = CLLocationCoordinate2DMake(yCoord, xCoord);
    }
    geometry = [MKPolyline polylineWithCoordinates:coords count:numberOfCoords];
    
    GEOSCoordSeq_destroy(sequence);
    
    return self;
    
}

@end

#pragma mark -

@implementation ShapeKitPolygon
@synthesize geometry,numberOfCoords;

-(id)initWithWKT:(NSString *) wkt {
    [super initWithWKT:wkt];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(GEOSGetExteriorRing(geosGeom)));
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    CLLocationCoordinate2D coords[numberOfCoords];
    
    for (int coord = 0; coord < numberOfCoords; coord++) {
        double xCoord = NULL;
        GEOSCoordSeq_getX(sequence, coord, &xCoord);
        
        double yCoord = NULL;
        GEOSCoordSeq_getY(sequence, coord, &yCoord);
        coords[coord] = CLLocationCoordinate2DMake(yCoord, xCoord);
    }
    geometry = [MKPolygon polygonWithCoordinates:coords count:numberOfCoords];
    
    GEOSCoordSeq_destroy(sequence);
    
    return self;
}

-(id)initWithGeosGeometry:(GEOSGeometry *)geom {
    [super initWithGeosGeometry:geom];
    GEOSCoordSequence *sequence = GEOSCoordSeq_clone(GEOSGeom_getCoordSeq(GEOSGetExteriorRing(geosGeom)));
    GEOSCoordSeq_getSize(sequence, &numberOfCoords);
    CLLocationCoordinate2D coords[numberOfCoords];
    
    for (int coord = 0; coord < numberOfCoords; coord++) {
        double xCoord = NULL;
        GEOSCoordSeq_getX(sequence, coord, &xCoord);
        
        double yCoord = NULL;
        GEOSCoordSeq_getY(sequence, coord, &yCoord);
        coords[coord] = CLLocationCoordinate2DMake(yCoord, xCoord);
    }
    geometry = [MKPolygon polygonWithCoordinates:coords count:numberOfCoords];
    
    GEOSCoordSeq_destroy(sequence);
    
    return self;
}

@end