Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created October 28, 2017 01:33
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vznvzn/126383eab0094130b9fecc96004fa506 to your computer and use it in GitHub Desktop.
Save vznvzn/126383eab0094130b9fecc96004fa506 to your computer and use it in GitHub Desktop.
def f2(n)
return n.odd? ? (n * 3 + 1) / 2 : n / 2
end
def dense(w, d)
w2 = w - 1
a = (0...w2).to_a
s = '0' * w2
(1..(d * w - 1)).map { a.delete_at(rand(a.size)) }.each { |x| s[x, 1] = '1' }
return ('1' + s).to_i(2)
end
def stat(l)
l = [0] if (l.empty?)
t = t2 = 0
l.each \
{
|x|
t += x
t2 += x ** 2
}
c = l.size
a = t.to_f / c
z = t2.to_f / c - a ** 2
sd = Math.sqrt(z < 0 ? 0 : z)
return a, sd, l.max.to_f, l.min.to_f
end
def stat2(l, t, n)
return Hash[[["a#{n}", "sd#{n}", "mx#{n}"], stat(l)[0..2].map { |x| x / t }].transpose]
end
def d(s)
c = s.split('').select { |x| x == '1' }.size
d = c.to_f / s.length
return d
end
def data(n)
ns = n.to_s(2)
nl = ns.length
m = nl / 2
nsh = ns[0..m]
nsl = ns[m..-1]
asdm1 = stat2(ns.split(/0+/).map { |x| x.length }, nl, 1)
l1 = ns.split(/1+/)
l1.shift
asdm0 = stat2(l1.map { |x| x.length }, nl, 0)
return {'n' => n, 'ns' => ns, 'nl' => nl, 'd' => d(ns), 'dh' => d(nsh), 'dl' => d(nsl)}.merge(asdm1).merge(asdm0)
end
def dist(c)
return (0...c).map { |i| data(dense(100, i.to_f / (c - 1))) }
end
def dot(x, z)
t = z['c']
(z.keys - ['c']).each { |v| t += z[v] * x[v] }
return t
end
def detect(x1, a, l1)
x1 = x1.dup
i = 0
n1 = x1['n']
loop \
{
x2 = data(f2(x1['n']))
x2['wr'] = x2['nl'].to_f / x1['nl'].to_f
x2b = {}
(a['v'] + ['wr']).each \
{
|vy|
x2b[vy] = dot(x1, a[vy].select { |k, x| (a['v'] + ['c']).member?(k) })
x2["#{vy}_e"] = x2b[vy] - x2[vy]
}
x1 = x2
l1 << x1
i += 1
break if (x1['n'] < n1)
}
return i
end
def sum(l)
t = 0.0
l.each { |x| t += x }
return t
end
def out(fn, a)
f = File.open(fn, 'a')
f.puts(a.keys.join("\t")) if (f.size == 0)
f.puts(a.values.join("\t"))
f.close
end
def plot(f, a, fn, c)
f.puts("plot \\")
(a['v'] + ['wr']).each \
{
|k|
f.puts("'#{fn}' using #{c}(column('#{k}_e')) with line title '#{k}',\\")
}
f.puts
f.puts("pause -1")
end
raise if (ARGV[0].nil?)
d = "out/#{ARGV[0]}"
l = Dir.glob("#{d}/*").select { |x| x =~ /out\d+.txt$/}
n = l.size
i = n - 1
a = Kernel.eval(File.open("#{d}/out#{i}.txt").readlines[0])
l = a['lb'].map { |x| x.size }
raise if (sum(l) != 0)
l = dist(100)
File.open(fn = "out.txt", 'w').close
l.each \
{
|x|
c = detect(x, a, l1 = [])
l1.each_with_index \
{
|y, i|
a2 = Hash[(a['v'] + ['wr']).map { |k| "#{k}_e" }.map { |k| [k, y[k]] }].merge({'i' => i, 'j' => l1.size - i})
out(fn, a2)
}
out(fn, {})
}
f = File.open('gnuplot.cmd', 'w')
f.puts("set colors classic;")
plot(f, a, fn, "(column('i')):")
plot(f, a, fn, "(column('j')):")
plot(f, a, fn, "")
f.close
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment