parent
f274c589c0
commit
28d2d2ca0c
@ -1,68 +1,79 @@
|
|||||||
require "strscan"
|
require "strscan"
|
||||||
|
|
||||||
def parse(ss)
|
Packet = Struct.new(:version, :type_id, :content) do
|
||||||
version = ss.scan(/.{3}/).to_i(2)
|
include Enumerable
|
||||||
type_id = ss.scan(/.{3}/).to_i(2)
|
|
||||||
|
|
||||||
content = case type_id
|
def self.parse(ss)
|
||||||
when 4 # literal value
|
version = ss.scan(/.{3}/).to_i(2)
|
||||||
value = ""
|
type_id = ss.scan(/.{3}/).to_i(2)
|
||||||
while group = ss.scan(/1/)
|
|
||||||
value << ss.scan(/.{4}/)
|
|
||||||
end
|
|
||||||
ss.scan(/0/)
|
|
||||||
value << ss.scan(/.{4}/)
|
|
||||||
value.to_i(2)
|
|
||||||
else # operator
|
|
||||||
length_type_id = ss.scan(/./)
|
|
||||||
case length_type_id
|
|
||||||
when ?0
|
|
||||||
sub_packet_length = ss.scan(/.{15}/).to_i(2)
|
|
||||||
sub_packet = ss.scan(/.{#{sub_packet_length}}/)
|
|
||||||
|
|
||||||
sss = StringScanner.new(sub_packet)
|
case type_id
|
||||||
sub_packets = []
|
when 4 # literal value
|
||||||
until sss.eos?
|
value = ss.scan(/(1.{4})*(0.{4})/).chars
|
||||||
sub_packets << parse(sss)
|
.each_slice(5).flat_map { _1[1..4] }.join
|
||||||
end
|
.to_i(2)
|
||||||
sub_packets
|
Literal.new(version, type_id, value)
|
||||||
when ?1
|
else # operator
|
||||||
sub_packet_num = ss.scan(/.{11}/).to_i(2)
|
sub_packets = case
|
||||||
sub_packet_num.times.map { parse(ss) }
|
when ss.scan(/0(.{15})/)
|
||||||
else
|
length = ss.captures[0].to_i(2)
|
||||||
fail
|
sub_packet = ss.scan(/.{#{length}}/)
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
{ version: version, type_id: type_id, content: content }
|
sss = StringScanner.new(sub_packet)
|
||||||
|
sub_packets = []
|
||||||
|
until sss.eos?
|
||||||
|
sub_packets << parse(sss)
|
||||||
|
end
|
||||||
|
sub_packets
|
||||||
|
when ss.scan(/1(.{11})/)
|
||||||
|
n = ss.captures[0].to_i(2)
|
||||||
|
n.times.map { parse(ss) }
|
||||||
|
else
|
||||||
|
fail
|
||||||
|
end
|
||||||
|
Operator.new(version, type_id, sub_packets)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def each
|
||||||
|
return enum_for(__method__) unless block_given?
|
||||||
|
|
||||||
|
yield self
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
message = ARGF.read.chomp.chars.map { _1.to_i(16).to_s(2).rjust(4, ?0) }.join
|
class Literal < Packet
|
||||||
ss = StringScanner.new(message)
|
def value = content
|
||||||
packet = parse(ss)
|
end
|
||||||
|
|
||||||
def each(packet, &block)
|
class Operator < Packet
|
||||||
return enum_for(__method__, packet) unless block_given?
|
def each(&block)
|
||||||
|
super
|
||||||
|
|
||||||
block.(packet)
|
|
||||||
if (content = packet.fetch(:content)).is_a?(Array)
|
|
||||||
content.each do |sub_packet|
|
content.each do |sub_packet|
|
||||||
each(sub_packet, &block)
|
sub_packet.each(&block)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
|
||||||
|
|
||||||
# p each(packet).sum { _1.fetch(:version) }
|
def value
|
||||||
|
values = content.map(&:value)
|
||||||
|
case type_id
|
||||||
|
when 0 then values.sum
|
||||||
|
when 1 then values.inject(:*)
|
||||||
|
when 2 then values.min
|
||||||
|
when 3 then values.max
|
||||||
|
when 4 then fail
|
||||||
|
when 5 then values.inject(:>) ? 1 : 0
|
||||||
|
when 6 then values.inject(:<) ? 1 : 0
|
||||||
|
when 7 then values.inject(:==) ? 1 : 0
|
||||||
|
else fail
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
value = {
|
message = ARGF.read.chomp.chars.map { _1.to_i(16).to_s(2).rjust(4, ?0) }.join
|
||||||
0 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.sum },
|
ss = StringScanner.new(message)
|
||||||
1 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.inject(:*) },
|
packet = Packet.parse(ss)
|
||||||
2 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.min },
|
|
||||||
3 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.max },
|
|
||||||
4 => ->(c) { c },
|
|
||||||
5 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.inject(:>) ? 1 : 0 },
|
|
||||||
6 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.inject(:<) ? 1 : 0 },
|
|
||||||
7 => ->(c) { c.map { value.fetch(_1.fetch(:type_id)).(_1.fetch(:content)) }.inject(:==) ? 1 : 0 },
|
|
||||||
}
|
|
||||||
|
|
||||||
p value.fetch(packet.fetch(:type_id)).(packet.fetch(:content))
|
p packet.sum(&:version)
|
||||||
|
p packet.value
|
||||||
|
Loading…
Reference in new issue