/**
 * Copyright (c) 2012 Muh Hon Cheng
 * Created by honcheng on 6/2/12.
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject
 * to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT
 * WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR
 * PURPOSE AND NONINFRINGEMENT. IN NO EVENT
 * SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR
 * IN CONNECTION WITH THE SOFTWARE OR
 * THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 * @author 		Muh Hon Cheng <honcheng@gmail.com>
 * @copyright	2012	Muh Hon Cheng
 * @version
 *
 */


#import "MultiFoldView.h"
#import "UIView+Screenshot.h"
#import <MapKit/MapKit.h>

@implementation MultiFoldView

#define FOLDVIEW_TAG 1000

- (id)initWithFrame:(CGRect)frame foldDirection:(FoldDirection)foldDirection folds:(int)folds pullFactor:(float)factor
{
    self = [super initWithFrame:frame];
    if (self) {
        
        _useOptimizedScreenshot = YES;
        _foldDirection = foldDirection;
        _numberOfFolds = folds;
        if (_numberOfFolds==1)
        {
            // no pull factor required if there is only one fold
            _pullFactor = 0;
        }
        else _pullFactor = factor;

        // create multiple FoldView next to each other
        for (int i=0; i<_numberOfFolds; i++)
        {
            if (_foldDirection==FoldDirectionHorizontalLeftToRight || _foldDirection==FoldDirectionHorizontalRightToLeft)
            {
                float foldWidth = frame.size.width/self.numberOfFolds;
                FoldView *foldView = [[FoldView alloc] initWithFrame:CGRectMake(foldWidth*i,0,foldWidth,frame.size.height) foldDirection:foldDirection];
                [foldView setTag:FOLDVIEW_TAG+i];
                [self addSubview:foldView];
            }
            else if (_foldDirection==FoldDirectionVertical)
            {
                float foldHeight = frame.size.height/self.numberOfFolds;
                FoldView *foldView = [[FoldView alloc] initWithFrame:CGRectMake(0,foldHeight*(self.numberOfFolds-i)-foldHeight,frame.size.width,foldHeight) foldDirection:foldDirection];
                [foldView setTag:FOLDVIEW_TAG+i];
                [self addSubview:foldView];
            }
        }
        [self setAutoresizesSubviews:YES];
    }
    return self;
}

- (void)setFrame:(CGRect)frame
{
    [super setFrame:frame];

    CGRect contentViewHolderFrame = [self.contentViewHolder frame];
    contentViewHolderFrame.size.height = frame.size.height;
    [self.contentViewHolder setFrame:contentViewHolderFrame];
    
    
    for (UIView *subview in self.subviews)
    {
        if ([subview isKindOfClass:[FoldView class]]) [subview removeFromSuperview];
    }
    
    for (int i=0; i<_numberOfFolds; i++)
    {
        if (_foldDirection==FoldDirectionHorizontalLeftToRight || _foldDirection==FoldDirectionHorizontalRightToLeft)
        {
            float foldWidth = frame.size.width/self.numberOfFolds;
            FoldView *foldView = [[FoldView alloc] initWithFrame:CGRectMake(foldWidth*i,0,foldWidth,frame.size.height) foldDirection:_foldDirection];
            [foldView setTag:FOLDVIEW_TAG+i];
            [self addSubview:foldView];
        }
        else if (_foldDirection==FoldDirectionVertical)
        {
            float foldHeight = frame.size.height/self.numberOfFolds;
            FoldView *foldView = [[FoldView alloc] initWithFrame:CGRectMake(0,foldHeight*(self.numberOfFolds-i)-foldHeight,frame.size.width,foldHeight) foldDirection:_foldDirection];
            [foldView setTag:FOLDVIEW_TAG+i];
            [self addSubview:foldView];
        }
    }
    
//    for (int i=0; i<_numberOfFolds; i++)
//    {
//        if (_foldDirection==FoldDirectionHorizontalLeftToRight || _foldDirection==FoldDirectionHorizontalRightToLeft)
//        {
//            float foldWidth = frame.size.width/self.numberOfFolds;
//            FoldView *foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+i];
//            [foldView setFrame:CGRectMake(foldWidth*i,0,foldWidth,frame.size.height)];
//        }
//        else if (_foldDirection==FoldDirectionVertical)
//        {
//            float foldHeight = frame.size.height/self.numberOfFolds;
//            FoldView *foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+i];
//            [foldView setFrame:CGRectMake(0,foldHeight*(self.numberOfFolds-i)-foldHeight,frame.size.width,foldHeight)];
//        }
//    }
}

- (id)initWithFrame:(CGRect)frame folds:(int)folds pullFactor:(float)factor
{
    return [self initWithFrame:frame foldDirection:FoldDirectionHorizontalRightToLeft folds:folds pullFactor:factor];
}

- (void)setContent:(UIView *)contentView
{
    if ([contentView isKindOfClass:NSClassFromString(@"MKMapView")]) _shouldTakeScreenshotBeforeUnfolding = YES;
    
    // set the content view
    self.contentViewHolder = [[UIView alloc] initWithFrame:CGRectMake(0,0,contentView.frame.size.width,contentView.frame.size.height)];
    //[self.contentView setFrame:CGRectMake(0,0,contentView.frame.size.width,contentView.frame.size.height)];
    [self.contentViewHolder setAutoresizingMask:UIViewAutoresizingFlexibleWidth|UIViewAutoresizingFlexibleHeight];
    // place content view below folds
    [self insertSubview:self.contentViewHolder atIndex:0];
    [self.contentViewHolder addSubview:contentView];
    // immediately take a screenshot of the content view to overlay in fold
    // if content view is a map view, screenshot will be a blank grid
    [self drawScreenshotOnFolds];
    [self.contentViewHolder setHidden:YES];
    
//    // set the content view
//    self.contentView = contentView;
//    [self.contentView setFrame:CGRectMake(0,0,contentView.frame.size.width,contentView.frame.size.height)];
//    [self.contentView setAutoresizingMask:UIViewAutoresizingFlexibleWidth|UIViewAutoresizingFlexibleWidth];
//    // place content view below folds
//    [self insertSubview:self.contentView atIndex:0];
//    // immediately take a screenshot of the content view to overlay in fold
//    // if content view is a map view, screenshot will be a blank grid
//    [self drawScreenshotOnFolds];
//    [self.contentView setHidden:YES];
}

- (void)drawScreenshotOnFolds
{
    UIImage *image = [self.contentViewHolder screenshotWithOptimization:self.useOptimizedScreenshot];
    [self setScreenshotImage:image];
    // get screenshot of content view, and splice the image to overlay in different folds
    
}

- (void)setScreenshotImage:(UIImage*)image
{
    if (self.foldDirection==FoldDirectionHorizontalLeftToRight || self.foldDirection==FoldDirectionHorizontalRightToLeft)
    {
        float foldWidth = image.size.width/self.numberOfFolds;
        
        for (int i=0; i<self.numberOfFolds; i++)
        {
            CGImageRef imageRef = CGImageCreateWithImageInRect([image CGImage], CGRectMake(foldWidth*i*image.scale, 0, foldWidth*image.scale, image.size.height*image.scale));
            if (imageRef)
            {
                UIImage *croppedImage = [UIImage imageWithCGImage:imageRef];
                CFRelease(imageRef);
                FoldView *foldView = nil;
                if (self.foldDirection==FoldDirectionHorizontalLeftToRight) {
                    foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG + (self.numberOfFolds - 1) - i];
                } else {
                    foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+i];
                }
                [foldView setImage:croppedImage];
            }
            
        }
    }
    else if (self.foldDirection==FoldDirectionVertical)
    {
        float foldHeight = image.size.height/self.numberOfFolds;
        for (int i=0; i<self.numberOfFolds; i++)
        {
            CGImageRef imageRef = CGImageCreateWithImageInRect([image CGImage], CGRectMake(0, foldHeight*(self.numberOfFolds-i-1)*image.scale, image.size.width*image.scale, foldHeight*image.scale));
            if (imageRef)
            {
                UIImage *croppedImage = [UIImage imageWithCGImage:imageRef];
                CFRelease(imageRef);
                FoldView *foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+i];
                [foldView setImage:croppedImage];
            }
        }
    }
}

// set fold states based on offset value
- (void)calculateFoldStateFromOffset:(float)offset
{
    CGFloat fraction = 0.0;
    if (self.foldDirection==FoldDirectionHorizontalRightToLeft || self.foldDirection==FoldDirectionHorizontalLeftToRight)
    {
        if (offset < 0)
            fraction = -1*offset/self.frame.size.width;
        else
            fraction = offset/self.frame.size.width;
    }
    else if (self.foldDirection==FoldDirectionVertical)
    {
        fraction = offset/self.frame.size.height;
    }

    if (_state==FoldStateClosed && fraction>0)
    {
        _state = FoldStateTransition;
        [self foldWillOpen];
    }
    else if (_state==FoldStateOpened && fraction<1)
    {
        _state = FoldStateTransition;
        [self foldWillClose];
    }
    else if (_state==FoldStateTransition)
    {
        if (fraction==0)
        {
            _state = FoldStateClosed;
            [self foldDidClosed];
        }
        else if (fraction>=1)
        {
            _state = FoldStateOpened;
            [self foldDidOpened];
        }
    }
}

// use the parent offset to calculate fraction
- (void)unfoldWithParentOffset:(float)offset
{
    [self calculateFoldStateFromOffset:offset];
    
    if (self.foldDirection==FoldDirectionHorizontalRightToLeft || self.foldDirection==FoldDirectionHorizontalLeftToRight)
    {
        float foldWidth = self.frame.size.width/self.numberOfFolds;
        CGFloat fraction;
        if (offset < 0) {
            if (offset<-1*(foldWidth+self.pullFactor*foldWidth))
            {
                offset = -1*(foldWidth+self.pullFactor*foldWidth);
            }
            fraction = offset /(-1*(foldWidth+self.pullFactor*foldWidth));
        } else {
            if (offset > (foldWidth+self.pullFactor*foldWidth))
            {
                offset = (foldWidth+self.pullFactor*foldWidth);
            }
            fraction = offset /(foldWidth+self.pullFactor*foldWidth);
        }
        //fraction = offset /(-1*(foldWidth+self.pullFactor*foldWidth));
        
        if (fraction < 0) fraction  = -1*fraction;//0;
        if (fraction > 1) fraction = 1;
        [self unfoldViewToFraction:fraction];
    }
    else if (self.foldDirection==FoldDirectionVertical)
    {
        float foldHeight = self.frame.size.height/self.numberOfFolds;
        
        if (offset>(foldHeight+self.pullFactor*foldHeight))
        {
            offset = (foldHeight+self.pullFactor*foldHeight);
        }
        
        CGFloat fraction = offset /(foldHeight+self.pullFactor*foldHeight);
        if (fraction < 0) fraction = -1*fraction;//0;
        if (fraction > 1) fraction = 1;
        
        [self unfoldViewToFraction:fraction];
    }
}

- (void)unfoldViewToFraction:(CGFloat)fraction
{
    // start the cascading effect of unfolding
    // with the first foldView with index FOLDVIEW_TAG+0
    FoldView *firstFoldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG];
    
    if (self.foldDirection==FoldDirectionHorizontalLeftToRight)
    {
        float offset = 0.0;
        if ([self.delegate respondsToSelector:@selector(displacementOfMultiFoldView:)])
        {
            offset = [self.delegate displacementOfMultiFoldView:self];
        }
        else
        {
            offset = self.superview.frame.origin.x;
        }
        if (offset<0) offset = -1*offset;
        [self unfoldView:firstFoldView toFraction:fraction withOffset:offset];
    }
    else [self unfoldView:firstFoldView toFraction:fraction withOffset:0];
}

- (void)unfoldView:(FoldView*)foldView toFraction:(CGFloat)fraction withOffset:(float)offset
{
    // unfold the subfold
    [foldView unfoldViewToFraction:fraction];
    
    if (self.foldDirection==FoldDirectionHorizontalLeftToRight || self.foldDirection==FoldDirectionHorizontalRightToLeft)
    {
        if (self.foldDirection==FoldDirectionHorizontalLeftToRight) {
            [foldView setFrame:CGRectMake(offset - 2*foldView.rightView.frame.size.width, 0, foldView.frame.size.width, foldView.frame.size.height)];
            
        }
        
        // check if there is another subfold beside this fold
        int index = [foldView tag] - FOLDVIEW_TAG;
        if (index < self.numberOfFolds-1)
        {
            FoldView *nextFoldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+index+1];
            // set the origin of the next foldView
            // set the origin of the next foldView
            if (self.foldDirection==FoldDirectionHorizontalLeftToRight) {
                [nextFoldView setFrame:CGRectMake(foldView.frame.origin.x - 2*nextFoldView.rightView.frame.size.width,0,nextFoldView.frame.size.width,nextFoldView.frame.size.height)];
            } else {
                [nextFoldView setFrame:CGRectMake(foldView.frame.origin.x + 2*foldView.leftView.frame.size.width,0,nextFoldView.frame.size.width,nextFoldView.frame.size.height)];
            }
            
            float foldWidth = self.frame.size.width/self.numberOfFolds;
            // calculate the offset between the right edge of the last subfold, and the edge of the screen
            // use this offset to readjust the fraction
            float displacement = 0.0;
            if ([self.delegate respondsToSelector:@selector(displacementOfMultiFoldView:)])
            {
                displacement = [self.delegate displacementOfMultiFoldView:self];
            }
            else
            {
                displacement = self.superview.frame.origin.x;
            }
            
            float x;
            if (self.foldDirection==FoldDirectionHorizontalLeftToRight) {
                //x = (foldView.frame.origin.x + (fraction * foldView.frame.size.width)) - 2*foldView.rightView.frame.size.width;
                x =  (foldView.frame.origin.x + (fraction * foldView.frame.size.width)) - 2*foldView.rightView.frame.size.width;
                //x = displacement - x;
                
            } else{
                x = displacement+foldView.frame.origin.x+2*foldView.leftView.frame.size.width;
                //x = self.superview.frame.origin.x+foldView.frame.origin.x+2*foldView.leftView.frame.size.width;
            }
            
            //float x = self.superview.frame.origin.x+foldView.frame.origin.x+2*foldView.leftView.frame.size.width;
            CGFloat adjustedFraction = 0;
            if (self.foldDirection==FoldDirectionHorizontalLeftToRight) {
                x = -x;
            }
            if (index+1==self.numberOfFolds-1)
            {
                // if this is the last fold, do not use the pull factor
                // so that the right edge of this subfold aligns with the right edge of the screen
                adjustedFraction = (-1*x)/(foldWidth);
            }
            else
            {
                // if this is not the last fold, use the pull factor
                adjustedFraction = (-1*x)/(foldWidth+self.pullFactor*foldWidth);
            }
            if (adjustedFraction < 0) adjustedFraction = 0;
            if (adjustedFraction > 1) adjustedFraction = 1;

            // unfold this foldView with the fraction
            // by calling the same function
            // this drills in to the next subfold in a cascading effect depending on the number of available folds
            
            //[self unfoldView:nextFoldView toFraction:adjustedFraction];
            //NSLog(@"%@ %f", foldView, foldView.frame.origin.x);
            [self unfoldView:nextFoldView toFraction:adjustedFraction withOffset:foldView.frame.origin.x];
        }
    }
    else if (self.foldDirection==FoldDirectionVertical)
    {
        // check if there is another subfold beside this fold
        int index = [foldView tag] - FOLDVIEW_TAG;
        
        if (index < self.numberOfFolds-1)
        {
            FoldView *nextFoldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+index+1];
            // set the origin of the next foldView
            [nextFoldView setFrame:CGRectMake(0,foldView.frame.origin.y - 2*foldView.bottomView.frame.size.height,nextFoldView.frame.size.width,nextFoldView.frame.size.height)];
            
            float foldHeight = self.frame.size.height/self.numberOfFolds;
            // calculate the offset between the right edge of the last subfold, and the edge of the screen
            // use this offset to readjust the fraction
            float displacement = 0.0;
            if ([self.delegate respondsToSelector:@selector(displacementOfMultiFoldView:)])
            {
                displacement = [self.delegate displacementOfMultiFoldView:self];
            }
            else
            {
                if ([self.superview isKindOfClass:[UIScrollView class]])
                {
                    displacement = -1*[(UIScrollView*)self.superview contentOffset].y;
                }
                else
                {
                    displacement = self.superview.frame.origin.y ;
                }
            }
            float y = displacement - 2*foldView.bottomView.frame.size.height;
            int _index = index - 1;
            while (_index>=0)
            {
                FoldView *prevFoldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+_index];
                y -= 2*prevFoldView.bottomView.frame.size.height;
                _index -= 1;
            }

            CGFloat adjustedFraction = 0;
            if (index+1==self.numberOfFolds-1)
            {
                // if this is the last fold, do not use the pull factor
                // so that the right edge of this subfold aligns with the right edge of the screen
                adjustedFraction = y/(foldHeight);
            }
            else
            {
                // if this is not the last fold, use the pull factor
                adjustedFraction = y/(foldHeight+self.pullFactor*foldHeight);
            }

            if (adjustedFraction < 0) adjustedFraction = 0;
            if (adjustedFraction > 1) adjustedFraction = 1;
            // unfold this foldView with the fraction
            // by calling the same function
            
            // this drills in to the next subfold in a cascading effect depending on the number of available folds
            [self unfoldView:nextFoldView toFraction:adjustedFraction withOffset:0];
        }
    }
    
}

// hide fold (when content view is visible) and show fold (when content view is hidden
- (void)showFolds:(BOOL)show
{
    for (int i=0; i<self.numberOfFolds; i++)
    {
        FoldView *foldView = (FoldView*)[self viewWithTag:FOLDVIEW_TAG+i];
        [foldView setHidden:!show];
    }
}

- (void)unfoldWithoutAnimation
{
    [self unfoldWithParentOffset:self.frame.size.width];
    [self foldDidOpened];
}

#pragma mark states

// when fold is completely opened, hide fold and show content view
- (void)foldDidOpened
{
    [self.contentViewHolder setHidden:NO];
    [self showFolds:NO];
}

// when fold is completely closed, hide content view and folds
- (void)foldDidClosed
{
    [self.contentViewHolder setHidden:YES];
    [self showFolds:YES];
}

// when fold is about to be opened, make sure content view is hidden, and show fold
- (void)foldWillOpen
{
    if (self.shouldTakeScreenshotBeforeUnfolding)
    {
        [self.contentViewHolder setHidden:NO];
        [self drawScreenshotOnFolds];
    }
    [self.contentViewHolder setHidden:YES];
    [self showFolds:YES];
}

// when fold is about to be closed, take a screenshot of the content view, hide it, and make sure fold is visible.
- (void)foldWillClose
{
    [self drawScreenshotOnFolds];
    [self.contentViewHolder setHidden:YES];
    [self showFolds:YES];
}



@end