{- | Fortran complex data type.

The complex data type is a simple layer on top of reals. We reuse the type and
value representation from reals, but for convenience, we provide a newtype
wrapper to enable writing a 'FKinded' instance for the complex type.

TODO candidate for improving. other ways of writing, name is long & poor.
alternatively, could enforce usage of this
-}

{-# LANGUAGE DerivingVia #-}

module Language.Fortran.Repr.Type.Scalar.Complex where

import Language.Fortran.Repr.Type.Scalar.Common
import Language.Fortran.Repr.Type.Scalar.Real

import GHC.Generics ( Generic )
import Data.Data ( Data )
import Data.Binary ( Binary )
import Text.PrettyPrint.GenericPretty ( Out )

newtype FTComplexWrapper = FTComplexWrapper { FTComplexWrapper -> FTReal
unFTComplexWrapper :: FTReal }
    deriving stock (Int -> FTComplexWrapper -> ShowS
[FTComplexWrapper] -> ShowS
FTComplexWrapper -> String
(Int -> FTComplexWrapper -> ShowS)
-> (FTComplexWrapper -> String)
-> ([FTComplexWrapper] -> ShowS)
-> Show FTComplexWrapper
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> FTComplexWrapper -> ShowS
showsPrec :: Int -> FTComplexWrapper -> ShowS
$cshow :: FTComplexWrapper -> String
show :: FTComplexWrapper -> String
$cshowList :: [FTComplexWrapper] -> ShowS
showList :: [FTComplexWrapper] -> ShowS
Show, (forall x. FTComplexWrapper -> Rep FTComplexWrapper x)
-> (forall x. Rep FTComplexWrapper x -> FTComplexWrapper)
-> Generic FTComplexWrapper
forall x. Rep FTComplexWrapper x -> FTComplexWrapper
forall x. FTComplexWrapper -> Rep FTComplexWrapper x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cfrom :: forall x. FTComplexWrapper -> Rep FTComplexWrapper x
from :: forall x. FTComplexWrapper -> Rep FTComplexWrapper x
$cto :: forall x. Rep FTComplexWrapper x -> FTComplexWrapper
to :: forall x. Rep FTComplexWrapper x -> FTComplexWrapper
Generic, Typeable FTComplexWrapper
Typeable FTComplexWrapper =>
(forall (c :: * -> *).
 (forall d b. Data d => c (d -> b) -> d -> c b)
 -> (forall g. g -> c g) -> FTComplexWrapper -> c FTComplexWrapper)
-> (forall (c :: * -> *).
    (forall b r. Data b => c (b -> r) -> c r)
    -> (forall r. r -> c r) -> Constr -> c FTComplexWrapper)
-> (FTComplexWrapper -> Constr)
-> (FTComplexWrapper -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
    Typeable t =>
    (forall d. Data d => c (t d)) -> Maybe (c FTComplexWrapper))
-> (forall (t :: * -> * -> *) (c :: * -> *).
    Typeable t =>
    (forall d e. (Data d, Data e) => c (t d e))
    -> Maybe (c FTComplexWrapper))
-> ((forall b. Data b => b -> b)
    -> FTComplexWrapper -> FTComplexWrapper)
-> (forall r r'.
    (r -> r' -> r)
    -> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r)
-> (forall r r'.
    (r' -> r -> r)
    -> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r)
-> (forall u.
    (forall d. Data d => d -> u) -> FTComplexWrapper -> [u])
-> (forall u.
    Int -> (forall d. Data d => d -> u) -> FTComplexWrapper -> u)
-> (forall (m :: * -> *).
    Monad m =>
    (forall d. Data d => d -> m d)
    -> FTComplexWrapper -> m FTComplexWrapper)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d)
    -> FTComplexWrapper -> m FTComplexWrapper)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d)
    -> FTComplexWrapper -> m FTComplexWrapper)
-> Data FTComplexWrapper
FTComplexWrapper -> Constr
FTComplexWrapper -> DataType
(forall b. Data b => b -> b)
-> FTComplexWrapper -> FTComplexWrapper
forall a.
Typeable a =>
(forall (c :: * -> *).
 (forall d b. Data d => c (d -> b) -> d -> c b)
 -> (forall g. g -> c g) -> a -> c a)
-> (forall (c :: * -> *).
    (forall b r. Data b => c (b -> r) -> c r)
    -> (forall r. r -> c r) -> Constr -> c a)
-> (a -> Constr)
-> (a -> DataType)
-> (forall (t :: * -> *) (c :: * -> *).
    Typeable t =>
    (forall d. Data d => c (t d)) -> Maybe (c a))
-> (forall (t :: * -> * -> *) (c :: * -> *).
    Typeable t =>
    (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a))
-> ((forall b. Data b => b -> b) -> a -> a)
-> (forall r r'.
    (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall r r'.
    (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r)
-> (forall u. (forall d. Data d => d -> u) -> a -> [u])
-> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u)
-> (forall (m :: * -> *).
    Monad m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> (forall (m :: * -> *).
    MonadPlus m =>
    (forall d. Data d => d -> m d) -> a -> m a)
-> Data a
forall u.
Int -> (forall d. Data d => d -> u) -> FTComplexWrapper -> u
forall u. (forall d. Data d => d -> u) -> FTComplexWrapper -> [u]
forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c FTComplexWrapper
forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> FTComplexWrapper -> c FTComplexWrapper
forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c FTComplexWrapper)
forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e))
-> Maybe (c FTComplexWrapper)
$cgfoldl :: forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> FTComplexWrapper -> c FTComplexWrapper
gfoldl :: forall (c :: * -> *).
(forall d b. Data d => c (d -> b) -> d -> c b)
-> (forall g. g -> c g) -> FTComplexWrapper -> c FTComplexWrapper
$cgunfold :: forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c FTComplexWrapper
gunfold :: forall (c :: * -> *).
(forall b r. Data b => c (b -> r) -> c r)
-> (forall r. r -> c r) -> Constr -> c FTComplexWrapper
$ctoConstr :: FTComplexWrapper -> Constr
toConstr :: FTComplexWrapper -> Constr
$cdataTypeOf :: FTComplexWrapper -> DataType
dataTypeOf :: FTComplexWrapper -> DataType
$cdataCast1 :: forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c FTComplexWrapper)
dataCast1 :: forall (t :: * -> *) (c :: * -> *).
Typeable t =>
(forall d. Data d => c (t d)) -> Maybe (c FTComplexWrapper)
$cdataCast2 :: forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e))
-> Maybe (c FTComplexWrapper)
dataCast2 :: forall (t :: * -> * -> *) (c :: * -> *).
Typeable t =>
(forall d e. (Data d, Data e) => c (t d e))
-> Maybe (c FTComplexWrapper)
$cgmapT :: (forall b. Data b => b -> b)
-> FTComplexWrapper -> FTComplexWrapper
gmapT :: (forall b. Data b => b -> b)
-> FTComplexWrapper -> FTComplexWrapper
$cgmapQl :: forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
gmapQl :: forall r r'.
(r -> r' -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
$cgmapQr :: forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
gmapQr :: forall r r'.
(r' -> r -> r)
-> r -> (forall d. Data d => d -> r') -> FTComplexWrapper -> r
$cgmapQ :: forall u. (forall d. Data d => d -> u) -> FTComplexWrapper -> [u]
gmapQ :: forall u. (forall d. Data d => d -> u) -> FTComplexWrapper -> [u]
$cgmapQi :: forall u.
Int -> (forall d. Data d => d -> u) -> FTComplexWrapper -> u
gmapQi :: forall u.
Int -> (forall d. Data d => d -> u) -> FTComplexWrapper -> u
$cgmapM :: forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
gmapM :: forall (m :: * -> *).
Monad m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
$cgmapMp :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
gmapMp :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
$cgmapMo :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
gmapMo :: forall (m :: * -> *).
MonadPlus m =>
(forall d. Data d => d -> m d)
-> FTComplexWrapper -> m FTComplexWrapper
Data)
    deriving (Int -> FTComplexWrapper
FTComplexWrapper -> Int
FTComplexWrapper -> [FTComplexWrapper]
FTComplexWrapper -> FTComplexWrapper
FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
FTComplexWrapper
-> FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
(FTComplexWrapper -> FTComplexWrapper)
-> (FTComplexWrapper -> FTComplexWrapper)
-> (Int -> FTComplexWrapper)
-> (FTComplexWrapper -> Int)
-> (FTComplexWrapper -> [FTComplexWrapper])
-> (FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper])
-> (FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper])
-> (FTComplexWrapper
    -> FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper])
-> Enum FTComplexWrapper
forall a.
(a -> a)
-> (a -> a)
-> (Int -> a)
-> (a -> Int)
-> (a -> [a])
-> (a -> a -> [a])
-> (a -> a -> [a])
-> (a -> a -> a -> [a])
-> Enum a
$csucc :: FTComplexWrapper -> FTComplexWrapper
succ :: FTComplexWrapper -> FTComplexWrapper
$cpred :: FTComplexWrapper -> FTComplexWrapper
pred :: FTComplexWrapper -> FTComplexWrapper
$ctoEnum :: Int -> FTComplexWrapper
toEnum :: Int -> FTComplexWrapper
$cfromEnum :: FTComplexWrapper -> Int
fromEnum :: FTComplexWrapper -> Int
$cenumFrom :: FTComplexWrapper -> [FTComplexWrapper]
enumFrom :: FTComplexWrapper -> [FTComplexWrapper]
$cenumFromThen :: FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
enumFromThen :: FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
$cenumFromTo :: FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
enumFromTo :: FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
$cenumFromThenTo :: FTComplexWrapper
-> FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
enumFromThenTo :: FTComplexWrapper
-> FTComplexWrapper -> FTComplexWrapper -> [FTComplexWrapper]
Enum, FTComplexWrapper -> FTComplexWrapper -> Bool
(FTComplexWrapper -> FTComplexWrapper -> Bool)
-> (FTComplexWrapper -> FTComplexWrapper -> Bool)
-> Eq FTComplexWrapper
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: FTComplexWrapper -> FTComplexWrapper -> Bool
== :: FTComplexWrapper -> FTComplexWrapper -> Bool
$c/= :: FTComplexWrapper -> FTComplexWrapper -> Bool
/= :: FTComplexWrapper -> FTComplexWrapper -> Bool
Eq, Eq FTComplexWrapper
Eq FTComplexWrapper =>
(FTComplexWrapper -> FTComplexWrapper -> Ordering)
-> (FTComplexWrapper -> FTComplexWrapper -> Bool)
-> (FTComplexWrapper -> FTComplexWrapper -> Bool)
-> (FTComplexWrapper -> FTComplexWrapper -> Bool)
-> (FTComplexWrapper -> FTComplexWrapper -> Bool)
-> (FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper)
-> (FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper)
-> Ord FTComplexWrapper
FTComplexWrapper -> FTComplexWrapper -> Bool
FTComplexWrapper -> FTComplexWrapper -> Ordering
FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
$ccompare :: FTComplexWrapper -> FTComplexWrapper -> Ordering
compare :: FTComplexWrapper -> FTComplexWrapper -> Ordering
$c< :: FTComplexWrapper -> FTComplexWrapper -> Bool
< :: FTComplexWrapper -> FTComplexWrapper -> Bool
$c<= :: FTComplexWrapper -> FTComplexWrapper -> Bool
<= :: FTComplexWrapper -> FTComplexWrapper -> Bool
$c> :: FTComplexWrapper -> FTComplexWrapper -> Bool
> :: FTComplexWrapper -> FTComplexWrapper -> Bool
$c>= :: FTComplexWrapper -> FTComplexWrapper -> Bool
>= :: FTComplexWrapper -> FTComplexWrapper -> Bool
$cmax :: FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper
max :: FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper
$cmin :: FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper
min :: FTComplexWrapper -> FTComplexWrapper -> FTComplexWrapper
Ord) via FTReal
    deriving anyclass (Get FTComplexWrapper
[FTComplexWrapper] -> Put
FTComplexWrapper -> Put
(FTComplexWrapper -> Put)
-> Get FTComplexWrapper
-> ([FTComplexWrapper] -> Put)
-> Binary FTComplexWrapper
forall t. (t -> Put) -> Get t -> ([t] -> Put) -> Binary t
$cput :: FTComplexWrapper -> Put
put :: FTComplexWrapper -> Put
$cget :: Get FTComplexWrapper
get :: Get FTComplexWrapper
$cputList :: [FTComplexWrapper] -> Put
putList :: [FTComplexWrapper] -> Put
Binary, Int -> FTComplexWrapper -> Doc
[FTComplexWrapper] -> Doc
FTComplexWrapper -> Doc
(Int -> FTComplexWrapper -> Doc)
-> (FTComplexWrapper -> Doc)
-> ([FTComplexWrapper] -> Doc)
-> Out FTComplexWrapper
forall a. (Int -> a -> Doc) -> (a -> Doc) -> ([a] -> Doc) -> Out a
$cdocPrec :: Int -> FTComplexWrapper -> Doc
docPrec :: Int -> FTComplexWrapper -> Doc
$cdoc :: FTComplexWrapper -> Doc
doc :: FTComplexWrapper -> Doc
$cdocList :: [FTComplexWrapper] -> Doc
docList :: [FTComplexWrapper] -> Doc
Out)

instance FKind FTComplexWrapper where
    parseFKind :: FKindLit -> Maybe FTComplexWrapper
parseFKind = \case FKindLit
8  -> FTComplexWrapper -> Maybe FTComplexWrapper
forall a. a -> Maybe a
Just (FTComplexWrapper -> Maybe FTComplexWrapper)
-> FTComplexWrapper -> Maybe FTComplexWrapper
forall a b. (a -> b) -> a -> b
$ FTReal -> FTComplexWrapper
FTComplexWrapper FTReal
FTReal4
                       FKindLit
16 -> FTComplexWrapper -> Maybe FTComplexWrapper
forall a. a -> Maybe a
Just (FTComplexWrapper -> Maybe FTComplexWrapper)
-> FTComplexWrapper -> Maybe FTComplexWrapper
forall a b. (a -> b) -> a -> b
$ FTReal -> FTComplexWrapper
FTComplexWrapper FTReal
FTReal8
                       FKindLit
_ -> Maybe FTComplexWrapper
forall a. Maybe a
Nothing
    printFKind :: FTComplexWrapper -> FKindLit
printFKind = \case FTComplexWrapper FTReal
FTReal4 -> FKindLit
8
                       FTComplexWrapper FTReal
FTReal8 -> FKindLit
16