/*  This code is based on Apple's "GeekGameBoard" sample code, version 1.0.

    http://developer.apple.com/samplecode/GeekGameBoard/

    Copyright © 2007 Apple Inc. Copyright © 2008 Jens Alfke. 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 CONTRI-

    BUTORS 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 "HexGrid.h"

#import "Game.h"

@implementation HexGrid

- (id) initWithRows: (unsigned)nRows columns: (unsigned)nColumns

            spacing: (CGSize)spacing

           position: (CGPoint)pos

{

    // Ignore given spacing.height; set it to make the hexagons regular.

    CGFloat capHeight = spacing.height / 2 * tan(M_PI/6);

    CGFloat side = spacing.height / 2 / cos(M_PI/6);

    spacing.height = side + capHeight;

    self = [super initWithRows: nRows columns: nColumns

                       spacing: spacing

                      position: pos];

    if( self ) {

        _capHeight = capHeight;

        _side = side;

        self.bounds = CGRectMake(-1, -1, 

                                 (nColumns+0.5)*spacing.width + 2,

                                 nRows*spacing.height+capHeight + 2);

        _cellClass = [Hex class];

    }

    return self;

}

- (id) initWithRows: (unsigned)nRows columns: (unsigned)nColumns

              frame: (CGRect)frame;

{

    // Compute the horizontal spacing:

    CGFloat s = floor(MIN( (frame.size.width -2.0)/nColumns,

                         (frame.size.height-2.0)/(nRows+0.5*tan(M_PI/6)) / (0.5*(tan(M_PI/6)+1/cos(M_PI/6))) ));

    self = [self initWithRows: nRows columns: nColumns

                      spacing: CGSizeMake(s,s)

                     position: frame.origin];

    if( self ) {

        // Center in frame:

        CGRect curFrame = self.frame;

        curFrame.origin.x = round( curFrame.origin.x + (frame.size.width - curFrame.size.width )/2.0f );

        curFrame.origin.y = round( curFrame.origin.y + (frame.size.height- curFrame.size.height)/2.0f );

        self.frame = curFrame;

    }

    return self;

}

- (void) dealloc

{

    CGPathRelease(_cellPath);

    [super dealloc];

}

- (void) addCellsInHexagon

{

    int size = _nRows - !(_nRows & 1);      // make it odd

    for( int row=0; row<_nRows; row++ ) {

        int n;                              // # of hexes remaining in this row

        if( row < size )

            n = size - abs(row-size/2);

        else

            n = 0;

        int c0 = floor(((int)_nRows+1-n -(row&1))/2.0);       // col of 1st remaining hex

        for( int col=0; col<_nColumns; col++ )

            if( col>=c0 && col<c0+n )

                [self addCellAtRow: row column: col];

    }

}

- (GridCell*) createCellAtRow: (unsigned)row column: (unsigned)col 

               suggestedFrame: (CGRect)frame

{

    // Overridden to stagger the odd-numbered rows

    BOOL stagger = (row & 1) != 0;

    if( _reversed && (_nRows & 1) )

        stagger = !stagger;

    if( stagger )

        frame.origin.x += _spacing.width/2;

    frame.size.height += _capHeight;

    return [super createCellAtRow: row column: col suggestedFrame: frame];

}

// Returns a hexagonal CG path defining a cell's outline. Used by cells when drawing & hit-testing.

- (CGPathRef) cellPath

{

    if( ! _cellPath ) {

        CGFloat x1 = _spacing.width/2;

        CGFloat x2 = _spacing.width;

        CGFloat y1 = _capHeight;

        CGFloat y2 = y1 + _side;

        CGFloat y3 = y2 + _capHeight;

        CGPoint p[6] = { {0,y1}, {x1,0}, {x2,y1}, {x2,y2}, {x1,y3}, {0,y2} };

        CGMutablePathRef path = CGPathCreateMutable();

        CGPathAddLines(path, NULL, p, 6);

        CGPathCloseSubpath(path);

        _cellPath = path;

    }

    return _cellPath;

}

@end

#pragma mark -

@implementation Hex

- (void) drawInParentContext: (CGContextRef)ctx fill: (BOOL)fill

{

    CGContextSaveGState(ctx);

    CGPoint pos = self.position;

    CGContextTranslateCTM(ctx, pos.x, pos.y);

    CGContextBeginPath(ctx);

    CGContextAddPath(ctx, [(HexGrid*)_grid cellPath]);

    CGContextDrawPath(ctx, (fill ?kCGPathFill :kCGPathStroke));

    if( !fill && self.highlighted ) {

        // Highlight by drawing my outline in the highlight color:

        CGContextSetStrokeColorWithColor(ctx, self.borderColor);

        CGContextSetLineWidth(ctx,6);

        CGContextBeginPath(ctx);

        CGContextAddPath(ctx, [(HexGrid*)_grid cellPath]);

        CGContextDrawPath(ctx, kCGPathStroke);

    }

    CGContextRestoreGState(ctx);

}

- (BOOL)containsPoint:(CGPoint)p

{

    return [super containsPoint: p]

        && CGPathContainsPoint( [(HexGrid*)_grid cellPath], NULL, p, NO );

}

- (void) setHighlighted: (BOOL)highlighted

{

    if( highlighted != self.highlighted ) {

        [super setHighlighted: highlighted];

        [_grid setNeedsDisplay];        // So I'll be asked to redraw myself

    }

}

#pragma mark -

#pragma mark NEIGHBORS:

- (NSArray*) neighbors

{

    NSMutableArray *neighbors = [NSMutableArray arrayWithCapacity: 6];

    Hex* n[6] = {self.nw, self.ne, self.w, self.e, self.sw, self.se};

    for( int i=0; i<6; i++ )

        if( n[i] )

            [neighbors addObject: n[i]];

    return neighbors;

}

- (Hex*) nw     {return (Hex*)[_grid cellAtRow: _row+1 column: _column - ((_row+1)&1)];}

- (Hex*) ne     {return (Hex*)[_grid cellAtRow: _row+1 column: _column + (_row&1)];}

- (Hex*) e      {return (Hex*)[_grid cellAtRow: _row   column: _column + 1];}

- (Hex*) se     {return (Hex*)[_grid cellAtRow: _row-1 column: _column + (_row&1)];}

- (Hex*) sw     {return (Hex*)[_grid cellAtRow: _row-1 column: _column - ((_row-1)&1)];}

- (Hex*) w      {return (Hex*)[_grid cellAtRow: _row   column: _column - 1];}

// Directions relative to the current player:

- (Hex*) fl     {return self.fwdIsN ?self.nw :self.se;}

- (Hex*) fr     {return self.fwdIsN ?self.ne :self.sw;}

- (Hex*) r      {return self.fwdIsN ?self.e  :self.w;}

- (Hex*) br     {return self.fwdIsN ?self.se :self.nw;}

- (Hex*) bl     {return self.fwdIsN ?self.sw :self.ne;}

- (Hex*) l      {return self.fwdIsN ?self.w  :self.e;}

@end