Skip to content

Commit

Permalink
Update recipe graph generation to take yields into account
Browse files Browse the repository at this point in the history
I noticed this when regenerating the recipe graph on the wiki and saw
that `tea leaves` was sitting there with no input.  `tea leaves` are
obtained by harvesting `tea plant`s (which now occur naturally in the
world) but we were not taking such "yield" pairs into account when
creating the graph.  This PR adds a directed edge for each entity that
`yields` another.

The image is already updated on the wiki with the output from this PR:

  https://github.com/swarm-game/swarm/wiki/Recipes-cheat-sheet
  • Loading branch information
byorgey committed Apr 30, 2024
1 parent 4739005 commit f2627fc
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions app/doc/Swarm/Doc/Gen.hs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ import Data.Containers.ListUtils (nubOrd)
import Data.Foldable (toList)
import Data.Map.Lazy (Map, (!))
import Data.Map.Lazy qualified as Map
import Data.Maybe (fromMaybe)
import Data.Maybe (fromMaybe, mapMaybe)
import Data.Set (Set)
import Data.Set qualified as Set
import Data.Text (Text, unpack)
Expand All @@ -33,7 +33,7 @@ import Swarm.Doc.Keyword
import Swarm.Doc.Pedagogy
import Swarm.Doc.Util
import Swarm.Doc.Wiki.Cheatsheet
import Swarm.Game.Entity (Entity, EntityMap (entitiesByName), entityName)
import Swarm.Game.Entity (Entity, EntityMap (entitiesByName), entityName, entityYields)
import Swarm.Game.Entity qualified as E
import Swarm.Game.Land
import Swarm.Game.Recipe (Recipe, recipeCatalysts, recipeInputs, recipeOutputs)
Expand Down Expand Up @@ -222,6 +222,10 @@ recipesToDot baseRobot classicTerm emap recipes = do
recipesToPairs f rs = both nid <$> nubOrd (concatMap f rs)
mapM_ (uncurry (.->.)) (recipesToPairs recipeInOut recipes)
mapM_ (uncurry (---<>)) (recipesToPairs recipeReqOut recipes)
-- --------------------------------------------------------------------------
-- also draw an edge for each entity that "yields" another entity
let yieldPairs = mapMaybe (\e -> (e ^. entityName,) <$> (e ^. entityYields)) . Map.elems $ entitiesByName emap
mapM_ (uncurry (.->.)) (both getE <$> yieldPairs)

-- ----------------------------------------------------------------------------
-- RECIPE LEVELS
Expand Down

0 comments on commit f2627fc

Please sign in to comment.