{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE TypeFamilies      #-}

module HaskellWorks.Data.Length
    ( Container(..)
    , Length(..)
    ) where

import Data.Int
import Data.Word
import HaskellWorks.Data.Container
import HaskellWorks.Data.Positioning
import Prelude                       hiding (length)

import qualified Data.ByteString      as BS
import qualified Data.List            as L
import qualified Data.Vector          as DV
import qualified Data.Vector.Storable as DVS

class Container v => Length v where
  length :: v -> Count
  end :: v -> Position
  end = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall v. Length v => v -> Word64
length
  {-# INLINE end #-}

instance Length [a] where
  length :: [a] -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (t :: * -> *) a. Foldable t => t a -> Int
L.length
  {-# INLINE length #-}

instance Length BS.ByteString where
  length :: ByteString -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. ByteString -> Int
BS.length
  {-# INLINE length #-}

instance Length (DV.Vector Word8) where
  length :: Vector Word8 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Word16) where
  length :: Vector Word16 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Word32) where
  length :: Vector Word32 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Word64) where
  length :: Vector Word64 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DVS.Vector Word8) where
  length :: Vector Word8 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Word16) where
  length :: Vector Word16 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Word32) where
  length :: Vector Word32 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Word64) where
  length :: Vector Word64 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DV.Vector Int8) where
  length :: Vector Int8 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Int16) where
  length :: Vector Int16 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Int32) where
  length :: Vector Int32 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DV.Vector Int64) where
  length :: Vector Int64 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Vector a -> Int
DV.length
  {-# INLINE length #-}

instance Length (DVS.Vector Int8) where
  length :: Vector Int8 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Int16) where
  length :: Vector Int16 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Int32) where
  length :: Vector Int32 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Int64) where
  length :: Vector Int64 -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}

instance Length (DVS.Vector Int) where
  length :: Vector Int -> Word64
length = forall a b. (Integral a, Num b) => a -> b
fromIntegral forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Storable a => Vector a -> Int
DVS.length
  {-# INLINE length #-}