{-# LANGUAGE PatternSynonyms, FlexibleContexts, QuasiQuotes #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE GADTs #-}
-- | Defines generation functions for SCS code packages.
module Language.Drasil.Code.Imperative.Generator (
  SomeProgGenerator(..),
  generator, generateCode, generateCodeProc, toFileLayout
) where

import Control.Lens ((^.))
import Control.Monad.State (get, evalState, runState)
import qualified Data.Set as Set (fromList)
import Data.Map (fromList, member, keys, elems)
import qualified Data.Map as M
import Data.Maybe (maybeToList, catMaybes)
import Text.PrettyPrint.HughesPJ (empty, isEmpty, vcat)

import Drasil.FileHandling (FileLayout, file, directory, exactFile, ps)
import Language.Drasil
import Drasil.GOOL (OOProg, VisibilityTag(..), headers, sources, mainMod,
  ProgData(..), initialState, FileData(..), modDoc)
import qualified Drasil.GOOL as OO (GSProgram, SFile, ProgramSym(..), unCI)
import Drasil.GProc (ProcProg)
import qualified Drasil.GProc as Proc (GSProgram, SFile, ProgramSym(..), unCI)
import Language.Drasil.Printers (SingleLine(OneLine), sentenceDoc, piSys, Notation (Scientific))
import Language.Drasil.Printing.Import (spec)
import Drasil.System (refTable, HasSystemMeta(..))

import Language.Drasil.Code.Imperative.ConceptMatch (chooseConcept)
import Language.Drasil.Code.Imperative.Descriptions (unmodularDesc)
import Language.Drasil.Code.Imperative.SpaceMatch (chooseSpace)
import Language.Drasil.Code.Imperative.GenerateGOOL (ClassType(..),
  genDoxConfig, genReadMe, genModuleWithImports, genModuleWithImportsProc)
import Language.Drasil.Code.Imperative.GenODE (chooseODELib)
import Language.Drasil.Code.Imperative.Helpers (liftS)
import Language.Drasil.Code.Imperative.Import (genModDef, genModDefProc,
  genModFuncs, genModFuncsProc, genModClasses)
import Language.Drasil.Code.Imperative.Modules (genInputMod, genInputModProc,
  genConstClass, genConstMod, checkConstClass, genInputClass,
  genInputConstraints, genInputConstraintsProc, genInputDerived,
  genInputDerivedProc, genInputFormat, genInputFormatProc, genMain, genMainProc,
  genMainFunc, genMainFuncProc, genCalcMod, genCalcModProc, genCalcFunc,
  genCalcFuncProc, genOutputFormat, genOutputFormatProc, genOutputMod,
  genOutputModProc, genSampleInput)
import Language.Drasil.Code.Imperative.DrasilState (GenState, DrasilState(..),
  ScopeType(..), designLog, modExportMap, clsDefMap, genICName,
  makeSoftwareDossierInfo, makeChoicesInfo, HasChoices(..))
import Language.Drasil.SoftwareDossier.SoftwareDossierSym (makeSds,
  SoftwareDossierSym(..))
import Language.Drasil.Code.Imperative.README.Core (ReadMeInfo(..))
import Language.Drasil.Code.PackageData (PackageData(..), pattern PackageData,
  package)
import Language.Drasil.SoftwareDossier.FileNames(sampleInputName)
import Language.Drasil.Code.ExtLibImport (auxMods, imports, modExports)
import Language.Drasil.Code.Lang (Lang(..))
import Language.Drasil.Choices (Choices(..), Modularity(..), Architecture(..),
  Visibility(..), DataInfo(..), Constraints(..), choicesSent, DocConfig(..),
  LogConfig(..), OptionalFeatures(..), InternalConcept(..))
import Language.Drasil.CodeSpec (CodeSpec, HasOldCodeSpec(..), getODE)

-- | Initializes the generator's 'DrasilState'.
-- 'String' parameter is a string representing the date.
-- \['Expr'\] parameter is the sample input values provided by the user.
generator :: Lang -> String -> [Expr] -> Choices -> CodeSpec -> DrasilState
generator :: Lang -> Name -> [Expr] -> Choices -> CodeSpec -> DrasilState
generator Lang
l Name
dt [Expr]
sd Choices
chs CodeSpec
cs = let
  sdsInfo :: SoftwareDossierInfo
sdsInfo = Verbosity -> [SoftwareDossierFile] -> [Expr] -> SoftwareDossierInfo
makeSoftwareDossierInfo
    (DocConfig -> Verbosity
doxVerbosity (DocConfig -> Verbosity) -> DocConfig -> Verbosity
forall a b. (a -> b) -> a -> b
$ OptionalFeatures -> DocConfig
docConfig (OptionalFeatures -> DocConfig) -> OptionalFeatures -> DocConfig
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs) (OptionalFeatures -> [SoftwareDossierFile]
auxFiles (OptionalFeatures -> [SoftwareDossierFile])
-> OptionalFeatures -> [SoftwareDossierFile]
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs) [Expr]
sd
  choices :: ChoicesInfo
choices = Modularity
-> ImplementationType
-> Structure
-> ConstantStructure
-> ConstantRepr
-> MatchedConceptMap
-> MatchedSpaces
-> ConstraintBehaviour
-> ConstraintBehaviour
-> [Comments]
-> Name
-> Name
-> [Logging]
-> (InternalConcept -> Name)
-> ChoicesInfo
makeChoicesInfo
    (Architecture -> Modularity
modularity (Architecture -> Modularity) -> Architecture -> Modularity
forall a b. (a -> b) -> a -> b
$ Choices -> Architecture
architecture Choices
chs)
    (Architecture -> ImplementationType
impType (Architecture -> ImplementationType)
-> Architecture -> ImplementationType
forall a b. (a -> b) -> a -> b
$ Choices -> Architecture
architecture Choices
chs)
    (DataInfo -> Structure
inputStructure (DataInfo -> Structure) -> DataInfo -> Structure
forall a b. (a -> b) -> a -> b
$ Choices -> DataInfo
dataInfo Choices
chs)
    (DataInfo -> ConstantStructure
constStructure (DataInfo -> ConstantStructure) -> DataInfo -> ConstantStructure
forall a b. (a -> b) -> a -> b
$ Choices -> DataInfo
dataInfo Choices
chs)
    (DataInfo -> ConstantRepr
constRepr (DataInfo -> ConstantRepr) -> DataInfo -> ConstantRepr
forall a b. (a -> b) -> a -> b
$ Choices -> DataInfo
dataInfo Choices
chs)
    MatchedConceptMap
mcm
    (Lang -> Choices -> MatchedSpaces
chooseSpace Lang
l Choices
chs)
    (Constraints -> ConstraintBehaviour
onSfwrConstraint (Constraints -> ConstraintBehaviour)
-> Constraints -> ConstraintBehaviour
forall a b. (a -> b) -> a -> b
$ Choices -> Constraints
srsConstraints Choices
chs)
    (Constraints -> ConstraintBehaviour
onPhysConstraint (Constraints -> ConstraintBehaviour)
-> Constraints -> ConstraintBehaviour
forall a b. (a -> b) -> a -> b
$ Choices -> Constraints
srsConstraints Choices
chs)
    (DocConfig -> [Comments]
comments (DocConfig -> [Comments]) -> DocConfig -> [Comments]
forall a b. (a -> b) -> a -> b
$ OptionalFeatures -> DocConfig
docConfig (OptionalFeatures -> DocConfig) -> OptionalFeatures -> DocConfig
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs)
    (Visibility -> Name
showDate (Visibility -> Name) -> Visibility -> Name
forall a b. (a -> b) -> a -> b
$ DocConfig -> Visibility
dates (DocConfig -> Visibility) -> DocConfig -> Visibility
forall a b. (a -> b) -> a -> b
$ OptionalFeatures -> DocConfig
docConfig (OptionalFeatures -> DocConfig) -> OptionalFeatures -> DocConfig
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs)
    (LogConfig -> Name
logFile (LogConfig -> Name) -> LogConfig -> Name
forall a b. (a -> b) -> a -> b
$ OptionalFeatures -> LogConfig
logConfig (OptionalFeatures -> LogConfig) -> OptionalFeatures -> LogConfig
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs)
    (LogConfig -> [Logging]
logging (LogConfig -> [Logging]) -> LogConfig -> [Logging]
forall a b. (a -> b) -> a -> b
$ OptionalFeatures -> LogConfig
logConfig (OptionalFeatures -> LogConfig) -> OptionalFeatures -> LogConfig
forall a b. (a -> b) -> a -> b
$ Choices -> OptionalFeatures
optFeats Choices
chs)
    (Choices -> InternalConcept -> Name
icNames Choices
chs)
  in DrasilState {
  -- constants
  codeSpec :: CodeSpec
codeSpec = CodeSpec
cs,
  printfo :: PrintingInformation
printfo = PrintingInformation
pinfo,
  _choices :: ChoicesInfo
_choices = ChoicesInfo
choices,
  modules :: [Mod]
modules = [Mod]
modules',
  extLibNames :: [(Name, Name)]
extLibNames = [(Name, Name)]
nms,
  extLibMap :: ExtLibMap
extLibMap = [(Name, ExtLibState)] -> ExtLibMap
forall k a. Ord k => [(k, a)] -> Map k a
fromList [(Name, ExtLibState)]
elmap,
  libPaths :: [Name]
libPaths = Maybe Name -> [Name]
forall a. Maybe a -> [a]
maybeToList Maybe Name
pth,
  eMap :: ModExportMap
eMap = ModExportMap
mem,
  libEMap :: ModExportMap
libEMap = ModExportMap
lem,
  clsMap :: ModExportMap
clsMap = ModExportMap
cdm,
  defSet :: Set Name
defSet = [Name] -> Set Name
forall a. Ord a => [a] -> Set a
Set.fromList ([Name] -> Set Name) -> [Name] -> Set Name
forall a b. (a -> b) -> a -> b
$ ModExportMap -> [Name]
forall k a. Map k a -> [k]
keys ModExportMap
mem [Name] -> [Name] -> [Name]
forall a. [a] -> [a] -> [a]
++ ModExportMap -> [Name]
forall k a. Map k a -> [k]
keys ModExportMap
cdm,
  getVal :: Int
getVal = Choices -> Int
folderVal Choices
chs,
  _softwareDossierInfo :: SoftwareDossierInfo
_softwareDossierInfo = SoftwareDossierInfo
sdsInfo,
  -- stateful
  currentModule :: Name
currentModule = Name
"",
  currentClass :: Name
currentClass = Name
"",
  _designLog :: Doc
_designLog = Doc
des,
  _loggedSpaces :: [(Space, CodeType)]
_loggedSpaces = [], -- Used to prevent duplicate logs added to design log
  currentScope :: ScopeType
currentScope = ScopeType
Global
}
  where pinfo :: PrintingInformation
pinfo = ChunkDB
-> Map UID Reference -> Stage -> Notation -> PrintingInformation
piSys (CodeSpec
cs CodeSpec -> Getting ChunkDB CodeSpec ChunkDB -> ChunkDB
forall s a. s -> Getting a s a -> a
^. Getting ChunkDB CodeSpec ChunkDB
forall c. HasSystemMeta c => Lens' c ChunkDB
Lens' CodeSpec ChunkDB
systemdb) (CodeSpec
cs CodeSpec
-> Getting (Map UID Reference) CodeSpec (Map UID Reference)
-> Map UID Reference
forall s a. s -> Getting a s a -> a
^. Getting (Map UID Reference) CodeSpec (Map UID Reference)
forall c. HasSmithEtAlSRS c => Lens' c (Map UID Reference)
Lens' CodeSpec (Map UID Reference)
refTable) Stage
Implementation Notation
Scientific
        (MatchedConceptMap
mcm, [Sentence]
concLog) = State [Sentence] MatchedConceptMap
-> [Sentence] -> (MatchedConceptMap, [Sentence])
forall s a. State s a -> s -> (a, s)
runState (Choices -> State [Sentence] MatchedConceptMap
chooseConcept Choices
chs) []
        showDate :: Visibility -> Name
showDate Visibility
Show = Name
dt
        showDate Visibility
Hide = Name
""
        ((Maybe Name
pth, [(Name, ExtLibState)]
elmap, (Name, Name)
lname), [Sentence]
libLog) = State [Sentence] ODEGenInfo
-> [Sentence] -> (ODEGenInfo, [Sentence])
forall s a. State s a -> s -> (a, s)
runState (Lang -> Maybe ODE -> State [Sentence] ODEGenInfo
chooseODELib Lang
l (Maybe ODE -> State [Sentence] ODEGenInfo)
-> Maybe ODE -> State [Sentence] ODEGenInfo
forall a b. (a -> b) -> a -> b
$ [ExtLib] -> Maybe ODE
getODE ([ExtLib] -> Maybe ODE) -> [ExtLib] -> Maybe ODE
forall a b. (a -> b) -> a -> b
$ Choices -> [ExtLib]
extLibs Choices
chs) []
        els :: [ExtLibState]
els = ((Name, ExtLibState) -> ExtLibState)
-> [(Name, ExtLibState)] -> [ExtLibState]
forall a b. (a -> b) -> [a] -> [b]
map (Name, ExtLibState) -> ExtLibState
forall a b. (a, b) -> b
snd [(Name, ExtLibState)]
elmap
        nms :: [(Name, Name)]
nms = [(Name, Name)
lname]
        mem :: ModExportMap
mem = OldCodeSpec -> Choices -> [Mod] -> ModExportMap
modExportMap (CodeSpec
cs CodeSpec -> Getting OldCodeSpec CodeSpec OldCodeSpec -> OldCodeSpec
forall s a. s -> Getting a s a -> a
^. Getting OldCodeSpec CodeSpec OldCodeSpec
forall c. HasOldCodeSpec c => Lens' c OldCodeSpec
Lens' CodeSpec OldCodeSpec
oldCodeSpec) Choices
chs [Mod]
modules'
        lem :: ModExportMap
lem = [(Name, Name)] -> ModExportMap
forall k a. Ord k => [(k, a)] -> Map k a
fromList ((ExtLibState -> [(Name, Name)]) -> [ExtLibState] -> [(Name, Name)]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (ExtLibState
-> Getting [(Name, Name)] ExtLibState [(Name, Name)]
-> [(Name, Name)]
forall s a. s -> Getting a s a -> a
^. Getting [(Name, Name)] ExtLibState [(Name, Name)]
Lens' ExtLibState [(Name, Name)]
modExports) [ExtLibState]
els)
        cdm :: ModExportMap
cdm = OldCodeSpec -> Choices -> [Mod] -> ModExportMap
clsDefMap (CodeSpec
cs CodeSpec -> Getting OldCodeSpec CodeSpec OldCodeSpec -> OldCodeSpec
forall s a. s -> Getting a s a -> a
^. Getting OldCodeSpec CodeSpec OldCodeSpec
forall c. HasOldCodeSpec c => Lens' c OldCodeSpec
Lens' CodeSpec OldCodeSpec
oldCodeSpec) Choices
chs [Mod]
modules'
        modules' :: [Mod]
modules' = (CodeSpec
cs CodeSpec -> Getting [Mod] CodeSpec [Mod] -> [Mod]
forall s a. s -> Getting a s a -> a
^. Getting [Mod] CodeSpec [Mod]
forall c. HasOldCodeSpec c => Lens' c [Mod]
Lens' CodeSpec [Mod]
modsO) [Mod] -> [Mod] -> [Mod]
forall a. [a] -> [a] -> [a]
++ (ExtLibState -> [Mod]) -> [ExtLibState] -> [Mod]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (ExtLibState -> Getting [Mod] ExtLibState [Mod] -> [Mod]
forall s a. s -> Getting a s a -> a
^. Getting [Mod] ExtLibState [Mod]
Lens' ExtLibState [Mod]
auxMods) [ExtLibState]
els
        nonPrefChs :: [Sentence]
nonPrefChs = Choices -> [Sentence]
choicesSent Choices
chs
        des :: Doc
des = [Doc] -> Doc
vcat ([Doc] -> Doc) -> ([Sentence] -> [Doc]) -> [Sentence] -> Doc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Sentence -> Doc) -> [Sentence] -> [Doc]
forall a b. (a -> b) -> [a] -> [b]
map (SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> (Sentence -> Spec) -> Sentence -> Doc
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PrintingInformation -> Sentence -> Spec
spec PrintingInformation
pinfo) ([Sentence] -> Doc) -> [Sentence] -> Doc
forall a b. (a -> b) -> a -> b
$
          ([Sentence]
nonPrefChs [Sentence] -> [Sentence] -> [Sentence]
forall a. [a] -> [a] -> [a]
++ [Sentence]
concLog [Sentence] -> [Sentence] -> [Sentence]
forall a. [a] -> [a] -> [a]
++ [Sentence]
libLog)

-- OO Versions --

data SomeProgGenerator where
  SomeProgGenerator :: forall repr. (OOProg repr) => (repr (OO.Program repr) -> ProgData) -> SomeProgGenerator

-- | Generates a package with the given 'DrasilState'. The passed
-- un-representation functions determine which target language the package will
-- be generated in.
generateCode :: (SoftwareDossierSym packRepr, Monad packRepr) =>
  Lang -> SomeProgGenerator ->
  (packRepr PackageData -> PackageData) -> DrasilState -> FileLayout
generateCode :: forall (packRepr :: * -> *).
(SoftwareDossierSym packRepr, Monad packRepr) =>
Lang
-> SomeProgGenerator
-> (packRepr PackageData -> PackageData)
-> DrasilState
-> FileLayout
generateCode Lang
l (SomeProgGenerator repr (Program repr) -> ProgData
unReprProg) packRepr PackageData -> PackageData
unReprPack DrasilState
g =
  let dirName :: Name
dirName = Lang -> Name
getDir Lang
l
      (packRepr PackageData
pckg, DrasilState
ds) = State DrasilState (packRepr PackageData)
-> DrasilState -> (packRepr PackageData, DrasilState)
forall s a. State s a -> s -> (a, s)
runState ((repr (Program repr) -> ProgData)
-> State DrasilState (packRepr PackageData)
forall (progRepr :: * -> *) (packRepr :: * -> *).
(OOProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
(progRepr (Program progRepr) -> ProgData)
-> GenState (packRepr PackageData)
genPackage repr (Program repr) -> ProgData
unReprProg) DrasilState
g
      (PackageData ProgData
prog [FileLayout]
progDossier) = packRepr PackageData -> PackageData
unReprPack packRepr PackageData
pckg
      designLogFile :: [FileLayout]
designLogFile = [PathSegment -> Doc -> FileLayout
forall doc. Writeable doc => PathSegment -> doc -> FileLayout
file [ps|designLog.txt|] (DrasilState
ds DrasilState -> Getting Doc DrasilState Doc -> Doc
forall s a. s -> Getting a s a -> a
^. Getting Doc DrasilState Doc
Lens' DrasilState Doc
designLog) |
                        Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$ Doc -> Bool
isEmpty (Doc -> Bool) -> Doc -> Bool
forall a b. (a -> b) -> a -> b
$ DrasilState
ds DrasilState -> Getting Doc DrasilState Doc -> Doc
forall s a. s -> Getting a s a -> a
^. Getting Doc DrasilState Doc
Lens' DrasilState Doc
designLog]
      initFile :: [FileLayout]
initFile = [PathSegment -> Doc -> FileLayout
forall doc. Writeable doc => PathSegment -> doc -> FileLayout
exactFile [ps|__init__.py|] Doc
empty | Lang
l Lang -> Lang -> Bool
forall a. Eq a => a -> a -> Bool
== Lang
Python]
      packageFiles :: [FileLayout]
packageFiles = [FileData] -> [FileLayout]
toFileLayout (ProgData -> [FileData]
progMods ProgData
prog) [FileLayout] -> [FileLayout] -> [FileLayout]
forall a. [a] -> [a] -> [a]
++ [FileLayout]
progDossier
        [FileLayout] -> [FileLayout] -> [FileLayout]
forall a. [a] -> [a] -> [a]
++ [FileLayout]
designLogFile [FileLayout] -> [FileLayout] -> [FileLayout]
forall a. [a] -> [a] -> [a]
++ [FileLayout]
initFile
  in
    PathSegment -> [FileLayout] -> FileLayout
forall (f :: * -> *).
Foldable f =>
PathSegment -> f FileLayout -> FileLayout
directory
      [ps|{dirName}|]
      [FileLayout]
packageFiles

-- | Internal: Converts a list of `FileData` to a `FileLayout`.
toFileLayout :: [FileData] -> [FileLayout]
toFileLayout :: [FileData] -> [FileLayout]
toFileLayout [FileData]
fc =
  let
    root :: Map Name (Entry Doc)
root = (Map Name (Entry Doc) -> FileData -> Map Name (Entry Doc))
-> Map Name (Entry Doc) -> [FileData] -> Map Name (Entry Doc)
forall b a. (b -> a -> b) -> b -> [a] -> b
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl (\Map Name (Entry Doc)
m FileData
f -> (Name, Doc) -> Map Name (Entry Doc) -> Map Name (Entry Doc)
forall a. (Name, a) -> Map Name (Entry a) -> Map Name (Entry a)
insertFile (FileData -> Name
filePath FileData
f, ModData -> Doc
modDoc (ModData -> Doc) -> ModData -> Doc
forall a b. (a -> b) -> a -> b
$ FileData -> ModData
fileMod FileData
f) Map Name (Entry Doc)
m) Map Name (Entry Doc)
forall k a. Map k a
M.empty [FileData]
fc

    entryToLayout :: (Name, Entry doc) -> FileLayout
entryToLayout (Name
n, File doc
d) = PathSegment -> doc -> FileLayout
forall doc. Writeable doc => PathSegment -> doc -> FileLayout
file [ps|{n}|] doc
d
    entryToLayout (Name
n, Folder Map Name (Entry doc)
m) = PathSegment -> [FileLayout] -> FileLayout
forall (f :: * -> *).
Foldable f =>
PathSegment -> f FileLayout -> FileLayout
directory [ps|{n}|] ([FileLayout] -> FileLayout) -> [FileLayout] -> FileLayout
forall a b. (a -> b) -> a -> b
$ ((Name, Entry doc) -> FileLayout)
-> [(Name, Entry doc)] -> [FileLayout]
forall a b. (a -> b) -> [a] -> [b]
map (Name, Entry doc) -> FileLayout
entryToLayout ([(Name, Entry doc)] -> [FileLayout])
-> [(Name, Entry doc)] -> [FileLayout]
forall a b. (a -> b) -> a -> b
$ Map Name (Entry doc) -> [(Name, Entry doc)]
forall k a. Map k a -> [(k, a)]
M.assocs Map Name (Entry doc)
m
  in
    ((Name, Entry Doc) -> FileLayout)
-> [(Name, Entry Doc)] -> [FileLayout]
forall a b. (a -> b) -> [a] -> [b]
map (Name, Entry Doc) -> FileLayout
forall {doc}. Writeable doc => (Name, Entry doc) -> FileLayout
entryToLayout (Map Name (Entry Doc) -> [(Name, Entry Doc)]
forall k a. Map k a -> [(k, a)]
M.assocs Map Name (Entry Doc)
root)

data Entry a = File a | Folder (M.Map String (Entry a))
  deriving (Int -> Entry a -> ShowS
[Entry a] -> ShowS
Entry a -> Name
(Int -> Entry a -> ShowS)
-> (Entry a -> Name) -> ([Entry a] -> ShowS) -> Show (Entry a)
forall a. Show a => Int -> Entry a -> ShowS
forall a. Show a => [Entry a] -> ShowS
forall a. Show a => Entry a -> Name
forall a.
(Int -> a -> ShowS) -> (a -> Name) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: forall a. Show a => Int -> Entry a -> ShowS
showsPrec :: Int -> Entry a -> ShowS
$cshow :: forall a. Show a => Entry a -> Name
show :: Entry a -> Name
$cshowList :: forall a. Show a => [Entry a] -> ShowS
showList :: [Entry a] -> ShowS
Show)

insertFile :: (String, a) -> M.Map String (Entry a) -> M.Map String (Entry a)
insertFile :: forall a. (Name, a) -> Map Name (Entry a) -> Map Name (Entry a)
insertFile (Name
p, a
d) Map Name (Entry a)
m =
  if Char
'/' Char -> Name -> Bool
forall a. Eq a => a -> [a] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` Name
p
    then
      let (Name
fname, Name
rest) = (Char -> Bool) -> Name -> (Name, Name)
forall a. (a -> Bool) -> [a] -> ([a], [a])
break (Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
'/') Name
p
          folderM :: Map Name (Entry a)
folderM = case Entry a -> Name -> Map Name (Entry a) -> Entry a
forall k a. Ord k => a -> k -> Map k a -> a
M.findWithDefault (Map Name (Entry a) -> Entry a
forall a. Map Name (Entry a) -> Entry a
Folder Map Name (Entry a)
forall k a. Map k a
M.empty) Name
fname Map Name (Entry a)
m of
                      File a
_   -> Name -> Map Name (Entry a)
forall {a}. Name -> a
dupError Name
fname
                      Folder Map Name (Entry a)
f -> Map Name (Entry a)
f
      in Name -> Entry a -> Map Name (Entry a) -> Map Name (Entry a)
forall k a. Ord k => k -> a -> Map k a -> Map k a
M.insert Name
fname (Map Name (Entry a) -> Entry a
forall a. Map Name (Entry a) -> Entry a
Folder (Map Name (Entry a) -> Entry a) -> Map Name (Entry a) -> Entry a
forall a b. (a -> b) -> a -> b
$ (Name, a) -> Map Name (Entry a) -> Map Name (Entry a)
forall a. (Name, a) -> Map Name (Entry a) -> Map Name (Entry a)
insertFile (Int -> ShowS
forall a. Int -> [a] -> [a]
drop Int
1 Name
rest, a
d) Map Name (Entry a)
folderM) Map Name (Entry a)
m
    else (Entry a -> Entry a -> Entry a)
-> Name -> Entry a -> Map Name (Entry a) -> Map Name (Entry a)
forall k a. Ord k => (a -> a -> a) -> k -> a -> Map k a -> Map k a
M.insertWith (\Entry a
_ -> Name -> Entry a -> Entry a
forall {a}. Name -> a
dupError Name
p) Name
p (a -> Entry a
forall a. a -> Entry a
File a
d) Map Name (Entry a)
m
  where
    dupError :: Name -> a
dupError Name
fname = Name -> a
forall a. HasCallStack => Name -> a
error (Name -> a) -> Name -> a
forall a b. (a -> b) -> a -> b
$ Name
"A file or folder with name '" Name -> ShowS
forall a. [a] -> [a] -> [a]
++ Name
fname Name -> ShowS
forall a. [a] -> [a] -> [a]
++ Name
"' already exists."

-- | Generates a package, including a Makefile, sample input file, and Doxygen
-- configuration file (all subject to the user's choices).
-- The passed un-representation function determines which target language the
-- package will be generated in.
-- GOOL's static code analysis interpreter is called to initialize the state
-- used by the language renderer.
genPackage :: (OOProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
  (progRepr (OO.Program progRepr) -> ProgData) -> GenState (packRepr PackageData)
genPackage :: forall (progRepr :: * -> *) (packRepr :: * -> *).
(OOProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
(progRepr (Program progRepr) -> ProgData)
-> GenState (packRepr PackageData)
genPackage progRepr (Program progRepr) -> ProgData
unRepr = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  State GOOLState (CodeInfoOO GOOLState)
ci <- StateT
  DrasilState Identity (State GOOLState (CodeInfoOO GOOLState))
GenState (GSProgram CodeInfoOO)
forall (r :: * -> *). OOProg r => GenState (GSProgram r)
genProgram
  State GOOLState (progRepr (Program progRepr))
p <- GenState (State GOOLState (progRepr (Program progRepr)))
forall (r :: * -> *). OOProg r => GenState (GSProgram r)
genProgram
  let info :: GOOLState
info = CodeInfoOO GOOLState -> GOOLState
forall a. CodeInfoOO a -> a
OO.unCI (CodeInfoOO GOOLState -> GOOLState)
-> CodeInfoOO GOOLState -> GOOLState
forall a b. (a -> b) -> a -> b
$ State GOOLState (CodeInfoOO GOOLState)
-> GOOLState -> CodeInfoOO GOOLState
forall s a. State s a -> s -> a
evalState State GOOLState (CodeInfoOO GOOLState)
ci GOOLState
initialState
      (progRepr (Program progRepr)
reprPD, GOOLState
s) = State GOOLState (progRepr (Program progRepr))
-> GOOLState -> (progRepr (Program progRepr), GOOLState)
forall s a. State s a -> s -> (a, s)
runState State GOOLState (progRepr (Program progRepr))
p GOOLState
info
      fileInfoState :: SoftwareDossierState
fileInfoState = [Name] -> [Name] -> Maybe Name -> SoftwareDossierState
makeSds (GOOLState
s GOOLState -> Getting [Name] GOOLState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] GOOLState [Name]
Lens' GOOLState [Name]
headers) (GOOLState
s GOOLState -> Getting [Name] GOOLState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] GOOLState [Name]
Lens' GOOLState [Name]
sources) (GOOLState
s GOOLState
-> Getting (Maybe Name) GOOLState (Maybe Name) -> Maybe Name
forall s a. s -> Getting a s a -> a
^. Getting (Maybe Name) GOOLState (Maybe Name)
Lens' GOOLState (Maybe Name)
mainMod)
      pd :: ProgData
pd = progRepr (Program progRepr) -> ProgData
unRepr progRepr (Program progRepr)
reprPD
      m :: packRepr FileLayout
m = [Name]
-> ImplementationType
-> [Comments]
-> SoftwareDossierState
-> ProgData
-> packRepr FileLayout
forall (r :: * -> *).
SoftwareDossierSym r =>
[Name]
-> ImplementationType
-> [Comments]
-> SoftwareDossierState
-> ProgData
-> r FileLayout
makefile (DrasilState -> [Name]
libPaths DrasilState
g) (DrasilState
g DrasilState
-> Getting ImplementationType DrasilState ImplementationType
-> ImplementationType
forall s a. s -> Getting a s a -> a
^. Getting ImplementationType DrasilState ImplementationType
forall a. HasChoices a => Lens' a ImplementationType
Lens' DrasilState ImplementationType
implType) (DrasilState
g DrasilState
-> Getting [Comments] DrasilState [Comments] -> [Comments]
forall s a. s -> Getting a s a -> a
^. Getting [Comments] DrasilState [Comments]
forall a. HasChoices a => Lens' a [Comments]
Lens' DrasilState [Comments]
commented) SoftwareDossierState
fileInfoState ProgData
pd
      as :: [Name]
as = (Person -> Name) -> [Person] -> [Name]
forall a b. (a -> b) -> [a] -> [b]
map Person -> Name
forall n. HasName n => n -> Name
fullName (DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Person] CodeSpec [Person] -> [Person]
forall s a. s -> Getting a s a -> a
^. Getting [Person] CodeSpec [Person]
forall c. HasOldCodeSpec c => Lens' c [Person]
Lens' CodeSpec [Person]
authorsO)
      cfp :: [RelativeFile]
cfp = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec
-> Getting [RelativeFile] CodeSpec [RelativeFile] -> [RelativeFile]
forall s a. s -> Getting a s a -> a
^. Getting [RelativeFile] CodeSpec [RelativeFile]
forall c. HasOldCodeSpec c => Lens' c [RelativeFile]
Lens' CodeSpec [RelativeFile]
configFilesO
      db :: PrintingInformation
db = DrasilState -> PrintingInformation
printfo DrasilState
g
      -- FIXME: The below code does `Doc -> String` conversion.
      prps :: Name
prps = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
purpose)
      bckgrnd :: Name
bckgrnd = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
background)
      mtvtn :: Name
mtvtn = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
motivation)
      scp :: Name
scp = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
scope)
  Maybe (packRepr FileLayout)
i <- GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
Applicative r =>
GenState (Maybe (r FileLayout))
genSampleInput
  Maybe (packRepr FileLayout)
d <- SoftwareDossierState -> GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
SoftwareDossierSym r =>
SoftwareDossierState -> GenState (Maybe (r FileLayout))
genDoxConfig SoftwareDossierState
fileInfoState
  Maybe (packRepr FileLayout)
rm <- ReadMeInfo -> GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
SoftwareDossierSym r =>
ReadMeInfo -> GenState (Maybe (r FileLayout))
genReadMe ReadMeInfo {
        langName :: Name
langName = Name
"",
        langVersion :: Name
langVersion = Name
"",
        invalidOS :: Maybe Name
invalidOS = Maybe Name
forall a. Maybe a
Nothing,
        implementType :: ImplementationType
implementType = DrasilState
g DrasilState
-> Getting ImplementationType DrasilState ImplementationType
-> ImplementationType
forall s a. s -> Getting a s a -> a
^. Getting ImplementationType DrasilState ImplementationType
forall a. HasChoices a => Lens' a ImplementationType
Lens' DrasilState ImplementationType
implType,
        extLibNV :: [(Name, Name)]
extLibNV = DrasilState -> [(Name, Name)]
extLibNames DrasilState
g,
        extLibFP :: [Name]
extLibFP = DrasilState -> [Name]
libPaths DrasilState
g,
        contributors :: [Name]
contributors = [Name]
as,
        configFP :: [RelativeFile]
configFP = [RelativeFile]
cfp,
        caseName :: Name
caseName = Name
"",
        examplePurpose :: Name
examplePurpose = Name
prps,
        exampleDescr :: Name
exampleDescr = Name
bckgrnd,
        exampleMotivation :: Name
exampleMotivation = Name
mtvtn,
        exampleScope :: Name
exampleScope = Name
scp,
        folderNum :: Int
folderNum = DrasilState -> Int
getVal DrasilState
g,
        inputOutput :: (Name, Name)
inputOutput = (Name
sampleInputName, Name
"output.txt")} -- This needs a more permanent solution
  packRepr PackageData -> GenState (packRepr PackageData)
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return (packRepr PackageData -> GenState (packRepr PackageData))
-> packRepr PackageData -> GenState (packRepr PackageData)
forall a b. (a -> b) -> a -> b
$ ProgData -> [packRepr FileLayout] -> packRepr PackageData
forall (r :: * -> *).
Monad r =>
ProgData -> [r FileLayout] -> r PackageData
package ProgData
pd (packRepr FileLayout
mpackRepr FileLayout
-> [packRepr FileLayout] -> [packRepr FileLayout]
forall a. a -> [a] -> [a]
:[Maybe (packRepr FileLayout)] -> [packRepr FileLayout]
forall a. [Maybe a] -> [a]
catMaybes [Maybe (packRepr FileLayout)
i,Maybe (packRepr FileLayout)
rm,Maybe (packRepr FileLayout)
d])

-- | Generates an SCS program based on the problem and the user's design choices.
genProgram :: (OOProg r) => GenState (OO.GSProgram r)
genProgram :: forall (r :: * -> *). OOProg r => GenState (GSProgram r)
genProgram = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  [FS (r (File r))]
ms <- Modularity -> StateT DrasilState Identity [FS (r (File r))]
forall (r :: * -> *). OOProg r => Modularity -> GenState [SFile r]
chooseModules (Modularity -> StateT DrasilState Identity [FS (r (File r))])
-> Modularity -> StateT DrasilState Identity [FS (r (File r))]
forall a b. (a -> b) -> a -> b
$ DrasilState
g DrasilState
-> Getting Modularity DrasilState Modularity -> Modularity
forall s a. s -> Getting a s a -> a
^. Getting Modularity DrasilState Modularity
forall a. HasChoices a => Lens' a Modularity
Lens' DrasilState Modularity
modular
  let n :: Name
n = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting Name CodeSpec Name -> Name
forall s a. s -> Getting a s a -> a
^. Getting Name CodeSpec Name
forall c. HasOldCodeSpec c => Lens' c Name
Lens' CodeSpec Name
pNameO
  let p :: Name
p = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec (DrasilState -> PrintingInformation
printfo DrasilState
g) (Sentence -> Spec) -> Sentence -> Spec
forall a b. (a -> b) -> a -> b
$ [Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
purpose
  GSProgram r -> GenState (GSProgram r)
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return (GSProgram r -> GenState (GSProgram r))
-> GSProgram r -> GenState (GSProgram r)
forall a b. (a -> b) -> a -> b
$ Name -> Name -> [FS (r (File r))] -> GSProgram r
forall (r :: * -> *).
ProgramSym r =>
Name -> Name -> [SFile r] -> GSProgram r
OO.prog Name
n Name
p [FS (r (File r))]
ms

-- | Generates either a single module or many modules, based on the users choice
-- of modularity.
chooseModules :: (OOProg r) => Modularity -> GenState [OO.SFile r]
chooseModules :: forall (r :: * -> *). OOProg r => Modularity -> GenState [SFile r]
chooseModules Modularity
Unmodular = State DrasilState (FS (r (File r)))
-> StateT DrasilState Identity [FS (r (File r))]
forall a b. State a b -> State a [b]
liftS State DrasilState (FS (r (File r)))
forall (r :: * -> *). OOProg r => GenState (SFile r)
genUnmodular
chooseModules Modularity
Modular = StateT DrasilState Identity [FS (r (File r))]
forall (r :: * -> *). OOProg r => GenState [SFile r]
genModules

-- | Generates an entire SCS program as a single module.
genUnmodular :: (OOProg r) => GenState (OO.SFile r)
genUnmodular :: forall (r :: * -> *). OOProg r => GenState (SFile r)
genUnmodular = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  Name
umDesc <- GenState Name
unmodularDesc
  Name
giName <- InternalConcept -> GenState Name
genICName InternalConcept
GetInput
  Name
dvName <- InternalConcept -> GenState Name
genICName InternalConcept
DerivedValuesFn
  Name
icName <- InternalConcept -> GenState Name
genICName InternalConcept
InputConstraintsFn
  let n :: Name
n = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting Name CodeSpec Name -> Name
forall s a. s -> Getting a s a -> a
^. Getting Name CodeSpec Name
forall c. HasOldCodeSpec c => Lens' c Name
Lens' CodeSpec Name
pNameO
      cls :: Bool
cls = (Name -> Bool) -> [Name] -> Bool
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any (Name -> ModExportMap -> Bool
forall k a. Ord k => k -> Map k a -> Bool
`member` DrasilState -> ModExportMap
clsMap DrasilState
g) [Name
giName, Name
dvName, Name
icName]
  Name
-> Name
-> [Name]
-> [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SClass r))]
-> GenState (SFile r)
forall (r :: * -> *).
OOProg r =>
Name
-> Name
-> [Name]
-> [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SClass r))]
-> GenState (SFile r)
genModuleWithImports Name
n Name
umDesc ((ExtLibState -> [Name]) -> [ExtLibState] -> [Name]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (ExtLibState -> Getting [Name] ExtLibState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] ExtLibState [Name]
Lens' ExtLibState [Name]
imports) (ExtLibMap -> [ExtLibState]
forall k a. Map k a -> [a]
elems (ExtLibMap -> [ExtLibState]) -> ExtLibMap -> [ExtLibState]
forall a b. (a -> b) -> a -> b
$ DrasilState -> ExtLibMap
extLibMap DrasilState
g))
    (GenState (Maybe (SMethod r))
forall (r :: * -> *). OOProg r => GenState (Maybe (SMethod r))
genMainFunc
      GenState (Maybe (SMethod r))
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. a -> [a] -> [a]
: (StateT DrasilState Identity (SMethod r)
 -> GenState (Maybe (SMethod r)))
-> [StateT DrasilState Identity (SMethod r)]
-> [GenState (Maybe (SMethod r))]
forall a b. (a -> b) -> [a] -> [b]
map ((SMethod r -> Maybe (SMethod r))
-> StateT DrasilState Identity (SMethod r)
-> GenState (Maybe (SMethod r))
forall a b.
(a -> b)
-> StateT DrasilState Identity a -> StateT DrasilState Identity b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap SMethod r -> Maybe (SMethod r)
forall a. a -> Maybe a
Just) ((CodeDefinition -> StateT DrasilState Identity (SMethod r))
-> [CodeDefinition] -> [StateT DrasilState Identity (SMethod r)]
forall a b. (a -> b) -> [a] -> [b]
map CodeDefinition -> StateT DrasilState Identity (SMethod r)
forall (r :: * -> *).
OOProg r =>
CodeDefinition -> GenState (SMethod r)
genCalcFunc (DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec
-> Getting [CodeDefinition] CodeSpec [CodeDefinition]
-> [CodeDefinition]
forall s a. s -> Getting a s a -> a
^. Getting [CodeDefinition] CodeSpec [CodeDefinition]
forall c. HasOldCodeSpec c => Lens' c [CodeDefinition]
Lens' CodeSpec [CodeDefinition]
execOrderO)
        [StateT DrasilState Identity (SMethod r)]
-> [StateT DrasilState Identity (SMethod r)]
-> [StateT DrasilState Identity (SMethod r)]
forall a. [a] -> [a] -> [a]
++ (Mod -> [StateT DrasilState Identity (SMethod r)])
-> [Mod] -> [StateT DrasilState Identity (SMethod r)]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap Mod -> [StateT DrasilState Identity (SMethod r)]
forall (r :: * -> *). OOProg r => Mod -> [GenState (SMethod r)]
genModFuncs (DrasilState -> [Mod]
modules DrasilState
g))
      [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. [a] -> [a] -> [a]
++ ((if Bool
cls then [] else [VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
OOProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputFormat VisibilityTag
Pub, VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
OOProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputDerived VisibilityTag
Pub,
        VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
OOProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputConstraints VisibilityTag
Pub]) [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. [a] -> [a] -> [a]
++ [GenState (Maybe (SMethod r))
forall (r :: * -> *). OOProg r => GenState (Maybe (SMethod r))
genOutputFormat]))
    ([ClassType -> GenState (Maybe (SClass r))
forall (r :: * -> *).
OOProg r =>
ClassType -> GenState (Maybe (SClass r))
genInputClass ClassType
Auxiliary, ClassType -> GenState (Maybe (SClass r))
forall (r :: * -> *).
OOProg r =>
ClassType -> GenState (Maybe (SClass r))
genConstClass ClassType
Auxiliary]
      [GenState (Maybe (SClass r))]
-> [GenState (Maybe (SClass r))] -> [GenState (Maybe (SClass r))]
forall a. [a] -> [a] -> [a]
++ (StateT DrasilState Identity (SClass r)
 -> GenState (Maybe (SClass r)))
-> [StateT DrasilState Identity (SClass r)]
-> [GenState (Maybe (SClass r))]
forall a b. (a -> b) -> [a] -> [b]
map ((SClass r -> Maybe (SClass r))
-> StateT DrasilState Identity (SClass r)
-> GenState (Maybe (SClass r))
forall a b.
(a -> b)
-> StateT DrasilState Identity a -> StateT DrasilState Identity b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap SClass r -> Maybe (SClass r)
forall a. a -> Maybe a
Just) ((Mod -> [StateT DrasilState Identity (SClass r)])
-> [Mod] -> [StateT DrasilState Identity (SClass r)]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap Mod -> [StateT DrasilState Identity (SClass r)]
forall (r :: * -> *). OOProg r => Mod -> [GenState (SClass r)]
genModClasses ([Mod] -> [StateT DrasilState Identity (SClass r)])
-> [Mod] -> [StateT DrasilState Identity (SClass r)]
forall a b. (a -> b) -> a -> b
$ DrasilState -> [Mod]
modules DrasilState
g))

-- | Generates all modules for an SCS program.
genModules :: (OOProg r) => GenState [OO.SFile r]
genModules :: forall (r :: * -> *). OOProg r => GenState [SFile r]
genModules = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  SFile r
mn     <- GenState (SFile r)
forall (r :: * -> *). OOProg r => GenState (SFile r)
genMain
  [SFile r]
inp    <- GenState [SFile r]
forall (r :: * -> *). OOProg r => GenState [SFile r]
genInputMod
  [SFile r]
con    <- GenState [SFile r]
forall (r :: * -> *). OOProg r => GenState [SFile r]
genConstMod
  SFile r
cal    <- GenState (SFile r)
forall (r :: * -> *). OOProg r => GenState (SFile r)
genCalcMod
  [SFile r]
out    <- GenState [SFile r]
forall (r :: * -> *). OOProg r => GenState [SFile r]
genOutputMod
  [SFile r]
moddef <- (Mod -> GenState (SFile r)) -> [Mod] -> GenState [SFile r]
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> [a] -> f [b]
traverse Mod -> GenState (SFile r)
forall (r :: * -> *). OOProg r => Mod -> GenState (SFile r)
genModDef (DrasilState -> [Mod]
modules DrasilState
g) -- hack ?
  [SFile r] -> GenState [SFile r]
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return ([SFile r] -> GenState [SFile r])
-> [SFile r] -> GenState [SFile r]
forall a b. (a -> b) -> a -> b
$ SFile r
mn SFile r -> [SFile r] -> [SFile r]
forall a. a -> [a] -> [a]
: [SFile r]
inp [SFile r] -> [SFile r] -> [SFile r]
forall a. [a] -> [a] -> [a]
++ [SFile r]
con [SFile r] -> [SFile r] -> [SFile r]
forall a. [a] -> [a] -> [a]
++ SFile r
cal SFile r -> [SFile r] -> [SFile r]
forall a. a -> [a] -> [a]
: [SFile r]
out [SFile r] -> [SFile r] -> [SFile r]
forall a. [a] -> [a] -> [a]
++ [SFile r]
moddef

-- Procedural Versions --

-- | Generates a package with the given 'DrasilState'. The passed
-- un-representation functions determine which target language the package will
-- be generated in.
generateCodeProc :: (ProcProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
  Lang -> (progRepr (Proc.Program progRepr) -> ProgData) ->
  (packRepr PackageData -> PackageData) ->
  DrasilState -> FileLayout
generateCodeProc :: forall (progRepr :: * -> *) (packRepr :: * -> *).
(ProcProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
Lang
-> (progRepr (Program progRepr) -> ProgData)
-> (packRepr PackageData -> PackageData)
-> DrasilState
-> FileLayout
generateCodeProc Lang
l progRepr (Program progRepr) -> ProgData
unReprProg packRepr PackageData -> PackageData
unReprPack DrasilState
g =
  let dirName :: Name
dirName = Lang -> Name
getDir Lang
l
      (packRepr PackageData
pckg, DrasilState
ds) = State DrasilState (packRepr PackageData)
-> DrasilState -> (packRepr PackageData, DrasilState)
forall s a. State s a -> s -> (a, s)
runState ((progRepr (Program progRepr) -> ProgData)
-> State DrasilState (packRepr PackageData)
forall (progRepr :: * -> *) (packRepr :: * -> *).
(ProcProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
(progRepr (Program progRepr) -> ProgData)
-> GenState (packRepr PackageData)
genPackageProc progRepr (Program progRepr) -> ProgData
unReprProg) DrasilState
g
      (PackageData ProgData
prog [FileLayout]
progDossier) = packRepr PackageData -> PackageData
unReprPack packRepr PackageData
pckg
      designLogFile :: [FileLayout]
designLogFile = [PathSegment -> Doc -> FileLayout
forall doc. Writeable doc => PathSegment -> doc -> FileLayout
file [ps|designLog.txt|] (DrasilState
ds DrasilState -> Getting Doc DrasilState Doc -> Doc
forall s a. s -> Getting a s a -> a
^. Getting Doc DrasilState Doc
Lens' DrasilState Doc
designLog) |
                        Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$ Doc -> Bool
isEmpty (Doc -> Bool) -> Doc -> Bool
forall a b. (a -> b) -> a -> b
$ DrasilState
ds DrasilState -> Getting Doc DrasilState Doc -> Doc
forall s a. s -> Getting a s a -> a
^. Getting Doc DrasilState Doc
Lens' DrasilState Doc
designLog]
      packageFiles :: [FileLayout]
packageFiles = [FileData] -> [FileLayout]
toFileLayout (ProgData -> [FileData]
progMods ProgData
prog) [FileLayout] -> [FileLayout] -> [FileLayout]
forall a. [a] -> [a] -> [a]
++ [FileLayout]
progDossier
        [FileLayout] -> [FileLayout] -> [FileLayout]
forall a. [a] -> [a] -> [a]
++ [FileLayout]
designLogFile
  in
    PathSegment -> [FileLayout] -> FileLayout
forall (f :: * -> *).
Foldable f =>
PathSegment -> f FileLayout -> FileLayout
directory
      [ps|{dirName}|]
      [FileLayout]
packageFiles

-- | Generates a package, including a Makefile, sample input file, and Doxygen
-- configuration file (all subject to the user's choices).
-- The passed un-representation function determines which target language the
-- package will be generated in.
-- GOOL's static code analysis interpreter is called to initialize the state
-- used by the language renderer.
genPackageProc :: (ProcProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
  (progRepr (Proc.Program progRepr) -> ProgData) ->
  GenState (packRepr PackageData)
genPackageProc :: forall (progRepr :: * -> *) (packRepr :: * -> *).
(ProcProg progRepr, SoftwareDossierSym packRepr, Monad packRepr) =>
(progRepr (Program progRepr) -> ProgData)
-> GenState (packRepr PackageData)
genPackageProc progRepr (Program progRepr) -> ProgData
unRepr = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  State GOOLState (CodeInfoProc GOOLState)
ci <- StateT
  DrasilState Identity (State GOOLState (CodeInfoProc GOOLState))
GenState (GSProgram CodeInfoProc)
forall (r :: * -> *). ProcProg r => GenState (GSProgram r)
genProgramProc
  State GOOLState (progRepr (Program progRepr))
p <- GenState (State GOOLState (progRepr (Program progRepr)))
forall (r :: * -> *). ProcProg r => GenState (GSProgram r)
genProgramProc
  let info :: GOOLState
info = CodeInfoProc GOOLState -> GOOLState
forall a. CodeInfoProc a -> a
Proc.unCI (CodeInfoProc GOOLState -> GOOLState)
-> CodeInfoProc GOOLState -> GOOLState
forall a b. (a -> b) -> a -> b
$ State GOOLState (CodeInfoProc GOOLState)
-> GOOLState -> CodeInfoProc GOOLState
forall s a. State s a -> s -> a
evalState State GOOLState (CodeInfoProc GOOLState)
ci GOOLState
initialState
      (progRepr (Program progRepr)
reprPD, GOOLState
s) = State GOOLState (progRepr (Program progRepr))
-> GOOLState -> (progRepr (Program progRepr), GOOLState)
forall s a. State s a -> s -> (a, s)
runState State GOOLState (progRepr (Program progRepr))
p GOOLState
info
      fileInfoState :: SoftwareDossierState
fileInfoState = [Name] -> [Name] -> Maybe Name -> SoftwareDossierState
makeSds (GOOLState
s GOOLState -> Getting [Name] GOOLState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] GOOLState [Name]
Lens' GOOLState [Name]
headers) (GOOLState
s GOOLState -> Getting [Name] GOOLState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] GOOLState [Name]
Lens' GOOLState [Name]
sources) (GOOLState
s GOOLState
-> Getting (Maybe Name) GOOLState (Maybe Name) -> Maybe Name
forall s a. s -> Getting a s a -> a
^. Getting (Maybe Name) GOOLState (Maybe Name)
Lens' GOOLState (Maybe Name)
mainMod)
      pd :: ProgData
pd = progRepr (Program progRepr) -> ProgData
unRepr progRepr (Program progRepr)
reprPD
      m :: packRepr FileLayout
m = [Name]
-> ImplementationType
-> [Comments]
-> SoftwareDossierState
-> ProgData
-> packRepr FileLayout
forall (r :: * -> *).
SoftwareDossierSym r =>
[Name]
-> ImplementationType
-> [Comments]
-> SoftwareDossierState
-> ProgData
-> r FileLayout
makefile (DrasilState -> [Name]
libPaths DrasilState
g) (DrasilState
g DrasilState
-> Getting ImplementationType DrasilState ImplementationType
-> ImplementationType
forall s a. s -> Getting a s a -> a
^. Getting ImplementationType DrasilState ImplementationType
forall a. HasChoices a => Lens' a ImplementationType
Lens' DrasilState ImplementationType
implType) (DrasilState
g DrasilState
-> Getting [Comments] DrasilState [Comments] -> [Comments]
forall s a. s -> Getting a s a -> a
^. Getting [Comments] DrasilState [Comments]
forall a. HasChoices a => Lens' a [Comments]
Lens' DrasilState [Comments]
commented) SoftwareDossierState
fileInfoState ProgData
pd
      as :: [Name]
as = (Person -> Name) -> [Person] -> [Name]
forall a b. (a -> b) -> [a] -> [b]
map Person -> Name
forall n. HasName n => n -> Name
fullName (DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Person] CodeSpec [Person] -> [Person]
forall s a. s -> Getting a s a -> a
^. Getting [Person] CodeSpec [Person]
forall c. HasOldCodeSpec c => Lens' c [Person]
Lens' CodeSpec [Person]
authorsO)
      cfp :: [RelativeFile]
cfp = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec
-> Getting [RelativeFile] CodeSpec [RelativeFile] -> [RelativeFile]
forall s a. s -> Getting a s a -> a
^. Getting [RelativeFile] CodeSpec [RelativeFile]
forall c. HasOldCodeSpec c => Lens' c [RelativeFile]
Lens' CodeSpec [RelativeFile]
configFilesO
      db :: PrintingInformation
db = DrasilState -> PrintingInformation
printfo DrasilState
g
      prps :: Name
prps = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
purpose)
      bckgrnd :: Name
bckgrnd = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
background)
      mtvtn :: Name
mtvtn = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
motivation)
      scp :: Name
scp = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec PrintingInformation
db ([Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
scope)
  Maybe (packRepr FileLayout)
i <- GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
Applicative r =>
GenState (Maybe (r FileLayout))
genSampleInput
  Maybe (packRepr FileLayout)
d <- SoftwareDossierState -> GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
SoftwareDossierSym r =>
SoftwareDossierState -> GenState (Maybe (r FileLayout))
genDoxConfig SoftwareDossierState
fileInfoState
  Maybe (packRepr FileLayout)
rm <- ReadMeInfo -> GenState (Maybe (packRepr FileLayout))
forall (r :: * -> *).
SoftwareDossierSym r =>
ReadMeInfo -> GenState (Maybe (r FileLayout))
genReadMe ReadMeInfo {
        langName :: Name
langName = Name
"",
        langVersion :: Name
langVersion = Name
"",
        invalidOS :: Maybe Name
invalidOS = Maybe Name
forall a. Maybe a
Nothing,
        implementType :: ImplementationType
implementType = DrasilState
g DrasilState
-> Getting ImplementationType DrasilState ImplementationType
-> ImplementationType
forall s a. s -> Getting a s a -> a
^. Getting ImplementationType DrasilState ImplementationType
forall a. HasChoices a => Lens' a ImplementationType
Lens' DrasilState ImplementationType
implType,
        extLibNV :: [(Name, Name)]
extLibNV = DrasilState -> [(Name, Name)]
extLibNames DrasilState
g,
        extLibFP :: [Name]
extLibFP = DrasilState -> [Name]
libPaths DrasilState
g,
        contributors :: [Name]
contributors = [Name]
as,
        configFP :: [RelativeFile]
configFP = [RelativeFile]
cfp,
        caseName :: Name
caseName = Name
"",
        examplePurpose :: Name
examplePurpose = Name
prps,
        exampleDescr :: Name
exampleDescr = Name
bckgrnd,
        exampleMotivation :: Name
exampleMotivation = Name
mtvtn,
        exampleScope :: Name
exampleScope = Name
scp,
        folderNum :: Int
folderNum = DrasilState -> Int
getVal DrasilState
g,
        inputOutput :: (Name, Name)
inputOutput = (Name
sampleInputName, Name
"output.txt")} -- This needs a more permanent solution
  packRepr PackageData -> GenState (packRepr PackageData)
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return (packRepr PackageData -> GenState (packRepr PackageData))
-> packRepr PackageData -> GenState (packRepr PackageData)
forall a b. (a -> b) -> a -> b
$ ProgData -> [packRepr FileLayout] -> packRepr PackageData
forall (r :: * -> *).
Monad r =>
ProgData -> [r FileLayout] -> r PackageData
package ProgData
pd (packRepr FileLayout
mpackRepr FileLayout
-> [packRepr FileLayout] -> [packRepr FileLayout]
forall a. a -> [a] -> [a]
:[Maybe (packRepr FileLayout)] -> [packRepr FileLayout]
forall a. [Maybe a] -> [a]
catMaybes [Maybe (packRepr FileLayout)
i,Maybe (packRepr FileLayout)
rm,Maybe (packRepr FileLayout)
d])

-- | Generates an SCS program based on the problem and the user's design choices.
genProgramProc :: (ProcProg r) => GenState (Proc.GSProgram r)
genProgramProc :: forall (r :: * -> *). ProcProg r => GenState (GSProgram r)
genProgramProc = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  [FS (r (File r))]
ms <- Modularity -> StateT DrasilState Identity [FS (r (File r))]
forall (r :: * -> *).
ProcProg r =>
Modularity -> GenState [SFile r]
chooseModulesProc (Modularity -> StateT DrasilState Identity [FS (r (File r))])
-> Modularity -> StateT DrasilState Identity [FS (r (File r))]
forall a b. (a -> b) -> a -> b
$ DrasilState
g DrasilState
-> Getting Modularity DrasilState Modularity -> Modularity
forall s a. s -> Getting a s a -> a
^. Getting Modularity DrasilState Modularity
forall a. HasChoices a => Lens' a Modularity
Lens' DrasilState Modularity
modular
  let n :: Name
n = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting Name CodeSpec Name -> Name
forall s a. s -> Getting a s a -> a
^. Getting Name CodeSpec Name
forall c. HasOldCodeSpec c => Lens' c Name
Lens' CodeSpec Name
pNameO
  let p :: Name
p = Doc -> Name
forall a. Show a => a -> Name
show (Doc -> Name) -> Doc -> Name
forall a b. (a -> b) -> a -> b
$ SingleLine -> Spec -> Doc
sentenceDoc SingleLine
OneLine (Spec -> Doc) -> Spec -> Doc
forall a b. (a -> b) -> a -> b
$ PrintingInformation -> Sentence -> Spec
spec (DrasilState -> PrintingInformation
printfo DrasilState
g) (Sentence -> Spec) -> Sentence -> Spec
forall a b. (a -> b) -> a -> b
$ [Sentence] -> Sentence
foldlSent ([Sentence] -> Sentence) -> [Sentence] -> Sentence
forall a b. (a -> b) -> a -> b
$ DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting [Sentence] CodeSpec [Sentence] -> [Sentence]
forall s a. s -> Getting a s a -> a
^. Getting [Sentence] CodeSpec [Sentence]
forall c. HasSystemMeta c => Lens' c [Sentence]
Lens' CodeSpec [Sentence]
purpose
  GSProgram r -> GenState (GSProgram r)
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return (GSProgram r -> GenState (GSProgram r))
-> GSProgram r -> GenState (GSProgram r)
forall a b. (a -> b) -> a -> b
$ Name -> Name -> [FS (r (File r))] -> GSProgram r
forall (r :: * -> *).
ProgramSym r =>
Name -> Name -> [SFile r] -> GSProgram r
Proc.prog Name
n Name
p [FS (r (File r))]
ms

-- | Generates either a single module or many modules, based on the users choice
-- of modularity.
chooseModulesProc :: (ProcProg r) => Modularity -> GenState [Proc.SFile r]
chooseModulesProc :: forall (r :: * -> *).
ProcProg r =>
Modularity -> GenState [SFile r]
chooseModulesProc Modularity
Unmodular = State DrasilState (FS (r (File r)))
-> StateT DrasilState Identity [FS (r (File r))]
forall a b. State a b -> State a [b]
liftS State DrasilState (FS (r (File r)))
forall (r :: * -> *). ProcProg r => GenState (SFile r)
genUnmodularProc
chooseModulesProc Modularity
Modular = StateT DrasilState Identity [FS (r (File r))]
forall (r :: * -> *). ProcProg r => GenState [SFile r]
genModulesProc

-- | Generates an entire SCS program as a single module.
genUnmodularProc :: (ProcProg r) => GenState (Proc.SFile r)
genUnmodularProc :: forall (r :: * -> *). ProcProg r => GenState (SFile r)
genUnmodularProc = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  Name
umDesc <- GenState Name
unmodularDesc
  Name
giName <- InternalConcept -> GenState Name
genICName InternalConcept
GetInput
  Name
dvName <- InternalConcept -> GenState Name
genICName InternalConcept
DerivedValuesFn
  Name
icName <- InternalConcept -> GenState Name
genICName InternalConcept
InputConstraintsFn
  let n :: Name
n = DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec -> Getting Name CodeSpec Name -> Name
forall s a. s -> Getting a s a -> a
^. Getting Name CodeSpec Name
forall c. HasOldCodeSpec c => Lens' c Name
Lens' CodeSpec Name
pNameO
      cls :: Bool
cls = (Name -> Bool) -> [Name] -> Bool
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Bool
any (Name -> ModExportMap -> Bool
forall k a. Ord k => k -> Map k a -> Bool
`member` DrasilState -> ModExportMap
clsMap DrasilState
g) [Name
giName, Name
dvName, Name
icName]
  if Bool
cls then Name -> GenState (SFile r)
forall a. HasCallStack => Name -> a
error Name
"genUnmodularProc: Procedural renderers do not support classes"
  else Name
-> Name
-> [Name]
-> [GenState (Maybe (SMethod r))]
-> GenState (SFile r)
forall (r :: * -> *).
ProcProg r =>
Name
-> Name
-> [Name]
-> [GenState (Maybe (SMethod r))]
-> GenState (SFile r)
genModuleWithImportsProc Name
n Name
umDesc ((ExtLibState -> [Name]) -> [ExtLibState] -> [Name]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (ExtLibState -> Getting [Name] ExtLibState [Name] -> [Name]
forall s a. s -> Getting a s a -> a
^. Getting [Name] ExtLibState [Name]
Lens' ExtLibState [Name]
imports) (ExtLibMap -> [ExtLibState]
forall k a. Map k a -> [a]
elems (ExtLibMap -> [ExtLibState]) -> ExtLibMap -> [ExtLibState]
forall a b. (a -> b) -> a -> b
$ DrasilState -> ExtLibMap
extLibMap DrasilState
g))
        (GenState (Maybe (SMethod r))
forall (r :: * -> *). SharedProg r => GenState (Maybe (SMethod r))
genMainFuncProc
          GenState (Maybe (SMethod r))
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. a -> [a] -> [a]
: (StateT DrasilState Identity (SMethod r)
 -> GenState (Maybe (SMethod r)))
-> [StateT DrasilState Identity (SMethod r)]
-> [GenState (Maybe (SMethod r))]
forall a b. (a -> b) -> [a] -> [b]
map ((SMethod r -> Maybe (SMethod r))
-> StateT DrasilState Identity (SMethod r)
-> GenState (Maybe (SMethod r))
forall a b.
(a -> b)
-> StateT DrasilState Identity a -> StateT DrasilState Identity b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap SMethod r -> Maybe (SMethod r)
forall a. a -> Maybe a
Just) ((CodeDefinition -> StateT DrasilState Identity (SMethod r))
-> [CodeDefinition] -> [StateT DrasilState Identity (SMethod r)]
forall a b. (a -> b) -> [a] -> [b]
map CodeDefinition -> StateT DrasilState Identity (SMethod r)
forall (r :: * -> *).
SharedProg r =>
CodeDefinition -> GenState (SMethod r)
genCalcFuncProc (DrasilState -> CodeSpec
codeSpec DrasilState
g CodeSpec
-> Getting [CodeDefinition] CodeSpec [CodeDefinition]
-> [CodeDefinition]
forall s a. s -> Getting a s a -> a
^. Getting [CodeDefinition] CodeSpec [CodeDefinition]
forall c. HasOldCodeSpec c => Lens' c [CodeDefinition]
Lens' CodeSpec [CodeDefinition]
execOrderO)
            [StateT DrasilState Identity (SMethod r)]
-> [StateT DrasilState Identity (SMethod r)]
-> [StateT DrasilState Identity (SMethod r)]
forall a. [a] -> [a] -> [a]
++ (Mod -> [StateT DrasilState Identity (SMethod r)])
-> [Mod] -> [StateT DrasilState Identity (SMethod r)]
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap Mod -> [StateT DrasilState Identity (SMethod r)]
forall (r :: * -> *). SharedProg r => Mod -> [GenState (SMethod r)]
genModFuncsProc (DrasilState -> [Mod]
modules DrasilState
g))
          [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. [a] -> [a] -> [a]
++ ([VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
SharedProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputFormatProc VisibilityTag
Pub, VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
SharedProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputDerivedProc VisibilityTag
Pub,
              VisibilityTag -> GenState (Maybe (SMethod r))
forall (r :: * -> *).
SharedProg r =>
VisibilityTag -> GenState (Maybe (SMethod r))
genInputConstraintsProc VisibilityTag
Pub] [GenState (Maybe (SMethod r))]
-> [GenState (Maybe (SMethod r))] -> [GenState (Maybe (SMethod r))]
forall a. [a] -> [a] -> [a]
++ [GenState (Maybe (SMethod r))
forall (r :: * -> *). SharedProg r => GenState (Maybe (SMethod r))
genOutputFormatProc]))

-- | Generates all modules for an SCS program.
genModulesProc :: (ProcProg r) => GenState [Proc.SFile r]
genModulesProc :: forall (r :: * -> *). ProcProg r => GenState [SFile r]
genModulesProc = do
  DrasilState
g <- StateT DrasilState Identity DrasilState
forall s (m :: * -> *). MonadState s m => m s
get
  SFile r
mn     <- GenState (SFile r)
forall (r :: * -> *). ProcProg r => GenState (SFile r)
genMainProc
  [SFile r]
inp    <- GenState [SFile r]
forall (r :: * -> *). ProcProg r => GenState [SFile r]
genInputModProc
  Bool
con    <- GenState Bool
checkConstClass
  SFile r
cal    <- GenState (SFile r)
forall (r :: * -> *). ProcProg r => GenState (SFile r)
genCalcModProc
  [SFile r]
out    <- GenState [SFile r]
forall (r :: * -> *). ProcProg r => GenState [SFile r]
genOutputModProc
  [SFile r]
moddef <- (Mod -> GenState (SFile r)) -> [Mod] -> GenState [SFile r]
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
forall (f :: * -> *) a b.
Applicative f =>
(a -> f b) -> [a] -> f [b]
traverse Mod -> GenState (SFile r)
forall (r :: * -> *). ProcProg r => Mod -> GenState (SFile r)
genModDefProc (DrasilState -> [Mod]
modules DrasilState
g) -- hack ?
  if Bool
con then Name -> GenState [SFile r]
forall a. HasCallStack => Name -> a
error Name
"genModulesProc: Procedural renderers do not support classes"
  else [SFile r] -> GenState [SFile r]
forall a. a -> StateT DrasilState Identity a
forall (m :: * -> *) a. Monad m => a -> m a
return ([SFile r] -> GenState [SFile r])
-> [SFile r] -> GenState [SFile r]
forall a b. (a -> b) -> a -> b
$ SFile r
mn SFile r -> [SFile r] -> [SFile r]
forall a. a -> [a] -> [a]
: [SFile r]
inp [SFile r] -> [SFile r] -> [SFile r]
forall a. [a] -> [a] -> [a]
++ SFile r
cal SFile r -> [SFile r] -> [SFile r]
forall a. a -> [a] -> [a]
: [SFile r]
out [SFile r] -> [SFile r] -> [SFile r]
forall a. [a] -> [a] -> [a]
++ [SFile r]
moddef

-- | Private utilities used in 'generateCode'.
getDir :: Lang -> String
getDir :: Lang -> Name
getDir Lang
Cpp = Name
"cpp"
getDir Lang
CSharp = Name
"csharp"
getDir Lang
Java = Name
"java"
getDir Lang
Python = Name
"python"
getDir Lang
Swift = Name
"swift"
getDir Lang
Julia = Name
"julia"
getDir Lang
Matlab = Name
"matlab"