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 outdir = "build"
  27 mudir = "mupdf"
  28 inOutDir s = outdir </> s
  29 egl = False
  30
  31 ocamlc = "ocamlc.opt"
  32 ocamlopt = "ocamlopt.opt"
  33 ocamldep = "ocamldep.opt"
  34 ocamlflags = "-warn-error +a -w +a -g -safe-string -strict-sequence"
  35 ocamlflagstbl = [("main", ("-I lablGL", "sed -f pp.sed", ["pp.sed"]))
  36                 ,("config", ("-I lablGL", "", []))
  37                 ]
  38 cflags = "-Wall -Werror -D_GNU_SOURCE -O\
  39          \ -g -std=c99 -pedantic-errors\
  40          \ -Wunused-parameter -Wsign-compare -Wshadow\
  41          \ -DVISAVIS -DCSS_HACK_TO_READ_EPUBS_COMFORTABLY"
  42          ++ (if egl then " -DUSE_EGL" else "")
  43 cflagstbl =
  44   [("link.o"
  45    ,"-I " ++ mudir ++ "/include -I "
  46     ++ mudir ++ "/thirdparty/freetype/include -Wextra")
  47   ]
  48 cclib ty =
  49   "-lGL -lX11 -lmupdf -lmupdfthird -lpthread -L" ++ mudir </> "build" </> ty
  50   ++ " -lcrypto" ++ (if egl then " -lEGL" else "")
  51 cclib_native = cclib "native"
  52 cclib_release = cclib "release"
  53
  54 getincludes :: [String] -> [String]
  55 getincludes [] = []
  56 getincludes ("-I":arg:tl) = arg : getincludes tl
  57 getincludes (_:tl) = getincludes tl
  58
  59 isabsinc :: String -> Bool
  60 isabsinc [] = False
  61 isabsinc (hd:_) = hd == '+' || hd == '/'
  62
  63 fixincludes [] = []
  64 fixincludes ("-I":d:tl)
  65   | isabsinc d = "-I":d:fixincludes tl
  66   | otherwise = "-I":inOutDir d:fixincludes tl
  67 fixincludes (e:tl) = e:fixincludes tl
  68
  69 ocamlKey comp tbl key
  70   | "lablGL/" `isPrefixOf` key =
  71     (comp, ocamlflags ++ " -w -44 -I lablGL", [], [])
  72   | otherwise = case lookup (dropExtension key) tbl of
  73     Nothing -> (comp, ocamlflags, [], [])
  74     Just (f, [], deps) -> (comp, ocamlflags ++ " " ++ f, [], deps)
  75     Just (f, pp, deps) -> (comp, ocamlflags ++ " " ++ f, ["-pp", pp], deps)
  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 fixppfile s ("File":_:tl) = ("File \"" ++ s ++ "\","):tl
  86 fixppfile _ l = l
  87
  88 fixpp :: String -> String -> String
  89 fixpp r s = unlines [unwords $ fixppfile r $ words x | x <- lines s]
  90
  91 ppppe ExitSuccess _ _ = return ()
  92 ppppe _ src emsg = error $ fixpp src emsg
  93
  94 needsrc key suff = do
  95   let src' = key -<.> suff
  96   let src = if src' == "help.ml" then inOutDir src' else src'
  97   need [src]
  98   return src
  99
 100 depscaml flags ppflags src = do
 101   (Stdout stdout, Stderr emsg, Exit ex) <-
 102         cmd ocamldep "-one-line" incs "-I" outdir ppflags src
 103   ppppe ex src emsg
 104   return stdout
 105   where flagl = words flags
 106         incs = unwords ["-I " ++ d | d <- getincludes flagl, not $ isabsinc d]
 107
 108 compilecaml comp flagl ppflags out src = do
 109   let fixedflags = fixincludes flagl
 110   (Stderr emsg, Exit ex) <-
 111     cmd comp "-c -I" outdir fixedflags "-o" out ppflags src
 112   ppppe ex src emsg
 113   return ()
 114
 115 cmio target suffix oracle ordoracle = do
 116   target %> \out -> do
 117     let key = dropDirectory1 out
 118     src <- needsrc key suffix
 119     (comp, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracle key
 120     let flagl = words flags
 121     let dep = out ++ "_dep"
 122     need $ dep : deps'
 123     ddep <- liftIO $ readFile dep
 124     let deps = deplist $ parseMakefile ddep
 125     need deps
 126     compilecaml comp flagl ppflags out src
 127   target ++ "_dep" %> \out -> do
 128     let ord = dropEnd 4 out
 129     let key = dropDirectory1 ord
 130     src <- needsrc key suffix
 131     (_, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracle key
 132     mkfiledeps <- depscaml flags ppflags src
 133     writeFileChanged out mkfiledeps
 134     let depo = deps ++ [dep -<.> ".cmo" | dep <- deps, fit dep]
 135           where
 136             deps = deplist $ parseMakefile mkfiledeps
 137             fit dep = ext == ".cmi" && base /= baseout
 138               where (base, ext) = splitExtension dep
 139                     baseout = dropExtension out
 140     need ((map (++ "_dep") depo) ++ deps')
 141     unit $ ordoracle $ OcamlOrdOracle ord
 142   where
 143     deplist [] = []
 144     deplist ((_, reqs) : _) =
 145       [if takeDirectory1 n == outdir then n else inOutDir n | n <- reqs]
 146
 147 cmx oracle ordoracle =
 148   "//*.cmx" %> \out -> do
 149     let key = dropDirectory1 out
 150     src <- needsrc key ".ml"
 151     (comp, flags, ppflags, deps') <- oracle $ OcamlCmdLineOracleN key
 152     let flagl = words flags
 153     mkfiledeps <- depscaml flags ppflags src
 154     need ((deplist $ parseMakefile mkfiledeps) ++ deps')
 155     unit $ ordoracle $ OcamlOrdOracleN out
 156     compilecaml comp flagl ppflags out src
 157   where
 158     deplist (_ : (_, reqs) : _) =
 159       [if takeDirectory1 n == outdir then n else inOutDir n | n <- reqs]
 160     deplist _ = []
 161
 162 binInOutDir globjs depln target =
 163   inOutDir target %> \out ->
 164   do
 165     need (globjs ++ map inOutDir ["link.o", "main.cmx", "help.cmx"])
 166     cmxs <- liftIO $ readMVar depln
 167     need cmxs
 168     unit $ cmd ocamlopt "-g -I lablGL -o" out
 169       "unix.cmxa str.cmxa" (reverse cmxs)
 170       (inOutDir "link.o") "-cclib" (cclib_release : globjs)
 171
 172 main = do
 173   depl <- newMVar ([] :: [String])
 174   depln <- newMVar ([] :: [String])
 175   envcflags <- lookupEnv "CFLAGS"
 176   shakeArgs shakeOptions { shakeFiles = outdir
 177                          , shakeVerbosity = Normal
 178                          , shakeChange = ChangeModtimeAndDigest } $ do
 179   want [inOutDir "llpp"]
 180
 181   gitDescribeOracle <- addOracle $ \(GitDescribeOracle ()) -> do
 182     Stdout out <- cmd "git describe --tags --dirty"
 183     return (out :: String)
 184
 185   ocamlOracle <- addOracle $ \(OcamlCmdLineOracle s) ->
 186     return $ ocamlKey ocamlc ocamlflagstbl s
 187
 188   ocamlOracleN <- addOracle $ \(OcamlCmdLineOracleN s) ->
 189     return $ ocamlKey ocamlopt ocamlflagstbl s
 190
 191   ocamlOrdOracle <- addOracle $ \(OcamlOrdOracle s) ->
 192     unless (takeExtension s == ".cmi") $
 193       liftIO $ modifyMVar_ depl $ \l -> return $ s:l
 194
 195   ocamlOrdOracleN <- addOracle $ \(OcamlOrdOracleN s) ->
 196     unless (takeExtension s == ".cmi") $
 197       liftIO $ modifyMVar_ depln $ \l -> return $ s:l
 198
 199   cOracle <- addOracle $ \(CCmdLineOracle s) -> return $ cKey envcflags s
 200
 201   inOutDir "help.ml" %> \out -> do
 202     version <- gitDescribeOracle $ GitDescribeOracle ()
 203     need ["mkhelp.sh", "KEYS"]
 204     Stdout f <- cmd "/bin/sh mkhelp.sh KEYS" version
 205     writeFileChanged out f
 206
 207   "//*.o" %> \out -> do
 208     let key = dropDirectory1 out
 209     flags <- cOracle $ CCmdLineOracle key
 210     let src = key -<.> ".c"
 211     let dep = out -<.> ".d"
 212     unit $ cmd ocamlc "-ccopt"
 213       [flags ++ " -MMD -MF " ++ dep ++ " -o " ++ out] "-c" src
 214     needMakefileDependencies dep
 215
 216   let globjs = map (inOutDir . (++) "lablGL/ml_") ["gl.o", "glarray.o", "raw.o"]
 217   inOutDir "llpp" %> \out -> do
 218     need (globjs ++ map inOutDir ["link.o", "main.cmo", "help.cmo"])
 219     cmos <- liftIO $ readMVar depl
 220     need cmos
 221     unit $ cmd ocamlc "-g -custom -I lablGL -o" out
 222       "unix.cma str.cma" (reverse cmos)
 223       (inOutDir "link.o") "-cclib" (cclib_native : globjs)
 224
 225   binInOutDir globjs depln "llpp.native"
 226   binInOutDir globjs depln "llpp.murel.native"
 227
 228   cmio "//*.cmi" ".mli" ocamlOracle ocamlOrdOracle
 229   cmio "//*.cmo" ".ml" ocamlOracle ocamlOrdOracle
 230   cmx ocamlOracleN ocamlOrdOracleN