summaryrefslogtreecommitdiff
path: root/Examples/test-suite/ruby/swig_gc.rb
blob: 30f6dfa42b518a769aab4f1e8037043d3eb5d48c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/usr/bin/env ruby
#
#
# VERY nice function from Robert Klemme to check memory leaks
# and check on what GC has collected since last call.
#
# Usage can be:
#
#     require 'swig_gc'
#
#     GC.stats
#     # do some stuff..
#     GC.start  # collect and report stats
#     # do some more...
#     GC.stats  # just report stats
#
# or:
#
#     require 'swig_gc'
#
#     GC.track_class = String  # track just String classes
#     GC.stats
#     # do some stuff..
#     GC.start  # collect and report stats
#     # do some more...
#     GC.stats  # just report stats
#
# 
# 
# 
#

module GC

  class << self
      
    attr          :last_stat
    attr_accessor :track_class

    alias :_start :start
    
    def start
      _start
      stats if $VERBOSE
    end
    
    def stats
      stats = Hash.new(0)
      ObjectSpace.each_object {|o| stats[o.class] += 1}
      
      if track_class
        v = stats[track_class]
        printf "\t%-30s  %10d", track_class.to_s, v
        if last_stat
          printf " | delta %10d", (v - last_stat[track_class])
        end
        puts
      else
        stats.sort {|(k1,v1),(k2,v2)| v2 <=> v1}.each do |k,v|
          printf "\t%-30s  %10d", k, v
          printf " | delta %10d", (v - last_stat[k]) if last_stat
          puts
        end
      end
        
      last_stat = stats
    end
  end
  
end