def verify_crop(fname, nshots, ims, crop_top, crop_bot, crop_lft, crop_rgt, show_plot=False):
"""
do another round of screenshots with the crop command this time and verify
that the correct region is cropped
"""
ims_crop_ref = ims[:, crop_top:, crop_lft:]
if crop_bot > 0:
ims_crop_ref = ims_crop_ref[:, :-crop_bot, :]
if crop_rgt > 0:
ims_crop_ref = ims_crop_ref[:, :, :-crop_rgt]
nshots, Ny, Nx = ims.shape
h = Ny - crop_top - crop_bot
w = Nx - crop_lft - crop_rgt
x = crop_lft
y = crop_top
mpv_args = [
"--vo-defaults=image:format=pgm",
"--vf-add=dsize",
"--vf-add=crop=%d:%d:%d:%d" % (w, h, x, y),
"--no-sub",
fname,
]
ims_crop_test = mpv_utils.sample_screenshots(nshots, mpv_args=mpv_args)
res = abs(ims_crop_test - ims_crop_ref)
if res.max() != 0:
print "ERROR: mpv didn't crop as expected"
for i, r in enumerate(res):
print i, "residual=", r.max()
if show_plot and r.max() > 0:
try:
import pylab as pl
except ImportError:
print import_err_msg % ("matplotlib", "python-matplotlib")
sys.exit(1)
pl.ion()
fig = pl.figure(1, dpi=150)
fig.clf()
ax = fig.add_subplot(221)
ax.imshow(ims_crop_ref[i], cmap="gray", interpolation="nearest")
ax = fig.add_subplot(222)
ax.imshow(ims_crop_test[i], cmap="gray", interpolation="nearest")
ax = fig.add_subplot(223)
ax.imshow(res[i], cmap="gray", interpolation="nearest")
pl.draw()
print "press enter to continue"
raw_input()
sys.exit(1)
else:
print "crop_verified!"
def get_crop_cmd(fname, nshots=11, thresh=0.02, pad=0, ignore_pixels=0, show_plot=False, verify=False):
"""
compute the appropriate crop command for a given file
"""
mpv_args = ["--vo-defaults=image:format=pgm", "--vf-add=dsize", "--no-sub", fname]
ims = mpv_utils.sample_screenshots(nshots, mpv_args=mpv_args)
nshots, Ny, Nx = ims.shape
imax = float(np.iinfo(ims.dtype).max)
ymx = ims.transpose((1, 0, 2)).reshape((Ny, -1)).max(axis=-1) / imax
xmx = ims.transpose((2, 1, 0)).reshape((Nx, -1)).max(axis=-1) / imax
# determine the cropping region where pixels average to greater than thresh
ygood = ymx >= thresh
xgood = xmx >= thresh
if ygood.any():
crop_top = np.flatnonzero(ygood[ignore_pixels:])[0]
if crop_top > 0:
crop_top += ignore_pixels
crop_bot = np.flatnonzero(ygood[::-1][ignore_pixels:])[0]
if crop_bot > 0:
crop_bot += ignore_pixels
else:
print "WARNING: Seems to be nothing here to crop from"
crop_top = crop_bot = 0
if xgood.any():
crop_lft = np.flatnonzero(xgood[ignore_pixels:])[0]
if crop_lft > 0:
crop_lft += ignore_pixels
crop_rgt = np.flatnonzero(xgood[::-1][ignore_pixels:])[0]
if crop_rgt > 0:
crop_rgt += ignore_pixels
else:
print "WARNING: Seems to be nothing here to crop from"
crop_lft = crop_rgt = 0
crop_top = max(0, crop_top - pad)
crop_bot = max(0, crop_bot - pad)
crop_rgt = max(0, crop_rgt - pad)
crop_lft = max(0, crop_lft - pad)
# these should be even to get exact results
# (probably b/c of chroma subsampling)
crop_top = np.floor(crop_top / 2) * 2
crop_lft = np.floor(crop_lft / 2) * 2
crop_bot = np.floor(crop_bot / 2) * 2
crop_rgt = np.floor(crop_rgt / 2) * 2
h = Ny - crop_top - crop_bot
w = Nx - crop_lft - crop_rgt
x = crop_lft
y = crop_top
print fname
print "crop_top=", crop_top
print "crop_bot=", crop_bot
print "crop_lft=", crop_lft
print "crop_rgt=", crop_rgt
if show_plot:
"""visually check the crop detection"""
try:
import pylab as pl
except ImportError:
print import_err_msg % ("matplotlib", "python-matplotlib")
sys.exit(1)
pl.ion()
fig = pl.figure(1, dpi=150)
fig.clf()
ax = fig.add_subplot(111)
ax.imshow(ims.max(axis=0), cmap="gray", interpolation="nearest")
ax.hlines((crop_top - 0.5, Ny - crop_bot - 0.5), -0.5, Nx - 0.5, color="g")
ax.vlines((crop_lft - 0.5, Nx - crop_rgt - 0.5), -0.5, Ny - 0.5, color="g")
fig = pl.figure(2)
fig.clf()
axs = map(lambda i: fig.add_subplot(2, 2, i), xrange(1, 5))
ii = np.arange(ymx[: Ny / 2].size)
axs[0].plot(ymx[: Ny / 2], -ii, ".r")
axs[0].hlines(-(crop_top - 0.5), 0, 1)
axs[0].set_xlim(0, 1)
axs[0].set_ylim((-ii).min(), (-ii).max())
ii = np.arange(ymx[Ny / 2 :].size)
axs[1].plot(ymx[Ny / 2 :], ii[::-1], ".r")
axs[1].hlines(crop_bot - 0.5, 0, 1)
axs[1].set_xlim(0, 1)
axs[1].set_ylim(ii.min(), ii.max())
ii = np.arange(xmx[: Nx / 2].size)
axs[2].plot(ii, xmx[: Nx / 2], ".b")
axs[2].vlines(crop_lft - 0.5, 0, 1)
axs[2].set_ylim(0, 1)
axs[2].set_xlim(ii.min(), ii.max())
ii = np.arange(xmx[Nx / 2 :].size)
axs[3].plot(-ii[::-1], xmx[Nx / 2 :], ".b")
axs[3].vlines(-(crop_rgt - 0.5), 0, 1)
axs[3].set_ylim(0, 1)
axs[3].set_xlim((-ii).min(), (-ii).max())
pl.draw()
print "press enter to continue"
raw_input()
if crop_top > 0 or crop_bot > 0 or crop_lft > 0 or crop_rgt > 0:
if verify:
verify_crop(fname, nshots, ims, crop_top, crop_bot, crop_lft, crop_rgt, show_plot=show_plot)
return ["--hwdec=no", "--vf=crop=%d:%d:%d:%d" % (w, h, x, y)]
else:
return []
