{-# LANGUAGE OverloadedStrings #-}

-- |
-- Module      :  ELynx.Tree.Export.Nexus
-- Description :  Export trees to Nexus files
-- Copyright   :  2021 Dominik Schrempf
-- License     :  GPL-3
--
-- Maintainer  :  dominik.schrempf@gmail.com
-- Stability   :  unstable
-- Portability :  portable
--
-- Creation date: Tue Apr 28 20:24:19 2020.
module ELynx.Tree.Export.Nexus
  ( toNexusTrees,
  )
where

import qualified Data.ByteString.Lazy.Char8 as BL
import ELynx.Export.Nexus
import ELynx.Tree.Export.Newick
import ELynx.Tree.Length
import ELynx.Tree.Name
import ELynx.Tree.Rooted
import ELynx.Tree.Support

-- | Export a list of (NAME, TREE) to a Nexus file.
toNexusTrees ::
  (HasMaybeLength e, HasMaybeSupport e, HasName a) =>
  [(BL.ByteString, Tree e a)] ->
  BL.ByteString
toNexusTrees :: forall e a.
(HasMaybeLength e, HasMaybeSupport e, HasName a) =>
[(ByteString, Tree e a)] -> ByteString
toNexusTrees [(ByteString, Tree e a)]
ts = ByteString -> [ByteString] -> ByteString
toNexus ByteString
"TREES" (forall a b. (a -> b) -> [a] -> [b]
map forall e a.
(HasMaybeLength e, HasMaybeSupport e, HasName a) =>
(ByteString, Tree e a) -> ByteString
tree [(ByteString, Tree e a)]
ts)

tree :: (HasMaybeLength e, HasMaybeSupport e, HasName a) => (BL.ByteString, Tree e a) -> BL.ByteString
tree :: forall e a.
(HasMaybeLength e, HasMaybeSupport e, HasName a) =>
(ByteString, Tree e a) -> ByteString
tree (ByteString
n, Tree e a
t) = ByteString
"  TREE " forall a. Semigroup a => a -> a -> a
<> ByteString
n forall a. Semigroup a => a -> a -> a
<> ByteString
" = " forall a. Semigroup a => a -> a -> a
<> forall e a.
(HasMaybeLength e, HasMaybeSupport e, HasName a) =>
Tree e a -> ByteString
toNewick Tree e a
t