-- |

module Main where

import System.Posix.Process

import System.Environment

import System.Exit

import System.FilePath

import System.Process

import System.IO (hPutStrLn, stderr)

import Control.Monad

main :: IO ()

main = getArgs >>= command

command :: [String] -> IO ()

command [helpString] | helpString `elem` ["-h", "--help", "help"] =

putStrLn usage

command ["help", "help"] =

putStrLn "Helps."

command ["help", "record"] = do

let args = ["+GhcEventsPerf", "-h"]

executeFile "ghc-events-perf-record" True args Nothing

command ["help", "convert"] =

putStr $ unlines $

[ "Usage: ghc-events-perf convert program-name [out-file perf-file eventlog-file]"

, " program-name the name of the profiled Haskell program"

, " out-file where to save the resulting compound eventlog"

, " perf-file path to the perf data file for the Haskell program"

, " eventlog-file path to the GHC RTS eventlog of the Haskell program"

]

command ("record" : args) = do

myPath <- getExecutablePath

let commonPath = fst $ splitFileName myPath

-- We assume ghc-events-perf-record is in the same directory

-- as ghc-events-perf. Needed for 'sudo', because root needn't

-- have user's ~/.cabal/bin in his PATH.

recordPath = combine commonPath "ghc-events-perf-record"

recordArgs = "--RTS" : args

executeFile recordPath True recordArgs Nothing

command ["convert", program_name, out_file, perf_file, eventlog_file] = do

let sync_eventlog_file = program_name ++ ".perf.eventlog"

syncArgs = [ program_name

, perf_file

, sync_eventlog_file

]

putStrLn "Translating and synchronizing..."

syncHandle <- runProcess "ghc-events-perf-sync" syncArgs

Nothing Nothing Nothing Nothing Nothing

void $ waitForProcess syncHandle

let mergeArgs = [ "merge"

, out_file

, eventlog_file

, sync_eventlog_file

]

putStrLn "Merging with the standard eventlog..."

executeFile "ghc-events" True mergeArgs Nothing

command ["convert", program_name] = do

let out_file = program_name ++ ".total.eventlog"

perf_file = "data.perf" -- the same as defaultPerfOutputFile in record

eventlog_file = program_name ++ ".eventlog"

command ["convert", program_name, out_file, perf_file, eventlog_file]

command _ = putStrLn usage >> Main.die "Unrecognized command"

usage :: String

usage = "Usage: ghc-events-perf command command-args\n\nThe available commands are: record convert help.\n\nSee 'ghc-events-perf help command' for more information on a specific command."

die :: String -> IO a

die s = hPutStrLn stderr s >> exitWith (ExitFailure 1)