-----------------------------------------------------------------------------
-- |
-- Module     : Data.Matrix.Banded.Class.Views
-- Copyright  : Copyright (c) , Patrick Perry <[email protected]>
-- License    : BSD3
-- Maintainer : Patrick Perry <[email protected]>
-- Stability  : experimental
--

module Data.Matrix.Banded.Class.Views (
    -- * Row and column views
    diagViewBanded,
    rowViewBanded,
    colViewBanded,
    
    unsafeDiagViewBanded,
    unsafeRowViewBanded,
    unsafeColViewBanded,
    
    ) where

import BLAS.Internal( checkedRow, checkedCol, checkedDiag, diagStart, diagLen )

import Data.Matrix.Banded.Class.Internal
import Data.Vector.Dense.Class
import Foreign

diagViewBanded :: (BaseBanded a x, Storable e) => 
    a mn e -> Int -> x k e
diagViewBanded a = checkedDiag (shape a) (unsafeDiagViewBanded a) 

rowViewBanded :: (BaseBanded a x, Storable e) => 
    a mn e -> Int -> (Int, x k e, Int)
rowViewBanded a = checkedRow (shape a) (unsafeRowViewBanded a) 

colViewBanded :: (BaseBanded a x, Storable e) => 
    a mn e -> Int -> (Int, x k e, Int)
colViewBanded a = checkedCol (shape a) (unsafeColViewBanded a)

unsafeDiagViewBanded :: (BaseBanded a x, Storable e) => 
    a mn e -> Int -> x k e
unsafeDiagViewBanded a d
    | isHermBanded a = conj $ unsafeDiagViewBanded a' (negate d)
    | otherwise =
        let (fp,p,m,n,_,_,ld,_) = arrayFromBanded a
            off = indexOfBanded a (diagStart d)
            p'  = p `advancePtr` off
            len = diagLen (m,n) d
            inc = ld
            c   = False
        in vectorViewArray fp p' len inc c
  where
    a' = (hermBanded . coerceBanded) a