{-
	Copyright (C) 2018 Dr. Alistair Ward

	This file is part of BishBosh.

	BishBosh is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	BishBosh is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with BishBosh.  If not, see <http://www.gnu.org/licenses/>.
-}
{- |
 [@AUTHOR@]	Dr. Alistair Ward

 [@DESCRIPTION@]

-}

module BishBosh.Property.Arboreal(
-- * Type-classes
	Prunable(..),
-- * Types
-- ** Type-synonyms
	Depth
) where

import qualified	Data.Tree

-- | A distance down the tree.
type Depth	= Int

-- | An interface which tree-like data can support.
class Prunable tree where
	prune	:: Depth -> tree -> tree	-- ^ Remove branches after the specified depth.

instance Prunable (Data.Tree.Tree a) where
	prune :: Depth -> Tree a -> Tree a
prune Depth
depth tree :: Tree a
tree@Data.Tree.Node { subForest :: forall a. Tree a -> Forest a
Data.Tree.subForest = Forest a
forest }
		| Depth
depth Depth -> Depth -> Bool
forall a. Ord a => a -> a -> Bool
< Depth
0	= [Char] -> Tree a
forall a. HasCallStack => [Char] -> a
error [Char]
"BishBosh.Property.Arboreal.prune:\tdepth can't be negative."
		| Bool
otherwise	= Tree a
tree {
			subForest :: Forest a
Data.Tree.subForest	= if Depth
depth Depth -> Depth -> Bool
forall a. Eq a => a -> a -> Bool
== Depth
0
				then []
				else (Tree a -> Tree a) -> Forest a -> Forest a
forall a b. (a -> b) -> [a] -> [b]
map (
					Depth -> Tree a -> Tree a
forall tree. Prunable tree => Depth -> tree -> tree
prune (Depth -> Tree a -> Tree a) -> Depth -> Tree a -> Tree a
forall a b. (a -> b) -> a -> b
$ Depth -> Depth
forall a. Enum a => a -> a
pred Depth
depth	-- Recurse.
				) Forest a
forest
		}