{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE TupleSections #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeFamilies #-}
module Brick.Widgets.Core
(
TextWidth(..)
, emptyWidget
, raw
, txt
, txtWrap
, txtWrapWith
, str
, strWrap
, strWrapWith
, fill
, hyperlink
, padLeft
, padRight
, padTop
, padBottom
, padLeftRight
, padTopBottom
, padAll
, (<=>)
, (<+>)
, hBox
, vBox
, hLimit
, hLimitPercent
, vLimit
, vLimitPercent
, setAvailableSize
, withDefAttr
, modifyDefAttr
, withAttr
, forceAttr
, overrideAttr
, updateAttrMap
, withBorderStyle
, joinBorders
, separateBorders
, freezeBorders
, showCursor
, Named(..)
, translateBy
, cropLeftBy
, cropRightBy
, cropTopBy
, cropBottomBy
, reportExtent
, clickable
, viewport
, visible
, visibleRegion
, unsafeLookupViewport
, cached
, addResultOffset
, cropToContext
)
where
#if !(MIN_VERSION_base(4,11,0))
import Data.Monoid ((<>))
#endif
#if !MIN_VERSION_base(4,8,0)
import Control.Applicative
import Data.Monoid ((<>), mempty)
#endif
import Lens.Micro ((^.), (.~), (&), (%~), to, _1, _2, each, to, Lens')
import Lens.Micro.Mtl (use, (%=))
import Control.Monad ((>=>),when)
import Control.Monad.Trans.State.Lazy
import Control.Monad.Trans.Reader
import Control.Monad.Trans.Class (lift)
import qualified Data.Foldable as F
import qualified Data.Text as T
import qualified Data.DList as DL
import qualified Data.Map as M
import qualified Data.Set as S
import qualified Data.IMap as I
import qualified Data.Function as DF
import Data.List (sortBy, partition)
import qualified Graphics.Vty as V
import Control.DeepSeq
import Text.Wrap (wrapTextToLines, WrapSettings, defaultWrapSettings)
import Brick.Types
import Brick.Types.Internal
import Brick.Widgets.Border.Style
import Brick.Util (clOffset, clamp)
import Brick.AttrMap
import Brick.Widgets.Internal
import qualified Brick.BorderMap as BM
class TextWidth a where
textWidth :: a -> Int
instance TextWidth T.Text where
textWidth = V.wcswidth . T.unpack
instance (F.Foldable f) => TextWidth (f Char) where
textWidth = V.wcswidth . F.toList
class Named a n where
getName :: a -> n
withBorderStyle :: BorderStyle -> Widget n -> Widget n
withBorderStyle bs p = Widget (hSize p) (vSize p) $ withReaderT (& ctxBorderStyleL .~ bs) (render p)
joinBorders :: Widget n -> Widget n
joinBorders p = Widget (hSize p) (vSize p) $ withReaderT (& ctxDynBordersL .~ True) (render p)
separateBorders :: Widget n -> Widget n
separateBorders p = Widget (hSize p) (vSize p) $ withReaderT (&ctxDynBordersL .~ False) (render p)
freezeBorders :: Widget n -> Widget n
freezeBorders p = Widget (hSize p) (vSize p) $ (bordersL .~ BM.empty) <$> render p
emptyWidget :: Widget n
emptyWidget = raw V.emptyImage
addResultOffset :: Location -> Result n -> Result n
addResultOffset off = addCursorOffset off .
addVisibilityOffset off .
addExtentOffset off .
addDynBorderOffset off
addVisibilityOffset :: Location -> Result n -> Result n
addVisibilityOffset off r = r & visibilityRequestsL.each.vrPositionL %~ (off <>)
addExtentOffset :: Location -> Result n -> Result n
addExtentOffset off r = r & extentsL.each %~ (\(Extent n l sz o) -> Extent n (off <> l) sz o)
addDynBorderOffset :: Location -> Result n -> Result n
addDynBorderOffset off r = r & bordersL %~ BM.translate off
reportExtent :: n -> Widget n -> Widget n
reportExtent n p =
Widget (hSize p) (vSize p) $ do
result <- render p
let ext = Extent n (Location (0, 0)) sz (Location (0, 0))
sz = ( result^.imageL.to V.imageWidth
, result^.imageL.to V.imageHeight
)
return $ result & extentsL %~ (ext:)
clickable :: n -> Widget n -> Widget n
clickable n p =
Widget (hSize p) (vSize p) $ do
clickableNamesL %= (n:)
render $ reportExtent n p
addCursorOffset :: Location -> Result n -> Result n
addCursorOffset off r =
let onlyVisible = filter isVisible
isVisible l = l^.locationColumnL >= 0 && l^.locationRowL >= 0
in r & cursorsL %~ (\cs -> onlyVisible $ (`clOffset` off) <$> cs)
unrestricted :: Int
unrestricted = 100000
takeColumns :: Int -> String -> String
takeColumns _ "" = ""
takeColumns numCols (c:cs) =
let w = V.safeWcwidth c
in if w == numCols
then [c]
else if w < numCols
then c : takeColumns (numCols - w) cs
else ""
strWrap :: String -> Widget n
strWrap = strWrapWith defaultWrapSettings
strWrapWith :: WrapSettings -> String -> Widget n
strWrapWith settings t = txtWrapWith settings $ T.pack t
safeTextWidth :: T.Text -> Int
safeTextWidth = V.safeWcswidth . T.unpack
txtWrap :: T.Text -> Widget n
txtWrap = txtWrapWith defaultWrapSettings
txtWrapWith :: WrapSettings -> T.Text -> Widget n
txtWrapWith settings s =
Widget Greedy Fixed $ do
c <- getContext
let theLines = fixEmpty <$> wrapTextToLines settings (c^.availWidthL) s
fixEmpty l | T.null l = " "
| otherwise = l
case force theLines of
[] -> return emptyResult
multiple ->
let maxLength = maximum $ safeTextWidth <$> multiple
padding = V.charFill (c^.attrL) ' ' (c^.availWidthL - maxLength) (length lineImgs)
lineImgs = lineImg <$> multiple
lineImg lStr = V.text' (c^.attrL)
(lStr <> T.replicate (maxLength - safeTextWidth lStr) " ")
in return $ emptyResult & imageL .~ (V.horizCat [V.vertCat lineImgs, padding])
str :: String -> Widget n
str s =
Widget Fixed Fixed $ do
c <- getContext
let theLines = fixEmpty <$> (dropUnused . lines) s
fixEmpty :: String -> String
fixEmpty [] = " "
fixEmpty l = l
dropUnused l = takeColumns (availWidth c) <$> take (availHeight c) l
case force theLines of
[] -> return emptyResult
[one] -> return $ emptyResult & imageL .~ (V.string (c^.attrL) one)
multiple ->
let maxLength = maximum $ V.safeWcswidth <$> multiple
lineImgs = lineImg <$> multiple
lineImg lStr = V.string (c^.attrL) (lStr ++ replicate (maxLength - V.safeWcswidth lStr) ' ')
in return $ emptyResult & imageL .~ (V.vertCat lineImgs)
txt :: T.Text -> Widget n
txt = str . T.unpack
hyperlink :: T.Text -> Widget n -> Widget n
hyperlink url p =
Widget (hSize p) (vSize p) $ do
c <- getContext
let attr = attrMapLookup (c^.ctxAttrNameL) (c^.ctxAttrMapL) `V.withURL` url
withReaderT (& ctxAttrMapL %~ setDefaultAttr attr) (render p)
padLeft :: Padding -> Widget n -> Widget n
padLeft padding p =
let (f, sz) = case padding of
Max -> (id, Greedy)
Pad i -> (hLimit i, hSize p)
in Widget sz (vSize p) $ do
c <- getContext
let lim = case padding of
Max -> c^.availWidthL
Pad i -> c^.availWidthL - i
result <- render $ hLimit lim p
render $ (f $ vLimit (result^.imageL.to V.imageHeight) $ fill ' ') <+>
(Widget Fixed Fixed $ return result)
padRight :: Padding -> Widget n -> Widget n
padRight padding p =
let (f, sz) = case padding of
Max -> (id, Greedy)
Pad i -> (hLimit i, hSize p)
in Widget sz (vSize p) $ do
c <- getContext
let lim = case padding of
Max -> c^.availWidthL
Pad i -> c^.availWidthL - i
result <- render $ hLimit lim p
render $ (Widget Fixed Fixed $ return result) <+>
(f $ vLimit (result^.imageL.to V.imageHeight) $ fill ' ')
padTop :: Padding -> Widget n -> Widget n
padTop padding p =
let (f, sz) = case padding of
Max -> (id, Greedy)
Pad i -> (vLimit i, vSize p)
in Widget (hSize p) sz $ do
c <- getContext
let lim = case padding of
Max -> c^.availHeightL
Pad i -> c^.availHeightL - i
result <- render $ vLimit lim p
render $ (f $ hLimit (result^.imageL.to V.imageWidth) $ fill ' ') <=>
(Widget Fixed Fixed $ return result)
padBottom :: Padding -> Widget n -> Widget n
padBottom padding p =
let (f, sz) = case padding of
Max -> (id, Greedy)
Pad i -> (vLimit i, vSize p)
in Widget (hSize p) sz $ do
c <- getContext
let lim = case padding of
Max -> c^.availHeightL
Pad i -> c^.availHeightL - i
result <- render $ vLimit lim p
render $ (Widget Fixed Fixed $ return result) <=>
(f $ hLimit (result^.imageL.to V.imageWidth) $ fill ' ')
padLeftRight :: Int -> Widget n -> Widget n
padLeftRight c w = padLeft (Pad c) $ padRight (Pad c) w
padTopBottom :: Int -> Widget n -> Widget n
padTopBottom r w = padTop (Pad r) $ padBottom (Pad r) w
padAll :: Int -> Widget n -> Widget n
padAll v w = padLeftRight v $ padTopBottom v w
fill :: Char -> Widget n
fill ch =
Widget Greedy Greedy $ do
c <- getContext
return $ emptyResult & imageL .~ (V.charFill (c^.attrL) ch (c^.availWidthL) (c^.availHeightL))
{-# NOINLINE vBox #-}
vBox :: [Widget n] -> Widget n
vBox [] = emptyWidget
vBox [a] = a
vBox pairs = renderBox vBoxRenderer pairs
{-# NOINLINE hBox #-}
hBox :: [Widget n] -> Widget n
hBox [] = emptyWidget
hBox [a] = a
hBox pairs = renderBox hBoxRenderer pairs
data BoxRenderer n =
BoxRenderer { contextPrimary :: Lens' Context Int
, contextSecondary :: Lens' Context Int
, imagePrimary :: V.Image -> Int
, imageSecondary :: V.Image -> Int
, limitPrimary :: Int -> Widget n -> Widget n
, limitSecondary :: Int -> Widget n -> Widget n
, primaryWidgetSize :: Widget n -> Size
, concatenatePrimary :: [V.Image] -> V.Image
, concatenateSecondary :: [V.Image] -> V.Image
, locationFromOffset :: Int -> Location
, padImageSecondary :: Int -> V.Image -> V.Attr -> V.Image
, loPrimary :: forall a. Lens' (Edges a) a
, hiPrimary :: forall a. Lens' (Edges a) a
, loSecondary :: forall a. Lens' (Edges a) a
, hiSecondary :: forall a. Lens' (Edges a) a
, locationFromPrimarySecondary :: Int -> Int -> Location
, splitLoPrimary :: Int -> V.Image -> V.Image
, splitHiPrimary :: Int -> V.Image -> V.Image
, splitLoSecondary :: Int -> V.Image -> V.Image
, splitHiSecondary :: Int -> V.Image -> V.Image
, lookupPrimary :: Int -> BM.BorderMap DynBorder -> I.IMap DynBorder
, insertSecondary :: Location -> I.Run DynBorder -> BM.BorderMap DynBorder -> BM.BorderMap DynBorder
}
vBoxRenderer :: BoxRenderer n
vBoxRenderer =
BoxRenderer { contextPrimary = availHeightL
, contextSecondary = availWidthL
, imagePrimary = V.imageHeight
, imageSecondary = V.imageWidth
, limitPrimary = vLimit
, limitSecondary = hLimit
, primaryWidgetSize = vSize
, concatenatePrimary = V.vertCat
, concatenateSecondary = V.horizCat
, locationFromOffset = Location . (0 ,)
, padImageSecondary = \amt img a ->
let p = V.charFill a ' ' amt (V.imageHeight img)
in V.horizCat [img, p]
, loPrimary = eTopL
, hiPrimary = eBottomL
, loSecondary = eLeftL
, hiSecondary = eRightL
, locationFromPrimarySecondary = \r c -> Location (c, r)
, splitLoPrimary = V.cropBottom
, splitHiPrimary = \n img -> V.cropTop (V.imageHeight img-n) img
, splitLoSecondary = V.cropRight
, splitHiSecondary = \n img -> V.cropLeft (V.imageWidth img-n) img
, lookupPrimary = BM.lookupRow
, insertSecondary = BM.insertH
}
hBoxRenderer :: BoxRenderer n
hBoxRenderer =
BoxRenderer { contextPrimary = availWidthL
, contextSecondary = availHeightL
, imagePrimary = V.imageWidth
, imageSecondary = V.imageHeight
, limitPrimary = hLimit
, limitSecondary = vLimit
, primaryWidgetSize = hSize
, concatenatePrimary = V.horizCat
, concatenateSecondary = V.vertCat
, locationFromOffset = Location . (, 0)
, padImageSecondary = \amt img a ->
let p = V.charFill a ' ' (V.imageWidth img) amt
in V.vertCat [img, p]
, loPrimary = eLeftL
, hiPrimary = eRightL
, loSecondary = eTopL
, hiSecondary = eBottomL
, locationFromPrimarySecondary = \c r -> Location (c, r)
, splitLoPrimary = V.cropRight
, splitHiPrimary = \n img -> V.cropLeft (V.imageWidth img-n) img
, splitLoSecondary = V.cropBottom
, splitHiSecondary = \n img -> V.cropTop (V.imageHeight img-n) img
, lookupPrimary = BM.lookupCol
, insertSecondary = BM.insertV
}
renderBox :: BoxRenderer n -> [Widget n] -> Widget n
renderBox br ws =
Widget (maximum $ hSize <$> ws) (maximum $ vSize <$> ws) $ do
c <- getContext
let pairsIndexed = zip [(0::Int)..] ws
(his, lows) = partition (\p -> (primaryWidgetSize br $ snd p) == Fixed)
pairsIndexed
let availPrimary = c^.(contextPrimary br)
availSecondary = c^.(contextSecondary br)
renderHis _ prev [] = return $ DL.toList prev
renderHis remainingPrimary prev ((i, prim):rest) = do
result <- render $ limitPrimary br remainingPrimary
$ limitSecondary br availSecondary
$ cropToContext prim
renderHis (remainingPrimary - (result^.imageL.(to $ imagePrimary br)))
(DL.snoc prev (i, result)) rest
renderedHis <- renderHis availPrimary DL.empty his
renderedLows <- case lows of
[] -> return []
ls -> do
let remainingPrimary = c^.(contextPrimary br) -
(sum $ (^._2.imageL.(to $ imagePrimary br)) <$> renderedHis)
primaryPerLow = remainingPrimary `div` length ls
rest = remainingPrimary - (primaryPerLow * length ls)
secondaryPerLow = c^.(contextSecondary br)
primaries = replicate rest (primaryPerLow + 1) <>
replicate (length ls - rest) primaryPerLow
let renderLow ((i, prim), pri) =
(i,) <$> (render $ limitPrimary br pri
$ limitSecondary br secondaryPerLow
$ cropToContext prim)
if remainingPrimary > 0 then mapM renderLow (zip ls primaries) else return []
let rendered = sortBy (compare `DF.on` fst) $ renderedHis ++ renderedLows
allResults = snd <$> rendered
allImages = (^.imageL) <$> allResults
allPrimaries = imagePrimary br <$> allImages
allTranslatedResults = (flip map) (zip [0..] allResults) $ \(i, result) ->
let off = locationFromOffset br offPrimary
offPrimary = sum $ take i allPrimaries
in addResultOffset off result
maxSecondary = maximum $ imageSecondary br <$> allImages
padImage img = padImageSecondary br (maxSecondary - imageSecondary br img)
img (c^.attrL)
(imageRewrites, newBorders) = catAllBorders br (borders <$> allTranslatedResults)
rewrittenImages = zipWith (rewriteImage br) imageRewrites allImages
paddedImages = padImage <$> rewrittenImages
cropResultToContext $ Result (concatenatePrimary br paddedImages)
(concat $ cursors <$> allTranslatedResults)
(concat $ visibilityRequests <$> allTranslatedResults)
(concat $ extents <$> allTranslatedResults)
newBorders
catDynBorder
:: Lens' (Edges BorderSegment) BorderSegment
-> Lens' (Edges BorderSegment) BorderSegment
-> DynBorder
-> DynBorder
-> Maybe DynBorder
catDynBorder towardsA towardsB a b
| dbStyle a == dbStyle b
&& dbAttr a == dbAttr b
&& a ^. dbSegmentsL.towardsB.bsAcceptL
&& b ^. dbSegmentsL.towardsA.bsOfferL
&& not (a ^. dbSegmentsL.towardsB.bsDrawL)
= Just (a & dbSegmentsL.towardsB.bsDrawL .~ True)
| otherwise = Nothing
catDynBorders
:: Lens' (Edges BorderSegment) BorderSegment
-> Lens' (Edges BorderSegment) BorderSegment
-> I.IMap DynBorder
-> I.IMap DynBorder
-> I.IMap DynBorder
catDynBorders towardsA towardsB am bm = I.mapMaybe id
$ I.intersectionWith (catDynBorder towardsA towardsB) am bm
catBorders
:: (border ~ BM.BorderMap DynBorder, rewrite ~ I.IMap V.Image)
=> BoxRenderer n -> border -> border -> ((rewrite, rewrite), border)
catBorders br r l = if lCoord + 1 == rCoord
then ((lRe, rRe), lr')
else ((I.empty, I.empty), lr)
where
lr = BM.expand (BM.coordinates r) l `BM.unsafeUnion`
BM.expand (BM.coordinates l) r
lr' = id
. mergeIMap lCoord lIMap'
. mergeIMap rCoord rIMap'
$ lr
lCoord = BM.coordinates l ^. hiPrimary br
rCoord = BM.coordinates r ^. loPrimary br
lIMap = lookupPrimary br lCoord l
rIMap = lookupPrimary br rCoord r
lIMap' = catDynBorders (loPrimary br) (hiPrimary br) lIMap rIMap
rIMap' = catDynBorders (hiPrimary br) (loPrimary br) rIMap lIMap
lRe = renderDynBorder <$> lIMap'
rRe = renderDynBorder <$> rIMap'
mergeIMap p imap bm = F.foldl'
(\bm' (s,v) -> insertSecondary br (locationFromPrimarySecondary br p s) v bm')
bm
(I.unsafeToAscList imap)
catAllBorders ::
BoxRenderer n ->
[BM.BorderMap DynBorder] ->
([(I.IMap V.Image, I.IMap V.Image)], BM.BorderMap DynBorder)
catAllBorders _ [] = ([], BM.empty)
catAllBorders br (bm:bms) = (zip ([I.empty]++los) (his++[I.empty]), bm') where
(rewrites, bm') = runState (traverse (state . catBorders br) bms) bm
(his, los) = unzip rewrites
rewriteEdge ::
(Int -> V.Image -> V.Image) ->
(Int -> V.Image -> V.Image) ->
([V.Image] -> V.Image) ->
I.IMap V.Image -> V.Image -> V.Image
rewriteEdge splitLo splitHi combine = (combine .) . go . offsets 0 . I.unsafeToAscList where
offsets _ [] = []
offsets n ((n', r):nrs) = (n'-n, r) : offsets (n'+I.len r) nrs
go [] old = [old]
go ((lo, I.Run len new):nrs) old
= [splitLo lo old]
++ replicate len new
++ go nrs (splitHi (lo+len) old)
rewriteImage :: BoxRenderer n -> (I.IMap V.Image, I.IMap V.Image) -> V.Image -> V.Image
rewriteImage br (loRewrite, hiRewrite) old = rewriteHi . rewriteLo $ old where
size = imagePrimary br old
go = rewriteEdge (splitLoSecondary br) (splitHiSecondary br) (concatenateSecondary br)
rewriteLo img
| I.null loRewrite = img
| otherwise = concatenatePrimary br
[ go loRewrite (splitLoPrimary br 1 img)
, splitHiPrimary br 1 img
]
rewriteHi img
| I.null hiRewrite = img
| otherwise = concatenatePrimary br
[ splitLoPrimary br (size-1) img
, go hiRewrite (splitHiPrimary br (size-1) img)
]
hLimit :: Int -> Widget n -> Widget n
hLimit w p =
Widget Fixed (vSize p) $
withReaderT (availWidthL %~ (min w)) $ render $ cropToContext p
hLimitPercent :: Int -> Widget n -> Widget n
hLimitPercent w' p =
Widget Fixed (vSize p) $ do
let w = clamp 0 100 w'
ctx <- getContext
let usableWidth = ctx^.availWidthL
widgetWidth = round (toRational usableWidth * (toRational w / 100))
withReaderT (availWidthL %~ (min widgetWidth)) $ render $ cropToContext p
vLimit :: Int -> Widget n -> Widget n
vLimit h p =
Widget (hSize p) Fixed $
withReaderT (availHeightL %~ (min h)) $ render $ cropToContext p
vLimitPercent :: Int -> Widget n -> Widget n
vLimitPercent h' p =
Widget (vSize p) Fixed $ do
let h = clamp 0 100 h'
ctx <- getContext
let usableHeight = ctx^.availHeightL
widgetHeight = round (toRational usableHeight * (toRational h / 100))
withReaderT (availHeightL %~ (min widgetHeight)) $ render $ cropToContext p
setAvailableSize :: (Int, Int) -> Widget n -> Widget n
setAvailableSize (w, h) p =
Widget Fixed Fixed $
withReaderT (\c -> c & availHeightL .~ h & availWidthL .~ w) $
render $ cropToContext p
withAttr :: AttrName -> Widget n -> Widget n
withAttr an p =
Widget (hSize p) (vSize p) $
withReaderT (& ctxAttrNameL .~ an) (render p)
modifyDefAttr :: (V.Attr -> V.Attr) -> Widget n -> Widget n
modifyDefAttr f p =
Widget (hSize p) (vSize p) $ do
c <- getContext
withReaderT (& ctxAttrMapL %~ (setDefaultAttr (f $ getDefaultAttr (c^.ctxAttrMapL)))) (render p)
withDefAttr :: AttrName -> Widget n -> Widget n
withDefAttr an p =
Widget (hSize p) (vSize p) $ do
c <- getContext
withReaderT (& ctxAttrMapL %~ (setDefaultAttr (attrMapLookup an (c^.ctxAttrMapL)))) (render p)
updateAttrMap :: (AttrMap -> AttrMap) -> Widget n -> Widget n
updateAttrMap f p =
Widget (hSize p) (vSize p) $
withReaderT (& ctxAttrMapL %~ f) (render p)
forceAttr :: AttrName -> Widget n -> Widget n
forceAttr an p =
Widget (hSize p) (vSize p) $ do
c <- getContext
withReaderT (& ctxAttrMapL .~ (forceAttrMap (attrMapLookup an (c^.ctxAttrMapL)))) (render p)
overrideAttr :: AttrName -> AttrName -> Widget n -> Widget n
overrideAttr targetName fromName =
updateAttrMap (mapAttrName fromName targetName)
raw :: V.Image -> Widget n
raw img = Widget Fixed Fixed $ return $ emptyResult & imageL .~ img
translateBy :: Location -> Widget n -> Widget n
translateBy off p =
Widget (hSize p) (vSize p) $ do
result <- render p
return $ addResultOffset off
$ result & imageL %~ (V.translate (off^.locationColumnL) (off^.locationRowL))
cropLeftBy :: Int -> Widget n -> Widget n
cropLeftBy cols p =
Widget (hSize p) (vSize p) $ do
result <- render p
let amt = V.imageWidth (result^.imageL) - cols
cropped img = if amt < 0 then V.emptyImage else V.cropLeft amt img
return $ addResultOffset (Location (-1 * cols, 0))
$ result & imageL %~ cropped
cropRightBy :: Int -> Widget n -> Widget n
cropRightBy cols p =
Widget (hSize p) (vSize p) $ do
result <- render p
let amt = V.imageWidth (result^.imageL) - cols
cropped img = if amt < 0 then V.emptyImage else V.cropRight amt img
return $ result & imageL %~ cropped
cropTopBy :: Int -> Widget n -> Widget n
cropTopBy rows p =
Widget (hSize p) (vSize p) $ do
result <- render p
let amt = V.imageHeight (result^.imageL) - rows
cropped img = if amt < 0 then V.emptyImage else V.cropTop amt img
return $ addResultOffset (Location (0, -1 * rows))
$ result & imageL %~ cropped
cropBottomBy :: Int -> Widget n -> Widget n
cropBottomBy rows p =
Widget (hSize p) (vSize p) $ do
result <- render p
let amt = V.imageHeight (result^.imageL) - rows
cropped img = if amt < 0 then V.emptyImage else V.cropBottom amt img
return $ result & imageL %~ cropped
showCursor :: n -> Location -> Widget n -> Widget n
showCursor n cloc p =
Widget (hSize p) (vSize p) $ do
result <- render p
return $ result & cursorsL %~ (CursorLocation cloc (Just n):)
hRelease :: Widget n -> Maybe (Widget n)
hRelease p =
case hSize p of
Fixed -> Just $ Widget Greedy (vSize p) $
withReaderT (& availWidthL .~ unrestricted) (render p)
Greedy -> Nothing
vRelease :: Widget n -> Maybe (Widget n)
vRelease p =
case vSize p of
Fixed -> Just $ Widget (hSize p) Greedy $
withReaderT (& availHeightL .~ unrestricted) (render p)
Greedy -> Nothing
cached :: (Ord n) => n -> Widget n -> Widget n
cached n w =
Widget (hSize w) (vSize w) $ do
result <- cacheLookup n
case result of
Just prevResult -> return prevResult
Nothing -> do
wResult <- render w
cacheUpdate n wResult
return wResult
cacheLookup :: (Ord n) => n -> RenderM n (Maybe (Result n))
cacheLookup n = do
cache <- lift $ gets (^.renderCacheL)
return $ M.lookup n cache
cacheUpdate :: (Ord n) => n -> Result n -> RenderM n ()
cacheUpdate n r = lift $ modify (& renderCacheL %~ M.insert n r)
viewport :: (Ord n, Show n)
=> n
-> ViewportType
-> Widget n
-> Widget n
viewport vpname typ p =
clickable vpname $ Widget Greedy Greedy $ do
let observeName :: (Ord n, Show n) => n -> RenderM n ()
observeName n = do
observed <- use observedNamesL
case S.member n observed of
False -> observedNamesL %= S.insert n
True ->
error $ "Error: while rendering the interface, the name " <> show n <>
" was seen more than once. You should ensure that all of the widgets " <>
"in each interface have unique name values. This means either " <>
"using a different name type or adding constructors to your " <>
"existing one and using those to name your widgets. For more " <>
"information, see the \"Resource Names\" section of the Brick User Guide."
observeName vpname
c <- getContext
let newVp = VP 0 0 newSize
newSize = (c^.availWidthL, c^.availHeightL)
doInsert (Just vp) = Just $ vp & vpSize .~ newSize
doInsert Nothing = Just newVp
lift $ modify (& viewportMapL %~ (M.alter doInsert vpname))
let release = case typ of
Vertical -> vRelease
Horizontal -> hRelease
Both -> vRelease >=> hRelease
released = case release p of
Just w -> w
Nothing -> case typ of
Vertical -> error $ "tried to embed an infinite-height " <>
"widget in vertical viewport " <> (show vpname)
Horizontal -> error $ "tried to embed an infinite-width " <>
"widget in horizontal viewport " <> (show vpname)
Both -> error $ "tried to embed an infinite-width or " <>
"infinite-height widget in 'Both' type " <>
"viewport " <> (show vpname)
initialResult <- render released
reqs <- lift $ gets $ (^.rsScrollRequestsL)
let relevantRequests = snd <$> filter (\(n, _) -> n == vpname) reqs
when (not $ null relevantRequests) $ do
mVp <- lift $ gets $ (^.viewportMapL.to (M.lookup vpname))
case mVp of
Nothing -> error $ "BUG: viewport: viewport name " <> show vpname <> " absent from viewport map"
Just vp -> do
let updatedVp = applyRequests relevantRequests vp
applyRequests [] v = v
applyRequests (rq:rqs) v =
case typ of
Horizontal -> scrollTo typ rq (initialResult^.imageL) $ applyRequests rqs v
Vertical -> scrollTo typ rq (initialResult^.imageL) $ applyRequests rqs v
Both -> scrollTo Horizontal rq (initialResult^.imageL) $
scrollTo Vertical rq (initialResult^.imageL) $
applyRequests rqs v
lift $ modify (& viewportMapL %~ (M.insert vpname updatedVp))
when (not $ null $ initialResult^.visibilityRequestsL) $ do
mVp <- lift $ gets $ (^.viewportMapL.to (M.lookup vpname))
case mVp of
Nothing -> error $ "BUG: viewport: viewport name " <> show vpname <> " absent from viewport map"
Just vp -> do
let rqs = initialResult^.visibilityRequestsL
updateVp vp' rq = case typ of
Both -> scrollToView Horizontal rq $ scrollToView Vertical rq vp'
Horizontal -> scrollToView typ rq vp'
Vertical -> scrollToView typ rq vp'
lift $ modify (& viewportMapL %~ (M.insert vpname $ foldl updateVp vp rqs))
mVp <- lift $ gets $ (^.viewportMapL.to (M.lookup vpname))
vp <- case mVp of
Nothing -> error $ "BUG: viewport: viewport name " <> show vpname <> " absent from viewport map"
Just v -> return v
let img = initialResult^.imageL
fixTop v = if V.imageHeight img < v^.vpSize._2
then v & vpTop .~ 0
else v
fixLeft v = if V.imageWidth img < v^.vpSize._1
then v & vpLeft .~ 0
else v
updateVp = case typ of
Both -> fixLeft . fixTop
Horizontal -> fixLeft
Vertical -> fixTop
lift $ modify (& viewportMapL %~ (M.insert vpname (updateVp vp)))
mVpFinal <- lift $ gets (M.lookup vpname . (^.viewportMapL))
vpFinal <- case mVpFinal of
Nothing -> error $ "BUG: viewport: viewport name " <> show vpname <> " absent from viewport map"
Just v -> return v
translated <- render $ translateBy (Location (-1 * vpFinal^.vpLeft, -1 * vpFinal^.vpTop))
$ Widget Fixed Fixed $ return initialResult
let translatedSize = ( translated^.imageL.to V.imageWidth
, translated^.imageL.to V.imageHeight
)
case translatedSize of
(0, 0) -> do
let spaceFill = V.charFill (c^.attrL) ' ' (c^.availWidthL) (c^.availHeightL)
return $ translated & imageL .~ spaceFill
& visibilityRequestsL .~ mempty
& extentsL .~ mempty
_ -> render $ cropToContext
$ padBottom Max
$ padRight Max
$ Widget Fixed Fixed
$ return $ translated & visibilityRequestsL .~ mempty
unsafeLookupViewport :: (Ord n) => n -> RenderM n (Maybe Viewport)
unsafeLookupViewport name = lift $ gets (M.lookup name . (^.viewportMapL))
scrollTo :: ViewportType -> ScrollRequest -> V.Image -> Viewport -> Viewport
scrollTo Both _ _ _ = error "BUG: called scrollTo on viewport type 'Both'"
scrollTo Vertical req img vp = vp & vpTop .~ newVStart
where
newVStart = clamp 0 (V.imageHeight img - vp^.vpSize._2) adjustedAmt
adjustedAmt = case req of
VScrollBy amt -> vp^.vpTop + amt
VScrollPage Up -> vp^.vpTop - vp^.vpSize._2
VScrollPage Down -> vp^.vpTop + vp^.vpSize._2
VScrollToBeginning -> 0
VScrollToEnd -> V.imageHeight img - vp^.vpSize._2
SetTop i -> i
_ -> vp^.vpTop
scrollTo Horizontal req img vp = vp & vpLeft .~ newHStart
where
newHStart = clamp 0 (V.imageWidth img - vp^.vpSize._1) adjustedAmt
adjustedAmt = case req of
HScrollBy amt -> vp^.vpLeft + amt
HScrollPage Up -> vp^.vpLeft - vp^.vpSize._1
HScrollPage Down -> vp^.vpLeft + vp^.vpSize._1
HScrollToBeginning -> 0
HScrollToEnd -> V.imageWidth img - vp^.vpSize._1
SetLeft i -> i
_ -> vp^.vpLeft
scrollToView :: ViewportType -> VisibilityRequest -> Viewport -> Viewport
scrollToView Both _ _ = error "BUG: called scrollToView on 'Both' type viewport"
scrollToView Vertical rq vp = vp & vpTop .~ newVStart
where
curStart = vp^.vpTop
curEnd = curStart + vp^.vpSize._2
reqStart = rq^.vrPositionL.locationRowL
reqEnd = rq^.vrPositionL.locationRowL + rq^.vrSizeL._2
newVStart :: Int
newVStart = if reqStart < vStartEndVisible
then reqStart
else vStartEndVisible
vStartEndVisible = if reqEnd < curEnd
then curStart
else curStart + (reqEnd - curEnd)
scrollToView Horizontal rq vp = vp & vpLeft .~ newHStart
where
curStart = vp^.vpLeft
curEnd = curStart + vp^.vpSize._1
reqStart = rq^.vrPositionL.locationColumnL
reqEnd = rq^.vrPositionL.locationColumnL + rq^.vrSizeL._1
newHStart :: Int
newHStart = if reqStart < hStartEndVisible
then reqStart
else hStartEndVisible
hStartEndVisible = if reqEnd < curEnd
then curStart
else curStart + (reqEnd - curEnd)
visible :: Widget n -> Widget n
visible p =
Widget (hSize p) (vSize p) $ do
result <- render p
let imageSize = ( result^.imageL.to V.imageWidth
, result^.imageL.to V.imageHeight
)
return $ if imageSize^._1 > 0 && imageSize^._2 > 0
then result & visibilityRequestsL %~ (VR (Location (0, 0)) imageSize :)
else result
visibleRegion :: Location -> V.DisplayRegion -> Widget n -> Widget n
visibleRegion vrloc sz p =
Widget (hSize p) (vSize p) $ do
result <- render p
return $ if sz^._1 > 0 && sz^._2 > 0
then result & visibilityRequestsL %~ (VR vrloc sz :)
else result
{-# NOINLINE (<+>) #-}
(<+>) :: Widget n
-> Widget n
-> Widget n
(<+>) a b = hBox [a, b]
{-# NOINLINE (<=>) #-}
(<=>) :: Widget n
-> Widget n
-> Widget n
(<=>) a b = vBox [a, b]
{-# RULES
"baseHbox" forall a b . a <+> b = hBox [a, b]
"hBox2" forall as bs . hBox [hBox as, hBox bs] = hBox (as ++ bs)
"hboxL" forall as b . hBox [hBox as, b] = hBox (as ++ [b])
"hboxR" forall a bs . hBox [a, hBox bs] = hBox (a : bs)
"baseVbox" forall a b . a <=> b = vBox [a, b]
"vBox2" forall as bs . vBox [vBox as, vBox bs] = vBox (as ++ bs)
"vboxL" forall as b . vBox [vBox as, b] = vBox (as ++ [b])
"vboxR" forall a bs . vBox [a, vBox bs] = vBox (a : bs)
#-}