//
//  RMMercatorToScreenProjection.m
//
// Copyright (c) 2008-2009, Route-Me Contributors
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of source code must retain the above copyright notice, this
//   list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright notice,
//   this list of conditions and the following disclaimer in the documentation
//   and/or other materials provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.

#import "RMGlobalConstants.h"
#import "RMMercatorToScreenProjection.h"
#include "RMProjection.h"

@implementation RMMercatorToScreenProjection

@synthesize projection;
@synthesize origin;
@synthesize metersPerPixel;

- (void)deepCopy:(RMMercatorToScreenProjection *)copy
{
	screenBounds = copy.screenBounds;
	projection = copy.projection;
	metersPerPixel = copy.metersPerPixel;
	origin = copy.origin;
}

- (id)initFromProjection:(RMProjection *)aProjection toScreenBounds:(CGRect)aScreenBounds
{
	if (!(self = [super init]))
		return nil;

	screenBounds = aScreenBounds;
	projection = [aProjection retain];
	metersPerPixel = 1;

	return self;
}

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

// Deltas in screen coordinates.
- (RMProjectedPoint)moveProjectedPoint:(RMProjectedPoint)aPoint by:(CGSize)delta
{
	RMProjectedSize XYDelta = [self projectScreenSizeToProjectedSize:delta];
	aPoint.easting += XYDelta.width;
	aPoint.northing += XYDelta.height;
	aPoint = [projection wrapPointHorizontally:aPoint];
	return aPoint;
}

- (RMProjectedRect)moveProjectedRect:(RMProjectedRect)aRect by:(CGSize)delta
{
	aRect.origin = [self moveProjectedPoint:aRect.origin by:delta];
	return aRect;
}

- (RMProjectedPoint)zoomProjectedPoint:(RMProjectedPoint)aPoint byFactor:(float)factor near:(CGPoint)aPixelPoint
{
	RMProjectedPoint XYPivot = [self projectScreenPointToProjectedPoint:aPixelPoint];
	RMProjectedPoint result = RMScaleProjectedPointAboutPoint(aPoint, factor, XYPivot);
	result = [projection wrapPointHorizontally:result];
	return result;
}

- (RMProjectedRect)zoomProjectedRect:(RMProjectedRect)aRect byFactor:(float)factor near:(CGPoint)aPixelPoint
{
	RMProjectedPoint XYPivot = [self projectScreenPointToProjectedPoint:aPixelPoint];
	RMProjectedRect result = RMScaleProjectedRectAboutPoint(aRect, factor, XYPivot);
	result.origin = [projection wrapPointHorizontally:result.origin];
	return result;
}

- (void)moveScreenBy:(CGSize)delta
{
	// Reverse the delta - if the screen's contents moves left, the origin moves right.
	// It makes sense if you think about it long enough and squint your eyes a bit.

	delta.width = -delta.width;
	delta.height = -delta.height;
	origin = [self moveProjectedPoint:origin by:delta];
}

- (void)zoomScreenByFactor:(float)factor near:(CGPoint)aPixelPoint
{
	// The result of this function should be the same as this:
	//RMMercatorPoint test = [self zoomPoint:origin ByFactor:1.0f / factor Near:pivot];

	// First we move the origin to the pivot...
	origin.easting += aPixelPoint.x * metersPerPixel;
	origin.northing += (screenBounds.size.height - aPixelPoint.y) * metersPerPixel;

	// Then scale by 1/factor
	metersPerPixel /= factor;

	// Then translate back
	origin.easting -= aPixelPoint.x * metersPerPixel;
	origin.northing -= (screenBounds.size.height - aPixelPoint.y) * metersPerPixel;

	origin = [projection wrapPointHorizontally:origin];
}

- (void)zoomBy:(float)factor
{
	metersPerPixel *= factor;
}

/*
 This method returns the pixel point based on the currently displayed map view converted from a RMProjectedPoint.
 
 origin is the top left projected point currently displayed in the view.  The range of this value is based
 on the planetBounds.  planetBounds in turn is based on an RMProjection.  For example 
 look at +(RMProjection *)googleProjection to see range of values for planetBounds/origin.
 
 The tricky part is when the current map view contains the divider for horizontally wrapping maps.  
 
 Note: tested only with googleProjection
 */
- (CGPoint)projectProjectedPoint:(RMProjectedPoint)aPoint withMetersPerPixel:(float)aScale
{
	CGPoint	aPixelPoint = { 0, 0 };

	RMProjectedRect projectedScreenBounds;
	projectedScreenBounds.origin = origin;
	projectedScreenBounds.size.width = screenBounds.size.width * aScale;
	projectedScreenBounds.size.height = screenBounds.size.height * aScale;

	RMProjectedRect planetBounds = [projection planetBounds];
	RMProjectedPoint planetEndPoint = {
        planetBounds.origin.easting + planetBounds.size.width,
		planetBounds.origin.northing + planetBounds.size.height
    };

	// Normalize coordinate system so there is no negative values
	RMProjectedRect normalizedProjectedScreenBounds;
	normalizedProjectedScreenBounds.origin.easting = projectedScreenBounds.origin.easting + planetEndPoint.easting;
	normalizedProjectedScreenBounds.origin.northing = projectedScreenBounds.origin.northing + planetEndPoint.northing;
	normalizedProjectedScreenBounds.size = projectedScreenBounds.size;

	RMProjectedPoint normalizedProjectedPoint;
	normalizedProjectedPoint.easting = aPoint.easting + planetEndPoint.easting;
	normalizedProjectedPoint.northing = aPoint.northing + planetEndPoint.northing;

	double rightMostViewableEasting;

	// check if world wrap divider is contained in view
	if ((normalizedProjectedScreenBounds.origin.easting + normalizedProjectedScreenBounds.size.width) > planetBounds.size.width)
    {
		rightMostViewableEasting = projectedScreenBounds.size.width - (planetBounds.size.width - normalizedProjectedScreenBounds.origin.easting);

		// Check if Right of divider but on screen still
		if (normalizedProjectedPoint.easting <= rightMostViewableEasting) {
			aPixelPoint.x = (planetBounds.size.width + normalizedProjectedPoint.easting - normalizedProjectedScreenBounds.origin.easting) / aScale;
		} else {
			// everywhere else is left of divider
			aPixelPoint.x = (normalizedProjectedPoint.easting - normalizedProjectedScreenBounds.origin.easting) / aScale;
		}
	}
	else {
		// Divider not contained in view
		aPixelPoint.x = (normalizedProjectedPoint.easting - normalizedProjectedScreenBounds.origin.easting) / aScale;
	}

	aPixelPoint.y = screenBounds.size.height - (normalizedProjectedPoint.northing - normalizedProjectedScreenBounds.origin.northing) / aScale;

	return aPixelPoint;
}

- (CGPoint)projectProjectedPoint:(RMProjectedPoint)aPoint
{
	return [self projectProjectedPoint:aPoint withMetersPerPixel:metersPerPixel];
}

- (CGRect)projectProjectedRect:(RMProjectedRect)aRect
{
	CGRect aPixelRect;
	aPixelRect.origin = [self projectProjectedPoint:aRect.origin];
	aPixelRect.size.width = aRect.size.width / metersPerPixel;
	aPixelRect.size.height = aRect.size.height / metersPerPixel;
	return aPixelRect;
}

- (RMProjectedPoint)projectScreenPointToProjectedPoint:(CGPoint)aPixelPoint withMetersPerPixel:(float)aScale
{
	RMProjectedPoint aPoint;
	aPoint.easting = origin.easting + aPixelPoint.x * aScale;
	aPoint.northing = origin.northing + (screenBounds.size.height - aPixelPoint.y) * aScale;

	origin = [projection wrapPointHorizontally:origin];

	return aPoint;
}

- (RMProjectedPoint)projectScreenPointToProjectedPoint:(CGPoint)aPixelPoint
{
	// I will assume the point is within the screenbounds rectangle.
	return [projection wrapPointHorizontally:[self projectScreenPointToProjectedPoint:aPixelPoint withMetersPerPixel:metersPerPixel]];
}

- (RMProjectedRect)projectScreenRectToProjectedRect:(CGRect)aPixelRect
{
	RMProjectedRect aRect;
	aRect.origin = [self projectScreenPointToProjectedPoint: aPixelRect.origin];
	aRect.size.width = aPixelRect.size.width * metersPerPixel;
	aRect.size.height = aPixelRect.size.height * metersPerPixel;
	return aRect;
}

- (RMProjectedSize)projectScreenSizeToProjectedSize:(CGSize)aPixelSize
{
	RMProjectedSize aSize;
	aSize.width = aPixelSize.width * metersPerPixel;
	aSize.height = -aPixelSize.height * metersPerPixel;
	return aSize;
}

- (RMProjectedRect)projectedBounds
{
	RMProjectedRect aRect;
	aRect.origin = origin;
	aRect.size.width = screenBounds.size.width * metersPerPixel;
	aRect.size.height = screenBounds.size.height * metersPerPixel;
	return aRect;
}

- (void)setProjectedBounds:(RMProjectedRect)aRect
{
	float scaleX = aRect.size.width / screenBounds.size.width;
	float scaleY = aRect.size.height / screenBounds.size.height;

	// I will pick a scale in between those two.
	metersPerPixel = (scaleX + scaleY) / 2;
	origin = [projection wrapPointHorizontally:aRect.origin];
}

- (RMProjectedPoint)projectedCenter
{
	RMProjectedPoint aPoint;
	aPoint.easting = origin.easting + screenBounds.size.width * metersPerPixel / 2;
	aPoint.northing = origin.northing + screenBounds.size.height * metersPerPixel / 2;
	aPoint = [projection wrapPointHorizontally:aPoint];
	return aPoint;
}

- (void)setProjectedCenter:(RMProjectedPoint)aPoint
{
	origin = [projection wrapPointHorizontally:aPoint];
	origin.easting -= screenBounds.size.width * metersPerPixel / 2;
	origin.northing -= screenBounds.size.height * metersPerPixel / 2;
}

- (void)setScreenBounds:(CGRect)rect
{
    screenBounds = rect;
}

- (CGRect)screenBounds
{
	return screenBounds;
}

- (float)metersPerPixel
{
	return metersPerPixel;
}

- (void)setMetersPerPixel:(float)newMPP
{
	// We need to adjust the origin - since the origin
	// is in the corner, it will change when we change the scale.
	RMProjectedPoint center = [self projectedCenter];
	metersPerPixel = newMPP;
	[self setProjectedCenter:center];
}

@end