def paint_aftr_w(color, angle, nx, ny, nr, useoil=False):
ref, before, after = get_roi(nx, ny, nr)
radius, srad = intrad(nr)
rb.compose(after, brush, x=radius, y=radius, rad=radius, srad=srad, angle=angle, color=color, useoil=useoil)
# if useoil here set to true: 2x slow down + instability
err_aftr = np.mean(diff(after, ref))
return err_aftr
def paint_final_w(color, angle, nr, useoil=False):
radius, srad = intrad(nr)
rb.compose(canvas,
brush,
x=x,
y=y,
rad=radius,
srad=srad,
angle=angle,
color=color,
usefloat=True,
useoil=useoil)
def add_stroke(img,
canvas,
brush,
color,
angle,
fatness,
x,
y,
r,
is_final=False,
useoil=False):
radius, srad = intrad(r, fatness)
# if brush placement is final, add to canvas and return
if is_final:
rb.compose(canvas,
brush,
x=x,
y=y,
rad=radius,
srad=srad,
angle=angle,
color=color,
useoil=useoil)
return None
# otherwise only run on ROI and get error
else:
# get ROI areas to calculate error
ref, before = get_roi(img, canvas, x, y, r)
after = np.array(before)
# if useoil here set to true: 2x slow down + instability
# we disable here, assuming is not influential on error estimation
rb.compose(after,
brush,
x=radius,
y=radius,
rad=radius,
srad=srad,
angle=angle,
color=color,
useoil=False)
err_aftr = np.mean(diff(after, ref))
return err_aftr