File: instance_assume.rb

package info (click to toggle)
ruby-facets 2.9.2-1
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 9,824 kB
  • sloc: ruby: 25,483; xml: 90; makefile: 20
file content (66 lines) | stat: -rw-r--r-- 1,666 bytes parent folder | download
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
# Deprecated b/c it can be done via instance_assign(obj.to_data).

module Kernel

  # Assign instance vars using another object.
  #
  #   class O
  #     attr_accessor :d
  #     def initialize( a, b, c, d)
  #       @a = a
  #       @b = b
  #       @c = c
  #       @d = d
  #     end
  #   end
  #   o1 = O.new(1,2,3,4)
  #   o2 = O.new(0,0,0,0)
  #
  #   o2.instance_assume( o1, '@a', '@b', '@c', '@d' )
  #   o2.instance_eval{ @a }  #=> 1
  #   o2.instance_eval{ @b }  #=> 2
  #   o2.instance_eval{ @c }  #=> 3
  #   o2.instance_eval{ @d }  #=> 4
  #
  #   CREDIT: Trans

  def instance_assume(obj, *vars)
    if vars.empty?
      vars = obj.instance_variables  # instance_vars | obj.instance_variables
    else
      vars = vars.collect do |k|
        var.to_s.slice(0,1) == '@' ? var : "@#{var}"
      end
      vars = vars | instance_variables | obj.instance_variables
    end
    vars.each do |var|
      instance_variable_set(var, obj.instance_variable_get(var))
    end
    return self  # ???
  end

  # As with #instance_assume, but forces the setting of the object's
  # instance varaibles even if the reciever doesn't have them defined.
  #
  # See #instance_assume.

  def instance_assume!(obj, *vars)
    if vars.empty?
      vars = obj.instance_variables
    else
      vars = vars.collect do |k|
        var.to_s.slice(0,1) == '@' ? var : "@#{var}"
      end
      vars = vars | obj.instance_variables
    end
    vars.each do |var|
      instance_variable_set(var, obj.instance_variable_get(var))
    end
    return self  # ???
  end

  # Shadow method for instance_assume.
  alias :__assume__ :instance_assume if defined?(instance_assume)

end