Shakefile.hs

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