Shakefile.hs

   1 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
   2 {-# OPTIONS_GHC -fno-warn-missing-signatures #-}
   3 import Data.List.Extra
   4 import System.Exit
   5 import Control.Monad
   6 import Control.Concurrent.MVar
   7 import Development.Shake
   8 import Development.Shake.Util
   9 import Development.Shake.Classes
  10 import Development.Shake.FilePath
  11 import System.Environment (lookupEnv)
  12
  13 newtype OcamlOrdOracle = OcamlOrdOracle String
  14                        deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  15 newtype OcamlOrdOracleN = OcamlOrdOracleN String
  16                         deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  17 newtype OcamlCmdLineOracle = OcamlCmdLineOracle String
  18                            deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  19 newtype OcamlCmdLineOracleN = OcamlCmdLineOracleN String
  20                             deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  21 newtype CCmdLineOracle = CCmdLineOracle String
  22                        deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  23 newtype GitDescribeOracle = GitDescribeOracle ()
  24                           deriving (Show,Typeable,Eq,Hashable,Binary,NFData)
  25
  26 data Bt = Native | Bytecode
  27
  28 outdir = "build"
  29 mudir = "mupdf"
  30 inOutDir s = outdir </> s
  31 egl = False
  32
  33 ocamlc = "ocamlc.opt"
  34 ocamlopt = "ocamlopt.opt"
  35 ocamldep = "ocamldep.opt"
  36 ocamlflags = "-warn-error +a -w +a -g -safe-string -strict-sequence"
  37 ocamlflagstbl = [("main", ("-I lablGL", "sed -f pp.sed", ["pp.sed"]))
  38                 ,("config", ("-I lablGL", "", []))
  39                 ]
  40 cflags = "-Wall -Werror -D_GNU_SOURCE -O\
  41          \ -g -std=c99 -pedantic-errors\
  42          \ -Wunused-parameter -Wsign-compare -Wshadow"
  43          ++ (if egl then " -DUSE_EGL" else "")
  44 cflagstbl =
  45   [("link.o"
  46    ,"-I " ++ mudir ++ "/include -I "
  47     ++ mudir ++ "/thirdparty/freetype/include -Wextra")
  48   ]
  49 cclib ty =
  50   "-lGL -lX11 -lmupdf -lmupdfthird -lpthread -L" ++ mudir </> "build" </> ty
  51   ++ " -lcrypto" ++ (if egl then " -lEGL" else "")
  52 cclibNative = cclib "native"
  53 cclibRelease = cclib "release"
  54
  55 getincludes :: [String] -> [String]
  56 getincludes [] = []
  57 getincludes ("-I":arg:tl) = arg : getincludes tl
  58 getincludes (_:tl) = getincludes tl
  59
  60 isabsinc :: String -> Bool
  61 isabsinc [] = False
  62 isabsinc (hd:_) = hd == '+' || hd == '/'
  63
  64 fixincludes [] = []
  65 fixincludes ("-I":d:tl)
  66   | isabsinc d = "-I":d:fixincludes tl
  67   | otherwise = "-I":inOutDir d:fixincludes tl
  68 fixincludes (e:tl) = e:fixincludes tl
  69
  70 ocamlKey comp tbl key
  71   | "lablGL/" `isPrefixOf` key =
  72     (comp, ocamlflags ++ " -w -44 -I lablGL", [], [])
  73   | otherwise = case lookup (dropExtension key) tbl of
  74     Nothing -> (comp, ocamlflags, [], [])
  75     Just (f, [], deps) -> (comp, ocamlflags ++ " " ++ f, [], deps)
  76     Just (f, pp, deps) -> (comp, ocamlflags ++ " " ++ f, ["-pp", pp], deps)
  77
  78 cKey1 key | "lablGL/" `isPrefixOf` key = "-Wno-pointer-sign -O2"
  79           | otherwise = case lookup key cflagstbl of
  80             Nothing -> cflags
  81             Just f -> f ++ " " ++ cflags
  82
  83 cKey Nothing key = cKey1 key
  84 cKey (Just flags) key = flags ++ " " ++ cKey1 key
  85
  86 fixppfile s ("File":_:tl) = ("File \"" ++ s ++ "\","):tl
  87 fixppfile _ l = l
  88
  89 fixpp :: String -> String -> String
  90 fixpp r s = unlines [unwords $ fixppfile r $ words x | x <- lines s]
  91
  92 ppppe ExitSuccess _ _ = return ()
  93 ppppe _ src emsg = error $ fixpp src emsg
  94
  95 needsrc key suff = do
  96   let src' = key -<.> suff
  97   let src = if src' == "help.ml" then inOutDir src' else src'
  98   need [src]
  99   return src
 100
 101 depscaml flags ppflags src = do
 102   (Stdout stdout, Stderr emsg, Exit ex) <-
 103         cmd ocamldep "-one-line" incs "-I" outdir ppflags src
 104   ppppe ex src emsg
 105   return stdout
 106   where flagl = words flags
 107         incs = unwords ["-I " ++ d | d <- getincludes flagl, not $ isabsinc d]
 108
 109 compilecaml comp flagl ppflags out src = do
 110   let fixedflags = fixincludes flagl
 111   (Stderr emsg, Exit ex) <-
 112     cmd comp "-c -I" outdir fixedflags "-o" out ppflags src
 113   ppppe ex src emsg
 114   return ()
 115
 116 deplistE reqs =
 117   [if takeDirectory1 n == outdir then n else inOutDir n | n <- reqs]
 118 deplist Native (_ : (_, reqs) : _) = deplistE reqs
 119 deplist Bytecode ((_, reqs) : _) = deplistE reqs
 120 deplist _ _ = []
 121
 122 cmio target suffix oracle ordoracle = do
 123   target %> \out -> do
 124     let key = dropDirectory1 out
 125     src <- needsrc key suffix
 126     (comp, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracle key
 127     let flagl = words flags
 128     let dep = out ++ "_dep"
 129     need $ dep : deps'
 130     ddep <- liftIO $ readFile dep
 131     let deps = deplist Bytecode $ parseMakefile ddep
 132     need deps
 133     compilecaml comp flagl ppflags out src
 134   target ++ "_dep" %> \out -> do
 135     let ord = dropEnd 4 out
 136     let key = dropDirectory1 ord
 137     src <- needsrc key suffix
 138     (_, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracle key
 139     mkfiledeps <- depscaml flags ppflags src
 140     writeFileChanged out mkfiledeps
 141     let depo = deps ++ [dep -<.> ".cmo" | dep <- deps, fit dep]
 142           where
 143             deps = deplist Bytecode $ parseMakefile mkfiledeps
 144             fit dep = ext == ".cmi" && base /= baseout
 145               where (base, ext) = splitExtension dep
 146                     baseout = dropExtension out
 147     need (map (++ "_dep") depo ++ deps')
 148     unit $ ordoracle $ OcamlOrdOracle ord
 149
 150 cmx oracle ordoracle =
 151   "//*.cmx" %> \out -> do
 152     let key = dropDirectory1 out
 153     src <- needsrc key ".ml"
 154     (comp, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracleN key
 155     let flagl = words flags
 156     mkfiledeps <- depscaml flags ppflags src
 157     need (deplist Native (parseMakefile mkfiledeps) ++ deps')
 158     unit $ ordoracle $ OcamlOrdOracleN out
 159     compilecaml comp flagl ppflags out src
 160
 161 binInOutDir globjs depln target =
 162   inOutDir target %> \out ->
 163   do
 164     let mulibs = [mudir </> "build" </> "native" </> "libmupdf.a"
 165                  ,mudir </> "build" </> "native" </> "libmupdfthird.a"]
 166     need (mulibs ++ globjs ++ map inOutDir ["link.o", "main.cmx", "help.cmx"])
 167     cmxs <- liftIO $ readMVar depln
 168     need cmxs
 169     unit $ cmd ocamlopt "-g -I lablGL -o" out
 170       "unix.cmxa str.cmxa" (reverse cmxs)
 171       (inOutDir "link.o") "-cclib" (cclibRelease : globjs)
 172
 173 main = do
 174   depl <- newMVar ([] :: [String])
 175   depln <- newMVar ([] :: [String])
 176   envcflags <- lookupEnv "CFLAGS"
 177   shakeArgs shakeOptions { shakeFiles = outdir
 178                          , shakeVerbosity = Normal
 179                          , shakeChange = ChangeModtimeAndDigest } $ do
 180   want [inOutDir "llpp"]
 181
 182   gitDescribeOracle <- addOracle $ \(GitDescribeOracle ()) -> do
 183     Stdout out <- cmd "git describe --tags --dirty"
 184     return (out :: String)
 185
 186   ocamlOracle <- addOracle $ \(OcamlCmdLineOracle s) ->
 187     return $ ocamlKey ocamlc ocamlflagstbl s
 188
 189   ocamlOracleN <- addOracle $ \(OcamlCmdLineOracleN s) ->
 190     return $ ocamlKey ocamlopt ocamlflagstbl s
 191
 192   ocamlOrdOracle <- addOracle $ \(OcamlOrdOracle s) ->
 193     unless (takeExtension s == ".cmi") $
 194       liftIO $ modifyMVar_ depl $ \l -> return $ s:l
 195
 196   ocamlOrdOracleN <- addOracle $ \(OcamlOrdOracleN s) ->
 197     unless (takeExtension s == ".cmi") $
 198       liftIO $ modifyMVar_ depln $ \l -> return $ s:l
 199
 200   cOracle <- addOracle $ \(CCmdLineOracle s) -> return $ cKey envcflags s
 201
 202   inOutDir "help.ml" %> \out -> do
 203     version <- gitDescribeOracle $ GitDescribeOracle ()
 204     need ["mkhelp.sh", "KEYS"]
 205     Stdout f <- cmd "/bin/sh mkhelp.sh KEYS" version
 206     writeFileChanged out f
 207
 208   "//*.o" %> \out -> do
 209     let key = dropDirectory1 out
 210     flags <- cOracle $ CCmdLineOracle key
 211     let src = key -<.> ".c"
 212     let dep = out -<.> ".d"
 213     unit $ cmd ocamlc "-ccopt"
 214       [flags ++ " -MMD -MF " ++ dep ++ " -o " ++ out] "-c" src
 215     needMakefileDependencies dep
 216
 217   let globjs = map (inOutDir . (++) "lablGL/ml_") ["gl.o", "glarray.o", "raw.o"]
 218
 219   let mulib ty name = do
 220         -- perhaps alwaysrerun is in order here?
 221         mudir </> "build" </> ty </> name %> \_ -> do
 222           unit $ cmd (Cwd "mupdf") ("make build=" ++ ty) "libs"
 223
 224   mulib "release" "libmupdf.a"
 225   mulib "release" "libmupdfthird.a"
 226   mulib "native" "libmupdf.a"
 227   mulib "native" "libmupdfthird.a"
 228
 229   inOutDir "llpp" %> \out -> do
 230     let mulibs = [mudir </> "build" </> "native" </> "libmupdf.a"
 231                  ,mudir </> "build" </> "native" </> "libmupdfthird.a"]
 232     need (mulibs ++ globjs ++ map inOutDir ["link.o", "main.cmo", "help.cmo"])
 233     cmos <- liftIO $ readMVar depl
 234     need cmos
 235     unit $ cmd ocamlc "-g -custom -I lablGL -o" out
 236       "unix.cma str.cma" (reverse cmos)
 237       (inOutDir "link.o") "-cclib" (cclibNative : globjs)
 238
 239   binInOutDir globjs depln "llpp.native"
 240   binInOutDir globjs depln "llpp.murel.native"
 241
 242   cmio "//*.cmi" ".mli" ocamlOracle ocamlOrdOracle
 243   cmio "//*.cmo" ".ml" ocamlOracle ocamlOrdOracle
 244   cmx ocamlOracleN ocamlOrdOracleN