Class: BERT::Encode

Inherits:
Object
  • Object
show all
Includes:
Types
Defined in:
lib/bert/encode.rb

Constant Summary

Constants included from Types

Types::ATOM, Types::BIN, Types::FLOAT, Types::FUN, Types::INT, Types::LARGE_BIGNUM, Types::LARGE_TUPLE, Types::LIST, Types::MAGIC, Types::MAX_INT, Types::MIN_INT, Types::NEW_FUN, Types::NIL, Types::SMALL_BIGNUM, Types::SMALL_INT, Types::SMALL_TUPLE, Types::STRING

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(out) ⇒ Encode

Returns a new instance of Encode.



7
8
9
 
# File 'lib/bert/encode.rb', line 7

def initialize(out)
  self.out = out
end

Instance Attribute Details

#out ⇒ Object

Returns the value of attribute out.



5
6
7
 
# File 'lib/bert/encode.rb', line 5

def out
  @out
end

Class Method Details

.encode(data) ⇒ Object 



11
12
13
14
15
16
17
18
19
 
# File 'lib/bert/encode.rb', line 11

def self.encode(data)
  io = StringIO.new
  self.new(io).write_any(data)
  if '1.9'.respond_to?(:force_encoding)
    io.string.force_encoding("ASCII-8BIT")
  else
    io.string
  end
end

Instance Method Details

#write_1(byte) ⇒ Object 



39
40
41
 
# File 'lib/bert/encode.rb', line 39

def write_1(byte)
  out.write([byte].pack("C"))
end

#write_2(short) ⇒ Object 



43
44
45
 
# File 'lib/bert/encode.rb', line 43

def write_2(short)
  out.write([short].pack("n"))
end

#write_4(long) ⇒ Object 



47
48
49
 
# File 'lib/bert/encode.rb', line 47

def write_4(long)
  out.write([long].pack("N"))
end

#write_any(obj) ⇒ Object 



21
22
23
24
 
# File 'lib/bert/encode.rb', line 21

def write_any obj
  write_1 MAGIC
  write_any_raw obj
end

#write_any_raw(obj) ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/bert/encode.rb', line 26

def write_any_raw obj
  case obj
    when Symbol then write_symbol(obj)
    when Fixnum, Bignum then write_fixnum(obj)
    when Float then write_float(obj)
    when Tuple then write_tuple(obj)
    when Array then write_list(obj)
    when String then write_binary(obj)
    else
      fail(obj)
  end
end

#write_bignum(num) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/bert/encode.rb', line 84

def write_bignum(num)
  n = (num.to_s(2).size / 8.0).ceil
  if n < 256
    write_1 SMALL_BIGNUM
    write_1 n
    write_bignum_guts(num)
  else
    write_1 LARGE_BIGNUM
    write_4 n
    write_bignum_guts(num)
  end
end

#write_bignum_guts(num) ⇒ Object 



97
98
99
100
101
102
103
104
105
 
# File 'lib/bert/encode.rb', line 97

def write_bignum_guts(num)
  write_1 (num >= 0 ? 0 : 1)
  num = num.abs
  while num != 0
    rem = num % 256
    write_1 rem
    num = num >> 8
  end
end

#write_binary(data) ⇒ Object 



130
131
132
133
134
 
# File 'lib/bert/encode.rb', line 130

def write_binary(data)
  write_1 BIN
  write_4 data.length
  write_string data
end

#write_boolean(bool) ⇒ Object 



55
56
57
 
# File 'lib/bert/encode.rb', line 55

def write_boolean(bool)
  write_symbol(bool.to_s.to_sym)
end

#write_fixnum(num) ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/bert/encode.rb', line 67

def write_fixnum(num)
  if num >= 0 && num < 256
    write_1 SMALL_INT
    write_1 num
  elsif num <= MAX_INT && num >= MIN_INT
    write_1 INT
    write_4 num
  else
    write_bignum num
  end
end

#write_float(float) ⇒ Object 



79
80
81
82
 
# File 'lib/bert/encode.rb', line 79

def write_float(float)
  write_1 FLOAT
  write_string format("%15.15e", float).ljust(31, "\000")
end

#write_list(data) ⇒ Object 



121
122
123
124
125
126
127
128
 
# File 'lib/bert/encode.rb', line 121

def write_list(data)
  fail(data) unless data.is_a? Array
  write_1 NIL and return if data.empty?
  write_1 LIST
  write_4 data.length
  data.each{|e| write_any_raw e }
  write_1 NIL
end

#write_string(string) ⇒ Object 



51
52
53
 
# File 'lib/bert/encode.rb', line 51

def write_string(string)
  out.write(string)
end

#write_symbol(sym) ⇒ Object 



59
60
61
62
63
64
65
 
# File 'lib/bert/encode.rb', line 59

def write_symbol(sym)
  fail(sym) unless sym.is_a?(Symbol)
  data = sym.to_s
  write_1 ATOM
  write_2 data.length
  write_string data
end

#write_tuple(data) ⇒ Object 



107
108
109
110
111
112
113
114
115
116
117
118
119
 
# File 'lib/bert/encode.rb', line 107

def write_tuple(data)
  fail(data) unless data.is_a? Array

  if data.length < 256
    write_1 SMALL_TUPLE
    write_1 data.length
  else
    write_1 LARGE_TUPLE
    write_4 data.length
  end

  data.each { |e| write_any_raw e }
end