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