* variable:
'sm' -> 'test_mode'

* files:
longmode.rb -> publicmode.rb
shortmode.rb -> testmode.rb

# Automatic editing of collection of source files.
# Does not provide backup files:
# to be safe, work on fresh git commit, and use 'git stash' to revert.

import edtools as et
import re, sys

def f(t):
    t = re.sub(r'--(\d)', r'+\1', t)
    t = re.sub(r'\+-(\d)', r'-\1', t)
    t = re.sub(r'\s*\*\s*deg', r'', t)

    t = f'{eval(t)}'

    t = re.sub('\.0$', '', t)

    if t != '0':
        t += '*deg'

    return t

def user_edit(fn, ti):

    t = re.sub(r'OldDetec2D(\s+\w+)?\((n\w*|\d+),\s*([-.0-9]+\s*\*\s*deg|0),\s*([-.0-9]+\s*\*\s*deg|0),\s*(n\w*|\d+),\s*([-.0-9]+\s*\*\s*deg|0),\s*([-.0-9]+\s*\*\s*deg|0)\)',
               r'NewDetec2D\1(@(\4-\3)@, @(\7-\6)@, \2, \5, @(\4+\3)/2@, @(\7+\6)/2@)', ti)
    if t == ti:
        return ti

#    t = re.sub(r'@(\(.*?\)(/2)?)@', f(r'\1'), t)

    for m in re.finditer(r'@(\(.*?\)(/2)?)@', t):
        s = f(m.group(1))
        t = re.sub(r'@(\(.*?\)(/2)?)@', s, t, 1)

    return t

et.ed_argfiles(user_edit)

def user_edit(fn, t):

    nmini = 11
    if re.search(r'(specular|varia)', fn):
        nmini = 50

    m = re.search(r'bp\.parse_args\(sim_n=(\d+)', t)
    if not m:
        return t
    n = m.group(1)
    t = re.sub(r"bp\.simargs\['n'\]", f'<%= sm ? {nmini} : {n} %>', t)

    t = re.sub(r'(bp\.parse_args\()sim_n=(\d+),\s*', r'\1', t)
    t = re.sub(r'(bp\.parse_args\()sim_n=(\d+)\)', r'\1)', t)

    if not re.search(r'^#!/usr/bin/env python3\n', t):
        raise Exception("missing shebang in "+fn)
    t = re.sub(r'(^#!.*?\n("""\n(.*\n)+"""\n)?)\n*',
               r'\1'+'<% sm = (e=ENV["SHORTMODE"] and e!="OFF") %>\n', t)

    return t

def user_edit(fn, t):

    t = re.sub(r'def get_simulation\(sample\):',
               'def get_simulation(sample, **simargs):', t)

    N = 0
    xx = r'\n((\s+?)\w+ = ba\.\w+?Detector)\((\d+), (.+?, .+?), \d+, (.+?, .+?)\)'
    m = re.search(xx, t)
    if m:
        N = int(m.group(3))
    t = re.sub(xx, r"\n\2n = simargs['n']\n\1(n, \4, n, \5)", t)

    xx = r'(\w+ = ba\.\w+?Detector)\((\d+), (.+?, .+?, .+?)\)'
    m = re.search(xx, t)
    if m:
        N = int(m.group(2))
    t = re.sub(xx, r"\1(simargs['n'], \3)", t)

    xx = r"(if __name__ == '__main__':\n)((.*?\n)+?)(\s+\w+ = get_simulation\(\w+)\)\n((.*?\n)*?)\s+bp\.plot_simulation_result\(simulation\.simulate\(\)\)"
    t = re.sub(xx, r"\1" +
               "    plotargs, simargs = bp.kwargs_from_cmdline(sim_n=" + str(N) + ")\n" +
               r"\2" +
               r"\4, **simargs)\n"
               r"\5" +
               "    result = simulation.simulate()\n" +
               "    bp.plot_simulation_result(result, **plotargs)", t)

    return t