git.iko.soy/rowtype-yoga/purescript-native
on 127af89 help sign in
24 files changed+918−1497
since splitsnapshots
unifiedsplit
.gitignore+1−2 README-go.md−48 README.md+54−25 app/Main.hs+46−161 default.nix−23 package.yaml+6−9 psgo.cabal−145 runtime/purescript.go+22 src/CodeGen/IL.hs+131−31 src/CodeGen/IL/Common.hs+130−104 src/CodeGen/IL/Optimizer.hs+93−34 src/CodeGen/IL/Optimizer/Common.hs−18 src/CodeGen/IL/Optimizer/Inliner.hs−435 src/CodeGen/IL/Optimizer/MagicDo.hs−146 src/CodeGen/IL/Optimizer/TCO.hs+17−15 src/CodeGen/IL/Printer.hs+209−221 src/Tests.hs+12−11 stack.nix+3−18 stack.yaml+65−16 support/Makefile+129 support/ffi-loader.go−30 support/go.mod.ffi-loader−2 support/go.mod.output−2 support/go.mod.working−1
.gitignore+1−2
@@ -12,12 +12,11 @@ cabal.sandbox.config
1212.test_modules/
1313bower_components/
1414node_modules
15+pscpp.cabal
1516tmp/
1617.stack-work/
17−stack.yaml.lock
1818output
1919examples/docs/docs/
2020core-tests/full-core-docs.md
2121.psc-ide-port
2222.vscode/
23−result
README-go.mddeleted−48
@@ -1,48 +0,0 @@
1−
2−### psgo
3−
4−#### Performance
5−
6−* No runtime system beyond some support types and functions
7−* Uses PureScript's normal tail call optimization techniques for generated Go code
8−
9−#### Differences from PureScript:
10−
11−* Foreign imports/exports are Go instead of JavaScript – see [standard library foreign implementations](https://github.com/andyarvanitis/purescript-native-go-ffi)
12−* No Go-specific REPL
13−
14−#### Other notes:
15−
16−* The provided `Any` type is just an alias for `interface{}`
17−* PureScript types are represented in Go with:
18−
19− | PureScript | Go |
20− |------------|----|
21− |`Array` | `[]Any`|
22− |`String` | `string`|
23− |`Number` | `float64`|
24− |`Int` | `int`|
25− |`Char` | `string` (single UTF-8 entity)|
26− |`Boolean` | `bool`|
27− |Records | `map[string]Any`|
28−
29−#### Future ideas:
30−
31−* Nice facilities (modules) for concurrency/parallelism, using goroutines
32−
33−
34−#### Getting Started
35−This assumes you are running macOS (OS X) or a Unix-like system (Linux, *BSD, etc.).
36−
37−1. Choose a working directory wherever you like. Then create a `src` subdirectory under it, which will be where you will place your own PureScript source files.
38−
39−1. Initialize your project with `spago init`, and install any dependencies with `spago install` (please see their instructions if you haven't used it before).
40−
41−1. Add the `backend = "psgo"` option to your `spago.dhall` file. That way `spago` will use the psgo backend when compiling.
42−
43−1. You should now be ready to build a PureScript program:
44− * As stated above, place your source file(s) in the working directory's `src` subdirectory and execute `spago build`.
45−
46− * This will generate the Go source tree for your program and then build an executable binary. The resulting executable will be in your working directory and will be named `Main`. The generated source files will be under `output/`.
47−
48−---
README.md+54−25
@@ -1,41 +1,70 @@
11[![PureScript](https://raw.githubusercontent.com/purescript/purescript/master/logo.png)](http://purescript.org)
22
3−This is a platform-native compiler backend for [PureScript](https://github.com/purescript/purescript). It attempts to generate "sane", debuggable, and portable [C++11](https://isocpp.org/wiki/faq/cpp11) (or later) or [Go](https://golang.org/) code as an intermediate language, which is then compiled to a native executable binary. This also enables easy interoperability with existing C/C++ or Go frameworks and libraries on a number of platforms.
3+This is an experimental C++11 (or later) compiler backend for [PureScript](https://github.com/purescript/purescript). It attempts to generate "sane", debuggable, and portable C++ code as an intermediate language, which is then compiled to a native executable binary. This enables easy interoperability with existing C/C++ frameworks and libraries on a number of platforms.
44
5−There are two utilities in the purescript-native "suite": `pscpp` and `psgo`. The source code in this branch is for `psgo`, the purescript-to-Go transpiler. For the source code of the `pscpp` purescript-to-C++ transpiler, please see the [cpp branch](https://github.com/andyarvanitis/purescript-native/tree/cpp).
5+---
6+
7+### **Please note that this project supersedes [pure11](https://github.com/pure11/pure11)**
8+
9+#### Please see [changes](https://github.com/andyarvanitis/purescript-native/wiki/changes) for more details.
10+
11+---
612
713#### Platforms
8−* Although purescript-native should work on any platform supporting PureScript and modern C++ or Go, the generated code and resulting binary executables have been tested on:
9− * macOS Mojave 10.14.5 – full test suite
10− * For C++, Xcode 10.2.1's `clang` was used
11− * For Go, `go` version 1.12.7 was used
12− * Windows 10 x64 – full test suite
13− * C++: Visual Studio 2017 and `clang`
14− * Go: `go` version 1.12.7
15− * Linux Debian 9.5 amd64 – full test suite (C++)
16− * Default versions of `clang` and `gcc`
17− * Linux Debian 10 amd64 – full test suite (Go)
18− * Default version of `go`
19− * Raspberry Pi 3 B+ Raspbian official build (ARM), default versions of `clang` and `gcc`
14+* Generated C++ code and resulting binary executables tested (to at least some extent) on:
15+ * macOS Mojave, Xcode 10.1's clang (full test suite)
16+ * Windows 10 x64 and x86, Visual Studio 2017
17+ * Linux Debian 9.5 amd64, default versions of clang and gcc/g++ (full test suite)
18+ * Raspberry Pi 3 B+ Raspbian official build (ARM), default versions of clang and gcc/g++
19+
20+#### Performance
21+
22+* No runtime system beyond some support classes and the standard C++11 (or later) runtime library
23+* For automatic memory management, uses native C++11 reference counting (`std::shared_ptr`)
24+* Uses PureScript's normal tail call optimization techniques for generated C++ code
25+
26+#### Differences from PureScript:
27+
28+* Foreign imports/exports are C++ instead of JavaScript – see [FFI notes](https://github.com/andyarvanitis/purescript-native/wiki/FFI) and [standard library foreign implementations](https://github.com/andyarvanitis/purescript-native-ffi)
29+* No C++-specific REPL
30+
31+#### Other notes:
32+
33+* PureScript arrays are implemented using [`std::vector`](http://en.cppreference.com/w/cpp/container/vector)
34+* `String` types are implemented with C++11 `u8` literals (UTF-8) and `std::string`
35+* `Number` is C++ `double`, `Int` is C++ `int`, `Char` is `std::string` (single UTF-8 entity), `Boolean` is `bool`
36+
37+#### Future ideas:
38+
39+* Nice facilities (modules) for concurrency/parallelism, using `std::thread`, `std::async`, etc. under the hood (output is already generally thread-safe for immutable values, thanks to `std::shared_ptr`)
40+
41+#### Requirements for building the compiler itself
42+
43+* [Haskell Stack](https://docs.haskellstack.org/en/stable/README/) (if you're running macOS 10.13+, you can use pre-built binaries from [here](https://github.com/andyarvanitis/purescript-native/releases/))
44+
45+#### Requirements for using it
46+
47+* A recent version (0.12+) of [purescript](https://github.com/purescript/purescript/releases).
2048
49+* A C++11-capable toolchain, e.g. recent versions of clang, gcc, Microsoft Visual Studio
50+* GNU Make is the default supported build tool, but you should be able to use your favorite C++ build system, tools, debuggers, etc.
2151
22−#### Requirements for building `pscpp` and `psgo`
52+#### Getting Started
53+This assumes you are running macOS (OS X) or a Unix-like system (Linux, *BSD, etc.).
2354
24−* [Haskell Stack](https://docs.haskellstack.org/en/stable/README/) (if you're running macOS 10.14.5+, you can use pre-built binaries from [here](https://github.com/andyarvanitis/purescript-native/releases/))
55+1. Make sure you have developer tools for your system installed. For macOS, you'll need a recent version of Xcode. For Linux, etc., you can use clang 3.5 or later, or gcc/g++ 4.9.2 or later.
2556
26−#### Requirements for using PureScript + purescript-native
57+2. Create a working directory wherever you like, and a `src` subdirectory under it, which will be where you will place your own PureScript source files.
2758
28−* A recent version (0.13+) of [purescript](https://github.com/purescript/purescript/releases).
59+3. Under your working directory, also create an `ffi` subdirectory, which will be where you will place C/C++ FFI source files. You should at least add the contents of [purescript-native-ffi](https://github.com/andyarvanitis/purescript-native-ffi) into this directory, in addition to any of your own foreign implementations.
2960
30−* For `pscpp`, a C++11-capable toolchain, e.g. recent versions of clang, gcc, Microsoft Visual Studio
31− * GNU Make + `psc-package` is the default supported build tool, but you should be able to use your favorite package manager, C++ build system, tools, debuggers, etc.
32−* For `psgo`, the [Go toolchain](https://golang.org) for your system
33− * You can use your favorite PureScript package manager and build tools – but for simplicity, [`spago`](https://github.com/spacchetti/spago) is recommended.
61+4. Generate the default GNU `Makefile` in your working directory by running `pscpp --makefile`.
3462
63+5. Use PureScript's standard [`psc-package`](https://psc-package.readthedocs.io/en/latest/) utility to add and manage package dependencies.
3564
36−#### For more information and a Getting Started guide, please see
37−* [`pscpp`](https://github.com/andyarvanitis/purescript-native/blob/cpp/README-cpp.md)
38−* [`psgo`](https://github.com/andyarvanitis/purescript-native/blob/golang/README-go.md)
65+6. You should now be ready to build a PureScript program:
66+ * As stated above, place your source file(s) in the working directory's `src` subdirectory and execute `make debug` or `make release`. If your build machine has multiple cores, you might want to append `-jN` to your *make* command, where `N` is the number of cores.
3967
68+ * This will generate the C++ source tree for your program and then build an executable binary. The resulting executable will be in the `bin` subdirectory under the output directory and called `main` (so `output/bin/main`, by default). Source files will be under `src` (`output/src/` by default).
4069
4170---
app/Main.hs+46−161
@@ -10,7 +10,7 @@ import Data.Aeson
1010import Data.Aeson.Types
1111import Data.Char
1212import Data.FileEmbed (embedFile)
13−import Data.List (delete, intercalate, isPrefixOf, nub, partition, (\\))
13+import Data.List (delete, isPrefixOf, partition)
1414import Data.Maybe
1515import Data.Monoid ((<>))
1616import Data.Version
@@ -19,10 +19,8 @@ import Control.Monad.Supply.Class
1919import Text.Printf
2020
2121import System.Environment
22−import System.Directory (createDirectoryIfMissing, doesDirectoryExist, doesFileExist, getCurrentDirectory, getModificationTime)
22+import System.Directory (createDirectoryIfMissing, doesDirectoryExist, doesFileExist, getModificationTime)
2323import System.FilePath ((</>), joinPath, splitDirectories, takeDirectory)
24−import System.FilePath.Find
25−import System.Process
2624
2725import Data.Text (Text)
2826import qualified Data.Text as T
@@ -31,8 +29,6 @@ import qualified Data.Text.Lazy as L
3129import qualified Data.Text.Lazy.Encoding as L
3230import qualified Data.ByteString as B
3331
34−import Development.GitRev
35−
3632import Language.PureScript.AST.Literals
3733import Language.PureScript.CoreFn
3834import Language.PureScript.CoreFn.FromJSON
@@ -43,8 +39,6 @@ import CodeGen.IL.Printer
4339
4440import Tests
4541
46−data Command = Build | Run (Maybe String)
47−
4842parseJson :: Text -> Value
4943parseJson text
5044 | Just fileJson <- decode . L.encodeUtf8 $ L.fromStrict text = fileJson
@@ -54,102 +48,43 @@ jsonToModule :: Value -> Module Ann
5448jsonToModule value =
5549 case parse moduleFromJSON value of
5650 Success (_, r) -> r
57− _ -> error "Failed parsing corefn. Psgo currently supports purescript <= 0.13.8"
58−
59−splitArgs :: [String] -> ([String], Maybe String, [String])
60−splitArgs args =
61− let (opts, files) = partition (isPrefixOf "--") args
62− opts' = (map . map) toLower opts
63− (moduleName, files') =
64− if null files
65− then (Nothing, files)
66− else
67− if "--run" `elem` opts' && length (files) >= 1
68− then do
69− let file0 = head files
70− files' = tail files
71− case file0 of
72− "Main.main" -> (Just "Main", files')
73− "Test.Main.main" -> (Just "Test.Main", files')
74− _ -> (Nothing, files)
75− else (Nothing, files)
76− in (opts, moduleName, files')
51+ _ -> error "failed"
7752
7853main :: IO ()
7954main = do
8055 args <- getArgs
81− -- let (opts, files) = partition (isPrefixOf "--") args
82− let (opts, moduleName, files) = splitArgs args
83− opts' = (map . map) toLower opts
84− if "--tests" `elem` opts'
85− then runTests
56+ if null args
57+ then return ()
8658 else do
87− if "--help" `elem` opts' || "--version" `elem` opts'
88− then do
89− when ("--help" `elem` opts') $ do
90− putStrLn help
91− when ("--version" `elem` opts') $ do
92− let branch = $(gitBranch)
93− details | branch == "golang" = "master, commit " ++ $(gitHash)
94− | otherwise = branch
95− putStrLn $ details ++ if $(gitDirty) then " (DIRTY)" else ""
59+ let (opts, files) = partition (isPrefixOf "--") args
60+ opts' = (map . map) toLower opts
61+ if "--tests" `elem` opts'
62+ then runTests
9663 else do
97− if null files
98− then do
99− currentDir <- getCurrentDirectory
100− processFiles opts' [currentDir]
101− else
102− processFiles opts' files
103− if "--run" `elem` opts
104− then runBuildTool $ Run moduleName
105− else when ("--no-build" `notElem` opts) $
106− runBuildTool Build
107− return ()
108− where
109− processFiles :: [String] -> [FilePath] -> IO ()
110− processFiles opts [file] = do
111− isDir <- doesDirectoryExist file
112− if isDir then do
113− files <- find always (fileName ==? corefn) file
114− if null files
115− then errorNoCorefnFiles
116− else generateFiles opts files
117− else generateFiles opts [file]
118− processFiles opts files = do
119− generateFiles opts files
120− basePath :: [FilePath] -> FilePath
121− basePath files =
122− let filepath = takeDirectory (head files) in
123− joinPath $ (init $ splitDirectories filepath)
124− generateFiles :: [String] -> [FilePath] -> IO ()
125− generateFiles opts files = do
126− let baseOutpath = basePath files
127− writeSupportFiles baseOutpath
128− Par.mapM (generateCode opts baseOutpath) files
129− return ()
130− runBuildTool :: Command -> IO ()
131− runBuildTool cmd = do
132− project <- projectEnv
133− case cmd of
134− Build -> do
135− callProcess "go" ["build", T.unpack project </> modPrefix' </> "Main"]
136− Run moduleName -> do
137− callProcess "go" ["run", T.unpack project </> modPrefix' </> fromMaybe "Main" moduleName]
138− return ()
64+ when ("--makefile" `elem` opts') $
65+ B.writeFile "Makefile" $(embedFile "support/Makefile")
66+ when ("--help" `elem` opts') $
67+ putStrLn help
68+ when (not $ null files) $ do
69+ let filepath = takeDirectory (head files)
70+ baseOutpath = joinPath $ (init $ splitDirectories filepath) ++ [outdir]
71+ writeRuntimeFiles baseOutpath
72+ Par.mapM (generateCode opts' baseOutpath) files
73+ return ()
74+ return ()
13975
14076generateCode :: [String] -> FilePath -> FilePath -> IO ()
14177generateCode opts baseOutpath jsonFile = do
14278 jsonModTime <- getModificationTime jsonFile
14379 let filepath = takeDirectory jsonFile
14480 dirparts = splitDirectories $ filepath
145− mname = last dirparts
81+ mname = (\c -> if c == '.' then '_' else c) <$> last dirparts
14682 basedir = joinPath $ init dirparts
147− mname' = T.pack mname
148− possibleFileName = basedir </> (T.unpack mname') </> implFileName mname'
149− exists <- doesFileExist possibleFileName
83+ possInterfaceFilename = basedir </> outdir </> mname </> interfaceFileName mname
84+ exists <- doesFileExist possInterfaceFilename
15085 if exists
15186 then do
152− modTime <- getModificationTime possibleFileName
87+ modTime <- getModificationTime possInterfaceFilename
15388 when (modTime < jsonModTime) $
15489 transpile opts baseOutpath jsonFile
15590 else transpile opts baseOutpath jsonFile
@@ -157,94 +92,44 @@ generateCode opts baseOutpath jsonFile = do
15792transpile :: [String] -> FilePath -> FilePath -> IO ()
15893transpile opts baseOutpath jsonFile = do
15994 jsonText <- T.decodeUtf8 <$> B.readFile jsonFile
160− project <- projectEnv
16195 let module' = jsonToModule $ parseJson jsonText
162− ((_, foreigns, asts, implHeader, implFooter), _) <- runSupplyT 5 (moduleToIL module' project)
163− let mn = moduleNameToIL' $ moduleName module'
96+ ((interface, foreigns, asts, implHeader, implFooter), _) <- runSupplyT 5 (moduleToIL module' Nothing)
97+ let mn = moduleNameToIL $ moduleName module'
16498 implementation = prettyPrintIL asts
16599 outpath = joinPath [baseOutpath, T.unpack mn]
100+ interfacePath = outpath </> interfaceFileName (T.unpack mn)
166101 implPath = outpath </> implFileName mn
102+ putStrLn interfacePath
167103 createDirectoryIfMissing True outpath
104+ B.writeFile interfacePath $ T.encodeUtf8 interface
168105 putStrLn implPath
169106 B.writeFile implPath $ T.encodeUtf8 (implHeader <> implementation <> implFooter)
170107
171−writeSupportFiles :: FilePath -> IO ()
172−writeSupportFiles baseOutpath = do
173− currentDir <- getCurrentDirectory
174− let outputdir = T.pack $ baseOutpath \\ currentDir
175− ffiOutpath = currentDir </> subdir
108+writeRuntimeFiles :: FilePath -> IO ()
109+writeRuntimeFiles baseOutpath = do
176110 createDirectoryIfMissing True baseOutpath
177− createDirectoryIfMissing True ffiOutpath
178− writeModuleFile outputdir currentDir $(embedFile "support/go.mod.working")
179− writeModuleFile outputdir baseOutpath $(embedFile "support/go.mod.output")
180− writeModuleFile outputdir ffiOutpath $(embedFile "support/go.mod.ffi-loader")
181− writeLoaderFile ffiOutpath $(embedFile "support/ffi-loader.go")
182− where
183− writeModuleFile :: Text -> FilePath -> B.ByteString -> IO ()
184− writeModuleFile outputdir path modText = do
185− let goModSource = path </> "go.mod"
186− goModSourceExists <- doesFileExist goModSource
187− when (not goModSourceExists) $ do
188− project <- projectEnv
189− let project' = T.unpack project
190− outputdir' = tail $ T.unpack outputdir
191− modText' = T.replace "$PROJECT" project $ T.decodeUtf8 modText
192− replaceLoader = project' </> ffiLoader' <> "=." </> subdir
193− replaceOutput = project' </> modPrefix' <> "=." </> outputdir'
194− B.writeFile goModSource $ T.encodeUtf8 modText'
195− callProcess "go" ["mod", "edit", "-replace", replaceLoader]
196− callProcess "go" ["mod", "edit", "-replace", replaceOutput]
197− callProcess "go" ["clean", "-modcache"]
198− writeLoaderFile :: FilePath -> B.ByteString -> IO ()
199− writeLoaderFile ffiOutpath loaderText = do
200− let loaderSource = ffiOutpath </> "ffi_loader.go"
201− loaderSourceExists <- doesFileExist loaderSource
202− when (not loaderSourceExists) $ do
203− B.writeFile loaderSource loaderText
111+ let runtimeSource = baseOutpath </> "purescript.go"
112+ runtimeExists <- doesFileExist runtimeSource
113+ when (not runtimeExists) $ do
114+ B.writeFile runtimeSource $(embedFile "runtime/purescript.go")
204115
205−implFileName :: Text -> FilePath
206−implFileName mn = ((\c -> if c == '.' then '_' else c) <$> T.unpack mn) <> ".go"
116+outdir :: FilePath
117+outdir = "src"
118+
119+interfaceFileName :: String -> FilePath
120+interfaceFileName mn = mn <> ".go"
207121
208−projectEnv :: IO Text
209−projectEnv = do
210− T.pack . fromMaybe defaultProject <$> lookupEnv goproject
122+implFileName :: Text -> FilePath
123+implFileName mn = T.unpack $ mn <> ".go"
211124
212125help :: String
213126help = "Usage: psgo OPTIONS COREFN-FILES\n\
214− \ PureScript to native (via go) compiler\n\n\
127+ \ PureScript-to-go compiler\n\n\
215128 \Available options:\n\
216129 \ --help Show this help text\n\n\
217− \ --version Show the version number\n\n\
218− \ --run Run the generated go code directly, without building an\n\
219− \ executable\n\
220− \ --no-build Generate go source files, but do not build an executable\n\
130+ \ --makefile Generate a GNU Makefile which can be used for compiling\n\
131+ \ a PureScript program and libraries to a native binary via\n\
132+ \ purs corefn output and C++\n\n\
221133 \ --tests Run test cases (under construction)\n\n\
222134 \See also:\n\
223135 \ purs compile --help\n"
224−
225−corefn :: String
226−corefn = "corefn.json"
227−
228−goSrc :: String
229−goSrc = ".go"
230−
231−goproject :: String
232−goproject = "GOPROJECT"
233−
234−defaultProject :: String
235−defaultProject = "project.localhost"
236−
237−modPrefix' :: String
238−modPrefix' = T.unpack modPrefix
239−
240−ffiLoader' :: String
241−ffiLoader' = T.unpack ffiLoader
242−
243−subdir :: String
244−subdir = T.unpack modLabel
245−
246−errorNoCorefnFiles :: IO ()
247−errorNoCorefnFiles = do
248− ioError . userError $ "no compiled purescript '" <> corefn <> "' files found –\n" <>
249− " make sure to use the '--codegen corefn' option with your purs\n" <>
250− " project build tool"
default.nixdeleted−23
@@ -1,23 +0,0 @@
1−let
2− haskellNix = import (builtins.fetchTarball https://github.com/input-output-hk/haskell.nix/archive/master.tar.gz) {};
3− nixpkgsSrc = haskellNix.sources.nixpkgs-1903; # has hsc2hs-0.68.4
4−
5− hsc2hsOverlay = self: super: {
6− buildPackages = super.buildPackages // {
7− hsc2hs = super.haskellPackages.hsc2hs;
8− };
9− };
10−
11− nixpkgsArgs = haskellNix.nixpkgsArgs // {
12− overlays = haskellNix.nixpkgsArgs.overlays ++ [ hsc2hsOverlay ];
13− };
14−in
15−{ pkgs ? import nixpkgsSrc nixpkgsArgs
16−}:
17−
18−pkgs.haskell-nix.stackProject {
19− src = pkgs.haskell-nix.haskellLib.cleanGit {
20− name = "purescript-native";
21− src = ./.;
22− };
23−}
package.yaml+6−9
@@ -9,8 +9,7 @@ copyright: "2018 Andy Arvanitis"
99extra-source-files:
1010- README.md
1111- ChangeLog.md
12−- stack.yaml
13−- support/*
12+- runtime/*.go
1413
1514# Metadata used when publishing your package
1615# synopsis: Short description of your package
@@ -24,7 +23,7 @@ description: Please see the README on Github at <https://github.com/andy
2423dependencies:
2524- base >= 4.7 && < 5
2625- bytestring >= 0.10.8.2 && < 0.11
27−- purescript
26+- purescript -any
2827- base-compat >=0.6.0
2928- protolude >=0.1.6
3029- text
@@ -35,12 +34,10 @@ dependencies:
3534- safe >=0.3.9 && <0.4
3635- transformers >=0.3.0 && <0.6
3736- mtl >=2.1.0 && <2.3.0
38−- aeson >=1.0 && <1.5
39−- directory
40−- process
41−- file-embed
42−- filemanip
43−- gitrev
37+- aeson >=1.0 && <1.4
38+- directory -any
39+- process -any
40+- file-embed -any
4441
4542library:
4643 source-dirs: src
psgo.cabaldeleted−145
@@ -1,145 +0,0 @@
1−cabal-version: 1.12
2−
3−-- This file has been generated from package.yaml by hpack version 0.35.0.
4−--
5−-- see: https://github.com/sol/hpack
6−
7−name: psgo
8−version: 0.1.0.0
9−description: Please see the README on Github at <https://github.com/andyarvanitis/purescript-native#readme>
10−homepage: https://github.com/andyarvanitis/purescript-native#readme
11−bug-reports: https://github.com/andyarvanitis/purescript-native/issues
12−author: Andy Arvanitis
13−maintainer: example@example.com
14−copyright: 2018 Andy Arvanitis
15−license: BSD3
16−license-file: LICENSE
17−build-type: Simple
18−extra-source-files:
19− README.md
20− ChangeLog.md
21− stack.yaml
22− support/ffi-loader.go
23− support/go.mod.ffi-loader
24− support/go.mod.output
25− support/go.mod.working
26−
27−source-repository head
28− type: git
29− location: https://github.com/andyarvanitis/purescript-native
30−
31−library
32− exposed-modules:
33− CodeGen.IL
34− CodeGen.IL.Common
35− CodeGen.IL.Optimizer
36− CodeGen.IL.Optimizer.Common
37− CodeGen.IL.Optimizer.Inliner
38− CodeGen.IL.Optimizer.MagicDo
39− CodeGen.IL.Optimizer.TCO
40− CodeGen.IL.Printer
41− Tests
42− other-modules:
43− Paths_psgo
44− hs-source-dirs:
45− src
46− default-extensions:
47− ConstraintKinds
48− DataKinds
49− DeriveFunctor
50− EmptyDataDecls
51− FlexibleContexts
52− KindSignatures
53− LambdaCase
54− MultiParamTypeClasses
55− NoImplicitPrelude
56− PatternGuards
57− PatternSynonyms
58− RankNTypes
59− RecordWildCards
60− OverloadedStrings
61− ScopedTypeVariables
62− TupleSections
63− ViewPatterns
64− build-depends:
65− aeson >=1.0 && <1.5
66− , base >=4.7 && <5
67− , base-compat >=0.6.0
68− , bytestring >=0.10.8.2 && <0.11
69− , containers
70− , directory
71− , file-embed
72− , filemanip
73− , filepath
74− , gitrev
75− , monad-parallel >=0.7.2.2 && <0.8
76− , mtl >=2.1.0 && <2.3.0
77− , pattern-arrows >=0.0.2 && <0.1
78− , process
79− , protolude >=0.1.6
80− , purescript
81− , safe >=0.3.9 && <0.4
82− , text
83− , transformers >=0.3.0 && <0.6
84− default-language: Haskell2010
85−
86−executable psgo
87− main-is: Main.hs
88− other-modules:
89− Paths_psgo
90− hs-source-dirs:
91− app
92− ghc-options: -threaded -rtsopts -with-rtsopts=-N
93− build-depends:
94− aeson >=1.0 && <1.5
95− , base >=4.7 && <5
96− , base-compat >=0.6.0
97− , bytestring >=0.10.8.2 && <0.11
98− , containers
99− , directory
100− , file-embed
101− , filemanip
102− , filepath
103− , gitrev
104− , monad-parallel >=0.7.2.2 && <0.8
105− , mtl >=2.1.0 && <2.3.0
106− , pattern-arrows >=0.0.2 && <0.1
107− , process
108− , protolude >=0.1.6
109− , psgo
110− , purescript
111− , safe >=0.3.9 && <0.4
112− , text
113− , transformers >=0.3.0 && <0.6
114− default-language: Haskell2010
115−
116−test-suite psgo-test
117− type: exitcode-stdio-1.0
118− main-is: Spec.hs
119− other-modules:
120− Paths_psgo
121− hs-source-dirs:
122− test
123− ghc-options: -threaded -rtsopts -with-rtsopts=-N
124− build-depends:
125− aeson >=1.0 && <1.5
126− , base >=4.7 && <5
127− , base-compat >=0.6.0
128− , bytestring >=0.10.8.2 && <0.11
129− , containers
130− , directory
131− , file-embed
132− , filemanip
133− , filepath
134− , gitrev
135− , monad-parallel >=0.7.2.2 && <0.8
136− , mtl >=2.1.0 && <2.3.0
137− , pattern-arrows >=0.0.2 && <0.1
138− , process
139− , protolude >=0.1.6
140− , psgo
141− , purescript
142− , safe >=0.3.9 && <0.4
143− , text
144− , transformers >=0.3.0 && <0.6
145− default-language: Haskell2010
runtime/purescript.gonew file+22
@@ -0,0 +1,22 @@
1+package purescript
2+
3+//go:generate purs --version
4+
5+type Any = interface{}
6+
7+type Fn = func(Any) Any
8+type EffFn = func() Any
9+
10+func Apply(f Any, args ...Any) Any {
11+ result := f
12+ for _, arg := range args {
13+ fn, _ := result.(Fn)
14+ result = fn(arg)
15+ }
16+ return result
17+}
18+
19+func Get(dict Any, key string) Any {
20+ d, _ := dict.(func(string) Any)
21+ return d(key)
22+}
src/CodeGen/IL.hs+131−31
@@ -48,19 +48,23 @@ moduleToIL
4848 -- . (Monad m, MonadReader Options m, MonadSupply m, MonadError MultipleErrors m)
4949 . (Monad m, MonadSupply m)
5050 => Module Ann
51− -> Text
51+ -> Maybe AST
5252 -> m (Text, [Text], [AST], Text, Text)
53−moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
53+moduleToIL (Module _ coms mn _ imps _ foreigns decls) _ =
5454 do
55− ilDecls <- mapM (bindToIL ModuleDecl) decls
55+ let usedNames = concatMap getNames decls
56+ let mnLookup = renameImports usedNames imps
57+ ilImports <- traverse (importToIL mnLookup) . (\\ (mn : C.primModules)) . (\\ [mn]) $ ordNub $ map snd imps
58+ interfaceImport <- importToIL (renameImports [] [(emptyAnn, mn)]) mn
59+ let decls' = renameModules mnLookup decls
60+ ilDecls <- mapM (bindToIL ModuleDecl) decls'
5661 optimized <- traverse (traverse (optimize modName')) ilDecls
5762 let optimized' = concat optimized
5863 values = annotValue <$> optimized'
59− foreigns' = moduleIdentToIL <$> foreigns
64+ foreigns' = identToIL <$> foreigns
6065 interface = interfaceSource modName values foreigns
61− imports = nub . concat $ importToIL <$> optimized'
62− implHeader = implHeaderSource modName imports project
63− implFooter = implFooterSource (runModuleName mn) foreigns
66+ implHeader = implHeaderSource modName ilImports interfaceImport
67+ implFooter = implFooterSource modName foreigns
6468 return $ (interface, foreigns', optimized', implHeader, implFooter)
6569 where
6670 modName = moduleNameToIL mn
@@ -77,16 +81,53 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
7781 getNames (NonRec _ ident _) = [ident]
7882 getNames (Rec vals) = map (snd . fst) vals
7983
80− -- | Generates IL code for a module import
81− --
82− importToIL :: AST -> [Text]
83− importToIL = AST.everything (++) modRef
84+ -- | Creates alternative names for each module to ensure they don't collide
85+ -- with declaration names.
86+ renameImports :: [Ident] -> [(Ann, ModuleName)] -> M.Map ModuleName (Ann, ModuleName)
87+ renameImports = go M.empty
88+ where
89+ go :: M.Map ModuleName (Ann, ModuleName) -> [Ident] -> [(Ann, ModuleName)] -> M.Map ModuleName (Ann, ModuleName)
90+ go acc used ((ann, mn') : mns') =
91+ let mni = Ident $ runModuleName mn'
92+ in if mn' /= mn && mni `elem` used
93+ then let newName = freshModuleName 1 mn' used
94+ in go (M.insert mn' (ann, newName) acc) (Ident (runModuleName newName) : used) mns'
95+ else go (M.insert mn' (ann, mn') acc) used mns'
96+ go acc _ [] = acc
97+
98+ freshModuleName :: Integer -> ModuleName -> [Ident] -> ModuleName
99+ freshModuleName i mn'@(ModuleName pns) used =
100+ let newName = ModuleName $ init pns ++ [ProperName $ runProperName (last pns) <> "_" <> T.pack (show i)]
101+ in if Ident (runModuleName newName) `elem` used
102+ then freshModuleName (i + 1) mn' used
103+ else newName
104+
105+ -- | Generates C++ code for a module import, binding the required module
106+ -- to the alternative
107+ importToIL :: M.Map ModuleName (Ann, ModuleName) -> ModuleName -> m Text
108+ importToIL mnLookup mn' = do
109+ let ((_, _, _, _), mnSafe) = fromMaybe (internalError "Missing value in mnLookup") $ M.lookup mn' mnLookup
110+ mname = moduleNameToIL mnSafe
111+ pure $ "#include \"" <> mname <> "/" <> mname <> ".h\"\n"
112+
113+ -- | Replaces the `ModuleName`s in the AST so that the generated code refers to
114+ -- the collision-avoiding renamed module imports.
115+ renameModules :: M.Map ModuleName (Ann, ModuleName) -> [Bind Ann] -> [Bind Ann]
116+ renameModules mnLookup binds =
117+ let (f, _, _) = everywhereOnValues id goExpr goBinder
118+ in map f binds
84119 where
85− modRef (AST.Indexer _ (AST.Var _ _) (AST.Var _ mname))
86− | not $ T.null mname = [extract mname]
87− modRef _ = []
88− extract :: Text -> Text
89− extract = T.replace "_" "." . T.dropWhileEnd (=='_')
120+ goExpr :: Expr a -> Expr a
121+ goExpr (Var ann q) = Var ann (renameQual q)
122+ goExpr e = e
123+ goBinder :: Binder a -> Binder a
124+ goBinder (ConstructorBinder ann q1 q2 bs) = ConstructorBinder ann (renameQual q1) (renameQual q2) bs
125+ goBinder b = b
126+ renameQual :: Qualified a -> Qualified a
127+ renameQual (Qualified (Just mn') a) =
128+ let (_,mnSafe) = fromMaybe (internalError "Missing value in mnLookup") $ M.lookup mn' mnLookup
129+ in Qualified (Just mnSafe) a
130+ renameQual q = q
90131
91132 -- |
92133 -- Generate code in the simplified intermediate representation for a declaration
@@ -110,10 +151,41 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
110151 else AST.Comment Nothing com <$> nonRecToIL dt a i (modifyAnn removeComments e)
111152 nonRecToIL ModuleDecl (ss, _, _, _) ident val = do
112153 il <- valueToIL val
113− pure $ AST.Function Nothing (Just $ moduleIdentToIL ident) [] (AST.Block Nothing [AST.Return Nothing il])
154+ pure $ AST.Function Nothing (Just $ identToIL ident) [] (AST.Block Nothing [AST.Return Nothing il])
114155 nonRecToIL RecLetDecl (ss, _, _, _) ident val = do
115− ilExpr <- valueToIL val
116− pure $ AST.Assignment Nothing (AST.Var Nothing $ identToIL ident) ilExpr
156+ il <- valueToIL val
157+ pure $
158+ let retainedName = identToIL ident
159+ retained = AST.Var Nothing $ retainedName
160+ unretainedName = retainedName <> unretainedSuffix
161+ unretained = AST.Var Nothing $ unretainedName
162+ boxed = AST.Var Nothing . -- Note: prevents optimizer from eliminating this
163+ prettyPrintIL1 $
164+ AST.VariableIntroduction
165+ Nothing
166+ retainedName
167+ (Just unretained)
168+ alias = AST.Var Nothing $ "auto& " <> retainedName <> " = " <> unretainedName
169+ in
170+ case il of
171+ AST.Function _ name args (AST.Block Nothing ils) ->
172+ AST.Assignment Nothing
173+ retained
174+ (AST.Function Nothing
175+ name
176+ args
177+ (AST.Block Nothing (boxed : ils)))
178+ _ ->
179+ AST.Assignment Nothing
180+ retained
181+ (AST.App Nothing
182+ (AST.Function Nothing
183+ Nothing
184+ []
185+ (AST.Block Nothing ([ alias
186+ , AST.Return Nothing il
187+ ])))
188+ [])
117189 nonRecToIL _ (ss, _, _, _) ident val = do
118190 il <- valueToIL val
119191 pure $ AST.VariableIntroduction Nothing (identToIL ident) (Just il)
@@ -137,7 +209,7 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
137209 updates <- mapM (sndM valueToIL) ps
138210 obj' <- freshName'
139211 let objVar = AST.Var Nothing obj'
140− copy = AST.VariableIntroduction Nothing obj' (Just $ copyDict obj)
212+ copy = AST.VariableIntroduction Nothing obj' (Just $ AST.App Nothing (AST.Var Nothing (unbox dictType)) [obj])
141213 assign (k, v) = AST.Assignment Nothing (accessorString k objVar) v
142214 sts = copy : (assign <$> updates) ++ [AST.Return Nothing objVar]
143215 return $ AST.App Nothing (AST.Function Nothing Nothing [] (AST.Block Nothing sts)) []
@@ -150,7 +222,6 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
150222 valueToIL e@App{} = do
151223 let (f, args) = unApp e []
152224 args' <- mapM valueToIL args
153− fn <- valueToIL f
154225 case f of
155226 Var (_, _, _, Just IsNewtype) _ -> return (head args')
156227 _ -> flip (foldl (\fn a -> AST.App Nothing fn [a])) args' <$> valueToIL f
@@ -158,20 +229,48 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
158229 unApp :: Expr Ann -> [Expr Ann] -> (Expr Ann, [Expr Ann])
159230 unApp (App _ val arg) args = unApp val (arg : args)
160231 unApp other args = (other, args)
232+ -- valueToIL (Var (_, _, _, Just IsForeign) qi@(Qualified (Just mn') ident)) =
233+ -- return $ if mn' == mn
234+ -- then AST.Var Nothing (moduleNameToIL mn' <> "::" <> identToIL ident)
235+ -- else varToIL qi
236+ -- valueToIL (Var (_, _, _, Just IsForeign) (Qualified (Just mn') ident)) =
237+ -- return $ AST.Var Nothing (moduleNameToIL mn' <> "::" <> identToIL ident)
238+ -- valueToIL (Var (_, _, _, Just IsForeign) qi@(Qualified (Just mn') ident)) =
239+ -- return $ if mn' == mn
240+ -- then foreignIdent ident
241+ -- else varToIL qi
242+ -- valueToIL (Var (_, _, _, Just IsForeign) ident) =
243+ -- internalError $ "Encountered an unqualified reference to a foreign ident " ++ T.unpack (showQualified showIdent ident)
161244 valueToIL (Var _ ident) = return $ varToIL ident
162245 valueToIL (Case (ss, _, _, _) values binders) = do
163246 vals <- mapM valueToIL values
164247 bindersToIL ss binders vals
165248 valueToIL (Let _ ds val) = do
166249 let recurs = concatMap getNames $ filter isRec ds
167− recurDs = (\v -> AST.VariableIntroduction Nothing (identToIL v) Nothing) <$> recurs
250+ recurDs = (\v -> AST.Var Nothing $ "boxed::recur " <> identToIL v) <$> recurs
251+ recurDsWeak = (\v -> AST.Var
252+ Nothing
253+ ("boxed::recur::weak " <> identToIL v <> unretainedSuffix <> "(" <> identToIL v <> ")")
254+ ) <$> recurs
255+ mutualRecurWarning = if any isMutRec $ filter isRec ds
256+ then [AST.Var Nothing "#pragma message(\"Mutually recursive lets will cause retain cycles (memory leaks)\") //"]
257+ else []
168258 ds' <- concat <$> mapM (bindToIL LetDecl) ds
169259 ret <- valueToIL val
170− return $ AST.App Nothing (AST.Function Nothing Nothing [] (AST.Block Nothing (recurDs ++ ds' ++ [AST.Return Nothing ret]))) []
260+ return $
261+ AST.App Nothing
262+ (AST.Function Nothing
263+ Nothing
264+ []
265+ (AST.Block Nothing (recurDs ++ recurDsWeak ++ mutualRecurWarning ++ ds' ++ [AST.Return Nothing ret])))
266+ []
171267 where
172268 isRec :: Bind Ann -> Bool
173269 isRec (Rec _) = True
174− isRec _ = False
270+ isRec _ = False
271+ isMutRec :: Bind Ann -> Bool
272+ isMutRec b | (length . nub $ getNames b) > 1 = True
273+ | otherwise = False
175274 valueToIL (Constructor (_, _, _, Just IsNewtype) _ (ProperName ctor) _) = error "IsNewtype"
176275 valueToIL (Constructor _ _ (ProperName ctor) fields) =
177276 let body = AST.ObjectLiteral Nothing $ (mkString ctor, AST.BooleanLiteral Nothing True) :
@@ -201,9 +300,9 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
201300 -- | Generate code in the simplified intermediate representation for a reference to a
202301 -- variable that may have a qualified name.
203302 qualifiedToIL :: (a -> Ident) -> Qualified a -> AST
204− qualifiedToIL f (Qualified (Just (ModuleName mn')) a) | mn' == C.prim = AST.Var Nothing . moduleIdentToIL $ f a
205− qualifiedToIL f (Qualified (Just mn') a) | mn /= mn' = AST.Indexer Nothing (AST.Var Nothing . moduleIdentToIL $ f a) (AST.Var Nothing (moduleNameToIL mn'))
206− qualifiedToIL f (Qualified _ a) = AST.Indexer Nothing (AST.Var Nothing . moduleIdentToIL $ f a) (AST.Var Nothing "")
303+ qualifiedToIL f (Qualified (Just (ModuleName [ProperName mn'])) a) | mn' == C.prim = AST.Var Nothing . runIdent $ f a
304+ qualifiedToIL f (Qualified (Just mn') a) = AST.Indexer Nothing (AST.Var Nothing . identToIL $ f a) (AST.Var Nothing (moduleNameToIL mn'))
305+ qualifiedToIL f (Qualified _ a) = AST.Var Nothing $ identToIL (f a)
207306
208307 -- foreignIdent :: Ident -> AST
209308 -- foreignIdent ident = accessorString (mkString $ runIdent ident) (AST.Var Nothing "$foreign")
@@ -228,7 +327,7 @@ moduleToIL (Module _ coms mn _ imps _ foreigns decls) project =
228327 go _ _ _ = internalError "Invalid arguments to bindersToIL"
229328
230329 failedPatternMessage :: Text
231− failedPatternMessage = "Failed pattern match at " <> runModuleName mn <> " " <> displayStartEndPos ss
330+ failedPatternMessage = "Failed pattern match at " <> runModuleName mn <> " " <> displayStartEndPos ss <> ": "
232331
233332 guardsToIL :: Either [(Guard Ann, Expr Ann)] (Expr Ann) -> m [AST]
234333 guardsToIL (Left gs) = traverse genGuard gs where
@@ -315,7 +414,8 @@ emptyAnn = (SourceSpan "" (SourcePos 0 0) (SourcePos 0 0), [], Nothing, Nothing)
315414
316415arrayLength :: AST -> AST
317416arrayLength a = AST.App Nothing (AST.Var Nothing arrayLengthFn) [a]
318−-- arrayLength a = AST.Var Nothing (arrayLengthFn <> "(" <> prettyPrintIL1 a <> ")")
319417
320−copyDict :: AST -> AST
321−copyDict a = AST.App Nothing (AST.Var Nothing copyDictFn) [a]
418+freshName' :: MonadSupply m => m Text
419+freshName' = do
420+ name <- freshName
421+ return $ T.replace "$" "_Local_" name
src/CodeGen/IL/Common.hs+130−104
@@ -3,9 +3,7 @@ module CodeGen.IL.Common where
33
44import Prelude.Compat
55
6−import Control.Monad.Supply.Class (MonadSupply, freshName)
76import Data.Char
8−import Data.Maybe (maybe)
97import Data.Monoid ((<>))
108import Data.Text (Text)
119import qualified Data.Text as T
@@ -13,82 +11,57 @@ import qualified Data.Text as T
1311import Language.PureScript.Crash
1412import Language.PureScript.Names
1513
16−import qualified Language.PureScript.Constants as C
17−
1814moduleNameToIL :: ModuleName -> Text
19−moduleNameToIL (ModuleName mn) =
20− let name = T.replace "." "_" mn
21− in if nameIsILBuiltIn name then (name <> moduleRenamerMarker) else name
22−
23−moduleNameToIL' :: ModuleName -> Text
24−moduleNameToIL' (ModuleName mn) = T.replace "_" "." mn
15+moduleNameToIL (ModuleName pns) =
16+ let name = T.intercalate "_" (runProperName `map` pns)
17+ in if nameIsILBuiltIn name then ("_" <> name <> "_") else name
2518
26−-- | Convert an 'Ident' into a valid IL identifier:
19+-- | Convert an 'Ident' into a valid C++ identifier:
2720--
2821-- * Alphanumeric characters are kept unmodified.
2922--
30−-- * Reserved IL identifiers are wrapped with '_'.
23+-- * Reserved C++ identifiers are wrapped with '_'.
3124--
3225identToIL :: Ident -> Text
3326identToIL UnusedIdent = unusedName
3427identToIL (Ident "$__unused") = unusedName
35−identToIL (Ident name) | name == C.undefined = undefinedName -- Note: done in Printer.hs too
3628identToIL (Ident name) = properToIL name
3729identToIL (GenIdent _ _) = internalError "GenIdent in identToIL"
3830
39−moduleIdentToIL :: Ident -> Text
40−moduleIdentToIL UnusedIdent = unusedName
41−moduleIdentToIL (Ident name) | name == C.undefined = undefinedName
42−moduleIdentToIL (Ident name) = moduleProperToIL name
43−moduleIdentToIL (GenIdent _ _) = internalError "GenIdent in identToIL"
44−
45−undefinedName :: Text
46−undefinedName = "Undefined"
47−
4831unusedName :: Text
49−unusedName = "_"
32+unusedName = "_Unused_"
5033
5134properToIL :: Text -> Text
5235properToIL name
53− | nameIsILReserved name = "ˉ" <> name
36+ | nameIsILReserved name || nameIsILBuiltIn name || prefixIsReserved name = "_" <> name <> "_"
5437 | otherwise = T.concatMap identCharToText name
5538
39+-- | Test if a string is a valid AST identifier without escaping.
40+identNeedsEscaping :: Text -> Bool
41+identNeedsEscaping s = s /= properToIL s || T.null s
42+
5643-- | Attempts to find a human-readable name for a symbol, if none has been specified returns the
5744-- ordinal value.
5845identCharToText :: Char -> Text
59−identCharToText 'ṩ' = "_ṩ"
6046identCharToText c | isAlphaNum c = T.singleton c
6147identCharToText '_' = "_"
62−identCharToText '\'' = "ꞌ" -- lowercase saltillo
63−identCharToText c = error (show (ord c)) -- TODO: should never occur now(?)
64−
65−moduleProperToIL :: Text -> Text
66−moduleProperToIL = T.concatMap identCharToText
48+identCharToText '\'' = "_Prime_"
49+identCharToText '$' = "$"
50+identCharToText c = "_Code_Point_" <> T.pack (show (ord c))
6751
68−-- | Checks whether an identifier name is reserved in IL.
52+-- | Checks whether an identifier name is reserved in C++.
6953nameIsILReserved :: Text -> Bool
7054nameIsILReserved name =
7155 name `elem` ilAnyReserved
7256
73−-- | Checks whether a name matches a built-in value in IL.
57+-- | Checks whether a name matches a built-in value in C++.
7458nameIsILBuiltIn :: Text -> Bool
7559nameIsILBuiltIn name =
7660 name `elem`
77− [ "Any"
78− , "Apply"
79− , "Contains"
80− , "CopyDict"
81− , "Dict"
82− , "EffFn"
83− , "Fn"
84− , "Foreign"
85− , "Get"
86− , "Index"
87− , "Is"
88− , "Length"
89− , "Once"
90− , "Run"
91− , "Undefined"
61+ [ "final"
62+ , "override"
63+ , "posix"
64+ , "std"
9265 ]
9366
9467ilAnyReserved :: [Text]
@@ -100,77 +73,135 @@ ilAnyReserved =
10073
10174ilKeywords :: [Text]
10275ilKeywords =
103− [ "break"
76+ [ "alignas"
77+ , "alignof"
78+ , "and"
79+ , "and_eq"
80+ , "asm"
81+ , "assert"
82+ , "atomic_cancel"
83+ , "atomic_commit"
84+ , "atomic_noexcept"
85+ , "auto"
86+ , "bitand"
87+ , "bitor"
88+ , "bool"
89+ , "break"
10490 , "case"
105− , "chan"
91+ , "catch"
92+ , "char"
93+ , "char16_t"
94+ , "char32_t"
95+ , "class"
96+ , "compl"
97+ , "concept"
10698 , "const"
99+ , "constexpr"
100+ , "const_cast"
107101 , "continue"
102+ , "concept"
103+ , "decltype"
108104 , "default"
109− , "defer"
105+ , "delete"
106+ , "do"
107+ , "double"
108+ , "dynamic_cast"
110109 , "else"
111− , "fallthrough"
110+ , "enum"
111+ , "explicit"
112+ , "export"
113+ , "extern"
114+ , "false"
115+ , "float"
112116 , "for"
113− , "func"
114− , "go"
117+ , "friend"
115118 , "goto"
116119 , "if"
117120 , "import"
118− , "interface"
119− , "map"
120− , "package"
121− , "range"
121+ , "inline"
122+ , "int"
123+ , "long"
124+ , "module"
125+ , "mutable"
126+ , "namespace"
127+ , "new"
128+ , "noexcept"
129+ , "not"
130+ , "not_eq"
131+ , "nullptr"
132+ , "nullptr_t"
133+ , "operator"
134+ , "or"
135+ , "or_eq"
136+ , "private"
137+ , "protected"
138+ , "public"
139+ , "register"
140+ , "reinterpret_cast"
141+ , "requires"
122142 , "return"
123− , "select"
143+ , "short"
144+ , "signed"
145+ , "sizeof"
146+ , "static"
147+ , "static_assert"
148+ , "static_cast"
124149 , "struct"
125150 , "switch"
126− , "type"
127− , "var"
151+ , "synchronized"
152+ , "template"
153+ , "this"
154+ , "thread_local"
155+ , "throw"
156+ , "true"
157+ , "try"
158+ , "typedef"
159+ , "typeid"
160+ , "typename"
161+ , "typeof"
162+ , "union"
163+ , "unsigned"
164+ , "using"
165+ , "virtual"
166+ , "void"
167+ , "volatile"
168+ , "wchar_t"
169+ , "while"
170+ , "xor"
171+ , "xor_eq"
128172 ]
129173
174+prefixIsReserved :: Text -> Bool
175+prefixIsReserved name =
176+ any (`T.isPrefixOf` name)
177+ [ "__"
178+ ]
179+
130180ilLiterals :: [Text]
131181ilLiterals =
132− [ "init"
133− , "nil"
134− , "int"
135− , "int8"
136− , "int16"
137− , "int32"
138− , "int64"
139− , "uint"
140− , "uint8"
141− , "uint16"
142− , "uint32"
143− , "uint64"
144− , "uintptr"
145− , "float32"
146− , "float64"
147− , "complex64"
148− , "complex128"
149− , "byte"
150− , "rune"
182+ [ "box"
183+ , "boxed"
184+ , "copy"
185+ , "errno"
186+ , "peek"
187+ , "purescript"
188+ , "runtime_error"
189+ , "setjmp"
151190 , "string"
191+ , "unbox"
152192 ]
153193
154−withPrefix :: Text -> Text
155−withPrefix s = "Ꞌ" <> s -- uppercase saltillo
156−
157−anyType :: Text
158−anyType = "Any"
159−
160194dictType :: Text
161−dictType = "Dict"
195+dictType = "dict_t"
162196
163197arrayType :: Text
164−arrayType = "[]" <> anyType
165−
166−uncurriedFnType :: Int -> Text
167−uncurriedFnType i = "Fn" <> (T.pack . show $ i)
198+arrayType = "array_t"
168199
169200int :: Text
170201int = "int"
171202
172203float :: Text
173−float = "float64"
204+float = "double"
174205
175206bool :: Text
176207bool = "bool"
@@ -178,19 +209,14 @@ bool = "bool"
178209string :: Text
179210string = "string"
180211
181−arrayLengthFn :: Text
182−arrayLengthFn = "Length"
183−
184−indexFn :: Text
185−indexFn = "Index"
212+unbox :: Text -> Text
213+unbox t = "unbox<" <> t <> ">"
186214
187−copyDictFn :: Text
188−copyDictFn = "CopyDict"
215+arrayLengthFn :: Text
216+arrayLengthFn = "array_length"
189217
190−freshName' :: MonadSupply m => m Text
191−freshName' = do
192− name <- freshName
193− return $ T.replace "$" "ṩ" name
218+unretainedSuffix :: Text
219+unretainedSuffix = "_Weak_"
194220
195−moduleRenamerMarker :: Text
196−moduleRenamerMarker = "__"
221+tcoLoop :: Text
222+tcoLoop = "_TCO_Loop_"
src/CodeGen/IL/Optimizer.hs+93−34
@@ -8,30 +8,31 @@ import Data.Text (Text)
88import Language.PureScript.CoreImp.AST
99import Language.PureScript.CoreImp.Optimizer.Blocks
1010import Language.PureScript.CoreImp.Optimizer.Common hiding (isDict)
11−import Language.PureScript.CoreImp.Optimizer.Inliner (etaConvert, evaluateIifes, inlineVariables, unThunk)
11+import Language.PureScript.CoreImp.Optimizer.Inliner hiding (inlineUnsafeCoerce, inlineUnsafePartial)
1212import Language.PureScript.CoreImp.Optimizer.Unused
13+import Language.PureScript.PSString (PSString, decodeString)
1314
1415import CodeGen.IL.Common (unusedName)
15−import CodeGen.IL.Optimizer.Inliner
16−import CodeGen.IL.Optimizer.MagicDo
1716import CodeGen.IL.Optimizer.TCO
1817
1918import qualified Language.PureScript.Constants as C
2019
21−-- | Apply a series of optimizer passes to simplified IL code
20+
21+-- | Apply a series of optimizer passes to simplified C++ code
2222optimize :: MonadSupply m => AST -> AST -> m AST
23+-- optimize = untilFixedPoint $ return . inlineUnsafeCoerce . inlineUnsafePartial . tidyUp . tco
2324optimize mn il = do
24− il' <- untilFixedPoint (inlineFnComposition . inlineUnsafeCoerce . inlineUnsafePartial . tidyUp) il
25− il'' <- untilFixedPoint (return . ignoreUnusedResults . inlineCommonValues . inlineCommonOperators . tidyUp) . tco mn . inlineST
26− =<< untilFixedPoint (return . magicDoST)
27− =<< untilFixedPoint (return . magicDoEff)
28− =<< untilFixedPoint (return . magicDoEffect) il'
29− untilFixedPoint (return . collapseVarDeclarations) il''
25+ il' <- untilFixedPoint (return . inlineApply . inlineUnsafeCoerce . inlineUnsafePartial . tidyUp) il
26+ untilFixedPoint (return . tidyUp) . tco mn
27+ =<< untilFixedPoint (return . magicDo')
28+ =<< untilFixedPoint (return . magicDo) il'
29+
30+
3031 where
3132 tidyUp :: AST -> AST
3233 tidyUp = applyAll
3334 [ collapseNestedBlocks
34− , collapseNestedIfs
35+ -- , collapseNestedIfs
3536 , collapseIfChecks
3637 , removeCodeAfterReturnStatements
3738 , unThunk
@@ -47,6 +48,34 @@ untilFixedPoint f = go
4748 a' <- f a
4849 if a' == a then return a' else go a'
4950
51+-- | Overridden from CoreImp
52+
53+inlineUnsafePartial :: AST -> AST
54+inlineUnsafePartial = everywhereTopDown convert where
55+ convert (App ss (Indexer _ (Var _ unsafePartial) (Var _ partialUnsafe)) [ comp ])
56+ | unsafePartial == C.unsafePartial && partialUnsafe == C.partialUnsafe
57+ -- Apply to undefined here, the application should be optimized away
58+ -- if it is safe to do so
59+ = App ss comp [ Var ss C.undefined ]
60+ convert other = other
61+
62+inlineUnsafeCoerce :: AST -> AST
63+inlineUnsafeCoerce = everywhereTopDown convert where
64+ convert (App _ (Indexer _ (Var _ unsafeCoerceFn) (Var _ unsafeCoerce)) [ comp ])
65+ | unsafeCoerceFn == C.unsafeCoerceFn && unsafeCoerce == C.unsafeCoerce
66+ = comp
67+ convert other = other
68+
69+inlineApply :: AST -> AST
70+inlineApply = everywhereTopDown convert where
71+ convert (App ss (App _ (Indexer _ (Var _ apply) (Var _ dataFunction)) [ f ]) [ arg ])
72+ | apply == C.apply && dataFunction == C.dataFunction
73+ = App ss f [ arg ]
74+ convert (App ss (App _ (Indexer _ (Var _ applyFlipped) (Var _ dataFunction)) [ arg ]) [ f ])
75+ | applyFlipped == C.applyFlipped && dataFunction == C.dataFunction
76+ = App ss f [ arg ]
77+ convert other = other
78+
5079collapseIfChecks :: AST -> AST
5180collapseIfChecks = everywhere collapse where
5281 collapse :: AST -> AST
@@ -55,27 +84,57 @@ collapseIfChecks = everywhere collapse where
5584 | prop == "otherwise" && val == "Data_Boolean" = exprs
5685 collapse exprs = exprs
5786
58−collapseVarDeclarations :: AST -> AST
59−collapseVarDeclarations = everywhere collapse where
60− collapse :: AST -> AST
61− collapse (Block ss sts) = Block ss (go Nothing sts)
62− collapse s = s
63−
64− go :: Maybe Text -> [AST] -> [AST]
65− go _ [] = []
66− go Nothing (decl@(VariableIntroduction _ var (Just _)) : sts) = decl : go (Just var) (go Nothing sts)
67− go (Just var') ((VariableIntroduction ss var (Just val)) : sts)
68− | var == var' = (Assignment ss (Var Nothing var) val) : go (Just var') sts
69− go var' (s:sts) = s : go var' sts
70−
71−ignoreUnusedResults :: AST -> AST
72−ignoreUnusedResults = everywhere $ removeFromBlock go
87+magicDo :: AST -> AST
88+magicDo = magicDo'' C.eff C.effDictionaries
89+
90+magicDo' :: AST -> AST
91+magicDo' = magicDo'' C.effect C.effectDictionaries
92+
93+magicDo'' :: Text -> C.EffectDictionaries -> AST -> AST
94+magicDo'' effectModule C.EffectDictionaries{..} = everywhereTopDown convert
7395 where
74− go :: [AST] -> [AST]
75− go [] = []
76− go (VariableIntroduction ss var (Just s) : sts)
77− | not $ any (everything (||) (isUsed var)) sts = sts'
78− where
79− sts' | App{} <- s = s : (go sts)
80− | otherwise = go sts
81− go (s:sts) = s : go sts
96+ -- Desugar monomorphic calls to >>= and return for the Eff monad
97+ convert :: AST -> AST
98+ -- Desugar pure
99+ convert (App _ (App _ pure' [val]) []) | isPure pure' = val
100+ -- Desugar discard
101+ convert (App _ (App _ bind [m]) [Function s1 Nothing [unused] (Block s2 il)]) | isDiscard bind && unused == unusedName =
102+ Function s1 Nothing [] $ Block s2 (App s2 m [] : map applyReturns il )
103+ -- Desugar bind
104+ convert (App _ (App _ bind [m]) [Function s1 Nothing [arg] (Block s2 il)]) | isBind bind =
105+ Function s1 Nothing [] $ Block s2 (VariableIntroduction s2 arg (Just (App s2 m [])) : map applyReturns il)
106+ -- Inline double applications
107+ convert (App _ (App s1 (Function s2 Nothing [] (Block ss body)) []) []) =
108+ App s1 (Function s2 Nothing [] (Block ss (applyReturns `fmap` body))) []
109+ convert other = other
110+
111+ -- Check if an expression represents a monomorphic call to >>= for the Eff monad
112+ isBind (App _ fn [dict]) | isDict (effectModule, edBindDict) dict && isBindPoly fn = True
113+ isBind _ = False
114+ -- Check if an expression represents a call to @discard@
115+ isDiscard (App _ (App _ fn [dict1]) [dict2])
116+ | isDict (C.controlBind, C.discardUnitDictionary) dict1 &&
117+ isDict (effectModule, edBindDict) dict2 &&
118+ isDiscardPoly fn = True
119+ isDiscard _ = False
120+ -- Check if an expression represents a monomorphic call to pure or return for the Eff applicative
121+ isPure (App _ fn [dict]) | isDict (effectModule, edApplicativeDict) dict && isPurePoly fn = True
122+ isPure _ = False
123+ -- Check if an expression represents the polymorphic >>= function
124+ isBindPoly = isDict (C.controlBind, C.bind)
125+ -- Check if an expression represents the polymorphic pure function
126+ isPurePoly = isDict (C.controlApplicative, C.pure')
127+ -- Check if an expression represents the polymorphic discard function
128+ isDiscardPoly = isDict (C.controlBind, C.discard)
129+
130+
131+ applyReturns :: AST -> AST
132+ applyReturns (Return ss ret) = Return ss (App ss ret [])
133+ applyReturns (Block ss ils) = Block ss (map applyReturns ils)
134+ applyReturns (IfElse ss cond t f) = IfElse ss cond (applyReturns t) (applyReturns `fmap` f)
135+ applyReturns other = other
136+
137+ isDict :: (Text, PSString) -> AST -> Bool
138+ isDict (moduleName, dictName) (Indexer _ (Var _ x) (Var _ y)) =
139+ Just x == decodeString dictName && y == moduleName
140+ isDict _ _ = False
src/CodeGen/IL/Optimizer/Common.hsdeleted−18
@@ -1,18 +0,0 @@
1−-- | Common functions used by the various optimizer phases
2−module CodeGen.IL.Optimizer.Common where
3−
4−import Prelude.Compat
5−
6−import Data.Text (Text)
7−
8−import Language.PureScript.CoreImp.AST
9−import Language.PureScript.PSString (PSString, decodeString)
10−
11−isDict :: (Text, PSString) -> AST -> Bool
12−isDict (moduleName, dictName) (Indexer _ (Var _ x) (Var _ y)) =
13− Just x == decodeString dictName && y == moduleName
14−isDict _ _ = False
15−
16−isDict' :: [(Text, PSString)] -> AST -> Bool
17−isDict' xs il = any (`isDict` il) xs
18−
src/CodeGen/IL/Optimizer/Inliner.hsdeleted−435
@@ -1,435 +0,0 @@
1−-- | This module performs basic inlining of known functions
2−module CodeGen.IL.Optimizer.Inliner
3− ( inlineCommonValues
4− , inlineCommonOperators
5− , inlineFnComposition
6− , inlineUnsafeCoerce
7− , inlineUnsafePartial
8− ) where
9−
10−import Prelude.Compat
11−
12−import Control.Monad.Supply.Class (MonadSupply)
13−
14−import Data.Either (rights)
15−import Data.Maybe (fromMaybe)
16−import Data.String (IsString, fromString)
17−import Data.Text (Text)
18−import qualified Data.Text as T
19−
20−import Language.PureScript.PSString (PSString, decodeString, mkString)
21−import Language.PureScript.CoreImp.AST
22−import Language.PureScript.CoreImp.Optimizer.Common hiding (isDict, isDict')
23−import Language.PureScript.AST (SourceSpan(..))
24−import qualified Language.PureScript.Constants as C
25−
26−import CodeGen.IL.Common
27−import CodeGen.IL.Optimizer.Common
28−
29−-- TODO: Potential bug:
30−-- Shouldn't just inline this case: { var x = 0; x.toFixed(10); }
31−-- Needs to be: { 0..toFixed(10); }
32−-- Probably needs to be fixed in pretty-printer instead.
33−shouldInline :: AST -> Bool
34−shouldInline (Var _ _) = True
35−shouldInline (NumericLiteral _ _) = True
36−shouldInline (StringLiteral _ _) = True
37−shouldInline (BooleanLiteral _ _) = True
38−shouldInline (Indexer _ index val) = shouldInline index && shouldInline val
39−shouldInline _ = False
40−
41−inlineCommonValues :: AST -> AST
42−inlineCommonValues = everywhere convert
43− where
44− convert :: AST -> AST
45− convert (App ss fn [dict])
46− | isDict' [semiringNumber, semiringInt] dict && isDict fnZero fn = NumericLiteral ss (Left 0)
47− | isDict' [semiringNumber, semiringInt] dict && isDict fnOne fn = NumericLiteral ss (Left 1)
48− | isDict boundedBoolean dict && isDict fnBottom fn = BooleanLiteral ss False
49− | isDict boundedBoolean dict && isDict fnTop fn = BooleanLiteral ss True
50− convert (App ss (App _ fn [dict]) [x])
51− | isDict ringInt dict && isDict fnNegate fn = Unary ss Negate (unbox x $ snd ringInt)
52− convert (App ss (App _ (App _ fn [dict]) [x]) [y])
53− | isDict semiringInt dict && isDict fnAdd fn = intOp ss Add semiringInt x y
54− | isDict semiringInt dict && isDict fnMultiply fn = intOp ss Multiply semiringInt x y
55− | isDict ringInt dict && isDict fnSubtract fn = intOp ss Subtract ringInt x y
56− convert other = other
57− fnZero = (C.dataSemiring, C.zero)
58− fnOne = (C.dataSemiring, C.one)
59− fnBottom = (C.dataBounded, C.bottom)
60− fnTop = (C.dataBounded, C.top)
61− fnAdd = (C.dataSemiring, C.add)
62− fnMultiply = (C.dataSemiring, C.mul)
63− fnSubtract = (C.dataRing, C.sub)
64− fnNegate = (C.dataRing, C.negate)
65− intOp ss op (_, d) x y = Binary ss op (unbox x d) (unbox y d)
66−
67−inlineCommonOperators :: AST -> AST
68−inlineCommonOperators = everywhereTopDown $ applyAll $
69− [ binary semiringNumber opAdd Add
70− , binary semiringNumber opMul Multiply
71−
72− , binary ringNumber opSub Subtract
73− , unary ringNumber opNegate Negate
74−
75− , binary euclideanRingNumber opDiv Divide
76−
77− -- , binary eqNumber opEq EqualTo
78− -- , binary eqNumber opNotEq NotEqualTo
79− , binary eqInt opEq EqualTo
80− , binary eqInt opNotEq NotEqualTo
81− , binary eqString opEq EqualTo
82− , binary eqString opNotEq NotEqualTo
83− , binary eqChar opEq EqualTo
84− , binary eqChar opNotEq NotEqualTo
85− , binary eqBoolean opEq EqualTo
86− , binary eqBoolean opNotEq NotEqualTo
87−
88− , binary ordBoolean opLessThan LessThan
89− , binary ordBoolean opLessThanOrEq LessThanOrEqualTo
90− , binary ordBoolean opGreaterThan GreaterThan
91− , binary ordBoolean opGreaterThanOrEq GreaterThanOrEqualTo
92− , binary ordChar opLessThan LessThan
93− , binary ordChar opLessThanOrEq LessThanOrEqualTo
94− , binary ordChar opGreaterThan GreaterThan
95− , binary ordChar opGreaterThanOrEq GreaterThanOrEqualTo
96− , binary ordInt opLessThan LessThan
97− , binary ordInt opLessThanOrEq LessThanOrEqualTo
98− , binary ordInt opGreaterThan GreaterThan
99− , binary ordInt opGreaterThanOrEq GreaterThanOrEqualTo
100− , binary ordNumber opLessThan LessThan
101− , binary ordNumber opLessThanOrEq LessThanOrEqualTo
102− , binary ordNumber opGreaterThan GreaterThan
103− , binary ordNumber opGreaterThanOrEq GreaterThanOrEqualTo
104− , binary ordString opLessThan LessThan
105− , binary ordString opLessThanOrEq LessThanOrEqualTo
106− , binary ordString opGreaterThan GreaterThan
107− , binary ordString opGreaterThanOrEq GreaterThanOrEqualTo
108−
109− , binary semigroupString opAppend Add
110−
111− , binary heytingAlgebraBoolean opConj And
112− , binary heytingAlgebraBoolean opDisj Or
113− , unary heytingAlgebraBoolean opNot Not
114−
115− , binary' C.dataIntBits C.or BitwiseOr
116− , binary' C.dataIntBits C.and BitwiseAnd
117− , binary' C.dataIntBits C.xor BitwiseXor
118− , binary' C.dataIntBits C.shl ShiftLeft
119− , binary' C.dataIntBits C.shr ShiftRight
120− , unary' C.dataIntBits C.complement BitwiseNot
121−
122− , inlineNonClassFunction (isModFn (C.dataFunction, C.apply)) $ \f x -> App Nothing f [x]
123− , inlineNonClassFunction (isModFn (C.dataFunction, C.applyFlipped)) $ \x f -> App Nothing f [x]
124− , inlineUnsafeIndex (isModFnWithDict (C.dataArray, C.unsafeIndex)) $ flip (Indexer Nothing)
125− ] ++
126− [ fn | i <- [0..10], fn <- [ mkFn i, runFn i ] ] ++
127− [ fn | i <- [0..10], fn <- [ mkEffFn C.controlMonadEffUncurried C.mkEffFn i, runEffFn C.controlMonadEffUncurried C.runEffFn i ] ] ++
128− [ fn | i <- [0..10], fn <- [ mkEffFn C.effectUncurried C.mkEffectFn i, runEffFn C.effectUncurried C.runEffectFn i ] ]
129− where
130− binary :: (Text, PSString) -> (Text, PSString) -> BinaryOperator -> AST -> AST
131− binary dict@(_, d) fns op = convert where
132− convert :: AST -> AST
133− convert (App ss (App _ (App _ fn [dict']) [x]) [y])
134− | isDict dict dict' && isDict fns fn = Binary ss op (unbox x d) (unbox y d)
135− convert other = other
136− binary' :: Text -> PSString -> BinaryOperator -> AST -> AST
137− binary' moduleName opString op = convert where
138− convert :: AST -> AST
139− convert (App ss (App _ fn [x]) [y]) | isDict (moduleName, opString) fn = Binary ss op (unbox x mn) (unbox y mn)
140− convert other = other
141− mn :: PSString
142− mn = mkString moduleName
143− unary :: (Text, PSString) -> (Text, PSString) -> UnaryOperator -> AST -> AST
144− unary dicts@(_, d) fns op = convert where
145− convert :: AST -> AST
146− convert (App ss (App _ fn [dict']) [x]) | isDict dicts dict' && isDict fns fn = Unary ss op (unbox x d)
147− convert other = other
148− unary' :: Text -> PSString -> UnaryOperator -> AST -> AST
149− unary' moduleName fnName op = convert where
150− convert :: AST -> AST
151− convert (App ss fn [x]) | isDict (moduleName, fnName) fn = Unary ss op (unbox x fnName)
152− convert other = other
153−
154− mkFn :: Int -> AST -> AST
155− mkFn = mkFn' C.dataFunctionUncurried C.mkFn $ \ss1 ss2 ss3 args il ->
156− Function ss1 Nothing args (Block ss2 [Return ss3 il])
157−
158− mkEffFn :: Text -> Text -> Int -> AST -> AST
159− mkEffFn modName fnName = mkFn' modName fnName $ \ss1 ss2 ss3 args il ->
160− Function ss1 Nothing args (Block ss2 [Return ss3 (App ss3 il [])])
161−
162− mkFn' :: Text -> Text -> (Maybe SourceSpan -> Maybe SourceSpan -> Maybe SourceSpan -> [Text] -> AST -> AST) -> Int -> AST -> AST
163− mkFn' modName fnName res 0 = convert where
164− convert :: AST -> AST
165− convert (App _ mkFnN [Function s1 Nothing [_] (Block s2 [Return s3 il])]) | isNFn modName fnName 0 mkFnN =
166− res s1 s2 s3 [] il
167− convert other = other
168− mkFn' modName fnName res n = convert where
169− convert :: AST -> AST
170− convert orig@(App ss mkFnN [fn]) | isNFn modName fnName n mkFnN =
171− case collectArgs n [] fn of
172− Just (args, [Return ss' ret]) -> res ss ss ss' args ret
173− _ -> orig
174− convert other = other
175− collectArgs :: Int -> [Text] -> AST -> Maybe ([Text], [AST])
176− collectArgs 1 acc (Function _ Nothing [oneArg] (Block _ il)) | length acc == n - 1 = Just (reverse (oneArg : acc), il)
177− collectArgs m acc (Function _ Nothing [oneArg] (Block _ [Return _ ret])) = collectArgs (m - 1) (oneArg : acc) ret
178− collectArgs _ _ _ = Nothing
179−
180− isNFn :: Text -> Text -> Int -> AST -> Bool
181− isNFn expectMod prefix n (Indexer _ (Var _ name) (Var _ modName)) | modName == expectMod =
182− name == fromString (T.unpack prefix <> show n)
183− isNFn _ _ _ _ = False
184−
185− runFn :: Int -> AST -> AST
186− runFn = runFn' C.dataFunctionUncurried C.runFn (\ss f args ->
187− let len = length args
188− typ = mkString $ uncurriedFnType len in
189− if len > 0
190− then App ss (App Nothing (StringLiteral Nothing typ) [f]) args
191− else App ss f args)
192− runEffFn :: Text -> Text -> Int -> AST -> AST
193− runEffFn modName fnName = runFn' modName fnName $ \ss fn acc ->
194− let typ = mkString . uncurriedFnType $ length acc in
195− Function ss Nothing [] (Block ss [Return ss (App ss (App Nothing (StringLiteral Nothing typ) [fn]) acc)])
196−
197− runFn' :: Text -> Text -> (Maybe SourceSpan -> AST -> [AST] -> AST) -> Int -> AST -> AST
198− runFn' modName runFnName res n = convert where
199− convert :: AST -> AST
200− convert il = fromMaybe il $ go n [] il
201−
202− go :: Int -> [AST] -> AST -> Maybe AST
203− go 0 acc (App ss runFnN [fn]) | isNFn modName runFnName n runFnN && length acc == n =
204− Just $ res ss fn acc
205− go m acc (App _ lhs [arg]) = go (m - 1) (arg : acc) lhs
206− go _ _ _ = Nothing
207−
208− inlineNonClassFunction :: (AST -> Bool) -> (AST -> AST -> AST) -> AST -> AST
209− inlineNonClassFunction p f = convert where
210− convert :: AST -> AST
211− convert (App _ (App _ op' [x]) [y]) | p op' = f x y
212− convert other = other
213−
214− inlineUnsafeIndex :: (AST -> Bool) -> (AST -> AST -> AST) -> AST -> AST
215− inlineUnsafeIndex p _ = convert where
216− convert :: AST -> AST
217− convert (App _ (App ss op' [x]) [y@NumericLiteral{}])
218− | p op' = Indexer ss y x
219− convert (App _ (App ss op' [x]) [y])
220− | p op' = App ss (Var Nothing indexFn) [x, y]
221− convert other = other
222−
223− isModFn :: (Text, PSString) -> AST -> Bool
224− isModFn (m, op) (Indexer _ (Var _ op') (Var _ m')) =
225− m == m' && decodeString op == Just op'
226− isModFn _ _ = False
227−
228− isModFnWithDict :: (Text, PSString) -> AST -> Bool
229− isModFnWithDict (m, op) (App _ (Indexer _ (Var _ op') (Var _ m')) [Var _ _]) =
230− m == m' && decodeString op == Just op'
231− isModFnWithDict _ _ = False
232−
233−-- (f <<< g $ x) = f (g x)
234−-- (f <<< g) = \x -> f (g x)
235−inlineFnComposition :: forall m. MonadSupply m => AST -> m AST
236−inlineFnComposition = everywhereTopDownM convert where
237− convert :: AST -> m AST
238− convert (App s1 (App s2 (App _ (App _ fn [dict']) [x]) [y]) [z])
239− | isFnCompose dict' fn = return $ App s1 x [App s2 y [z]]
240− | isFnComposeFlipped dict' fn = return $ App s2 y [App s1 x [z]]
241− convert app@(App ss (App _ (App _ fn [dict']) _) _)
242− | isFnCompose dict' fn || isFnComposeFlipped dict' fn = mkApps ss <$> goApps app <*> freshName'
243− convert other = return other
244−
245− mkApps :: Maybe SourceSpan -> [Either AST (Text, AST)] -> Text -> AST
246− mkApps ss fns a = App ss (Function ss Nothing [] (Block ss $ vars <> [Return Nothing comp])) []
247− where
248− vars = uncurry (VariableIntroduction ss) . fmap Just <$> rights fns
249− comp = Function ss Nothing [a] (Block ss [Return Nothing apps])
250− apps = foldr (\fn acc -> App ss (mkApp fn) [acc]) (Var ss a) fns
251−
252− mkApp :: Either AST (Text, AST) -> AST
253− mkApp = either id $ \(name, arg) -> Var (getSourceSpan arg) name
254−
255− goApps :: AST -> m [Either AST (Text, AST)]
256− goApps (App _ (App _ (App _ fn [dict']) [x]) [y])
257− | isFnCompose dict' fn = mappend <$> goApps x <*> goApps y
258− | isFnComposeFlipped dict' fn = mappend <$> goApps y <*> goApps x
259− goApps app@(App {}) = pure . Right . (,app) <$> freshName'
260− goApps other = pure [Left other]
261−
262− isFnCompose :: AST -> AST -> Bool
263− isFnCompose dict' fn = isDict semigroupoidFn dict' && isDict fnCompose fn
264−
265− isFnComposeFlipped :: AST -> AST -> Bool
266− isFnComposeFlipped dict' fn = isDict semigroupoidFn dict' && isDict fnComposeFlipped fn
267−
268− fnCompose :: forall a b. (IsString a, IsString b) => (a, b)
269− fnCompose = (C.controlSemigroupoid, C.compose)
270−
271− fnComposeFlipped :: forall a b. (IsString a, IsString b) => (a, b)
272− fnComposeFlipped = (C.controlSemigroupoid, C.composeFlipped)
273−
274−inlineUnsafeCoerce :: AST -> AST
275−inlineUnsafeCoerce = everywhereTopDown convert where
276− convert (App _ (Indexer _ (Var _ unsafeCoerceFn) (Var _ unsafeCoerce)) [ comp ])
277− | unsafeCoerceFn == C.unsafeCoerceFn && unsafeCoerce == C.unsafeCoerce
278− = comp
279− convert other = other
280−
281−inlineUnsafePartial :: AST -> AST
282−inlineUnsafePartial = everywhereTopDown convert where
283− convert (App ss fn [ comp ])
284− | isDict fnUnsafePartial fn
285− = App ss comp []
286− where
287− fnUnsafePartial :: forall a b. (IsString a, IsString b) => (a, b)
288− fnUnsafePartial = (C.partialUnsafe, C.unsafePartial)
289− convert other = other
290−
291−semiringNumber :: forall a b. (IsString a, IsString b) => (a, b)
292−semiringNumber = (C.dataSemiring, C.semiringNumber)
293−
294−semiringInt :: forall a b. (IsString a, IsString b) => (a, b)
295−semiringInt = (C.dataSemiring, C.semiringInt)
296−
297−ringNumber :: forall a b. (IsString a, IsString b) => (a, b)
298−ringNumber = (C.dataRing, C.ringNumber)
299−
300−ringInt :: forall a b. (IsString a, IsString b) => (a, b)
301−ringInt = (C.dataRing, C.ringInt)
302−
303−euclideanRingNumber :: forall a b. (IsString a, IsString b) => (a, b)
304−euclideanRingNumber = (C.dataEuclideanRing, C.euclideanRingNumber)
305−
306−-- eqNumber :: forall a b. (IsString a, IsString b) => (a, b)
307−-- eqNumber = (C.dataEq, C.eqNumber)
308−
309−eqInt :: forall a b. (IsString a, IsString b) => (a, b)
310−eqInt = (C.dataEq, C.eqInt)
311−
312−eqString :: forall a b. (IsString a, IsString b) => (a, b)
313−eqString = (C.dataEq, C.eqString)
314−
315−eqChar :: forall a b. (IsString a, IsString b) => (a, b)
316−eqChar = (C.dataEq, C.eqChar)
317−
318−eqBoolean :: forall a b. (IsString a, IsString b) => (a, b)
319−eqBoolean = (C.dataEq, C.eqBoolean)
320−
321−ordBoolean :: forall a b. (IsString a, IsString b) => (a, b)
322−ordBoolean = (C.dataOrd, C.ordBoolean)
323−
324−ordNumber :: forall a b. (IsString a, IsString b) => (a, b)
325−ordNumber = (C.dataOrd, C.ordNumber)
326−
327−ordInt :: forall a b. (IsString a, IsString b) => (a, b)
328−ordInt = (C.dataOrd, C.ordInt)
329−
330−ordString :: forall a b. (IsString a, IsString b) => (a, b)
331−ordString = (C.dataOrd, C.ordString)
332−
333−ordChar :: forall a b. (IsString a, IsString b) => (a, b)
334−ordChar = (C.dataOrd, C.ordChar)
335−
336−semigroupString :: forall a b. (IsString a, IsString b) => (a, b)
337−semigroupString = (C.dataSemigroup, C.semigroupString)
338−
339−boundedBoolean :: forall a b. (IsString a, IsString b) => (a, b)
340−boundedBoolean = (C.dataBounded, C.boundedBoolean)
341−
342−heytingAlgebraBoolean :: forall a b. (IsString a, IsString b) => (a, b)
343−heytingAlgebraBoolean = (C.dataHeytingAlgebra, C.heytingAlgebraBoolean)
344−
345−semigroupoidFn :: forall a b. (IsString a, IsString b) => (a, b)
346−semigroupoidFn = (C.controlSemigroupoid, C.semigroupoidFn)
347−
348−opAdd :: forall a b. (IsString a, IsString b) => (a, b)
349−opAdd = (C.dataSemiring, C.add)
350−
351−opMul :: forall a b. (IsString a, IsString b) => (a, b)
352−opMul = (C.dataSemiring, C.mul)
353−
354−opEq :: forall a b. (IsString a, IsString b) => (a, b)
355−opEq = (C.dataEq, C.eq)
356−
357−opNotEq :: forall a b. (IsString a, IsString b) => (a, b)
358−opNotEq = (C.dataEq, C.notEq)
359−
360−opLessThan :: forall a b. (IsString a, IsString b) => (a, b)
361−opLessThan = (C.dataOrd, C.lessThan)
362−
363−opLessThanOrEq :: forall a b. (IsString a, IsString b) => (a, b)
364−opLessThanOrEq = (C.dataOrd, C.lessThanOrEq)
365−
366−opGreaterThan :: forall a b. (IsString a, IsString b) => (a, b)
367−opGreaterThan = (C.dataOrd, C.greaterThan)
368−
369−opGreaterThanOrEq :: forall a b. (IsString a, IsString b) => (a, b)
370−opGreaterThanOrEq = (C.dataOrd, C.greaterThanOrEq)
371−
372−opAppend :: forall a b. (IsString a, IsString b) => (a, b)
373−opAppend = (C.dataSemigroup, C.append)
374−
375−opSub :: forall a b. (IsString a, IsString b) => (a, b)
376−opSub = (C.dataRing, C.sub)
377−
378−opNegate :: forall a b. (IsString a, IsString b) => (a, b)
379−opNegate = (C.dataRing, C.negate)
380−
381−opDiv :: forall a b. (IsString a, IsString b) => (a, b)
382−opDiv = (C.dataEuclideanRing, C.div)
383−
384−opConj :: forall a b. (IsString a, IsString b) => (a, b)
385−opConj = (C.dataHeytingAlgebra, C.conj)
386−
387−opDisj :: forall a b. (IsString a, IsString b) => (a, b)
388−opDisj = (C.dataHeytingAlgebra, C.disj)
389−
390−opNot :: forall a b. (IsString a, IsString b) => (a, b)
391−opNot = (C.dataHeytingAlgebra, C.not)
392−
393−unbox :: AST -> PSString -> AST
394−unbox e d = select . inlineCommonValues $ inlineCommonOperators e
395− where
396− select :: AST -> AST
397− select e' =
398− case e' of
399− Var{} -> unbox' e'
400− App{} -> unbox' e'
401− Indexer{} -> unbox' e'
402− _ -> e'
403− unbox' :: AST -> AST
404− unbox' e' = maybe e' (\t -> App Nothing (StringLiteral Nothing t) [e']) $ unboxType d
405−
406−unboxType :: PSString -> Maybe PSString
407−unboxType s = do
408− t <- decodeString s
409− t' <- unboxType' t
410− return $ mkString t'
411− where
412− unboxType' :: Text -> Maybe Text
413− unboxType' t
414− | t == C.eqInt ||
415− t == C.semiringInt ||
416− t == C.ringInt ||
417− t == C.ordInt ||
418− t == C.dataIntBits
419− = Just int
420− | t == C.semiringNumber ||
421− t == C.ringNumber ||
422− t == C.ordNumber ||
423− t == C.euclideanRingNumber
424− = Just float
425− | t == C.eqBoolean ||
426− t == C.ordBoolean ||
427− t == C.heytingAlgebraBoolean
428− = Just bool
429− | t == C.eqChar ||
430− t == C.eqString ||
431− t == C.ordChar ||
432− t == C.ordString ||
433− t == C.semigroupString
434− = Just string
435− | otherwise = Nothing
src/CodeGen/IL/Optimizer/MagicDo.hsdeleted−146
@@ -1,146 +0,0 @@
1−-- | This module implements the "Magic Do" optimization, which inlines calls to return
2−-- and bind for the Eff monad, as well as some of its actions.
3−module CodeGen.IL.Optimizer.MagicDo (magicDoEffect, magicDoEff, magicDoST, inlineST) where
4−
5−import Prelude.Compat
6−import Protolude (ordNub)
7−
8−import Data.Maybe (fromJust, isJust)
9−import Data.Text (Text)
10−
11−import Language.PureScript.CoreImp.AST
12−import Language.PureScript.CoreImp.Optimizer.Common hiding (isDict)
13−import Language.PureScript.PSString (PSString, decodeString, mkString)
14−import qualified Language.PureScript.Constants as C
15−
16−import CodeGen.IL.Common (undefinedName, unusedName)
17−import CodeGen.IL.Optimizer.Common
18−
19−-- | Inline type class dictionaries for >>= and return for the Eff monad
20−--
21−-- E.g.
22−--
23−-- Prelude[">>="](dict)(m1)(function(x) {
24−-- return ...;
25−-- })
26−--
27−-- becomes
28−--
29−-- function __do {
30−-- var x = m1();
31−-- ...
32−-- }
33−magicDoEff :: AST -> AST
34−magicDoEff = magicDo C.eff C.effDictionaries
35−
36−magicDoEffect :: AST -> AST
37−magicDoEffect = magicDo C.effect C.effectDictionaries
38−
39−magicDoST :: AST -> AST
40−magicDoST = magicDo C.st C.stDictionaries
41−
42−magicDo :: Text -> C.EffectDictionaries -> AST -> AST
43−magicDo effectModule C.EffectDictionaries{..} = everywhereTopDown convert
44− where
45− -- Desugar monomorphic calls to >>= and return for the Eff monad
46− convert :: AST -> AST
47− -- Desugar pure
48− convert (App _ (App _ pure' [val]) []) | isPure pure' = val
49− -- Desugar discard
50− convert (App _ (App _ bind [m]) [Function s1 Nothing [unused] (Block s2 il)]) | isDiscard bind && unused == unusedName =
51− Function s1 Nothing [] $ Block s2 (App s2 m [] : map applyReturns il )
52− -- Desugar bind
53− convert (App _ (App _ bind [m]) [Function s1 Nothing [arg] (Block s2 il)]) | isBind bind =
54− Function s1 Nothing [] $ Block s2 (VariableIntroduction s2 arg (Just (App s2 m [])) : map applyReturns il)
55− -- Desugar untilE
56− convert (App s1 (App _ f [arg]) []) | isEffFunc edUntil f =
57− App s1 (Function s1 Nothing [] (Block s1 [ While s1 (Unary s1 Not (App s1 arg [])) (Block s1 []), Return s1 $ ObjectLiteral s1 []])) []
58− -- Desugar whileE
59− convert (App _ (App _ (App s1 f [arg1]) [arg2]) []) | isEffFunc edWhile f =
60− App s1 (Function s1 Nothing [] (Block s1 [ While s1 (App s1 arg1 []) (Block s1 [ App s1 arg2 [] ]), Return s1 $ ObjectLiteral s1 []])) []
61− -- -- Inline __do returns
62− -- convert (Return _ (App _ (Function _ (Just ident) [] body) [])) | ident == fnName = body
63− -- Inline double applications
64− convert (App _ (App s1 (Function s2 Nothing [] (Block ss body)) []) []) =
65− App s1 (Function s2 Nothing [] (Block ss (applyReturns `fmap` body))) []
66− convert other = other
67− -- Check if an expression represents a monomorphic call to >>= for the Eff monad
68− isBind (App _ fn [dict]) | isDict (effectModule, edBindDict) dict && isBindPoly fn = True
69− isBind _ = False
70− -- Check if an expression represents a call to @discard@
71− isDiscard (App _ (App _ fn [dict1]) [dict2])
72− | isDict (C.controlBind, C.discardUnitDictionary) dict1 &&
73− isDict (effectModule, edBindDict) dict2 &&
74− isDiscardPoly fn = True
75− isDiscard _ = False
76−
77− -- Check if an expression represents a monomorphic call to pure or return for the Eff applicative
78− isPure (App _ fn [dict]) | isDict (effectModule, edApplicativeDict) dict && isPurePoly fn = True
79− isPure _ = False
80− -- Check if an expression represents the polymorphic >>= function
81− isBindPoly = isDict (C.controlBind, C.bind)
82− -- Check if an expression represents the polymorphic pure function
83− isPurePoly = isDict (C.controlApplicative, C.pure')
84− -- Check if an expression represents the polymorphic discard function
85− isDiscardPoly = isDict (C.controlBind, C.discard)
86− -- Check if an expression represents a function in the Effect module
87− isEffFunc name (Indexer _ (StringLiteral _ name') (Var _ eff)) = eff == effectModule && name == name'
88− isEffFunc _ _ = False
89−
90− applyReturns :: AST -> AST
91− applyReturns (Return ss ret) = Return ss (App ss ret [])
92− applyReturns (Block ss ils) = Block ss (map applyReturns ils)
93− applyReturns (While ss cond il) = While ss cond (applyReturns il)
94− applyReturns (For ss v lo hi il) = For ss v lo hi (applyReturns il)
95− applyReturns (ForIn ss v xs il) = ForIn ss v xs (applyReturns il)
96− applyReturns (IfElse ss cond t f) = IfElse ss cond (applyReturns t) (applyReturns `fmap` f)
97− applyReturns other = other
98−
99−-- | Inline functions in the ST module
100−inlineST :: AST -> AST
101−inlineST = everywhere convertBlock
102− where
103− -- Look for runST blocks and inline the STRefs there.
104− -- If all STRefs are used in the scope of the same runST, only using { read, write, modify }STRef then
105− -- we can be more aggressive about inlining, and actually turn STRefs into local variables.
106− convertBlock (App s1 f [arg]) | isSTFunc C.runST f =
107− let refs = ordNub . findSTRefsIn $ arg
108− usages = findAllSTUsagesIn arg
109− allUsagesAreLocalVars = all (\u -> let v = toVar u in isJust v && fromJust v `elem` refs) usages
110− localVarsDoNotEscape = all (\r -> length (r `appearingIn` arg) == length (filter (\u -> let v = toVar u in v == Just r) usages)) refs
111− in App s1 (everywhere (convert (allUsagesAreLocalVars && localVarsDoNotEscape)) arg) []
112− convertBlock other = other
113− -- Convert a block in a safe way, preserving object wrappers of references,
114− -- or in a more aggressive way, turning wrappers into local variables depending on the
115− -- agg(ressive) parameter.
116− convert agg (App s1 f [arg]) | isSTFunc C.newSTRef f =
117− Function s1 Nothing [] (Block s1 [Return s1 $ if agg then arg else ObjectLiteral s1 [(mkString C.stRefValue, arg)]])
118− convert agg (App _ (App s1 f [ref]) []) | isSTFunc C.readSTRef f =
119− if agg then ref else Indexer s1 (StringLiteral s1 C.stRefValue) ref
120− convert agg (App _ (App _ (App s1 f [arg]) [ref]) []) | isSTFunc C.writeSTRef f =
121− if agg then Assignment s1 ref arg else Assignment s1 (Indexer s1 (StringLiteral s1 C.stRefValue) ref) arg
122− convert agg (App _ (App _ (App s1 f [func]) [ref]) []) | isSTFunc C.modifySTRef f =
123− if agg then Assignment s1 ref (App s1 func [ref]) else Assignment s1 (Indexer s1 (StringLiteral s1 C.stRefValue) ref) (App s1 func [Indexer s1 (StringLiteral s1 C.stRefValue) ref])
124− convert _ other = other
125− -- Check if an expression represents a function in the ST module
126− isSTFunc name (Indexer _ (StringLiteral _ name') (Var _ st)) = st == C.st && name == name'
127− isSTFunc _ _ = False
128− -- Find all ST Refs initialized in this block
129− findSTRefsIn = everything (++) isSTRef
130− where
131− isSTRef (VariableIntroduction _ ident (Just (App _ (App _ f [_]) []))) | isSTFunc C.newSTRef f = [ident]
132− isSTRef _ = []
133− -- Find all STRefs used as arguments to readSTRef, writeSTRef, modifySTRef
134− findAllSTUsagesIn = everything (++) isSTUsage
135− where
136− isSTUsage (App _ (App _ f [ref]) []) | isSTFunc C.readSTRef f = [ref]
137− isSTUsage (App _ (App _ (App _ f [_]) [ref]) []) | isSTFunc C.writeSTRef f || isSTFunc C.modifySTRef f = [ref]
138− isSTUsage _ = []
139− -- Find all uses of a variable
140− appearingIn ref = everything (++) isVar
141− where
142− isVar e@(Var _ v) | v == ref = [e]
143− isVar _ = []
144− -- Convert a AST value to a String if it is a Var
145− toVar (Var _ v) = Just v
146− toVar _ = Nothing
src/CodeGen/IL/Optimizer/TCO.hs+17−15
@@ -1,4 +1,4 @@
1−module CodeGen.IL.Optimizer.TCO (tco, tcoLoop) where
1+module CodeGen.IL.Optimizer.TCO (tco) where
22
33import Prelude.Compat
44
@@ -11,22 +11,19 @@ import CodeGen.IL.Common
1111
1212import qualified Language.PureScript.Constants as C
1313
14−tcoLoop :: Text
15−tcoLoop = "ṩtco_loop"
16−
1714tco :: AST -> AST -> AST
1815tco mn = everywhere convert where
1916 tcoVar :: Text -> Text
20− tcoVar arg = "ṩtco_var_" <> arg
17+ tcoVar arg = "_TCO_" <> arg <> "_"
2118
2219 copyVar :: Text -> Text
23− copyVar arg = "ṩcopy_" <> arg
20+ copyVar arg = "_TCO_Copy_" <> arg <> "_"
2421
2522 tcoDone :: Text
26− tcoDone = "ṩtco_done"
23+ tcoDone = "_TCO_Done_"
2724
2825 tcoResult :: Text
29− tcoResult = "ṩtco_result"
26+ tcoResult = "_TCO_Result_"
3027
3128 convert :: AST -> AST
3229 convert fn@(Function ss (Just name) args _)
@@ -40,6 +37,7 @@ tco mn = everywhere convert where
4037 | isTailRecursive name body'
4138 = Assignment ss (Var ss name) (replace (toLoop name outerArgs innerArgs body'))
4239 where
40+ -- fn = Function ss' Nothing args (Block ss' stmts)
4341 innerArgs = headDef [] argss
4442 outerArgs = concat . reverse $ tailSafe argss
4543 (argss, body', replace) = collectAllFunctionArgs [] id fn
@@ -92,14 +90,18 @@ tco mn = everywhere convert where
9290 toLoop :: Text -> [Text] -> [Text] -> AST -> AST
9391 toLoop ident outerArgs innerArgs js =
9492 Block rootSS $
95− concatMap (\arg -> [ VariableIntroduction rootSS (tcoVar arg) . Just $
96− Var rootSS (copyVar arg) ]) (outerArgs ++ innerArgs) ++
97− [ Assignment rootSS (Var rootSS $ "var " <> tcoDone) $ BooleanLiteral Nothing False
93+ concatMap (\arg -> [ VariableIntroduction rootSS (tcoVar arg) Nothing
94+ , Assignment rootSS (Var rootSS (tcoVar arg)) (Var rootSS (copyVar arg)) ]) (outerArgs ++ innerArgs) ++
95+ -- [ VariableIntroduction rootSS (blockDecl <> tcoDone) Nothing
96+ [ Var rootSS ("bool " <> tcoDone <> " = false")
97+ -- , Assignment rootSS (Var rootSS tcoDone) (BooleanLiteral rootSS False)
98+ -- , Assignment rootSS (Var rootSS tcoDone) (Var rootSS "NO")
9899 , VariableIntroduction rootSS tcoResult Nothing
99− , Assignment rootSS (Var rootSS $ "var " <> tcoLoop) (Function rootSS (Just tcoLoop) (outerArgs ++ innerArgs) (Block rootSS [loopify js]))
100+ , Assignment rootSS (Var rootSS ("const auto " <> tcoLoop)) (Function rootSS (Just tcoLoop) (outerArgs ++ innerArgs) (Block rootSS [loopify js]))
101+ -- , While rootSS (Binary Nothing EqualTo (Var rootSS tcoDone) (BooleanLiteral Nothing False))
100102 , While rootSS (Unary Nothing Not (Var rootSS tcoDone))
101103 (Block rootSS
102− [(Assignment rootSS (Var rootSS tcoResult) (App rootSS (Var rootSS tcoLoop) ((map (Var rootSS . tcoVar) outerArgs) ++ (map (Var rootSS . copyVar) innerArgs))))])
104+ [(Assignment rootSS (Var rootSS tcoResult) (App rootSS (Var rootSS tcoLoop) ((map (Var rootSS . tcoVar) outerArgs) ++ (map (Var rootSS . tcoVar) innerArgs))))])
103105 , Return rootSS (Var rootSS tcoResult)
104106 ]
105107 where
@@ -115,7 +117,7 @@ tco mn = everywhere convert where
115117 zipWith (\val arg ->
116118 Assignment ss (Var ss (tcoVar arg)) val) allArgumentValues outerArgs
117119 ++ zipWith (\val arg ->
118− Assignment ss (Var ss (copyVar arg)) val) (drop (length outerArgs) allArgumentValues) innerArgs
120+ Assignment ss (Var ss (tcoVar arg)) val) (drop (length outerArgs) allArgumentValues) innerArgs
119121 ++ [ ReturnNoResult ss ]
120122 | otherwise = Block ss [ markDone ss, Return ss ret ]
121123 loopify (ReturnNoResult ss) = Block ss [ markDone ss, ReturnNoResult ss ]
@@ -128,6 +130,7 @@ tco mn = everywhere convert where
128130
129131 markDone :: Maybe SourceSpan -> AST
130132 markDone ss = Assignment ss (Var ss tcoDone) (BooleanLiteral ss True)
133+ -- markDone ss = Assignment ss (Var ss tcoDone) (Var ss "YES")
131134
132135 collectArgs :: [[AST]] -> AST -> [[AST]]
133136 collectArgs acc (App _ fn args') = collectArgs (args' : acc) fn
@@ -135,7 +138,6 @@ tco mn = everywhere convert where
135138
136139 isSelfCall :: Text -> AST -> Bool
137140 isSelfCall ident (App _ (Var _ ident') _) = ident == ident'
138− isSelfCall ident (App _ (Indexer _ (Var _ ident') (Var _ mn')) _) = mn' == "" && ident == ident'
139141 isSelfCall ident (App _ (Indexer _ (Var _ ident') mn') _) = mn' == mn && ident == ident'
140142 isSelfCall ident (App _ fn _) = isSelfCall ident fn
141143 isSelfCall _ _ = False
src/CodeGen/IL/Printer.hs+209−221
@@ -7,9 +7,6 @@ module CodeGen.IL.Printer
77 , implHeaderSource
88 , implFooterSource
99 , isLiteral
10− , modLabel
11− , modPrefix
12− , ffiLoader
1310 ) where
1411
1512import Prelude.Compat
@@ -18,6 +15,7 @@ import Control.Arrow ((<+>))
1815import Control.Monad (forM, mzero)
1916import Control.Monad.State (StateT, evalStateT)
2017import Control.PatternArrows
18+import Data.Char (isAscii)
2119import Data.List (filter, (\\))
2220import qualified Control.Arrow as A
2321
@@ -33,9 +31,7 @@ import Language.PureScript.Crash
3331import Language.PureScript.Names (Ident, runIdent)
3432import Language.PureScript.Pretty.Common
3533import Language.PureScript.PSString (PSString, decodeString, mkString)
36−
3734import CodeGen.IL.Common
38−import CodeGen.IL.Optimizer.TCO (tcoLoop)
3935
4036import qualified Language.PureScript.Constants as C
4137
@@ -53,23 +49,26 @@ literals = mkPattern' match'
5349 match (BooleanLiteral _ True) = return $ emit "true"
5450 match (BooleanLiteral _ False) = return $ emit "false"
5551 match (ArrayLiteral _ xs) = mconcat <$> sequence
56− [ return . emit $ arrayType <> "{"
52+ [ return . emit $ arrayType <> "{ "
5753 , intercalate (emit ", ") <$> forM xs prettyPrintIL'
58− , return $ emit "}"
54+ , return $ emit " }"
5955 ]
6056 -- match (ObjectLiteral _ []) = return $ emit "std::initializer_list<std::pair<const string, boxed>>{}"
6157 match (ObjectLiteral _ ps) = mconcat <$> sequence
62− [ return . emit $ dictType <> "{"
58+ [ return . emit $ dictType <> "{\n"
6359 , withIndent $ do
6460 ils <- forM ps $ \(key, value) -> do
6561 value' <- prettyPrintIL' value
66− return $ objectPropertyToString key <> value'
67− return $ intercalate (emit " ") $ map (<> emit ",") ils
62+ return $ emit "{ " <> objectPropertyToString key <> value' <> emit " }"
63+ indentString <- currentIndent
64+ return $ intercalate (emit ",\n") $ map (indentString <>) ils
65+ , return $ emit "\n"
66+ , currentIndent
6867 , return $ emit "}"
6968 ]
7069 where
7170 objectPropertyToString :: (Emit gen) => PSString -> gen
72− objectPropertyToString s = emit $ stringLiteral s <> ": "
71+ objectPropertyToString s = emit $ stringLiteral s <> ", "
7372 match (Block _ sts) = mconcat <$> sequence
7473 [ return $ emit "{\n"
7574 , withIndent $ prettyStatements sts
@@ -77,10 +76,9 @@ literals = mkPattern' match'
7776 , currentIndent
7877 , return $ emit "}"
7978 ]
80− match (Var _ ident) | ident == C.undefined = return $ emit undefinedName
8179 match (Var _ ident) = return $ emit ident
8280 match (VariableIntroduction _ ident value) = mconcat <$> sequence
83− [ return . emit $ varDecl <> " " <> ident <> " " <> anyType
81+ [ return $ emit $ "boxed " <> ident
8482 , maybe (return mempty) (fmap (emit " = " <>) . prettyPrintIL') value
8583 ]
8684 match (Assignment _ target value) = mconcat <$> sequence
@@ -89,195 +87,163 @@ literals = mkPattern' match'
8987 , prettyPrintIL' value
9088 ]
9189 match (App _ val []) = mconcat <$> sequence
92− [ return $ emit "Run("
93− , prettyPrintIL' val
94− , return $ emit ")"
95− ]
96− match (App _ (StringLiteral _ u) [arg])
97− | Just ty <- decodeString u = mconcat <$> sequence
98− [ prettyPrintIL' arg
99− , return $ emit ".("
100− , return $ emit ty
101− , return $ emit ")"
102− ]
103− match (App _ (Var _ fn) [arg]) | fn == arrayLengthFn = mconcat <$> sequence
104− [ return $ emit fn
105− , return $ emit "("
106− , prettyPrintIL' arg
107− , return $ emit ")"
108− ]
109− match (App _ (Var _ fn) args) | fn == indexFn = mconcat <$> sequence
110− [ return $ emit fn
111− , return $ emit "("
112− , intercalate (emit ", ") <$> forM args prettyPrintIL'
113− , return $ emit ")"
114− ]
115− match (App _ (Var _ fn) args) | fn == tcoLoop = mconcat <$> sequence
116− [ return $ emit fn
117− , return $ emit "("
118− , intercalate (emit ", ") <$> forM args prettyPrintIL'
119− , return $ emit ")"
120− ]
121− match (App _ (App Nothing (StringLiteral Nothing fnx) [fn@Function{}]) args)
122− | Just ftype <- decodeString fnx
123− , "Fn" `T.isPrefixOf` ftype = mconcat <$> sequence
124− [ prettyPrintIL' fn
125− , return $ emit "("
126− , intercalate (emit ", ") <$> forM args prettyPrintIL'
127− , return $ emit ")"
128− ]
129− match (App _ (App Nothing (StringLiteral Nothing fnx) [fn]) args)
130− | Just ftype <- decodeString fnx
131− , "Fn" `T.isPrefixOf` ftype = mconcat <$> sequence
132− [ prettyPrintIL' fn
133− , return $ emit ".("
134− , return $ emit ftype
135− , return $ emit ")("
136− , intercalate (emit ", ") <$> forM args prettyPrintIL'
137− , return $ emit ")"
90+ [ prettyPrintIL' val
91+ , return $ emit "()"
13892 ]
139− match app@(App{})
140− | (val, args) <- unApp app []
141− , length args > 1
93+ match (App ss (App _ (App _ (Indexer _ (Var _ fn) (Var _ fnMod)) [Indexer _ (Var _ dict) (Var _ dictMod)]) [x]) [y])
94+ | fnMod == dictMod
95+ , dictMod == C.dataEq || dictMod == C.dataSemiring || dictMod == C.dataRing ||
96+ dictMod == C.dataEuclideanRing || dictMod == C.dataHeytingAlgebra
97+ , Just op <- renderOp fn
98+ , Just t <- unboxType dict
14299 = mconcat <$> sequence
143− [ return $ emit "Apply("
144− , prettyPrintIL' val
145− , return $ emit ", "
146− , intercalate (emit ", ") <$> forM args prettyPrintIL'
147− , return $ emit ")"
100+ [ return $ emit $ unbox' t x
101+ , return $ emit op
102+ , return $ emit $ unbox' t y
103+ ]
104+ match (App _ (App _ (Indexer _ (Var _ fn) (Var _ fnMod)) [Indexer _ (Var _ dict) (Var _ dictMod)]) [n])
105+ | fnMod == dictMod
106+ , dictMod == C.dataRing
107+ , fn == C.negate
108+ , Just t <- unboxType dict
109+ = mconcat <$> sequence
110+ [ return $ emit "-"
111+ , return $ emit $ unbox' t n
112+ ]
113+ match (App _ (App _ (Indexer _ (Var _ fn) (Var _ fnMod)) [Indexer _ (Var _ dict) (Var _ dictMod)]) [n])
114+ | fnMod == dictMod
115+ , dictMod == C.dataHeytingAlgebra
116+ , fn == properToIL C.not
117+ , Just t <- unboxType dict
118+ = mconcat <$> sequence
119+ [ return $ emit "!"
120+ , return $ emit $ unbox' t n
148121 ]
149122 match (App _ val args) = mconcat <$> sequence
150− [ return $ emit "Apply("
151− , prettyPrintIL' val
152− , return $ emit ", "
123+ [ prettyPrintIL' val
124+ , return $ emit "("
153125 , intercalate (emit ", ") <$> forM args prettyPrintIL'
154126 , return $ emit ")"
155127 ]
156128 match (Function _ (Just name) [] (Block _ [Return _ ret]))
157− | isComplexLiteral ret = mconcat <$> sequence
158− [ return $ emit "\n"
159− , return . emit $ varDecl <> " " <> initName name <> " Once\n"
160− , prettyPrintIL' $ VariableIntroduction Nothing (valueName name) Nothing
161− , return $ emit "\n\n"
162− , return $ emit "func "
163− , return . emit $ withPrefix name
164− , return . emit $ "() " <> anyType <> " "
129+ | isLiteral ret = mconcat <$> sequence
130+ [ return $ emit "auto "
131+ , return $ emit name
132+ , return $ emit "() -> const boxed& "
165133 , return $ emit "{\n"
166134 , withIndent $ do
167− indentString <- currentIndent
168− return $ indentString <> (emit $ initName name <> ".Do(func() {\n") <>
169− indentString <> indentString <> (emit $ valueName name <> " = ")
135+ indentString <- currentIndent
136+ return $ indentString <> emit staticValueBegin
170137 , withIndent $ do
171138 case ret of
172139 Function _ Nothing args body -> mconcat <$> sequence
173− [ return $ emit "func("
174− , return . emit $ intercalate ", " (renderArg anyType <$> args)
175− , return . emit $ ") " <> anyType <> " "
140+ [ return $ emit "[]("
141+ , return $ emit $ intercalate ", " (renderArg <$> args)
142+ , return $ emit ") -> boxed "
176143 , prettyPrintIL' body
177144 ]
178145 _ -> do
179146 prettyPrintIL' ret
180− , return $ emit "\n"
147+ , return $ emit ";\n"
181148 , withIndent $ do
182− indentString <- currentIndent
183− return $ indentString <> (emit $ "})\n" <> "return " <> valueName name <>"\n")
184− , return $ emit "}\n\n"
149+ indentString <- currentIndent
150+ return $ indentString <> emit staticValueEnd
151+ , return $ emit "\n}"
185152 ]
186153 match (Function _ (Just name) [] ret) = mconcat <$> sequence
187− [ return $ emit "func "
188− , return . emit $ withPrefix name
189− , return . emit $ "() " <> anyType <> " "
154+ [ return $ emit "auto "
155+ , return $ emit name
156+ , return $ emit "() -> boxed "
190157 , prettyPrintIL' ret
191158 ]
192− match (Function _ _ args ret) = mconcat <$> sequence
193− [ return $ emit "func("
194− , return . emit $ intercalate ", " (renderArg anyType <$> args)
195− , return . emit $ ") " <> anyType <> " "
159+ match (Function _ name args ret) = mconcat <$> sequence
160+ [ return $ emit captures
161+ , return $ emit "("
162+ , return $ emit $ intercalate ", " (render <$> args)
163+ , return $ emit ") -> boxed "
196164 , prettyPrintIL' ret
197165 ]
198− match (Indexer _ (Var _ name) (Var _ "")) = mconcat <$> sequence
199− [ prettyPrintIL' (Var Nothing $ withPrefix name)
200− , return $ emit "()"
201− ]
202− match (Indexer _ (Var _ name) val) = mconcat <$> sequence
166+ where
167+ (captures, render)
168+ | name == Just tcoLoop = ("[&]", renderArgByVal)
169+ | otherwise = ("[=]", renderArg)
170+ match (Indexer _ prop@(Var _ name) val) = mconcat <$> sequence
203171 [ prettyPrintIL' val
204− , return $ emit "."
205− , prettyPrintIL' (Var Nothing $ withPrefix name)
172+ , return $ emit "::"
173+ , prettyPrintIL' prop
206174 , return $ emit "()"
207175 ]
208− match (Indexer _ prop@StringLiteral{} val@ObjectLiteral{}) = mconcat <$> sequence
176+ match (Indexer _ prop val) = mconcat <$> sequence
209177 [ prettyPrintIL' val
210178 , return $ emit "["
211179 , prettyPrintIL' prop
212180 , return $ emit "]"
213181 ]
214− match (Indexer _ prop@StringLiteral{} val) = mconcat <$> sequence
215− [ prettyPrintIL' val
216− , return $ emit ".("
217− , return $ emit dictType
218− , return $ emit ")["
219− , prettyPrintIL' prop
220− , return $ emit "]"
221− ]
222− match (Indexer _ index val@ArrayLiteral{}) = mconcat <$> sequence
223− [ prettyPrintIL' val
224− , return $ emit "["
225− , prettyPrintIL' index
226− , return $ emit "]"
227− ]
228− match (Indexer _ index val) = mconcat <$> sequence
229− [ prettyPrintIL' val
230− , return $ emit ".("
231− , return $ emit arrayType
232− , return $ emit ")["
233− , prettyPrintIL' index
234− , return $ emit "]"
235− ]
236182 match (InstanceOf _ val ty) = mconcat <$> sequence
237− [ return $ emit "Is("
238− , prettyPrintIL' val
239− , return $ emit ", "
183+ [ return . emit $ unbox' dictType val
184+ , return $ emit ".contains("
240185 , prettyPrintIL' ty
241186 , return $ emit ")"
242187 ]
243188 match (While _ cond sts) = mconcat <$> sequence
244− [ return $ emit "for "
189+ [ return $ emit "while ("
245190 , prettyPrintIL' cond
246− , return $ emit " "
191+ , return $ emit ") "
247192 , prettyPrintIL' sts
248193 ]
249194 match (For _ ident start end sts) = mconcat <$> sequence
250− [ return $ emit $ "for var " <> ident <> " = "
195+ [ return $ emit $ "for (boxed " <> ident <> " = "
251196 , prettyPrintIL' start
252197 , return $ emit $ "; " <> ident <> " < "
253198 , prettyPrintIL' end
254− , return $ emit $ "; " <> ident <> "++ "
199+ , return $ emit $ "; " <> ident <> "++) "
255200 , prettyPrintIL' sts
256201 ]
257202 match (ForIn _ ident obj sts) = mconcat <$> sequence
258− [ return $ emit $ "for var " <> ident <> " in "
203+ [ return $ emit $ "for (boxed " <> ident <> " in "
259204 , prettyPrintIL' obj
260− , return $ emit " "
205+ , return $ emit ") "
261206 , prettyPrintIL' sts
262207 ]
263− match (IfElse _ (Binary _ EqualTo cond@Binary{} (BooleanLiteral Nothing True)) thens elses) = mconcat <$> sequence
264− [ return $ emit "if "
265− , prettyPrintIL' cond
266− , return $ emit " "
208+ match (IfElse _ (Binary _ EqualTo cond (BooleanLiteral Nothing True)) thens elses) = mconcat <$> sequence
209+ [ return $ emit "if ("
210+ , return $ emit $ unbox' "bool" cond
211+ , return $ emit ") "
267212 , prettyPrintIL' thens
268213 , maybe (return mempty) (fmap (emit " else " <>) . prettyPrintIL') elses
269214 ]
270− match (IfElse _ (Binary _ EqualTo cond@Binary{} (BooleanLiteral Nothing False)) thens elses) = mconcat <$> sequence
271− [ return $ emit "if !("
272− , prettyPrintIL' cond
215+ match (IfElse _ (Binary _ EqualTo cond (BooleanLiteral Nothing False)) thens elses) = mconcat <$> sequence
216+ [ return $ emit "if (!("
217+ , return $ emit $ unbox' "bool" cond
218+ , return $ emit ")) "
219+ , prettyPrintIL' thens
220+ , maybe (return mempty) (fmap (emit " else " <>) . prettyPrintIL') elses
221+ ]
222+ match (IfElse _ (Binary _ EqualTo x y@(NumericLiteral _ n)) thens elses) = mconcat <$> sequence
223+ [ return $ emit "if ("
224+ , return $ emit $ unbox' t x
225+ , return $ emit $ " == "
226+ , prettyPrintIL' y
227+ , return $ emit ") "
228+ , prettyPrintIL' thens
229+ , maybe (return mempty) (fmap (emit " else " <>) . prettyPrintIL') elses
230+ ]
231+ where
232+ t | Left _ <- n = int
233+ | Right _ <- n = float
234+ match (IfElse _ (Binary _ EqualTo a b@StringLiteral{}) thens elses) = mconcat <$> sequence
235+ [ return $ emit "if ("
236+ , return $ emit $ unbox' string a
237+ , return $ emit $ " == "
238+ , prettyPrintIL' b
273239 , return $ emit ") "
274240 , prettyPrintIL' thens
275241 , maybe (return mempty) (fmap (emit " else " <>) . prettyPrintIL') elses
276242 ]
277243 match (IfElse _ cond thens elses) = mconcat <$> sequence
278− [ return $ emit "if "
244+ [ return $ emit "if ("
279245 , prettyPrintIL' cond
280− , return $ emit " "
246+ , return $ emit ") "
281247 , prettyPrintIL' thens
282248 , maybe (return mempty) (fmap (emit " else " <>) . prettyPrintIL') elses
283249 ]
@@ -285,10 +251,11 @@ literals = mkPattern' match'
285251 [ return $ emit "return "
286252 , prettyPrintIL' value
287253 ]
288− match (ReturnNoResult _) = return . emit $ "return " <> undefinedName
254+ match (ReturnNoResult _) = return $ emit "return undefined"
289255 -- match (Throw _ _) = return mempty
290256 match (Throw _ value) = mconcat <$> sequence
291− [ return $ emit "panic("
257+ [ return $ emit "THROW_("
258+ , return $ emit "\"PatternMatchFailure: \""
292259 , prettyPrintIL' value
293260 , return $ emit ")"
294261 ]
@@ -328,7 +295,7 @@ literals = mkPattern' match'
328295 Nothing -> ""
329296
330297unary' :: (Emit gen) => UnaryOperator -> (AST -> Text) -> Operator PrinterState AST gen
331−unary' op mkStr = Wrap match (<>)
298+unary' op mkStr = Wrap match (const)
332299 where
333300 match :: (Emit gen) => Pattern PrinterState AST (gen, AST)
334301 match = mkPattern match'
@@ -337,7 +304,7 @@ unary' op mkStr = Wrap match (<>)
337304 match' _ = Nothing
338305
339306unary :: (Emit gen) => UnaryOperator -> Text -> Operator PrinterState AST gen
340−unary op str = unary' op (const str)
307+unary op str = unary' op (\v -> str <> "(" <> prettyPrintIL1 v <> ")")
341308
342309negateOperator :: (Emit gen) => Operator PrinterState AST gen
343310negateOperator = unary' Negate (\v -> if isNegate v then "- " else "-")
@@ -363,21 +330,11 @@ isLiteral ObjectLiteral{} = True
363330isLiteral ArrayLiteral{} = True
364331isLiteral _ = False
365332
366−isComplexLiteral :: AST -> Bool
367−-- isComplexLiteral Function{} = True
368−isComplexLiteral ObjectLiteral{} = True
369−isComplexLiteral ArrayLiteral{} = True
370−isComplexLiteral _ = False
371−
372−unApp :: AST -> [AST] -> (AST, [AST])
373−unApp (App _ val [arg]) args = unApp val (arg : args)
374−unApp other args = (other, args)
375−
376333prettyStatements :: (Emit gen) => [AST] -> StateT PrinterState Maybe gen
377334prettyStatements sts = do
378335 ils <- forM sts prettyPrintIL'
379336 indentString <- currentIndent
380− return $ intercalate (emit "\n") $ map (indentString <>) ils
337+ return $ intercalate (emit "\n") $ map ((<> emit ";") . (indentString <>)) ils
381338
382339-- | Generate a pretty-printed string representing a collection of C++ expressions at the same indentation level
383340prettyPrintILWithSourceMaps :: [AST] -> (Text, [SMap])
@@ -396,7 +353,7 @@ prettyPrintIL' = A.runKleisli $ runPattern matchValue
396353 matchValue = buildPrettyPrinter operators (literals <+> fmap parensPos matchValue)
397354 operators :: (Emit gen) => OperatorTable PrinterState AST gen
398355 operators =
399− OperatorTable [ [ unary New "?" ]
356+ OperatorTable [ [ unary New "box" ]
400357 , [ unary Not "!"
401358 , unary BitwiseNot "~"
402359 , unary Positive "+"
@@ -407,7 +364,8 @@ prettyPrintIL' = A.runKleisli $ runPattern matchValue
407364 , [ binary Add "+"
408365 , binary Subtract "-" ]
409366 , [ binary ShiftLeft "<<"
410− , binary ShiftRight ">>" ]
367+ , binary ShiftRight ">>"
368+ , binary ZeroFillShiftRight ">>>" ]
411369 , [ binary LessThan "<"
412370 , binary LessThanOrEqualTo "<="
413371 , binary GreaterThan ">"
@@ -425,12 +383,15 @@ prettyPrintIL1 :: AST -> Text
425383prettyPrintIL1 = maybe (internalError "Incomplete pattern") runPlainString . flip evalStateT (PrinterState 0) . prettyPrintIL'
426384
427385stringLiteral :: PSString -> Text
428−stringLiteral pss | Just s <- decodeString pss = stringLiteral' s
386+stringLiteral pss | Just s <- decodeString pss =
387+ (if T.all isAscii s
388+ then ""
389+ else ("u8")) <> stringLiteral' s
429390 where
430391 stringLiteral' :: Text -> Text
431392 stringLiteral' s = "\"" <> T.concatMap encodeChar s <> "\""
432393 encodeChar :: Char -> Text
433− encodeChar '\0' = "\\x00"
394+ encodeChar '\0' = "\\0"
434395 encodeChar '\b' = "\\b"
435396 encodeChar '\t' = "\\t"
436397 encodeChar '\n' = "\\n"
@@ -442,29 +403,62 @@ stringLiteral pss | Just s <- decodeString pss = stringLiteral' s
442403 encodeChar c = T.singleton $ c
443404stringLiteral _ = "\"\\uFFFD\""
444405
406+unbox' :: Text -> AST -> Text
407+unbox' _ v@(NumericLiteral{}) = prettyPrintIL1 v
408+unbox' _ v@(BooleanLiteral{}) = prettyPrintIL1 v
409+unbox' _ v@(StringLiteral{}) = prettyPrintIL1 v
410+unbox' t v = unbox t <> "(" <> prettyPrintIL1 v <> ")"
411+
412+unboxType :: Text -> Maybe Text
413+unboxType t
414+ | t == C.eqInt ||
415+ t == C.semiringInt ||
416+ t == C.ringInt
417+ = Just int
418+ | t == C.semiringNumber ||
419+ t == C.ringNumber ||
420+ t == C.euclideanRingNumber
421+ = Just float
422+ | t == C.heytingAlgebraBoolean
423+ = Just bool
424+ | otherwise = Nothing
425+
426+renderOp :: Text -> Maybe Text
427+renderOp op
428+ | op == C.eq = Just " == "
429+ | op == C.notEq = Just " != "
430+ | op == C.add = Just " + "
431+ | op == C.sub = Just " - "
432+ | op == C.mul = Just " * "
433+ | op == C.div = Just " / "
434+ | op == C.conj = Just " && "
435+ | op == C.disj = Just "|| "
436+ | otherwise = Nothing
437+
445438interfaceSource :: Text -> [(Text,Bool)] -> [Ident] -> Text
446−interfaceSource _ _ _ = ""
439+interfaceSource mn values foreigns =
440+ let foreigns' = identToIL <$> foreigns
441+ valueNames = fst <$> values
442+ values' = filter ((`notElem` foreigns') . fst) values ++
443+ zip (foreigns' \\ valueNames) (repeat True)
444+ in
445+ "// Generated by psil compiler\n\n" <>
446+ "#ifndef " <> mn <> "_H\n" <>
447+ "#define " <> mn <> "_H\n\n" <>
448+ "#include \"purescript.h\"\n\n" <>
449+ "namespace " <> mn <> " {\n\n" <>
450+ "using namespace purescript;\n\n" <>
451+ (T.concat $ (\(export, static) ->
452+ let rty = if static then "const boxed&" else "boxed" in
453+ "auto " <> export <> "() -> " <> rty <> ";\n") <$> values') <> "\n" <>
454+ "} // end namespace " <> mn <> "\n\n" <>
455+ "#endif // " <> mn <> "_H\n\n"
447456
448457implHeaderSource :: Text -> [Text] -> Text -> Text
449−implHeaderSource mn imports otherPrefix =
450− "// Code generated by psgo. DO NOT EDIT.\n\n" <>
451− "package " <> (if mn == "Main" || mn == "Test_Main" then "main" else mn) <> "\n\n" <>
452− "import . \"" <> runtime <> "\"\n" <>
453− (if mn == "Main" || mn == "Test_Main"
454− then "import _ \"" <> otherPrefix <> "/" <> ffiLoader <> "\"\n"
455− else "\n") <>
456− "import (\n" <>
457− (T.concat $ formatImport <$> imports) <> ")\n\n" <>
458− "type _ = Any\n\n"
459− where
460− formatImport :: Text -> Text
461− formatImport s =
462− "\t\"" <> otherPrefix <> "/" <> modPrefix <> "/" <> moduleRenamer s <> "\"\n"
463− moduleRenamer :: Text -> Text
464− moduleRenamer s
465− | (s', sfx) <- T.breakOn moduleRenamerMarker s,
466− not $ T.null sfx = s'
467− moduleRenamer s = s
458+implHeaderSource mn imports interfaceImport =
459+ T.concat imports <> "\n"<>
460+ interfaceImport <> "\n\n" <>
461+ "namespace " <> mn <> " {\n\n"
468462
469463implFooterSource :: Text -> [Ident] -> Text
470464implFooterSource mn foreigns =
@@ -472,56 +466,50 @@ implFooterSource mn foreigns =
472466 (if null foreigns
473467 then ""
474468 else ("// Foreign values\n\n" <>
475− "var " <> foreignDict <> " = "<> foreignMod <> "(\"" <> mn <> "\")\n\n" <>
476− (T.concat $ (\foreign' ->
477− let name = moduleIdentToIL foreign' in
478− varDecl <> " " <> initName name <> " Once\n" <>
479− varDecl <> " " <> valueName name <> " " <> anyType <> "\n\n" <>
480− "func " <>
481− withPrefix name <>
482− "() Any { \n" <>
483− " " <> initName name <> ".Do(func() {\n" <>
484− " " <> valueName name <> " = " <>
485− "Get(" <> foreignDict <> ", " <>
486− (stringLiteral . mkString $ runIdent foreign') <> ")\n" <>
487− " })\n" <>
488− " return " <> valueName name <> "\n" <>
489− "}\n\n") <$> foreigns))) <>
490− if mn == "Main" || mn == "Test.Main" then mainSource else "\n"
469+ "DEFINE_FOREIGN_DICTIONARY_AND_ACCESSOR()\n\n" <>
470+ (T.concat $ (\foreign' -> "auto " <>
471+ identToIL foreign' <>
472+ "() -> const boxed& { " <>
473+ staticValueBegin <>
474+ foreignDict <> "().at(" <>
475+ (stringLiteral . mkString $ runIdent foreign') <> "); " <>
476+ staticValueEnd <>
477+ " };\n") <$> foreigns) <>
478+ "\n")) <>
479+ "} // end namespace " <> mn <>
480+ "\n\n" <>
481+ if mn == "Main" then mainSource else "\n"
491482 where
492483 mainSource :: Text
493484 mainSource = "\
494− \func main() {\n\
495− \ Run(Ꞌmain())\n\
485+ \int main(int argc, const char * argv[]) {\n\
486+ \ Main::main()();\n\
487+ \ return 0;\n\
496488 \}\n\n\
497489 \"
498490
499491varDecl :: Text
500−varDecl = "var"
492+varDecl = "const boxed"
501493
502−renderArg :: Text -> Text -> Text
503−renderArg decl arg = arg <> " " <> decl
494+varRefDecl :: Text
495+varRefDecl = varDecl <> "&"
504496
505−foreignMod :: Text
506−foreignMod = "Foreign"
497+renderArg' :: Text -> Text -> Text
498+renderArg' decl arg
499+ | arg == unusedName = decl
500+ | otherwise = decl <> " " <> arg
507501
508−foreignDict :: Text
509−foreignDict = "foreign"
502+renderArg :: Text -> Text
503+renderArg = renderArg' varRefDecl
510504
511−modLabel :: Text
512−modLabel = "purescript-native"
505+renderArgByVal :: Text -> Text
506+renderArgByVal = renderArg' varDecl
513507
514−modPrefix :: Text
515−modPrefix = modLabel <> "/output"
516−
517−runtime :: Text
518−runtime = "github.com/" <> modLabel <> "/go-runtime"
519−
520−ffiLoader :: Text
521−ffiLoader = modLabel <> "/ffi-loader"
508+foreignDict :: Text
509+foreignDict = "foreign"
522510
523−initName :: Text -> Text
524−initName s = "ᛌ" <> s <> "_once"
511+staticValueBegin :: Text
512+staticValueBegin = "static const boxed _ = "
525513
526−valueName :: Text -> Text
527−valueName s = "ᛧ" <> s
514+staticValueEnd :: Text
515+staticValueEnd = "return _;"
src/Tests.hs+12−11
@@ -31,13 +31,18 @@ runTests = do
3131 let srcDir = outputDir </> "src"
3232 createDirectory srcDir
3333
34− callProcess "git" ["clone", "--branch", "v0.13.8", "--depth", "1", "https://github.com/purescript/purescript.git"]
34+ callProcess "git" ["clone", "--branch", "v0.13.0", "--depth", "1", "https://github.com/purescript/purescript.git"]
3535 let passingDir = baseDir </> "purescript" </> "tests" </> "purs" </> "passing"
3636 passingTestCases <- sort . filter (".purs" `isSuffixOf`) <$> getDirectoryContents passingDir
3737
3838 setCurrentDirectory outputDir
39+ callProcess "pscpp" ["--makefile"]
3940
4041 fetchPackages
42+ callProcess "git" ["clone", "--depth", "1", "https://github.com/andyarvanitis/purescript-cpp-ffi.git", "ffi"]
43+
44+ callProcess "rm" ["-rf", ".psc-package/psc-0.13.0/prelude/v4.1.0"]
45+ callProcess "git" ["clone", "--branch", "v4.1.1", "--depth", "1", "https://github.com/purescript/purescript-prelude.git", ".psc-package/psc-0.13.0/prelude/v4.1.0"]
4146
4247 let tests = filter (`notElem` skipped) passingTestCases
4348
@@ -60,15 +65,14 @@ runTests = do
6065 testCaseDirExists <- doesDirectoryExist testCaseDir
6166 when testCaseDirExists $ callProcess "cp" ["-R", testCaseDir, srcDir]
6267
63− callProcess "rm" ["-rf", "output"]
64− callProcess "rm" ["-rf", "Main"]
65− callProcess "spago" ["build"]
68+ callProcess "make" ["clean"]
69+ callProcess "make" ["debug", "-j8"]
6670 --
67− -- Run executable file
71+ -- Run C++ files
6872 --
6973 outputFile <- openFile (tmp </> logpath </> logfile) AppendMode
7074 hPutStrLn outputFile ("\n" ++ inputFile ++ ":")
71− proc <- runProcess ("./" </> "Main") [] Nothing Nothing Nothing (Just outputFile) Nothing
75+ proc <- runProcess ("output" </> "bin" </> "main") [] Nothing Nothing Nothing (Just outputFile) Nothing
7276
7377 removeFile (srcDir </> inputFile)
7478 when testCaseDirExists $ callProcess "rm" ["-rf", srcDir </> (takeWhile (/='.') inputFile)]
@@ -112,11 +116,8 @@ packages =
112116fetchPackages :: IO ()
113117-------------------------------------------------------------------------------
114118fetchPackages = do
115− callProcess "spago" ["init"]
116− callCommand "sed '$s/\\(}\\)/, backend = \"psgo\"\\1/' < spago.dhall | dhall format > spago.tmp"
117− callCommand "mv spago.tmp spago.dhall"
118−
119− mapM (callProcess "spago" . (\p -> ["install", p])) packages
119+ callProcess "psc-package" ["init"]
120+ mapM (callProcess "psc-package" . (\p -> ["install", p])) packages
120121 return ()
121122
122123-------------------------------------------------------------------------------
stack.nix+3−18
@@ -1,23 +1,8 @@
1−#
2−# You may need to manually run `unset STACK_IN_NIX_SHELL`
3−# (doesn't work from a `shellHook`; see: https://github.com/commercialhaskell/stack/issues/5000
4−#
5−#
6−# You may need to set LC_ALL when building, e.g.
7−# LC_ALL=en_US.iso88591 stack --nix build; see e.g. https://github.com/koalaman/shellcheck/issues/324
8−#
9−#
10−{ pkgs ?
11− import (builtins.fetchTarball {
12− name = "nixos-release-19.03";
13− url = https://github.com/nixos/nixpkgs/archive/34c7eb7545d155cc5b6f499b23a7cb1c96ab4d59.tar.gz;
14− sha256 = "11z6ajj108fy2q5g8y4higlcaqncrbjm3dnv17pvif6avagw4mcb";
15− }) {}
16−}:
1+{ pkgs ? import <nixpkgs> {}, ghc ? pkgs.ghc }:
172
183pkgs.haskell.lib.buildStackProject {
194 name = "default-stack-shell";
20− ghc = pkgs.haskell.compiler.ghc865;
21− buildInputs = with pkgs; [git git-lfs gmp ncurses zlib];
5+ inherit ghc;
6+ buildInputs = with pkgs; [git git-lfs gmp zlib];
227 LANG = "en_US.UTF-8";
238}
stack.yaml+65−16
@@ -1,24 +1,73 @@
1−resolver: lts-13.26
1+# This file was automatically generated by 'stack init'
2+#
3+# Some commonly used options have been documented as comments in this file.
4+# For advanced use and comprehensive documentation of the format, please see:
5+# https://docs.haskellstack.org/en/stable/yaml_configuration/
26
7+# Resolver to choose a 'specific' stackage snapshot or a compiler version.
8+# A snapshot resolver dictates the compiler version and the set of packages
9+# to be used for project dependencies. For example:
10+#
11+# resolver: lts-3.5
12+# resolver: nightly-2015-09-21
13+# resolver: ghc-7.10.2
14+# resolver: ghcjs-0.1.0_ghc-7.10.2
15+# resolver:
16+# name: custom-snapshot
17+# location: "./custom-snapshot.yaml"
18+resolver: lts-12.0
19+
20+# User packages to be built.
21+# Various formats can be used as shown in the example below.
22+#
23+# packages:
24+# - some-directory
25+# - https://example.com/foo/bar/baz-0.0.2.tar.gz
26+# - location:
27+# git: https://github.com/commercialhaskell/stack.git
28+# commit: e7b331f14bcffb8367cd58fbfc8b40ec7642100a
29+# - location: https://github.com/commercialhaskell/stack/commit/e7b331f14bcffb8367cd58fbfc8b40ec7642100a
30+# extra-dep: true
31+# subdirs:
32+# - auto-update
33+# - wai
34+#
35+# A package marked 'extra-dep: true' will only be built if demanded by a
36+# non-dependency (i.e. a user package), and its test suites and benchmarks
37+# will not be run. This is useful for tweaking upstream packages.
338packages:
439- .
5−
40+# Dependency packages to be pulled from upstream that are not in the resolver
41+# (e.g., acme-missiles-0.3)
642extra-deps:
7−- git: https://github.com/purescript/purescript
8− commit: 9cad73ed8ea7df3011032ddbd2f3de5a0c08629c
9−- happy-1.19.9
10−- language-javascript-0.7.0.0
11−- network-3.0.1.1
12−- these-1.0.1
13−- semialign-1
43+- git: https://github.com/purescript/purescript
44+ commit: b35412a9ea49a46d315a0214cd2c3908442ab7a7
45+- pipes-http-1.0.5
46+
47+# Override default flag values for local packages and extra-deps
48+# flags: {}
49+
50+# Extra package databases containing global packages
51+# extra-package-dbs: []
52+
53+# Control whether we use the GHC we find on the path
54+# system-ghc: true
55+#
56+# Require a specific version of stack, using version ranges
57+# require-stack-version: -any # Default
58+# require-stack-version: ">=1.6"
59+#
60+# Override the architecture used by stack, especially useful on Windows
61+# arch: i386
62+# arch: x86_64
63+#
64+# Extra directories used by stack for building
65+# extra-include-dirs: [/path/to/dir]
66+# extra-lib-dirs: [/path/to/dir]
67+#
68+# Allow a newer minor version of GHC than the snapshot specifies
69+# compiler-check: newer-minor
1470
1571nix:
16− enable: false
1772 shell-file: stack.nix
1873
19−flags:
20− aeson-pretty:
21− lib-only: true
22− these:
23− assoc: false
24− quickcheck: false
support/Makefilenew file+129
@@ -0,0 +1,129 @@
1+# PureScript to native binary (via C++) Makefile
2+#
3+# Run 'make' or 'make release' to build an optimized release build
4+# Run 'make debug' to build a non-optimized build suitable for debugging
5+#
6+# You can also perform a parallel build with 'make -jN', where N is the
7+# number of cores to use.
8+#
9+# PURS, SRC, OUTPUT, and BIN can all be overridden with the
10+# command itself. For example: 'make BIN=myutil'
11+#
12+# Flags can be added to either the codegen or native build phases.
13+# For example: 'make PURSFLAGS=--codegen,js CXXFLAGS=-DDEBUG LDFLAGS=lgmp'
14+#
15+# You can also edit the generated version of this file directly.
16+#
17+PURS := purs
18+PSC_PACKAGE := psc-package
19+PSCPP := pscpp
20+SRC := src
21+OUTPUT := output
22+CC_SRC := $(OUTPUT)/src
23+FFI_SRC := ffi
24+BIN := main
25+
26+override PURSFLAGS += compile --codegen corefn
27+override CXXFLAGS += --std=c++11
28+
29+CXXVERSION = $(shell $(CXX) --version)
30+ifneq (,$(findstring g++,$(CXXVERSION)))
31+ PSCPPFLAGS += "--ucns"
32+endif
33+
34+DEBUG := "-DDEBUG -g"
35+RELEASE := "-DNDEBUG -O3"
36+
37+INCLUDES := -I $(CC_SRC)
38+BIN_DIR := $(OUTPUT)/bin
39+
40+PACKAGE_SOURCES = $(subst \,/,$(shell $(PSC_PACKAGE) sources))
41+PURESCRIPT_PKGS := $(firstword $(subst /, ,$(PACKAGE_SOURCES)))
42+
43+## Not all environments support globstar (** dir pattern)
44+rwildcard=$(wildcard $1$2) $(foreach d,$(wildcard $1*),$(call rwildcard,$d/,$2))
45+
46+unspecified:
47+ @echo
48+ @echo "Please specify the type of build you want: 'debug' or 'release'"
49+ @echo
50+ @echo " debug => not optimized (your app will run slower), good for"
51+ @echo " running in a source-level debugger, more (descriptive)"
52+ @echo " runtime exceptions"
53+ @echo
54+ @echo " release => optimized (your app will run faster), less helpful when debugging"
55+ @echo
56+
57+debug: codegen
58+ @$(MAKE) $(BIN) CFLAGS+=$(DEBUG) CXXFLAGS+=$(DEBUG)
59+
60+release: codegen
61+ @$(MAKE) $(BIN) CFLAGS+=$(RELEASE) CXXFLAGS+=$(RELEASE)
62+
63+.PHONY: corefn
64+corefn: PURESCRIPT_PKG_SRCS=$(foreach d,$(PACKAGE_SOURCES),$(call rwildcard,$(firstword $(subst *, ,$(d))),*.purs))
65+corefn: PURESCRIPT_SRCS=$(call rwildcard,$(SRC)/,*.purs)
66+corefn: $(PURESCRIPT_PKGS)
67+ @$(PURS) $(PURSFLAGS) --output $(OUTPUT) $(PURESCRIPT_PKG_SRCS) $(PURESCRIPT_SRCS)
68+
69+.PHONY: codegen
70+codegen: COREFN_SRCS=$(call rwildcard,$(OUTPUT)/,corefn.json)
71+codegen: corefn
72+ @$(PSCPP) $(PSCPPFLAGS) $(COREFN_SRCS)
73+
74+$(PURESCRIPT_PKGS):
75+ @echo "Getting packages using" $(PSC_PACKAGE) "..."
76+ @$(PSC_PACKAGE) update
77+
78+SRCS := $(call rwildcard,$(CC_SRC)/,*.cpp)
79+SRCS += $(call rwildcard,$(FFI_SRC)/,*.cpp)
80+SRCS += $(call rwildcard,$(FFI_SRC)/,*.cc)
81+SRCS += $(call rwildcard,$(FFI_SRC)/,*.mm)
82+SRCS += $(call rwildcard,$(FFI_SRC)/,*.c)
83+SRCS += $(call rwildcard,$(FFI_SRC)/,*.m)
84+
85+OBJS1 = $(SRCS:.cpp=.o)
86+OBJS2 = $(OBJS1:.cc=.o)
87+OBJS3 = $(OBJS2:.mm=.o)
88+OBJS4 = $(OBJS3:.c=.o)
89+OBJS = $(OBJS4:.m=.o)
90+DEPS = $(OBJS:.o=.d)
91+
92+$(BIN): $(OBJS)
93+ @echo "Linking" $(BIN_DIR)/$(BIN)
94+ @mkdir -p $(BIN_DIR)
95+ @$(CXX) $^ -o $(BIN_DIR)/$@ $(LDFLAGS)
96+
97+-include $(DEPS)
98+
99+%.o: %.cpp
100+ @echo "Creating" $@ "(C++)"
101+ @$(CXX) $(CXXFLAGS) $(INCLUDES) -MMD -MP -c $< -o $@
102+
103+%.o: %.cc
104+ @echo "Creating" $@ "(C++)"
105+ @$(CXX) $(CXXFLAGS) $(INCLUDES) -MMD -MP -c $< -o $@
106+
107+%.o: %.mm
108+ @echo "Creating" $@ "(Objective-C++)"
109+ @$(CXX) $(CXXFLAGS) $(INCLUDES) -MMD -MP -c $< -o $@
110+
111+%.o: %.c
112+ @echo "Creating" $@ "(C)"
113+ @$(CXX) $(CFLAGS) $(INCLUDES) -MMD -MP -c $< -o $@
114+
115+%.o: %.m
116+ @echo "Creating" $@ "(Objective-C)"
117+ @$(CXX) $(CFLAGS) $(INCLUDES) -MMD -MP -c $< -o $@
118+
119+.PHONY: all
120+all: release
121+
122+.PHONY: clean
123+clean:
124+ @-rm -f $(OBJS) $(DEPS)
125+ @-rm -rf $(OUTPUT)
126+
127+.PHONY: run
128+run:
129+ @$(BIN_DIR)/$(BIN) $(ARGS)
support/ffi-loader.godeleted−30
@@ -1,30 +0,0 @@
1−package ffi_loader
2−
3−// Load PureScript standard library FFI packages. Comment out the ones you don't need for
4−// faster/smaller builds.
5−
6−import (
7− _ "github.com/purescript-native/go-ffi/purescript-arrays"
8− _ "github.com/purescript-native/go-ffi/purescript-assert"
9− _ "github.com/purescript-native/go-ffi/purescript-console"
10− _ "github.com/purescript-native/go-ffi/purescript-effect"
11− _ "github.com/purescript-native/go-ffi/purescript-enums"
12− _ "github.com/purescript-native/go-ffi/purescript-exceptions"
13− _ "github.com/purescript-native/go-ffi/purescript-foldable-traversable"
14− _ "github.com/purescript-native/go-ffi/purescript-functions"
15− _ "github.com/purescript-native/go-ffi/purescript-globals"
16− _ "github.com/purescript-native/go-ffi/purescript-integers"
17− _ "github.com/purescript-native/go-ffi/purescript-lazy"
18− _ "github.com/purescript-native/go-ffi/purescript-math"
19− _ "github.com/purescript-native/go-ffi/purescript-partial"
20− _ "github.com/purescript-native/go-ffi/purescript-prelude"
21− _ "github.com/purescript-native/go-ffi/purescript-random"
22− _ "github.com/purescript-native/go-ffi/purescript-refs"
23− _ "github.com/purescript-native/go-ffi/purescript-st"
24− _ "github.com/purescript-native/go-ffi/purescript-strings"
25− _ "github.com/purescript-native/go-ffi/purescript-unfoldable"
26−)
27−
28−// Add your own FFI packages here.
29−
30−import ()
support/go.mod.ffi-loaderdeleted−2
@@ -1,2 +0,0 @@
1−module $PROJECT/purescript-native/ffi-loader
2−
support/go.mod.outputdeleted−2
@@ -1,2 +0,0 @@
1−module $PROJECT/purescript-native/output
2−
support/go.mod.workingdeleted−1
@@ -1 +0,0 @@
1−module $PROJECT/example