restrict enum conversion to parts

This commit is contained in:
Zachary Snow 2021-02-18 17:10:41 -05:00
parent ec766657a8
commit 357b2921b3
1 changed files with 1 additions and 12 deletions

View File

@ -36,24 +36,13 @@ convert :: [AST] -> [AST]
convert = map $ concatMap convertDescription
convertDescription :: Description -> [Description]
convertDescription (Package ml name items) =
[Package ml name $ concatMap convertPackageItem items]
convertDescription (description @ Part{}) =
[Part attrs extern kw lifetime name ports items']
where
items' = inject enumItems items -- only keep what's used
Part attrs extern kw lifetime name ports items = description'
(description', enumItems) = convertDescription' description
convertDescription (PackageItem item) =
map PackageItem $ convertPackageItem item
-- explode a package item with its corresponding enum items
convertPackageItem :: PackageItem -> [PackageItem]
convertPackageItem item = do
item' : enumItems
where
(PackageItem item', enumItems) =
convertDescription' $ PackageItem item
convertDescription other = [other]
-- replace and collect the enum types in a description
convertDescription' :: Description -> (Description, [PackageItem])