Update to 2.0.0 tree from current Fremantle build
[opencv] / samples / swig_python / logpolar.py
1 #!/usr/bin/python
2 import sys
3 from opencv.cv import *
4 from opencv.highgui import *
5
6 src=None
7 dst=None
8 src2=None
9
10 def on_mouse( event, x, y, flags, param ):
11
12     if( not src ):
13         return;
14
15     if event==CV_EVENT_LBUTTONDOWN:
16         cvLogPolar( src, dst, cvPoint2D32f(x,y), 40, CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS );
17         cvLogPolar( dst, src2, cvPoint2D32f(x,y), 40, CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS+CV_WARP_INVERSE_MAP );
18         cvShowImage( "log-polar", dst );
19         cvShowImage( "inverse log-polar", src2 );
20
21 if __name__ == "__main__":
22     
23     filename = "../c/fruits.jpg"
24     if len(sys.argv)>1:
25         filename=argv[1]
26     
27     src = cvLoadImage(filename,1)
28     if not src:
29         print "Could not open %s" % filename
30         sys.exit(-1)
31         
32     cvNamedWindow( "original",1 );
33     cvNamedWindow( "log-polar", 1 );
34     cvNamedWindow( "inverse log-polar", 1 );
35   
36     
37     dst = cvCreateImage( cvSize(256,256), 8, 3 );
38     src2 = cvCreateImage( cvGetSize(src), 8, 3 );
39     
40     cvSetMouseCallback( "original", on_mouse );
41     on_mouse( CV_EVENT_LBUTTONDOWN, src.width/2, src.height/2, None, None)
42     
43     cvShowImage( "original", src );
44     cvWaitKey();