I'm trying to do a polar transform on the first image below and end up with the second. However my result is the third image. I have a feeling it has to do with what location I choose as my "origin" but am unsure.
radius = sqrt(width**2 + height**2)
nheight = int(ceil(radius)/2)
nwidth = int(ceil(radius/2))
for y in range(0, height):
for x in range(0, width):
t = int(atan(y/x))
r = int(sqrt(x**2+y**2)/2)
color = getColor(getPixel(pic, x, y))
setColor( getPixel(radial,r,t), color)
There are a few differences / errors:
- They use the centre of the image as the origin
- They scale the axis appropriately. In your example, you're plotting your angle (between 0 and in your case, pi), instead of utilising the full height of the image.
- You're using the wrong atan function (atan2 works a lot better in this situation :))
- Not amazingly important, but you're rounding unnecessarily quite a lot, which throws off accuracy a little and can slow things down.
This is the code combining my suggested improvements. It's not massively efficient, but it should hopefully work :)
maxradius = sqrt(width**2 + height**2)/2
rscale = width / maxradius
tscale = height / (2*math.pi)
for y in range(0, height):
dy = y - height/2
for x in range(0, width):
dx = x - width/2
t = atan2(dy,dx)%(2*math.pi)
r = sqrt(dx**2+dy**2)
color = getColor(getPixel(pic, x, y))
setColor( getPixel(radial,int(r*rscale),int(t*tscale)), color)
In particular, it fixes the above problems in the following ways:
- We use
dx = x - width / 2
as a measure of distance from the centre, and similarly with dy
. We then use these in replace of x
, y
throughout the computation.
- We will have our
r
satisfying 0 <= r <= sqrt( (width/2)^2 +(height/2)^2 )
, and our t
eventually satisfying 0 < t <= 2 pi
so, I create the appropriate scale factors to put r
and t
along the x
and y
axes respectively.
- Normal
atan
can only distinguish based on gradients, and is computationally unstable near vertical lines... Instead, atan2
(see http://en.wikipedia.org/wiki/Atan2) solves both problems, and accepts (y,x)
pairs to give an angle. atan2
returns an angle -pi < t <= pi
, so we can find the remainder modulo 2 * math.pi
to it to get it in the range 0 < t <= 2pi
ready for scaling.
- I've only rounded at the end, when the new pixels get set.
Any questions, just ask!