Skip to content

Commit

Permalink
make it into a playground
Browse files Browse the repository at this point in the history
  • Loading branch information
@BrianHicks
BrianHicks committed Jan 10, 2020
1 parent 633c5c1 commit 2a688ce
Show file tree
Hide file tree
Showing 3 changed files with 155 additions and 87 deletions.
30 changes: 29 additions & 1 deletion src/Image.elm
View file
@@ -1,4 +1,4 @@
module Image exposing (Image, bars, recurse, view, viewColor)
module Image exposing (Image, bars, recurse, view, viewColor, waves)

import Array exposing (Array)
import Color.Transparent as Color exposing (Color)
Expand Down Expand Up @@ -101,3 +101,31 @@ bars =

Err problem ->
Debug.todo (Debug.toString problem)


waves : Image
waves =
let
l =
Color.fromColor Color.opaque Tango.skyBlue1

d =
Color.fromColor Color.opaque Tango.skyBlue3
in
case
Grid.fromRowsAndColumns
[ [ d, d, d, d, d, d, d, l ]
, [ l, d, d, d, d, d, d, d ]
, [ d, l, l, l, d, d, d, d ]
, [ d, d, d, d, l, l, d, d ]
, [ d, d, d, d, d, d, l, l ]
, [ l, l, l, d, d, d, d, d ]
, [ d, d, d, l, l, d, d, d ]
, [ d, d, d, d, d, l, l, d ]
]
of
Ok grid ->
grid

Err problem ->
Debug.todo (Debug.toString problem)
195 changes: 122 additions & 73 deletions src/Main.elm
View file
Expand Up @@ -34,6 +34,7 @@ type alias Model =
, weights : Dict Int Int
, rules : Adjacency.Rules Int
, showEntropy : Bool
, dimUnknown : Bool
}


Expand All @@ -43,17 +44,20 @@ type Msg
| Stop
| Step
| ToggleEntropy
| ToggleDimUnknown
| Load Image { width : Int, height : Int }


init : () -> ( Model, Cmd Msg )
init _ =
let
image =
Image.bars
( load Image.waves { width = 3, height = 3 }
, Cmd.none
)

windowSize =
{ width = 3, height = 3 }

load : Image -> { width : Int, height : Int } -> Model
load image windowSize =
let
windows =
Grid.windows windowSize image

Expand Down Expand Up @@ -111,20 +115,19 @@ init _ =
|> Adjacency.fromIds
|> Adjacency.finalize
in
( { image = image
, windowSize = windowSize
, windows = windows
, wave = Wave.init rules weights { width = 10, height = 10 }
, waveSize = { width = 10, height = 10 }
, seed = Random.initialSeed 0
, running = False
, indexes = indexes
, weights = weights
, rules = rules
, showEntropy = False
}
, Cmd.none
)
{ image = image
, windowSize = windowSize
, windows = windows
, wave = Wave.init rules weights { width = 10, height = 10 }
, waveSize = { width = 10, height = 10 }
, seed = Random.initialSeed 0
, running = False
, indexes = indexes
, weights = weights
, rules = rules
, showEntropy = False
, dimUnknown = False
}


update : Msg -> Model -> ( Model, Cmd Msg )
Expand Down Expand Up @@ -160,6 +163,12 @@ update msg model =
ToggleEntropy ->
( { model | showEntropy = not model.showEntropy }, Cmd.none )

ToggleDimUnknown ->
( { model | dimUnknown = not model.dimUnknown }, Cmd.none )

Load image dimensions ->
( load image dimensions, Cmd.none )


main : Program () Model Msg
main =
Expand Down Expand Up @@ -187,6 +196,20 @@ view model =
, Reset.borderBoxV201408
, h1 [ Html.text "Wave Function Collapse" ]
, h2 [ Html.text "Source Image" ]
, [ ( "Waves", Image.waves ), ( "Bars", Image.bars ), ( "Recurse", Image.recurse ) ]
|> List.map
(\( name, image ) ->
[ { width = 1, height = 1 }
, { width = 2, height = 2 }
, { width = 3, height = 3 }
]
|> List.map
(\dimensions ->
Html.button [ Events.onClick (Load image dimensions) ] [ Html.text (name ++ "(" ++ String.fromInt dimensions.height ++ "x" ++ String.fromInt dimensions.width ++ ")") ]
)
|> Html.p []
)
|> Html.div []
, Image.view [] model.image
, Html.details []
[ Html.summary [] [ Html.text "Windows" ]
Expand All @@ -204,64 +227,90 @@ view model =
)
]
, h2 [ Html.text "Wave" ]
, if model.running then
Html.button [ Events.onClick Stop ] [ Html.text "Stop" ]
, Html.p []
[ if model.running then
Html.button [ Events.onClick Stop ] [ Html.text "Stop" ]

else
Html.button [ Events.onClick Start ] [ Html.text "Start" ]
, Html.button [ Events.onClick Step ] [ Html.text "Step" ]
, Html.button [ Events.onClick (Reset { width = 2, height = 2 }) ] [ Html.text "Reset (2x2)" ]
, Html.button [ Events.onClick (Reset { width = 5, height = 5 }) ] [ Html.text "Reset (5x5)" ]
, Html.button [ Events.onClick (Reset { width = 10, height = 10 }) ] [ Html.text "Reset (10x10)" ]
, Html.button [ Events.onClick (Reset { width = 20, height = 20 }) ] [ Html.text "Reset (20x20)" ]
, Html.button [ Events.onClick (Reset { width = 50, height = 50 }) ] [ Html.text "Reset (50x50)" ]
, Html.button [ Events.onClick ToggleEntropy ] [ Html.text "Toggle Entropy (debug)" ]
else
Html.button [ Events.onClick Start ] [ Html.text "Start" ]
, Html.button [ Events.onClick Step ] [ Html.text "Step" ]
]
, Html.p []
[ Html.button [ Events.onClick (Reset { width = 2, height = 2 }) ] [ Html.text "Reset (2x2)" ]
, Html.button [ Events.onClick (Reset { width = 5, height = 5 }) ] [ Html.text "Reset (5x5)" ]
, Html.button [ Events.onClick (Reset { width = 10, height = 10 }) ] [ Html.text "Reset (10x10)" ]
, Html.button [ Events.onClick (Reset { width = 20, height = 20 }) ] [ Html.text "Reset (20x20)" ]
, Html.button [ Events.onClick (Reset { width = 50, height = 50 }) ] [ Html.text "Reset (50x50)" ]
]
, Html.p []
[ Html.button [ Events.onClick ToggleEntropy ] [ Html.text "Toggle Entropy (debug)" ]
, Html.button [ Events.onClick ToggleDimUnknown ] [ Html.text "Toggle Dimming Uknown Cells (debug)" ]
]
, Wave.view
(\indexes ->
let
{ reds, blues, greens, opacities } =
indexes
|> Set.toList
|> List.filterMap (\i -> Dict.get i model.indexes)
|> List.filterMap Grid.topLeft
|> List.foldl
(\color soFar ->
let
rgba =
Color.toRGBA color
in
{ reds = rgba.red :: soFar.reds
, blues = rgba.blue :: soFar.blues
, greens = rgba.green :: soFar.greens
, opacities = Color.opacityToFloat rgba.alpha :: soFar.opacities
(\cell ->
case cell of
Wave.Collapsed tile ->
case Dict.get tile model.indexes |> Maybe.andThen Grid.topLeft of
Just color ->
Image.viewColor [] color

Nothing ->
Html.text "invalid index"

Wave.Open indexes ->
let
{ reds, blues, greens, opacities } =
indexes
|> Set.toList
|> List.filterMap (\i -> Dict.get i model.indexes)
|> List.filterMap Grid.topLeft
|> List.foldl
(\color soFar ->
let
rgba =
Color.toRGBA color
in
{ reds = rgba.red :: soFar.reds
, blues = rgba.blue :: soFar.blues
, greens = rgba.green :: soFar.greens
, opacities = Color.opacityToFloat rgba.alpha :: soFar.opacities
}
)
{ reds = [], greens = [], blues = [], opacities = [] }

average items =
List.sum items / toFloat (List.length items)
in
if Set.isEmpty indexes then
Html.td
[ css
[ Css.lineHeight (Css.px 10)
, Css.fontSize (Css.px 10)
, Css.textAlign Css.center
]
, Attributes.width 10
, Attributes.height 10
]
[ Html.text "×" ]

else
Image.viewColor
[ Attributes.attribute "data-count" (String.fromInt (Set.size indexes))
, style "opacity"
(if model.dimUnknown then
"0.5"

else
"1"
)
]
(Color.fromRGBA
{ red = average reds
, green = average greens
, blue = average blues
, alpha = Color.customOpacity (average opacities)
}
)
{ reds = [], greens = [], blues = [], opacities = [] }

average items =
List.sum items / toFloat (List.length items)
in
if Set.isEmpty indexes then
Html.td
[ css
[ Css.lineHeight (Css.px 10)
, Css.fontSize (Css.px 10)
, Css.textAlign Css.center
]
, Attributes.width 10
, Attributes.height 10
]
[ Html.text "×" ]

else
Image.viewColor [ Attributes.attribute "data-count" (String.fromInt (Set.size indexes)) ]
(Color.fromRGBA
{ red = average reds
, green = average greens
, blue = average blues
, alpha = Color.customOpacity (average opacities)
}
)
)
model.wave
, if model.showEntropy then
Expand Down
17 changes: 4 additions & 13 deletions src/Wave.elm
View file
@@ -1,4 +1,4 @@
module Wave exposing (Wave, getEntropy, init, step, view)
module Wave exposing (Cell(..), Wave, getEntropy, init, step, view)

import Adjacency
import Dict exposing (Dict)
Expand Down Expand Up @@ -248,15 +248,6 @@ entropy probabilities possibilities =
|> List.sum


view : (Set comparable -> Html msg) -> Wave comparable -> Html msg
view fn (Wave { items }) =
Grid.view
(\cell ->
case cell of
Collapsed a ->
fn (Set.singleton a)

Open remaining ->
fn remaining
)
items
view : (Cell comparable -> Html msg) -> Wave comparable -> Html msg
view viewCell (Wave { items }) =
Grid.view viewCell items

0 comments on commit 2a688ce

Please sign in to comment.