Ignore:
Timestamp:
Apr 18, 2012 1:51:32 PM (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:
2dda7ae1
Parents:
279e371
Message:

Fixing code style problems

File:
1 edited

Legend:

Unmodified
Added
Removed

  • sansmodels/src/sans/models/ReflectivityIIModel.py

    r1352c78 r499fe7a  
    55from math import floor 
    66from math import fabs 
    7 from scipy.special import erf 
    87func_list = {'Erf(|nu|*z)':0, 'RPower(z^|nu|)':1, 'LPower(z^|nu|)':2, \ 
    98                     'RExp(-|nu|*z)':3, 'LExp(-|nu|*z)':4} 
    … …  
    1514    """ 
    1615    def __init__(self, multfactor=1): 
     16        """ 
     17            :param multfactor: number of layers in the model,  
     18            assumes 0<= n_layers <=10. 
     19        """ 
    1720        BaseComponent.__init__(self) 
    18         """ 
    19         :param multfactor: number of layers in the model,  
    20         assumes 0<= n_layers <=10. 
    21         """ 
    2221 
    2322        ## Setting  model name model description 
    24         self.description="" 
     23        self.description = "" 
    2524        model = ReflAdvModel() 
    2625        self.model = model 
    2726        self.name = "ReflectivityIIModel" 
    28         self.description=model.description 
     27        self.description = model.description 
    2928        self.n_layers = int(multfactor) 
    3029        ## Define parameters 
    … …  
    5453        # [int(maximum no. of functionality),"str(Titl), 
    5554        # [str(name of function0),...], [str(x-asix name of sld),...]] 
    56         self.multiplicity_info = [max_nshells,"No. of Layers:",[],['Depth']] 
     55        self.multiplicity_info = [max_nshells, "No. of Layers:", [], ['Depth']] 
    5756        ## independent parameter name and unit [string] 
    5857        self.input_name = "Q" 
    … …  
    9291        """ 
    9392        # rearrange the parameters for the given # of shells 
    94         for name , value in self.model.params.iteritems(): 
     93        for name, value in self.model.params.iteritems(): 
    9594            n = 0 
    9695            pos = len(name.split('_'))-1 
    … …  
    103102                continue 
    104103            elif first_name == 'func': 
    105                 n= -1 
    106                 while n<self.n_layers: 
     104                n = -1 
     105                while n < self.n_layers: 
    107106                    n += 1 
    108107                    if last_name == 'inter%s' % str(n):  
    109                         self.params[name]=value 
     108                        self.params[name] = value 
    110109                        continue 
    111110             
    112111                #continue 
    113112            elif last_name[0:5] == 'inter': 
    114                 n= -1 
    115                 while n<self.n_layers: 
     113                n = -1 
     114                while n < self.n_layers: 
    116115                    n += 1 
    117116                    if last_name == 'inter%s' % str(n): 
    118                         self.params[name]= value 
     117                        self.params[name] = value 
    119118                        continue 
    120119            elif last_name[0:4] == 'flat': 
    121                 while n<self.n_layers: 
     120                while n < self.n_layers: 
    122121                    n += 1 
    123122                    if last_name == 'flat%s' % str(n): 
    124                         self.params[name]= value 
     123                        self.params[name] = value 
    125124                        continue 
    126125            elif name == 'n_layers': 
    127126                continue 
    128127            else: 
    129                 self.params[name]= value 
     128                self.params[name] = value 
    130129   
    131130        self.model.params['n_layers'] = self.n_layers     
    … …  
    139138        this model details  
    140139        """ 
    141         for name ,detail in self.model.details.iteritems(): 
     140        for name, detail in self.model.details.iteritems(): 
    142141            if name in self.params.iterkeys(): 
    143                 self.details[name]= detail 
     142                self.details[name] = detail 
    144143             
    145144     
    … …  
    155154                    continue 
    156155                if  key.split('_')[0] == 'func':  
    157                         self.model.setParam(key, 0) 
    158                         continue 
     156                    self.model.setParam(key, 0) 
     157                    continue 
    159158 
    160159                for nshell in range(self.n_layers,max_nshells): 
    … …  
    166165                                value = self.model.params['sldIM_medium'] 
    167166                            self.model.setParam(key, value) 
    168                         except: pass 
     167                        except: 
     168                            message = "ReflectivityIIModel evaluation problem" 
     169                            raise RuntimeError, message 
    169170     
    170171    def _get_func_list(self): 
    171172        """ 
    172         Get the list of functions in each layer (shell)  
    173         """ 
    174         #func_list = {} 
     173            Get the list of functions in each layer (shell)  
     174        """ 
    175175        return func_list 
    176176         
    … …  
    186186        z = [] 
    187187        beta = [] 
    188         sub_range = int(floor(n_sub/2.0)) 
    189188        z.append(0) 
    190189        beta.append(self.params['sld_bottom0'])  
    … …  
    193192        dz = 0.0 
    194193        # for layers from the top 
    195         for n_lyr in range(1,self.n_layers+2): 
     194        for n_lyr in range(1, self.n_layers+2): 
    196195            i = n_lyr 
    197196            # j=0 for interface, j=1 for flat layer  
    198             for j in range(0,2): 
     197            for j in range(0, 2): 
    199198                # interation for sub-layers 
    200                 for n_s in range(0,n_sub): 
     199                for n_s in range(0, n_sub): 
    201200                    # for flat layer 
    202                     if j==1: 
    203                         if i==self.n_layers+1: 
     201                    if j == 1: 
     202                        if i == self.n_layers+1: 
    204203                            break 
    205204                        # shift half sub thickness for the first point 
    206205                        z0 -= dz/2.0 
    207206                        z.append(z0) 
    208                         sld_i = self.params['sld_flat%s'% str(i)] 
     207                        sld_i = self.params['sld_flat%s' % str(i)] 
    209208                        beta.append(sld_i) 
    210                         dz = self.params['thick_flat%s'% str(i)] 
     209                        dz = self.params['thick_flat%s' % str(i)] 
    211210                        z0 += dz 
    212211                    else: 
    213                         dz = self.params['thick_inter%s'% str(i-1)]/n_sub 
    214                         nu = fabs(self.params['nu_inter%s'% str(i-1)]) 
     212                        dz = self.params['thick_inter%s' % str(i-1)]/n_sub 
     213                        nu = fabs(self.params['nu_inter%s' % str(i-1)]) 
    215214                        if n_s == 0: 
    216215                            # shift half sub thickness for the point 
    … …  
    220219                            sld_l = self.params['sld_bottom0'] 
    221220                        else: 
    222                             sld_l = self.params['sld_flat%s'% str(i-1)] 
     221                            sld_l = self.params['sld_flat%s' % str(i-1)] 
    223222                        if i == self.n_layers+1: 
    224223                            sld_r = self.params['sld_medium'] 
    225224                        else: 
    226                             sld_r = self.params['sld_flat%s'% str(i)] 
     225                            sld_r = self.params['sld_flat%s' % str(i)] 
    227226                        if sld_r == sld_l: 
    228227                            sld_i = sld_r 
    229228                        else: 
    230                             func_idx = self.params['func_inter%s'% str(i-1)] 
     229                            func_idx = self.params['func_inter%s' % str(i-1)] 
    231230                            # calculate the sld 
    232231                            sld_i = self._get_sld(func_idx, n_sub, n_s+0.5, nu, 
    … …  
    235234                    z.append(z0) 
    236235                    beta.append(sld_i) 
    237                     if j==1: break 
    238                     else: z0 += dz 
     236                    if j == 1:  
     237                        break 
     238                    else:  
     239                        z0 += dz 
    239240        # put substrate and superstrate profile 
    240241        z.append(z0) 
    … …  
    246247        z.append(z0+z_ext) 
    247248        beta.append(self.params['sld_medium'])  
    248         z.insert(0,-z_ext) 
    249         beta.insert(0,self.params['sld_bottom0'])  
     249        z.insert(0, -z_ext) 
     250        beta.insert(0, self.params['sld_bottom0'])  
    250251        # rearrange the profile for NR sld profile style 
    251252        z = [z0 - x for x in z] 
    … …  
    269270        sld_cal = SLDCalFunc() 
    270271        # set params 
    271         sld_cal.setParam('fun_type',func_idx) 
    272         sld_cal.setParam('npts_inter',n_sub) 
    273         sld_cal.setParam('shell_num',n_s) 
    274         sld_cal.setParam('nu_inter',nu) 
    275         sld_cal.setParam('sld_left',sld_l) 
    276         sld_cal.setParam('sld_right',sld_r) 
     272        sld_cal.setParam('fun_type', func_idx) 
     273        sld_cal.setParam('npts_inter', n_sub) 
     274        sld_cal.setParam('shell_num', n_s) 
     275        sld_cal.setParam('nu_inter', nu) 
     276        sld_cal.setParam('sld_left', sld_l) 
     277        sld_cal.setParam('sld_right', sld_r) 
    277278        # return sld value 
    278279        return sld_cal.run() 
    … …  
    289290         
    290291        ## setParam to model  
    291         if name=='sld_medium': 
     292        if name == 'sld_medium': 
    292293            # the sld_*** model.params not in params must set  
    293294            # to value of sld_solv 
    294295            for key in self.model.params.iterkeys(): 
    295296                if key not in self.params.keys()and key.split('_')[0] == 'sld': 
    296                         self.model.setParam(key, value)    
     297                    self.model.setParam(key, value)    
    297298         
    298299        self.model.setParam( name, value) 
    … …  
    342343    ## Now (May27,10) directly uses the model eval function  
    343344    ## instead of the for-loop in Base Component. 
    344     def evalDistribution(self, x = []): 
     345    def evalDistribution(self, x): 
    345346        """  
    346347        Evaluate the model in cartesian coordinates 
Note: See TracChangeset for help on using the changeset viewer.