Changeset 077207b in sasview for sanscalculator/src

Timestamp:

Jan 4, 2013 9:30:31 AM (12 years ago)

Author:

Jae Cho <jhjcho@…>

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:

d7b6913

Parents:

ab7e3a4

Message:

added pdb reader

File:

• sanscalculator/src/sans/calculator/sans_gen.py (modified) (5 diffs)

sanscalculator/src/sans/calculator/sans_gen.py

@@ -4 +4 @@
 from sans.models.BaseComponent import BaseComponent
 import sans.models.sans_extension.sld2i as mod
+from periodictable import formula
 import numpy
 import os
@@ -54 +55 @@
         self.params['scale']       = 1.0
         self.params['background']  = 0.0
-        self.params['Up_frac_i']     = 1.0
-        self.params['Up_frac_f']    = 1.0
+        self.params['Up_frac_i']     = 0.5
+        self.params['Up_frac_f']    = 0.5
         self.params['Up_theta']  = 0.0
         self.description='GenSAS'
@@ -440 +441 @@
     ## List of allowed extensions
     ext = ['.pdb', '.PDB']
-
+    
     def read(self, path):
         """
@@ -458 +459 @@
         sld_my = numpy.zeros(0)
         sld_mz = numpy.zeros(0)
-        pix_symbol = numpy.zeros(0)#numpy.array([])
+        pix_symbol = numpy.zeros(0)
         try:
             input_f = open(path, 'rb')
             buff = input_f.read()
-            #buff.replace(' ', '')
             lines = buff.split('\n')
             input_f.close()
             for line in lines:
                 try:
-                    toks = line.split()
+                    #toks = line.split()
                     # check if line starts with "ATOM"
-                    if toks[0].count('ATM') > 0 or toks[0] == 'ATOM':
+                    if line[0:6].strip().count('ATM') > 0 or \
+                                line[0:6].strip() == 'ATOM':
                         # define fields of interest
-                        atom_id = toks[1]
-                        atom_name = toks[2]
-                        if line[12] == '' or len(atom_name) == 4:
-                            atom_name = atom_name[0].upper() 
-                        else:
-                            atom_name = atom_name[0].upper() + \
-                                        atom_name[1].lower()
+                        atom_id = line[6:11].strip()
+                        atom_name = line[12:16].strip()
                         try:
-                            float(toks[3])
-                            ind = 3
+                            float(line[12])
+                            atom_name = atom_name[1].upper()
                         except:
-                            ind = 4
-                        res_name = toks[ind]
-                        #chain_name = toks[3].strip()
-                        #residue_id = toks[4].strip()
-                        try:
-                            float(toks[4])
-                        except:
-                            ind += 1
-                        ind += 1
-                        _pos_x = float(toks[ind])
-                        ind += 1
-                        _pos_y = float(toks[ind])
-                        ind += 1
-                        _pos_z = float(toks[ind])
+                            if len(atom_name) == 4:
+                                atom_name = atom_name[0].upper()
+                            elif line[12] != ' ': 
+                                atom_name = atom_name[0].upper() + \
+                                        atom_name[1].lower() 
+                            else:
+                                atom_name = atom_name[0].upper()
+                        #res_name = line[17:20].strip()
+                        #chain_name = line[21:22].strip()
+                        _pos_x = float(line[30:38].strip())
+                        _pos_y = float(line[38:46].strip())
+                        _pos_z = float(line[46:54].strip())
+                        
                         pos_x = numpy.append(pos_x, _pos_x)
                         pos_y = numpy.append(pos_y, _pos_y)
                         pos_z = numpy.append(pos_z, _pos_z)
-                        sld_n = numpy.append(sld_n, 7e-06)
+                        try:
+                            #from periodictable import nsf
+                            #val = nsf.neutron_sld(atom_name)
+                            val = formula(atom_name).atoms.keys()[0].neutron.b_c
+                            # sld in Ang unit (from fm)
+                            val = float(val)
+                            val *= 1.0e-5
+                            sld_n = numpy.append(sld_n, val)
+                        except:
+                            print "Error: set the sld of %s to zero"% atom_name
+                            sld_n = numpy.append(sld_n,  0.0)                     
                         sld_mx = numpy.append(sld_mx, 0)
                         sld_my = numpy.append(sld_my, 0)
@@ -883 +888 @@
             self.ynodes = None
             self.znodes = None
-        
+      
     def set_stepsize(self):
         """