#!/usr/bin/env python3
import numpy as np
import matplotlib.pyplot as plt
from slicer import *

w = 50
ps = [.1, .2, .3, .4, .5, .6, .7, .8]
LP = [20, 25, 30, 35, 40]

date = datetime.today().strftime("%Y-%m-%d")

job = Job()
for j,lp in enumerate(LP):
    for (i, p) in enumerate(ps):
        name = f"{date}\\n\\n vert, lp={lp}, p={p}"
        filename = f"vert,p={p}.gwl"

        ml = board2x2(w, p, 0, 0)
        stage = Stage(ml, xo = 150*i, yo=0 + j*700, lp=lp, name=name, filename=filename)
        job.append(stage)

    for (i, p) in enumerate(ps):
        name = f"{date}\\n\\n diag1, lp={lp}, p={p}"
        filename = f"diag1,p={p}.gwl"

        ml = board2x2_45(w, p, 0, 0)
        stage = Stage(ml, xo = 150*i, yo=150 + j*700, lp=lp, name=name, filename=filename)
        job.append(stage)

    for (i, p) in enumerate(ps):
        name = f"{date}\\n\\n diag2, lp={lp}, p={p}"
        filename = f"diag2,p={p}.gwl"

        ml = board2x2_135(w, p, 0, 0)
        stage = Stage(ml, xo = 150*i, yo=300 + j*700, lp=lp, name=name, filename=filename)
        job.append(stage)

    for (i, p) in enumerate(ps):
        name = f"{date}\\n\\n circ, lp={lp}, p={p}"
        filename = f"circ, p={p}.gwl"

        ml = board_circ(w, p, 0, 0)
        stage = Stage(ml, xo = 150*i, yo=450 + j*700, lp=lp, name=name, filename=filename)
        job.append(stage)

    p1 = 0.300
    p2 = 0.900
    name = f"{date}\\n\\n lp={lp}, vp, big"
    filename = f"vp_big.gwl"
    ml = board_variable_pitch(140, p1, p2, 0, 0)
    stage = Stage(ml, xo=1300, yo=100 + j*700, lp=lp, name=name, filename=filename)
    job.append(stage)

    name = f"{date}\\n\\n lp={lp}, vp, small"
    filename = f"vp_small.gwl"
    ml = board_variable_pitch(20, p1, p2, 0, 0)
    stage = Stage(ml, xo=1300, yo=250 + j*700, lp=lp, name=name, filename=filename)
    job.append(stage)


write_job("job.gwl", job)
draw_job(job)
plt.show()