Changeset 2

Show
Ignore:
Timestamp:
09/23/07 03:58:45 (1 year ago)
Author:
steve
Message:

Renamed WaveletTransform? to WaveletImage?
added .cleanup() method to convert image back to RGB

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved

  • test.input

    r1 r2  
    11import wavelet 
    2 w1 = wavelet.WaveletTransform("hiero.jpg") 
     2w1 = wavelet.WaveletImage("hiero.jpg") 
    33w1.signature() 
    4 w2=wavelet.WaveletTransform("hiero21600.jpg") 
     4w2=wavelet.WaveletImage("hiero21600.jpg") 
    55w2.signature() 
    6 w3= wavelet.WaveletTransform("null.jpg") 
     6w3= wavelet.WaveletImage("null.jpg") 
    77w3.signature() 
    8 w4= wavelet.WaveletTransform("blueships.jpg") 
     8w4= wavelet.WaveletImage("blueships.jpg") 
    99w4.signature() 
    10 w5=wavelet.WaveletTransform("bluexmas2k2.jpg") 
     10w5=wavelet.WaveletImage("bluexmas2k2.jpg") 
    1111w5.signature() 
    12 w6=wavelet.WaveletTransform("bluehibiscus.jpg") 
     12w6=wavelet.WaveletImage("bluehibiscus.jpg") 
    1313w6.signature() 
    14 w7=wavelet.WaveletTransform("bluehibiscus1600.jpg") 
     14w7=wavelet.WaveletImage("bluehibiscus1600.jpg") 
    1515w7.signature() 
    1616 

  • wavelet.py

    r1 r2  
    1313import config 
    1414 
    15 class WaveletTransform(object): 
     15class WaveletImage(object): 
    1616        """A lazy wavelet transform class""" 
    1717        def __init__(self, path): 
    18                 super(WaveletTransform, self).__init__() 
     18                super(WaveletImage, self).__init__() 
    1919                 
    2020                """ 
    …… 
    3232                self.wavelets = None 
    3333                self.sig = None 
     34                self.mode = "YIQ" 
     35         
     36        def cleanup(self): 
     37                """frees wavelets data and reverts image to rgb""" 
     38                self.wavelets = None 
    3439                 
     40                if self.mode == "YIQ": 
     41                        yiq2rgb = ( 
     42                                1,   0.956,      0.621, 0, 
     43                                1,      -0.272, -0.647, 0, 
     44                                1,      -1.105,  1.702, 0 
     45                                ) 
     46                 
     47                        self.im = self.im.convert("RGB", yiq2rgb) 
     48                        self.mode="RGB" 
     49                         
    3550        def pix_sum(self, x,y): 
    3651                """returns a tuple which is the sum of the tuples given"""