def remove_nyquist(self, a, aft):
ny, nx = a.shape
if mm.iseven(ny):
aft[ny/2,:] = 0
if mm.iseven(nx):
aft[:,nx/2] = 0
a = real(ifft2(aft))
return a, aft
def remove_nyquist(self, a, aft):
ny, nx = a.shape
if mm.iseven(ny):
aft[ny / 2, :] = 0
if mm.iseven(nx):
aft[:, nx / 2] = 0
a = real(ifft2(aft))
return a, aft
def setUp(self):
## pick random odd upsampling
upsample = randint(1, 16)
if mm.iseven(upsample):
half = int((upsample - 1)/2.0)
else:
half = int(upsample/2.0) - 1
# setup random area to calculate idft on
nx = randint(10, 100)
ny = randint(10, 100)
height = randint(4, ny*upsample)
width = randint(4, nx*upsample)
try:
top = randint(0, ny*upsample - height)
except:
top = 0
try:
left = randint(0, nx*upsample - width)
except:
left = 0
# generate a random image
aa = rand(ny, nx)
aa[:-5, :-5] += 1
aa[:-2, :] += 2
a = fft2(aa)
aa, a = self.remove_nyquist(aa, a)
## attach data to test case
self.aa = aa
self.a = a
self.nx = nx
self.ny = ny
self.height = height
self.width = width
self.top = top
self.left = left
self.upsample = upsample
def setUp(self):
## pick random odd upsampling
upsample = randint(1, 16)
if mm.iseven(upsample):
half = int((upsample - 1) / 2.0)
else:
half = int(upsample / 2.0) - 1
# setup random area to calculate idft on
nx = randint(10, 100)
ny = randint(10, 100)
height = randint(4, ny * upsample)
width = randint(4, nx * upsample)
try:
top = randint(0, ny * upsample - height)
except:
top = 0
try:
left = randint(0, nx * upsample - width)
except:
left = 0
# generate a random image
aa = rand(ny, nx)
aa[:-5, :-5] += 1
aa[:-2, :] += 2
a = fft2(aa)
aa, a = self.remove_nyquist(aa, a)
## attach data to test case
self.aa = aa
self.a = a
self.nx = nx
self.ny = ny
self.height = height
self.width = width
self.top = top
self.left = left
self.upsample = upsample
