Merge pull request #371 from paultag/paultag/feature/profile
Add profile macro.
This commit is contained in:
commit
f6887ed7c2
43
hy/contrib/profile.hy
Normal file
43
hy/contrib/profile.hy
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
;;; Hy profiling macros
|
||||||
|
;;
|
||||||
|
;; Copyright (c) 2013 Paul R. Tagliamonte <tag@pault.ag>
|
||||||
|
;;
|
||||||
|
;; Permission is hereby granted, free of charge, to any person obtaining a
|
||||||
|
;; copy of this software and associated documentation files (the "Software"),
|
||||||
|
;; to deal in the Software without restriction, including without limitation
|
||||||
|
;; the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||||
|
;; and/or sell copies of the Software, and to permit persons to whom the
|
||||||
|
;; Software is furnished to do so, subject to the following conditions:
|
||||||
|
;;
|
||||||
|
;; The above copyright notice and this permission notice shall be included in
|
||||||
|
;; all copies or substantial portions of the Software.
|
||||||
|
;;
|
||||||
|
;; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
;; IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
;; FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||||
|
;; THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
;; LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||||
|
;; FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||||
|
;; DEALINGS IN THE SOFTWARE.
|
||||||
|
;;
|
||||||
|
;;; These macros make debugging where bottlenecks exist easier.
|
||||||
|
|
||||||
|
|
||||||
|
(defmacro/g! profile [&rest body]
|
||||||
|
" Profile a bit of code "
|
||||||
|
`(do
|
||||||
|
(import cProfile pstats)
|
||||||
|
|
||||||
|
(if-python2
|
||||||
|
(import [StringIO [StringIO]])
|
||||||
|
(import [io [StringIO]]))
|
||||||
|
|
||||||
|
(setv ~g!hy-pr (.Profile cProfile))
|
||||||
|
(.enable ~g!hy-pr)
|
||||||
|
(do ~@body)
|
||||||
|
(.disable ~g!hy-pr)
|
||||||
|
(setv ~g!hy-s (StringIO))
|
||||||
|
(setv ~g!hy-ps
|
||||||
|
(.sort-stats (kwapply (.Stats pstats ~g!hy-pr) {"stream" ~g!hy-s})))
|
||||||
|
(.print-stats ~g!hy-ps)
|
||||||
|
(print (.getvalue ~g!hy-s))))
|
Loading…
x
Reference in New Issue
Block a user