{-# LANGUAGE EmptyDataDecls        #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE TypeFamilies          #-}

module FPNLA.Operations.BLAS.Strategies.DataTypes (
    DefSeq,
    CBindSeq,
    BlocksByCols,
    BlocksByRows,
    BlocksSeq,

    DefPar_ST,
    ColsPar_ST,
    StrassenPar_ST,
    BlocksPar_ST,

    DefPar_MP,
    ColsPar_MP,
    StrassenPar_MP,
    StrassenMPar_MP,
    BlocksPar_MP,

    DefPar_Acc,

    NullContext,
    BlockContext(getBlockDim),
    SqrBlockContext(getSqrBlockDim),
    StrassenContext(getStrassenLimit),

    newNullContext,
    newBlockContext,
    newSqrBlockContext,
    newStrassenContext,
) where

import           FPNLA.Operations.Parameters (StratCtx)

-- Contextos:
data NullContext = NullContext deriving (Show)
data BlockContext = BlockContext {
        getBlockDim :: (Int, Int)
    } deriving (Show)
data SqrBlockContext = SqrBlockContext {
        getSqrBlockDim :: Int
    } deriving (Show)
data StrassenContext = StrassenContext {
        getStrassenExponentialLimit :: Int,
        getStrassenLimit            :: Int
    } deriving (Show)

newNullContext :: NullContext
newNullContext = NullContext

newBlockContext :: Int -> Int -> BlockContext
newBlockContext m n = BlockContext { getBlockDim = (m,n) }

newSqrBlockContext :: Int -> SqrBlockContext
newSqrBlockContext d = SqrBlockContext { getSqrBlockDim = d }

newStrassenContext :: Int -> StrassenContext
newStrassenContext e = StrassenContext { getStrassenExponentialLimit = e, getStrassenLimit = 2 ^ e }

-- Estrategias:
data DefSeq
data CBindSeq
data BlocksSeq gs
data BlocksByCols gs ts
data BlocksByRows gs ts

-- Strategies
data DefPar_ST
data ColsPar_ST s
data StrassenPar_ST gs
data BlocksPar_ST gs

-- Monad-Par
data DefPar_MP
data ColsPar_MP s
data StrassenPar_MP gs
data StrassenMPar_MP gs
data BlocksPar_MP gs

data DefPar_Acc

type instance StratCtx DefSeq = NullContext
type instance StratCtx CBindSeq = NullContext
type instance StratCtx (BlocksByCols gs ts) = (SqrBlockContext, StratCtx gs, StratCtx ts)
type instance StratCtx (BlocksByRows gs ts) = (SqrBlockContext, StratCtx gs, StratCtx ts)
type instance StratCtx (BlocksSeq gs) = (BlockContext, StratCtx gs)

type instance StratCtx DefPar_ST = NullContext
type instance StratCtx (ColsPar_ST s) = StratCtx s
type instance StratCtx (StrassenPar_ST gs)= (StrassenContext, StratCtx gs)
type instance StratCtx (BlocksPar_ST gs) = (BlockContext, StratCtx gs)

type instance StratCtx DefPar_MP = NullContext
type instance StratCtx (ColsPar_MP s) = StratCtx s
type instance StratCtx (StrassenPar_MP gs)= (StrassenContext, StratCtx gs)
type instance StratCtx (StrassenMPar_MP gs)= (StrassenContext, StratCtx gs)
type instance StratCtx (BlocksPar_MP gs) = (BlockContext, StratCtx gs)

type instance StratCtx DefPar_Acc = NullContext