Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created July 28, 2020 23:45
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/93c0dd7333d0bb84f07883be1cba4a47 to your computer and use it in GitHub Desktop.
Save vznvzn/93c0dd7333d0bb84f07883be1cba4a47 to your computer and use it in GitHub Desktop.
def f2(n)
return n.odd? ? (n * 3 + 1) / 2 : n / 2
end
def seq(n)
l = [n]
while (n != 1)
n = f2(n)
l << n
end
return l
end
def addgap(a, b)
d = b - a
$g[d] = $g.fetch(d, []) + [[a, b]]
$mx = [$mx, d].compact.max
end
def rmgap(a, b)
d = b - a
$g[d] -= [[a, b]]
if ($g[d].empty?) then
$g.delete(d)
$mx = $g.keys.max if (d == $mx)
end
end
def gap(z)
i = (0...$x.size).bsearch { |x| $x[x] >= z }
if (i.nil?) then
$x << z
return if ($x.size == 1)
addgap(*$x[-2, 2])
return
end
return if ($x[i] == z)
rmgap(*$x[i - 1, 2]) if (i > 0)
$x[i, 0] = z
addgap(*$x[i, 2])
addgap(*$x[i - 1, 2]) if (i > 0)
end
def gapmx(z)
gap(z)
return $mx.nil? ? z : $mx
end
def count(ns, l2 = [])
l = seq(n = ns.to_i(2))
cg = l.index { |x| x < l[0] }
cg = l.size - 1 if (cg.nil?)
cm = (0..cg).max_by { |x| l[x] }
c = l.size
return {'ns' => ns,
'cm' => cm,
'c' => c,
'l' => l
}.merge({'g' => gapmx(c)})
end
def out(fn, a, d, a2 = {}, t = nil)
k = a.keys
if ($f1.nil?) then
$f1 = File.open(fn, 'w')
$f1.puts(k.join("\t"))
f2 = File.open('gnuplot.cmd', 'w')
st = t.nil? ? 'unset title; ' : "set title '#{t}'; "
f2.puts("set ytics nomirror; set y2tics; set colors classic; set key top left box opaque; #{st}; plot \\")
(a.keys - ['t']).each \
{
|x|
f2.puts("'#{fn}' using (column('t')):(column('#{x}')) every #{d} with line title '#{x}' #{a2[x]}, \\")
}
f2.puts
f2.close
end
$f1.puts(a.values_at(*k).join("\t"))
$f1.flush
end
def rank(x)
return -x['z']
end
def adj(x, wt, km)
z = 0
km.keys.each { |k| z += (x[k] - wt["#{k}_a"]) / wt["#{k}_sd"] * km[k] }
x['z'] = z
raise [x, wt].to_s if (!z.finite?)
return x
end
def fmt(a)
['ns', 'l'].each { |x| a.delete(x) }
return a
end
def add(bs, km, k, h, ns, l1, wt, t, d)
j = ns.length
h[j] = [] if (!h.member?(j))
x = adj(count(ns), wt, km)
a, b = [h[j][0], h[j][-1]]
return if (h[j].map { |x| x['ns'] }.member?(x['ns']))
x['t'] = t
h[j] << x
h[j].sort_by! { |x| rank(x) }
h[j][bs..-1] = [] if (h[j].size > bs)
out('out.txt', fmt(x.dup), d, {k => 'axis x1y2', 'z' => 'axes x1y2'})
l1 << x if (!l1.nil?)
end
def init(w, m)
return '1' * w
end
def initd(w, m)
d = m.to_f / w
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)
end
def initb(w, m)
r = m.to_f / w
return '1' + (1...w).map { rand < r ? '0' : '1' }.join
end
def initw(w, m)
s = initd(w, w / 2)
i = rand(w - m) + 1
m = [w - i, m].max
s[i, m] = rand(2).to_s * m
return s
end
def mut(ns)
ns = ns.dup
i = rand(ns.length - 1) + 1
ns[i, 1] = (ns[i, 1].to_i ^ 1).to_s
return ns
end
def inner(ns1, ns2)
a, b = [ns1, ns2].sort_by { |x| x.length }
d = b.length - a.length
i = rand(d + 1)
c = b.dup
c[i, a.length] = a
ns1.replace(b)
ns2.replace(c)
end
def mix(ns1, ns2)
ns1 = ns1.dup
ns2 = ns2.dup
inner(ns1, ns2)
return (0...ns1.length).map { |x| [ns1, ns2][rand(2)][x, 1] }.join
end
def cut(ns1, ns2)
ns1 = ns1.dup
ns2 = ns2.dup
inner(ns1, ns2)
i = rand(ns1.length)
return ns1[0...i] + ns2[i..-1]
end
def ext(ns)
ns = ns.dup
(rand(3) + 1).times { ns[rand(ns.length) + 1, 0] = rand(2).to_s }
return ns
end
def short(ns)
return ns if (ns.length == 1)
ns = ns.dup
(rand([ns.length - 1, 3].min) + 1).times { ns[rand(ns.length - 1) + 1, 1] = '' }
return ns
end
def top(h1)
return h1[rand([h1.size / 4, h1.size].max)]
end
def stat(l)
return 0, 1.0 if (l.empty?)
t = l.inject { |a, x| a + x }
t2 = l.inject(0) { |a, x| a + x ** 2 }
c = l.size
a = t.to_f / c
z = t2.to_f / c - a ** 2
sd = Math.sqrt(z < 0 ? 0 : z)
raise if (sd.nan?)
return a, sd
end
def dist(km, h, wt)
l = h.values.flatten
km.keys.each \
{
|k|
a, sd = stat(l.map { |x| x[k] })
sd = 1.0 if (sd == 0)
wt.merge!({"#{k}_a" => a, "#{k}_sd" => sd})
}
end
def nonnil(x, y)
return x.nil? ? y : x
end
def opt(km, w1, k, l1, n)
h = {}
ops = ['init0', 'initd0', 'initb0', 'initw0', 'mut1', 'mix2', 'cut2']
# ops += ['ext1', 'short1']
bs = 1000
bc = 50
t = 0
d = n / 2000
d = 1 if (d == 0)
wt = {}
n.times \
{
if (t % 100 == 0) then
dist(km, h, wt)
h.values.flatten.each { |x| adj(x, wt, km) }
end
opc = ops[rand(ops.size)]
c = opc[-1..-1].to_i
op = opc[0..-2]
ks = h.keys.sort_by { |x| rank(h[x][0]) }
ks = ks[0...[ks.size, ks.size / 4].max]
case (c)
when 0
w = ks.empty? ? w1 : ks[rand(ks.size)]
ns = send(op, w, rand(w))
when 1, 2
redo if (h.empty?)
h.delete(ks.pop) if (ks.size > bc)
arg = (0...c).map { ks[rand(ks.size)] }
l = arg.map { |x| top(h[x])['ns'] }
ns = send(op, *l)
ns[0, 1] = '1'
end
add(bs, km, k, h, ns, l1, wt, t, d)
if (t % 100 == 0) then
kmx = nonnil(h.values.flatten.max_by { |x| x[k] }, {}).dup
h0 = nonnil(h.fetch(ks[0], [])[0], {}).dup
k0 = "#{k}_0"
msg = Hash[[['t', 'hmn', 'hmx', 'hc', 'h_0', k0], [t, ks.min, ks.max, h.size, h0, kmx]].transpose]
['h_0', k0].each { |x| fmt(msg[x]) }
$stderr.puts(msg)
end
t += 1
}
return h
end
def outa(f, l, a = {}, t = '')
f.puts('$dat << eof')
k = l[0].keys
f.puts(k.join("\t"))
l.each { |x| f.puts(x.values.join("\t")) }
f.puts('eof')
f.puts("set colors classic; set key top left opaque; set title '#{t}'; ")
# f.puts("set ytics nomirror; set y2tics;")
f.puts("plot \\")
k, ct = [k - ['t'], "(column('t')):"] if (k.member?('t'))
k.each \
{
|x|
next if (a.member?(x) && a[x].nil?)
opt = a.fetch(x, '')
opt += ' lw 2 ' if (!opt.include?('lw'))
opt += ' with line ' if (!opt.include?('with') && !opt.include?('pt'))
f.puts("'$dat' using #{ct}(column('#{x}')) #{opt} title '#{x}',\\")
}
f.puts
# f.puts("reset; pause -1;")
end
def outafn(l, a = {}, fn = nil)
outa(f = File.open(fn.nil? ? 'gnuplot.cmd' : fn, 'w'), l, a)
f.close
end
def d(s)
c = s.split('').select { |x| x == '1' }.size
d = c.to_f / s.length
return d
end
def len(ns, p)
l = ns.split(p)
l = [] if (l.nil?)
l.shift if (l[0] == '')
return l.map { |x| x.length }
end
def len01(ns)
return len(ns, /0+/), len(ns, /1+/)
end
def e(ns)
return len01(ns).flatten.size.to_f / ns.length
end
def fmt1(l, f)
w = l.max_by { |x| x.length }.length
f.puts("set colors classic; set ytics nomirror; set y2tics; set key top left opaque;\\")
f.puts("plot \\")
f.puts("[0:#{l.size - 1}][0:#{w}] '-' matrix using 2:1:3 with image title '', \\")
f.puts("'-' using (column('d')) with line axes x1y2 dt 2 lw 3, \\")
f.puts("'-' using 1 with line title 'e' axes x1y2 dt 2 lw 3")
l1 = []
l.each_with_index \
{
|z, i|
l1 << [d(z), e(z)]
c = w - z.length
z[z.length...w] = '0' * c
f.puts(z.split('').join(' '))
}
f.puts("eof")
f.puts("eof")
f.puts("d")
l1.each { |x| f.puts(x[0]) }
f.puts("eof")
l1.each { |x| f.puts(x[1]) }
f.puts("eof")
end
def grid(l)
f = File.open(fn = "gnuplot3.cmd", 'w')
f.puts("set palette negative grayscale; unset colorbox;")
fmt1(l.map { |x| x.reverse }, f)
f.close
end
def sample(l, c)
return l if (l.size < c)
r = l.size / c
l2 = *(0...l.size).select { |x| x % r == 0 }
l2[c..-1] = []
return l.values_at(*l2)
end
def g(x)
gapmx(x)
p({'x' => x, '$x' => $x, '$g' => $g, '$mx' => $mx})
end
def testgap()
[5, 6, 1, 2, 9, 1, 3].each { |x| g(x) }
exit
end
$x = []
$g = {}
$mx = nil
# testgap()
c = 2e3
h = opt({'c' => 1, 'g' => -1}, 200, nil, l1 = [], c.to_i)
outafn(h.values[0].map { |x| fmt(x.dup) }.sort_by { |x| x['t']}, {}, 'gnuplot1.cmd')
a = {'t1' => 'pt 5 ps 0.25'}
a = {'t1' => nil}
outafn(l1.sort_by! { |x| x['c'] }.map { |x| x1 = fmt(x.dup); x1['t1'] = x1.delete('t'); x1 }, a, 'gnuplot2.cmd')
grid(l = sample(l1, 250).map { |x| x['ns'] })
f = File.open('out2.txt', 'w')
l.each_with_index \
{
|ns, j|
c = 0
ns.reverse.split('').each_with_index { |x1, i| c += [-1, 1][x1.to_i]; f.puts([i, c, j].join("\t")) }
f.puts
}
f.close
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment