{- Copyright (C) 2010 Dr. Alistair Ward This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <http://www.gnu.org/licenses/>. -} {- | [@AUTHOR@] Dr. Alistair Ward [@DESCRIPTION@] Determines the CPU-time, required to evaluate the specified pure expression. -} module ToolShed.System.TimePure( -- * Functions getCPUSeconds, printCPUSeconds ) where import qualified Control.DeepSeq import qualified System.CPUTime import qualified System.IO {- | * Time the specified pure expression, returning the required number of CPU-seconds and the result, as a 'Pair'. * CAVEAT: as a side-effect, the expression is /deep/ evaluated. -} getCPUSeconds :: (Fractional seconds, Control.DeepSeq.NFData expression) => expression -- ^ Arbitrary polymorphic expression. -> IO (seconds, expression) -- ^ The original expression, tagged with the CPU-seconds taken. getCPUSeconds expression = do start <- System.CPUTime.getCPUTime end <- expression `Control.DeepSeq.deepseq` System.CPUTime.getCPUTime return {-to IO-monad-} (fromInteger (end - start) / 1e12 {-convert from pico-seconds-}, expression) {- | * Print the time required by the specified pure expression. * CAVEAT: as a side-effect, the expression is /deep/ evaluated. -} printCPUSeconds :: Control.DeepSeq.NFData expression => expression -> IO expression printCPUSeconds expression = do (cpuSeconds, result) <- getCPUSeconds expression System.IO.hPutStrLn System.IO.stderr $ "CPU-seconds:\t" ++ show (cpuSeconds :: Double) return {-to IO-monad-} result