Ignore:
Timestamp:
Apr 27, 2012 7:22:40 AM (13 years ago)
Author:
Mathieu Doucet <doucetm@…>
Branches:
master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload
Children:
10bfeb3
Parents:
34f3ad0
Message:

make pylint happier

File:
1 edited

Legend:

Unmodified
Added
Removed

  • sanscalculator/src/sans/calculator/instrument.py

    rf93b6a1e r8a621ac  
    11""" 
    2 This module is a small tool to allow user to  
    3 control instrumental parameters  
     2This module is a small tool to allow user to 
     3control instrumental parameters 
    44""" 
    55import numpy 
    … …  
    77# defaults in cgs unit 
    88_SAMPLE_A_SIZE = [1.27] 
    9 _SOURCE_A_SIZE = [3.81]  
    10 _SAMPLE_DISTANCE = [1627, 0]  
    11 _SAMPLE_OFFSET = [0, 0]  
    12 _SAMPLE_SIZE = [2.54]  
    13 _SAMPLE_THICKNESS = 0.2   
    14 _D_DISTANCE = [1000, 0]  
     9_SOURCE_A_SIZE = [3.81] 
     10_SAMPLE_DISTANCE = [1627, 0] 
     11_SAMPLE_OFFSET = [0, 0] 
     12_SAMPLE_SIZE = [2.54] 
     13_SAMPLE_THICKNESS = 0.2 
     14_D_DISTANCE = [1000, 0] 
    1515_D_SIZE = [128, 128] 
    16 _D_PIX_SIZE = [0.5, 0.5]   
     16_D_PIX_SIZE = [0.5, 0.5] 
    1717 
    1818_MIN = 0.0 
    … …  
    2121_WAVE_LENGTH = 6.0 
    2222_WAVE_SPREAD = 0.125 
    23 _MASS = 1.67492729E-24 #[gr] 
    24 _LAMBDA_ARRAY = [[0, 1e+16],[_INTENSITY, _INTENSITY]] 
     23_MASS = 1.67492729E-24  # [gr] 
     24_LAMBDA_ARRAY = [[0, 1e+16], [_INTENSITY, _INTENSITY]] 
     25 
    2526 
    2627class Aperture(object): 
    … …  
    3637        self.sample_distance = _SAMPLE_DISTANCE 
    3738         
    38     def set_source_size(self, size =[]): 
     39    def set_source_size(self, size=[]): 
    3940        """ 
    4041        Set the source aperture size 
    4142        """ 
    4243        if len(size) == 0: 
    43             self.source_size = 0.0  
     44            self.source_size = 0.0 
    4445        else: 
    4546            self.source_size = size 
    4647            validate(size[0]) 
    47     def set_sample_size(self, size =[]): 
     48             
     49    def set_sample_size(self, size=[]): 
    4850        """ 
    4951        Set the sample aperture size 
    5052        """ 
    5153        if len(size) == 0: 
    52             self.sample_size = 0.0  
     54            self.sample_size = 0.0 
    5355        else: 
    5456            self.sample_size = size 
    5557            validate(size[0]) 
    5658             
    57     def set_sample_distance(self, distance = []): 
     59    def set_sample_distance(self, distance=[]): 
    5860        """ 
    5961        Set the sample aperture distance 
    6062        """ 
    6163        if len(distance) == 0: 
    62             self.sample_distance = 0.0  
     64            self.sample_distance = 0.0 
    6365        else: 
    6466            self.sample_distance = distance 
    … …  
    7880        self.thickness = _SAMPLE_THICKNESS 
    7981 
    80      
    81     def set_size(self, size =[]): 
     82    def set_size(self, size=[]): 
    8283        """ 
    8384        Set the sample size 
    8485        """ 
    8586        if len(size) == 0: 
    86             self.sample_size = 0.0  
     87            self.sample_size = 0.0 
    8788        else: 
    8889            self.sample_size = size 
    8990            validate(size[0]) 
    9091             
    91     def set_thickness(self, thickness = 0.0): 
     92    def set_thickness(self, thickness=0.0): 
    9293        """ 
    9394        Set the sample thickness 
    … …  
    9697        validate(thickness) 
    9798     
    98     def set_distance(self, distance = []): 
     99    def set_distance(self, distance=[]): 
    99100        """ 
    100101        Set the sample distance 
    101102        """ 
    102103        if len(distance) == 0: 
    103             self.distance = 0.0  
     104            self.distance = 0.0 
    104105        else: 
    105106            self.distance = distance 
    … …  
    122123     
    123124         
    124     def set_size(self, size =[]): 
     125    def set_size(self, size=[]): 
    125126        """ 
    126127        Set the detector  size 
    … …  
    132133            validate(size[0]) 
    133134             
    134     def set_pix_size(self, size = []): 
     135    def set_pix_size(self, size=[]): 
    135136        """ 
    136137        Set the detector pix_size 
    137138        """ 
    138139        if len(size) == 0: 
    139             self.pix_size = 0  
     140            self.pix_size = 0 
    140141        else: 
    141142            self.pix_size = size 
    142143            validate(size[0]) 
    143144     
    144     def set_distance(self, distance = []): 
     145    def set_distance(self, distance=[]): 
    145146        """ 
    146147        Set the detector distance 
    … …  
    166167        # wavelength spread (FWHM) 
    167168        self.wavelength_spread = _WAVE_SPREAD 
    168         # wavelength spectrum  
    169         self.spectrum = self.get_default_spectrum()         
     169        # wavelength spectrum 
     170        self.spectrum = self.get_default_spectrum() 
    170171        # intensity in counts/sec 
    171         self.intensity = numpy.interp(self.wavelength,  
     172        self.intensity = numpy.interp(self.wavelength, 
    172173                                      self.spectrum[0], 
    173174                                      self.spectrum[1], 
    … …  
    188189        """ 
    189190        self.band = self.spectrum 
    190     def set_spectrum(self, spectrum):   
     191         
     192    def set_spectrum(self, spectrum): 
    191193        """ 
    192194        Set spectrum 
    … …  
    194196        :param spectrum: numpy array 
    195197        """ 
    196         self.spectrum = spectrum  
     198        self.spectrum = spectrum 
    197199        self.setup_spectrum() 
    198200          
    199201    def setup_spectrum(self): 
    200202        """ 
    201         To set the wavelength spectrum, and intensity, assumes  
     203        To set the wavelength spectrum, and intensity, assumes 
    202204        wavelength is already within the spectrum 
    203205        """ 
    204206        spectrum = self.spectrum 
    205         intensity = numpy.interp(self.wavelength,  
     207        intensity = numpy.interp(self.wavelength, 
    206208                                      spectrum[0], 
    207209                                      spectrum[1], 
    … …  
    213215        self.max = max(self.spectrum[0]) 
    214216        # set default band 
    215         self.set_band([self.min,self.max]) 
     217        self.set_band([self.min, self.max]) 
    216218         
    217219    def set_band(self, band=[]): 
    … …  
    224226        if min(band) < self.min or\ 
    225227                max(band) > self.max: 
    226             raise  
     228            raise 
    227229        self.band = band 
    228  
    229230          
    230     def set_intensity(self, intensity = 368428): 
     231    def set_intensity(self, intensity=368428): 
    231232        """ 
    232233        Sets the intensity in counts/sec 
    233234        """ 
    234         self.intensity = intensity   
    235         validate(intensity)  
    236              
    237     def set_wavelength(self, wavelength = _WAVE_LENGTH): 
     235        self.intensity = intensity 
     236        validate(intensity) 
     237             
     238    def set_wavelength(self, wavelength=_WAVE_LENGTH): 
    238239        """ 
    239240        Sets the wavelength 
    … …  
    242243        if wavelength < min(self.band) or\ 
    243244                wavelength > max(self.band): 
    244             raise  
     245            raise 
    245246        self.wavelength = wavelength 
    246247        validate(wavelength) 
    247         self.intensity = numpy.interp(self.wavelength,  
     248        self.intensity = numpy.interp(self.wavelength, 
    248249                                  self.spectrum[0], 
    249250                                  self.spectrum[1], 
    … …  
    251252                                  0.0) 
    252253 
    253  
    254     def set_mass(self, mass = _MASS): 
     254    def set_mass(self, mass=_MASS): 
    255255        """ 
    256256        Sets the wavelength 
    … …  
    259259        validate(mass) 
    260260         
    261     def set_wavelength_spread(self, spread = _WAVE_SPREAD): 
     261    def set_wavelength_spread(self, spread=_WAVE_SPREAD): 
    262262        """ 
    263263        Sets the wavelength spread 
    … …  
    302302        """ 
    303303        return self.spectrum 
     304     
    304305    def get_default_spectrum(self): 
    305306        """ 
    … …  
    307308        """ 
    308309        return numpy.array(_LAMBDA_ARRAY) 
     310     
    309311    def get_band(self): 
    310312        """ 
    311313        To get the wavelength band 
    312314        """ 
    313         return self.band  
     315        return self.band 
    314316     
    315317    def plot_spectrum(self): 
    … …  
    320322        try: 
    321323            import matplotlib.pyplot as plt 
    322             plt.plot(self.spectrum[0], self.spectrum[1], linewidth = 2, color = 'r') 
    323             plt.legend(['Spectrum'], loc = 'best') 
     324            plt.plot(self.spectrum[0], self.spectrum[1], linewidth=2, color='r') 
     325            plt.legend(['Spectrum'], loc='best') 
    324326            plt.show() 
    325327        except: 
    … …  
    345347        get list of the intensity wrt wavelength_list 
    346348        """ 
    347         out = numpy.interp(self.wavelength_list,  
     349        out = numpy.interp(self.wavelength_list, 
    348350                                      self.spectrum[0], 
    349351                                      self.spectrum[1], 
    … …  
    375377         
    376378         
    377 def validate(value = None): 
     379def validate(value=None): 
    378380    """ 
    379381    Check if the value is folat > 0.0 
    … …  
    389391    except: 
    390392        val = False 
    391     #if not val: 
    392     #    raise ValueError, "Got improper value..." 
Note: See TracChangeset for help on using the changeset viewer.