/*
* Copyright (C) 2015 - 2016, Daniel Dahan and CosmicMind, Inc. <http://cosmicmind.io>.
* 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.
*
*	*	Neither the name of CosmicMind nor the names of its
*		contributors may be used to endorse or promote products derived from
*		this software without specific prior written permission.
*
* 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 HOLDER 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 UIKit

public enum GridAxisDirection {
	case None
	case Horizontal
	case Vertical
}

public class GridAxis {
	/// Grid reference.
	unowned var grid: Grid
	
	/// Inherit grid rows and columns.
	public var inherited: Bool = false
	
	/// The direction the grid layouts its views out.
	public var direction: GridAxisDirection = .Horizontal
	
	/// The rows size.
	public var rows: Int {
		didSet {
			grid.reloadLayout()
		}
	}
	
	/// The columns size.
	public var columns: Int {
		didSet {
			grid.reloadLayout()
		}
	}
	
	/**
	Initializer.
	- Parameter grid: The Grid reference used for offset values.
	- Parameter rows: The number of rows, Vertical axis the grid will use.
	- Parameter columns: The number of columns, Horizontal axis the grid will use.
	*/
	public init(grid: Grid, rows: Int = 12, columns: Int = 12) {
		self.grid = grid
		self.rows = rows
		self.columns = columns
	}
}

public class GridOffset {
	/// Grid reference.
	unowned var grid: Grid
	
	/// The rows size.
	public var rows: Int {
		didSet {
			grid.reloadLayout()
		}
	}
	
	/// The columns size.
	public var columns: Int {
		didSet {
			grid.reloadLayout()
		}
	}
	
	/**
	Initializer.
	- Parameter grid: The Grid reference used for offset values.
	- Parameter rows: The number of rows, Vertical axis the grid will use.
	- Parameter columns: The number of columns, Horizontal axis the grid will use.
	*/
	public init(grid: Grid, rows: Int = 0, columns: Int = 0) {
		self.grid = grid
		self.rows = rows
		self.columns = columns
	}
}

public class Grid {
	/// The rows size.
	public var rows: Int {
		didSet {
			reloadLayout()
		}
	}
	
	/// The columns size.
	public var columns: Int {
		didSet {
			reloadLayout()
		}
	}
	
	/// Offsets for rows and columns.
	public private(set) var offset: GridOffset!
	
	/// The axis in which the Grid is laying out its views.
	public private(set) var axis: GridAxis!
	
	/// Preset inset value for grid.
	public var layoutInsetPreset: MaterialEdgeInset = .None {
		didSet {
			layoutInset = MaterialEdgeInsetToValue(contentInsetPreset)
		}
	}
	
	/// Insets value for grid.
	public var layoutInset: UIEdgeInsets = MaterialEdgeInsetToValue(.None) {
		didSet {
			reloadLayout()
		}
	}
	
	/// Preset inset value for grid.
	public var contentInsetPreset: MaterialEdgeInset = .None {
		didSet {
			contentInset = MaterialEdgeInsetToValue(contentInsetPreset)
		}
	}
	
	/// Insets value for grid.
	public var contentInset: UIEdgeInsets = MaterialEdgeInsetToValue(.None) {
		didSet {
			reloadLayout()
		}
	}
	
	/// A preset wrapper around interimSpace.
	public var interimSpacePreset: InterimSpace = .none {
		didSet {
			interimSpace = InterimSpaceToValue(interimSpacePreset)
		}
	}
	
	/// The space between grid columnss.
	public var interimSpace: CGFloat {
		didSet {
			reloadLayout()
		}
	}
	
	/// An Array of UIButtons.
	public var views: Array<UIView>? {
		didSet {
			reloadLayout()
		}
	}
	
	/**
	Initializer.
	- Parameter rows: The number of rows, Vertical axis the grid will use.
	- Parameter columns: The number of columns, Horizontal axis the grid will use.
	- Parameter interimSpace: The interimSpace between rows or columns.
	*/
	public init(rows: Int = 12, columns: Int = 12, interimSpace: CGFloat = 0) {
		self.rows = rows
		self.columns = columns
		self.interimSpace = interimSpace
		offset = GridOffset(grid: self)
		axis = GridAxis(grid: self)
	}
	
	/// Reload the button layout.
	public func reloadLayout() {
		if let v: Array<UIView> = views {
			let gc: Int = axis.inherited ? columns : axis.columns
			let gr: Int = axis.inherited ? rows : axis.rows
			var n: Int = 0
			for i in 0..<v.count {
				let view: UIView = v[i]
				if let sv: UIView = view.superview {
					sv.layoutIfNeeded()
					switch axis.direction {
					case .Horizontal:
						let w: CGFloat = (sv.bounds.width - contentInset.left - contentInset.right - layoutInset.left - layoutInset.right + interimSpace) / CGFloat(gc)
						let c: Int = view.grid.columns
						let co: Int = view.grid.offset.columns
						let vh: CGFloat = sv.bounds.height - contentInset.top - contentInset.bottom - layoutInset.top - layoutInset.bottom
						let vl: CGFloat = CGFloat(i + n + co) * w + contentInset.left + layoutInset.left
						let vw: CGFloat = w * CGFloat(c) - interimSpace
						view.frame = CGRectMake(vl, contentInset.top + layoutInset.top, vw, vh)
						n += c + co - 1
					case .Vertical:
						let h: CGFloat = (sv.bounds.height - contentInset.top - contentInset.bottom - layoutInset.top - layoutInset.bottom + interimSpace) / CGFloat(gr)
						let r: Int = view.grid.rows
						let ro: Int = view.grid.offset.rows
						let vw: CGFloat = sv.bounds.width - contentInset.left - contentInset.right - layoutInset.left - layoutInset.right
						let vt: CGFloat = CGFloat(i + n + ro) * h + contentInset.top + layoutInset.top
						let vh: CGFloat = h * CGFloat(r) - interimSpace
						view.frame = CGRectMake(contentInset.left + layoutInset.left, vt, vw, vh)
						n += r + ro - 1
					case .None:
						let w: CGFloat = (sv.bounds.width - contentInset.left - contentInset.right - layoutInset.left - layoutInset.right + interimSpace) / CGFloat(gc)
						let c: Int = view.grid.columns
						let co: Int = view.grid.offset.columns
						let h: CGFloat = (sv.bounds.height - contentInset.top - contentInset.bottom - layoutInset.top - layoutInset.bottom + interimSpace) / CGFloat(gr)
						let r: Int = view.grid.rows
						let ro: Int = view.grid.offset.rows
						let vt: CGFloat = CGFloat(ro) * h + contentInset.top + layoutInset.top
						let vl: CGFloat = CGFloat(co) * w + contentInset.left + layoutInset.left
						let vh: CGFloat = h * CGFloat(r) - interimSpace
						let vw: CGFloat = w * CGFloat(c) - interimSpace
						view.frame = CGRectMake(vl, vt, vw, vh)
					}
				}
			}
		}
	}
}

/// A memory reference to the Grid instance for UIView extensions.
private var GridKey: UInt8 = 0

/// Grid extension for UIView.
public extension UIView {
	/// Grid reference.
	public private(set) var grid: Grid {
		get {
			return associatedObject(self, key: &GridKey) {
				return Grid()
			}
		}
		set(value) {
			associateObject(self, key: &GridKey, value: value)
		}
	}
}
