//
//  RouteMeTests.m
//  MapView
//
//  Created by Hal Mueller on 4/6/09.
//  Copyright 2009 Route-Me Contributors. All rights reserved.
//

#import "RouteMeTests.h"
#import "RMMapView.h"
#import "RMCloudMadeMapSource.h"
#import "RMMarker.h"
#import "RMMarkerManager.h"

@implementation RouteMeTests
#define kAccuracyThreshold .0001
#define kAccuracyThresholdForGeographicCoordinates .00001
#define kAccuracyThresholdForPixelCoordinates kAccuracyThreshold

- (void)setUp {
    [super setUp];

	CGRect appRect = [[UIScreen mainScreen] applicationFrame];
	contentView = [[UIView alloc] initWithFrame:appRect];
	contentView.backgroundColor = [UIColor greenColor];
	
	initialCenter.latitude = 66.44;
	initialCenter.longitude = -178.0;
	
	mapView = [[RMMapView alloc] initWithFrame:CGRectMake(10,20,200,300)
								  WithLocation:initialCenter];
	[contentView addSubview:mapView];
}

-(void)tearDown {
    [mapView release]; mapView = nil;
	[super tearDown];
}

- (void)testObjectCreation 
{
	
	STAssertNotNil((mapView = [[RMMapView alloc] init]), @"mapView alloc/init failed");
	STAssertNoThrow([mapView release], @"mapView release failed");
	mapView = nil;
	
	id myTilesource;
	STAssertNotNil((myTilesource = [[RMCloudMadeMapSource alloc] initWithAccessKey:@"0199bdee456e59ce950b0156029d6934" styleNumber:999]),
				   @"tilesource creation failed");
	STAssertNoThrow([myTilesource release], @"tilesource release failed");
	STAssertNil((myTilesource = [[RMCloudMadeMapSource alloc] initWithAccessKey:nil styleNumber:999]),
				@"empty CloudMade key does not trigger error");
	STAssertNoThrow([myTilesource release], @"tilesource release failed");
	STAssertThrows((myTilesource = [[RMCloudMadeMapSource alloc] initWithAccessKey:@"0199bdee456e59ce950b0156029d693" styleNumber:999]),
				@"bogus CloudMade key does not trigger error");
	STAssertNoThrow([myTilesource release], @"tilesource release failed");
}

- (void)testProgrammaticViewCreation
{
	STAssertNotNil(mapView, @"mapview creation failed");
	STAssertNotNil([mapView contents], @"mapView contents should not be nil");
}

- (void)testMarkerCreation
{
	// create markers from -183 to -169 longitude 
	initialCenter.longitude = -178.0;

	CLLocationCoordinate2D markerPosition;
	NSUInteger nRows = 1;
	NSUInteger nColumns = 8;
	double columnSpacing = 2.0;
	
	UIImage *markerImage = [UIImage imageNamed:@"marker-red.png"];
	STAssertNotNil(markerImage, @"testMarkerCreation marker image did not load");
	markerPosition.latitude = initialCenter.latitude - ((nRows - 1)/2.0 * columnSpacing);
	NSUInteger i, j;
	for (i = 0; i < nRows; i++) {
		markerPosition.longitude = initialCenter.longitude - ((nColumns - 1)/2.0 * columnSpacing);
		for (j = 0; j < nColumns; j++) {
			markerPosition.longitude += columnSpacing;
			RMMarker *newMarker = [[RMMarker alloc] initWithUIImage:markerImage];
			STAssertNotNil(newMarker, @"testMarkerCreation marker creation failed");
			[newMarker setUserInfo:[NSArray arrayWithObjects:[NSNumber numberWithDouble:markerPosition.longitude],[NSNumber numberWithDouble:markerPosition.latitude],nil]];
			[mapView.contents.markerManager addMarker:newMarker
			 atLatLong:markerPosition];
		}
		markerPosition.latitude += columnSpacing;
	}
}

- (void)testMarkerCoordinatesFarEast
{
	[mapView.contents setZoom:3.0];

	// create markers from +177 to +191 longitude 
	initialCenter.longitude = +176.0;
	CLLocationCoordinate2D markerPosition;
	
	NSUInteger nColumns = 8;
	double columnSpacing = 2.0;
	
	UIImage *markerImage = [UIImage imageNamed:@"marker-red.png"];
	markerPosition.latitude = initialCenter.latitude;
	markerPosition.longitude = initialCenter.longitude - ((nColumns - 1)/2.0 * columnSpacing);
	NSUInteger j;
	NSMutableArray *testMarkers = [NSMutableArray arrayWithCapacity:nColumns];
	for (j = 0; j < nColumns; j++) {
		markerPosition.longitude += columnSpacing;
		RMMarker *newMarker = [[RMMarker alloc] initWithUIImage:markerImage];
		[testMarkers addObject:newMarker];
		[newMarker setUserInfo:[NSArray arrayWithObjects:[NSNumber numberWithDouble:markerPosition.longitude],[NSNumber numberWithDouble:markerPosition.latitude],nil]];
		[mapView.contents.markerManager addMarker:newMarker
		 atLatLong:markerPosition];
	}
	STAssertGreaterThan(columnSpacing, 0.0, @"this test requires positive columnSpacing");

	RMMarkerManager *mangler = [[mapView contents] markerManager];
	
	[[mapView contents] moveBy:CGSizeMake(-5.0, 0.0)];
#ifdef DEBUG
	RMSphericalTrapezium screenLimitsDegrees = [[mapView contents] latitudeLongitudeBoundingBoxForScreen];
	RMLog(@"screen limits west: %4.1f east %4.1f", screenLimitsDegrees.southwest.longitude, screenLimitsDegrees.northeast.longitude);
	RMLog(@"screen limits south: %4.1f north %4.1f", screenLimitsDegrees.southwest.latitude, screenLimitsDegrees.northeast.latitude);
#endif
	
	for (j = 1; j < nColumns; j++) {
		RMMarker *leftMarker = [testMarkers objectAtIndex:j - 1];
		RMMarker *rightMarker = [testMarkers objectAtIndex:j];
		CGPoint leftScreenPosition = [mangler screenCoordinatesForMarker:leftMarker];
		CGPoint rightScreenPosition = [mangler screenCoordinatesForMarker:rightMarker];
		CLLocationCoordinate2D leftMarkerCoordinate, rightMarkerCoordinate;
		leftMarkerCoordinate.longitude = [[(NSArray *)leftMarker.userInfo objectAtIndex:0] doubleValue];
		leftMarkerCoordinate.latitude = [[(NSArray *)leftMarker.userInfo objectAtIndex:1] doubleValue];
		rightMarkerCoordinate.longitude = [[(NSArray *)rightMarker.userInfo objectAtIndex:0] doubleValue];
		rightMarkerCoordinate.latitude = [[(NSArray *)rightMarker.userInfo objectAtIndex:1] doubleValue];
		STAssertLessThan(leftScreenPosition.x, rightScreenPosition.x, 
						 @"screen position calculation failed (markers %d, %d): left (%f, %f) right (%f, %f) mapped to left (%f, %f) right (%f, %f)",
						 j-1, j,
// write these out as longitude/latitude instead of standard latitude/longitude to make comparisons easier
						 leftMarkerCoordinate.longitude, leftMarkerCoordinate.latitude,
						 rightMarkerCoordinate.longitude, rightMarkerCoordinate.latitude,
						 leftScreenPosition.x, leftScreenPosition.y, rightScreenPosition.x, rightScreenPosition.y);
		CLLocationCoordinate2D computedLatitudeLongitude = 
		[mangler latitudeLongitudeForMarker:leftMarker];
		STAssertEqualsWithAccuracy(leftMarkerCoordinate.longitude, computedLatitudeLongitude.longitude, kAccuracyThresholdForGeographicCoordinates,
								   @"round-trip computation of longitude failed %f %f",
								   leftMarkerCoordinate.longitude, computedLatitudeLongitude.longitude);
		STAssertEqualsWithAccuracy(leftMarkerCoordinate.latitude, computedLatitudeLongitude.latitude, kAccuracyThresholdForGeographicCoordinates,
								   @"round-trip computation of latitude failed %f %f",
								   leftMarkerCoordinate.latitude, computedLatitudeLongitude.latitude);
	}
	
}

- (void)testMarkerCoordinatesFarWest
{
	[mapView.contents setZoom:3.0];

	// create markers from -177 to -169 longitude
	initialCenter.longitude = -178.0;
	CLLocationCoordinate2D markerPosition;
	
	NSUInteger nColumns = 8;
	double columnSpacing = 2.0;
	
	UIImage *markerImage = [UIImage imageNamed:@"marker-red.png"];
	markerPosition.latitude = initialCenter.latitude;
	markerPosition.longitude = initialCenter.longitude - ((nColumns - 1)/2.0 * columnSpacing);
	NSUInteger j;
	NSMutableArray *testMarkers = [NSMutableArray arrayWithCapacity:nColumns];
	for (j = 0; j < nColumns; j++) {
		markerPosition.longitude += columnSpacing;
		RMMarker *newMarker = [[RMMarker alloc] initWithUIImage:markerImage];
		[testMarkers addObject:newMarker];
		[newMarker setUserInfo:[NSArray arrayWithObjects:[NSNumber numberWithDouble:markerPosition.longitude],[NSNumber numberWithDouble:markerPosition.latitude],nil]];
		[mapView.contents.markerManager addMarker:newMarker
		 atLatLong:markerPosition];
	}
	STAssertGreaterThan(columnSpacing, 0.0, @"this test requires positive columnSpacing");

	RMMarkerManager *mangler = [[mapView contents] markerManager];
	
	[[mapView contents] moveBy:CGSizeMake(-5.0, 0.0)];
#ifdef DEBUG
	RMSphericalTrapezium screenLimitsDegrees = [[mapView contents] latitudeLongitudeBoundingBoxForScreen];
	RMLog(@"screen limits west: %4.1f east %4.1f", screenLimitsDegrees.southwest.longitude, screenLimitsDegrees.northeast.longitude);
	RMLog(@"screen limits south: %4.1f north %4.1f", screenLimitsDegrees.southwest.latitude, screenLimitsDegrees.northeast.latitude);
#endif
	
	for (j = 1; j < nColumns; j++) {
		RMMarker *leftMarker = [testMarkers objectAtIndex:j - 1];
		RMMarker *rightMarker = [testMarkers objectAtIndex:j];
		CGPoint leftScreenPosition = [mangler screenCoordinatesForMarker:leftMarker];
		CGPoint rightScreenPosition = [mangler screenCoordinatesForMarker:rightMarker];
		CLLocationCoordinate2D leftMarkerCoordinate, rightMarkerCoordinate;
		leftMarkerCoordinate.longitude = [[(NSArray *)leftMarker.userInfo objectAtIndex:0] doubleValue];
		leftMarkerCoordinate.latitude = [[(NSArray *)leftMarker.userInfo objectAtIndex:1] doubleValue];
		rightMarkerCoordinate.longitude = [[(NSArray *)rightMarker.userInfo objectAtIndex:0] doubleValue];
		rightMarkerCoordinate.latitude = [[(NSArray *)rightMarker.userInfo objectAtIndex:1] doubleValue];
		STAssertLessThan(leftScreenPosition.x, rightScreenPosition.x, 
						 @"screen position calculation failed (markers %d, %d): left (%f, %f) right (%f, %f) mapped to left (%f, %f) right (%f, %f)",
						 j-1, j,
						 leftMarkerCoordinate.longitude, leftMarkerCoordinate.latitude,
						 rightMarkerCoordinate.longitude, rightMarkerCoordinate.latitude,
						 leftScreenPosition.x, leftScreenPosition.y, rightScreenPosition.x, rightScreenPosition.y);
		CLLocationCoordinate2D computedLatitudeLongitude = 
		[mangler latitudeLongitudeForMarker:leftMarker];
		STAssertEqualsWithAccuracy(leftMarkerCoordinate.longitude, computedLatitudeLongitude.longitude, kAccuracyThresholdForGeographicCoordinates,
								   @"round-trip computation of longitude failed %f %f",
								   leftMarkerCoordinate.longitude, computedLatitudeLongitude.longitude);
		STAssertEqualsWithAccuracy(leftMarkerCoordinate.latitude, computedLatitudeLongitude.latitude, kAccuracyThresholdForGeographicCoordinates,
								   @"round-trip computation of latitude failed %f %f",
								   leftMarkerCoordinate.latitude, computedLatitudeLongitude.latitude);
	}
	
}

- (void)testScreenCoordinatesPacificNorthwest
{
	[[mapView contents] setZoom: 10];
	CLLocationCoordinate2D coord = {45.5,-121};
	[mapView moveToCoordinate:coord];
	
	CGPoint point1 = [mapView coordinateToPixel:coord];
	
	coord.longitude -= .125;
	CGPoint point2 = [mapView coordinateToPixel:coord];
	
	coord.longitude -= .125;
	CGPoint point3 = [mapView coordinateToPixel:coord];
	
	STAssertEqualsWithAccuracy(point1.y, point2.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertEqualsWithAccuracy(point2.y, point3.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertLessThan(point3.x, point2.x,
					 @"X pixel coordinates should be increasing left to right");
	STAssertLessThan(point2.x, point1.x,
					 @"X pixel coordinates should be increasing left to right");
}

- (void)testScreenCoordinatesFarEast
{
	[[mapView contents] setZoom: 10];
	CLLocationCoordinate2D coord = {45.5,179.9};
	[mapView moveToCoordinate:coord];
	
	CGPoint point1 = [mapView coordinateToPixel:coord];
	
	coord.longitude += .125;
	CGPoint point2 = [mapView coordinateToPixel:coord];
	
	coord.longitude += .125;
	CGPoint point3 = [mapView coordinateToPixel:coord];
	
	STAssertEqualsWithAccuracy(point1.y, point2.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertEqualsWithAccuracy(point2.y, point3.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertLessThan(point1.x, point2.x,
					 @"X pixel coordinates should be increasing left to right");
	STAssertLessThan(point2.x, point3.x,
					 @"X pixel coordinates should be increasing left to right");
}

- (void)testScreenCoordinatesFarWest
{
	[[mapView contents] setZoom: 10];
	CLLocationCoordinate2D coord = {45.5,-179.9};
	[mapView moveToCoordinate:coord];
	
	CGPoint point1 = [mapView coordinateToPixel:coord];
	
	coord.longitude -= .125;
	CGPoint point2 = [mapView coordinateToPixel:coord];
	
	coord.longitude -= .125;
	CGPoint point3 = [mapView coordinateToPixel:coord];
	
	STAssertEqualsWithAccuracy(point1.y, point2.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertEqualsWithAccuracy(point2.y, point3.y, kAccuracyThresholdForPixelCoordinates,
							   @"Y pixel values should be equal");
	STAssertLessThan(point3.x, point2.x,
					 @"X pixel coordinates should be increasing left to right");
	STAssertLessThan(point2.x, point1.x,
					 @"X pixel coordinates should be increasing left to right");
}

- (void)testZoomBounds
{
	double contentsMaxZoom, tilesourceMaxZoom, contentsZoom;
	contentsMaxZoom = [[mapView contents] maxZoom];
	tilesourceMaxZoom = [[[mapView contents] tileSource] maxZoom];
	contentsZoom = [[mapView contents] zoom];
	STAssertLessThanOrEqual(contentsMaxZoom, tilesourceMaxZoom, @"map's maxZoom exceeds tilesource's maxZoom");
	STAssertLessThanOrEqual(contentsZoom, tilesourceMaxZoom, @"map's zoom exceeds tilesource's maxZoom");
	
	double targetZoom = tilesourceMaxZoom + 0.4;
	[[mapView contents] setZoom:targetZoom]; // try to exceed tilesource limit
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, @"map's zoom wrong after trying to exceed tilesource's maxZoom");
	
	targetZoom = tilesourceMaxZoom + 0.5;
	[[mapView contents] setZoom:targetZoom]; // try to exceed tilesource limit
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, @"map's zoom wrong after trying to exceed tilesource's maxZoom");
	
	targetZoom = tilesourceMaxZoom - 1.6;
	[[mapView contents] setZoom:targetZoom];
	CGPoint pivotPoint = CGPointMake(5., 5.);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom - 1.0, kAccuracyThreshold, 
							   @"map's zoom %f wrong after zoomInToNextNativeZoomAt: for maxZoom-1 %f",
							   contentsZoom, tilesourceMaxZoom);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, 
							   @"map's zoom [%f] wrong after zoomInToNextNativeZoomAt: for maxZoom %f (first)",
							   contentsZoom, tilesourceMaxZoom);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after zoomInToNextNativeZoomAt: for maxZoom %f (second)",
							   contentsZoom, tilesourceMaxZoom);
	
	
	targetZoom = tilesourceMaxZoom - 1.5;
	[[mapView contents] setZoom:targetZoom];
	pivotPoint = CGPointMake(5., 5.);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom - 1.0, kAccuracyThreshold, 
							   @"map's zoom %f wrong after zoomInToNextNativeZoomAt: for maxZoom-1 %f",
							   contentsZoom, tilesourceMaxZoom);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, 
							   @"map's zoom [%f] wrong after zoomInToNextNativeZoomAt: for maxZoom %f (first)",
							   contentsZoom, tilesourceMaxZoom);
	[[mapView contents] zoomInToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, tilesourceMaxZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after zoomInToNextNativeZoomAt: for maxZoom %f (second)",
							   contentsZoom, tilesourceMaxZoom);
	
	
	double contentsMinZoom = 3.0;
	[[mapView contents] setMinZoom:contentsMinZoom];
	targetZoom = contentsMinZoom + 0.6;	
	[[mapView contents] setZoom:targetZoom];
	contentsZoom = [[mapView contents] zoom];
	NSLog(@"zoom: %f minZoom: %f", contentsZoom, [[mapView contents] minZoom]);
	[[mapView contents] zoomOutToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, contentsMinZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after first zoomOutToNextNativeZoomAt: for minZoom %f",
							   contentsZoom, contentsMinZoom);
	[[mapView contents] zoomOutToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, contentsMinZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after second zoomOutToNextNativeZoomAt: for minZoom %f",
							   contentsZoom, contentsMinZoom);
	
	contentsMinZoom = 3.0;
	[[mapView contents] setMinZoom:contentsMinZoom];
	targetZoom = contentsMinZoom + 1.6;	
	[[mapView contents] setZoom:targetZoom];
	contentsZoom = [[mapView contents] zoom];
	NSLog(@"zoom: %f minZoom: %f", contentsZoom, [[mapView contents] minZoom]);
	[[mapView contents] zoomOutToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, contentsMinZoom+1.0, kAccuracyThreshold, 
							   @"map's zoom %f wrong after first zoomOutToNextNativeZoomAt: for minZoom %f",
							   contentsZoom, contentsMinZoom);
	[[mapView contents] zoomOutToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, contentsMinZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after second zoomOutToNextNativeZoomAt: for minZoom %f",
							   contentsZoom, contentsMinZoom);
	[[mapView contents] zoomOutToNextNativeZoomAt:pivotPoint];
	contentsZoom = [[mapView contents] zoom];
	STAssertEqualsWithAccuracy(contentsZoom, contentsMinZoom, kAccuracyThreshold, 
							   @"map's zoom %f wrong after second zoomOutToNextNativeZoomAt: for minZoom %f",
							   contentsZoom, contentsMinZoom);
	
	
}

@end