Skip to content

Instantly share code, notes, and snippets.

@vznvzn
Created September 30, 2020 00:16
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/65ee42dde6257319952213305a5898b1 to your computer and use it in GitHub Desktop.
Save vznvzn/65ee42dde6257319952213305a5898b1 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 initgap()
$x = []
$g = {}
$gd = {}
end
def addgap(i)
a, b = $x[i, 2]
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 recalc(r)
r.to_a.select { |x| x >= 0 }.each \
{
|x|
$gd[$x[x]] = [x > 0 ? $x[x - 1] : nil, $x[x + 1]].compact.map { |x1| ($x[x] - x1).abs }.max
}
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.size - 2)
recalc(($x.size - 2)..($x.size - 1))
return
end
return if ($x[i] == z)
rmgap(*$x[i - 1, 2]) if (i > 0)
$x[i, 0] = z
addgap(i)
addgap(i - 1) if (i > 0)
recalc((i - 1)..(i + 1))
end
def gapmx(z)
gap(z)
x = $gd[z]
return x.nil? ? 0 : x
end
def fmt(a)
['ns', 'l'].each { |x| a.delete(x) }
return a
end
def axes2(x2)
return x2.inject({}) { |h, x| h.merge({x => 'axes x1y2' }) }
end
def x1y2()
return $a2 if (!$a2.nil?)
$a2 = axes2(['nw', 'c', 'cg', 'cm']) if ($a2.nil?)
return $a2
end
def midpt(ns)
w2 = ns.length / 4
l = ns.split('')
i = j = 0
while (i < w2 && j < l.size)
i += l[j].to_i
j += 1
end
return j.to_f / ns.length
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 len1(ns)
return len(ns, /0+/)
end
def len0(ns)
return len(ns, /1+/)
end
def len01(ns)
return len1(ns), len0(ns)
end
def d(s)
c = s.split('').select { |x| x == '1' }.size
d = c.to_f / s.length
return d
end
def e(ns)
return len01(ns).flatten.size.to_f / ns.length
end
def features(ns, k)
d = (d(ns) - 0.5).abs
e = (e(ns) - 0.5).abs
nw = ns.length
nw2 = nw / 2
nshi = ns[0...nw2]
nslo = ns[nw2..-1]
dlo = (0.5 - d(nslo)).abs
dhi = (0.5 - d(nshi)).abs
elo = (0.5 - e(nslo)).abs
ehi = (0.5 - e(nshi)).abs
mp = (0.5 - midpt(ns)).abs
return {
"d#{k}" => d,
"e#{k}" => e,
"dlo#{k}" => dlo,
"dhi#{k}" => dhi,
"elo#{k}" => elo,
"ehi#{k}" => ehi,
"mp#{k}" => mp
}
end
def count(ns)
return if (d(ns) != 0.5)
l = seq(n = ns.to_i(2))
return if ((x = l[$z]).nil?)
ns2 = x.to_s(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
nw = ns.length
return {'ns' => ns,
'nw' => nw,
'c' => c,
'cg' => cg,
'cm' => cm,
# 'l' => l,
}.merge(features(ns, '')).merge(features(ns2, '2'))
end
def out(fn, a, d, a2 = {}, t = nil)
if ($f1.nil?) then
$f1 = File.open(fn, 'w')
$f1.puts(a.keys.join("\t"))
$c = 0
end
$f1.puts(a.values.join("\t"))
$f1.flush
$c += 1
return if ($c % 1000 != 0)
d = ($c / 2000).to_i
d = 1 if (d == 0)
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 bottom left box opaque; #{st}; plot \\")
(a.keys - ['t']).each \
{
|x|
next if (a2.member?(x) && a2[x].nil?)
f2.puts("'#{fn}' using (column('t')):(column('#{x}')) every #{d} with line title '#{x}' #{a2[x]}, \\")
}
f2.puts
f2.close
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 stat1(k, l)
a, sd = stat(l)
return {"#{k}_a" => a,
"#{k}_s" => sd}
end
def add(bs, km, h, ns, l1, wt, t, d, r)
return 0 if ((a1 = count(ns)).nil?)
a1['r'] = r
# a1.merge!(stat1('h', h.keys))
j = ns.length
h[j] = [] if (!h.member?(j))
x = adj(a1, wt, km)
a, b = [h[j][0], h[j][-1]]
return 0 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)
x1 = x.dup
x1.delete('l')
out('out.txt', x1, d, x1y2().merge({'ns' => nil, 'r' => nil, 'z' => nil}))
l1 << x.dup if (!l1.nil?)
return 1
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)
end
def initd(w, m)
return(dense(w, m.to_f / w))
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 = dense(w, 0.5)
i = rand(w - m) + 1
s[i, m] = rand(2).to_s * m
return s
end
def initr(w, m)
s = '1'
b = 0
while (s.length < w)
z = w - s.length
s += b.to_s * [rand(m - 1) + 1, z].min
b ^= 1
end
return s
end
def init(w, m)
ns1 = dense(w, 0.5)
ns2 = initr(w, m)
ns1, ns2 = [ns2, ns1] if (rand(2) == 0)
i = rand(w - m) + 1
ns1[i, m] = ns2[i, m]
return ns1
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 mutb(ns)
ns = ns.dup
c = rand(ns.length - 1) + 1
a = (0...ns.length).to_a
c.times \
{
|x|
i = a[x]
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, l1, n)
h = {}
ops = ['init0', 'initd0', 'initb0', 'initw0', 'initr0', 'mut1', 'mutb1', 'mix2', 'cut2']
# ops += ['ext1', 'short1']
bs = 500
bc = 50
t = c1 = 0
wt = {}
while (c1 < n)
if (c1 % 100 == 0) then
dist(km, h, wt)
h.values.flatten.each { |x| adj(x, wt, km) }
end
opc = ops[r = rand(ops.size)]
p = 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 (p)
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...p).map { ks[rand(ks.size)] }
l = arg.map { |x| top(h[x])['ns'] }
ns = send(op, *l)
ns[0, 1] = '1'
end
# ns[-1, 1] = '1'
d = (t.to_f / (c1 + 1) * n / 2000).to_i
c1 += add(bs, km, h, ns, l1, wt, t, d, r)
if (t % 100 == 0) then
msg = {'t' => t,
'c1' => c1,
'f' => (c1.to_f / t * 100).round(2),
'r' => r,
'hmn' => ks.min,
'hmx' => ks.max,
'hc' => h.size}
h0 = nonnil(h.fetch(ks[0], [])[0], {}).dup
msg2 = { 'h0' => h0}
['h0'].each { |x| fmt(msg2[x]) }
$stderr.puts(msg)
$stderr.puts("\t" + msg2.inspect) if (t % 1000 == 0)
end
t += 1
end
return h
end
def outd(f, l)
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')
return k
end
def outa(f, l, a = {}, t = '')
k = outd(f, l)
f.puts("set colors classic; set key top left opaque; set title '#{t}'; ")
f.puts("set ytics nomirror; set y2tics;")
f.puts("plot \\")
ct = ''
# 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, fn, t = '', a = {})
fn = 'gnuplot.cmd' if (fn.nil?)
outa(f = File.open(fn, 'w'), l, a, t)
f.close
$stderr.puts([fn, l.size, t].inspect)
end
def out2(l, fn, t)
l.sort_by! { |x| x['z'] }
outafn(l, fn, t, a = {'t' => nil, 'nw' => nil, 'c' => nil, 'cg' => nil, 'cm' => nil, 'r' => nil, 'z' => 'axes x1y2', 'ns' => nil})
end
$z = ARGV[0].nil? ? 5 : ARGV[0].to_i
w = 400
c = 3e3
a = {
'd' => -1, 'e' => -1, 'dlo' => -1, 'dhi' => -1, 'elo' => -1, 'ehi' => -1, 'mp' => -1,
'd2' => -1, 'e2' => -1, 'dlo2' => -1, 'dhi2' => -1, 'elo2' => -1, 'ehi2' => -1, 'mp2' => -1,
}
h = opt(a, w, l1 = [], c.to_i)
t = "z = #{$z}"
out2(h.values[0], 'gnuplot1.cmd', t)
out2(l1, 'gnuplot2.cmd', t)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment