Changeset 580ee7a in sasview for calculator/instrument.py

Timestamp:

Aug 5, 2011 12:18:09 PM (13 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:

148f0dd

Parents:

1bf1bc0

Message:

extended rescal to tof

File:

• calculator/instrument.py (modified) (8 diffs)

calculator/instrument.py

@@ -17 +17 @@
 
 _MIN = 0.0
-_MAX = 30.0
+_MAX = 50.0
 _INTENSITY = 368428
 _WAVE_LENGTH = 6.0
 _WAVE_SPREAD = 0.125
 _MASS = 1.67492729E-24 #[gr]
-X_VAL = numpy.linspace(0, 30, 310)
-_LAMBDA_ARRAY = [[0, 5, 10, 15, 20, 25, 30], [1, 1, 1, 1, 1, 1, 1]]
+_LAMBDA_ARRAY = [[0, 1e+16],[_INTENSITY, _INTENSITY]]
 
 class Aperture(object):
@@ -152 +151 @@
             self.distance = distance
             validate(distance[0])
-    
-        
+
+            
 class Neutron(object):
     """
@@ -160 +159 @@
     def __init__(self):
         
-        # intensity in counts/sec
-        self.intensity = _INTENSITY
         # neutron mass in cgs unit
         self.mass = _MASS
-        # wavelength spectrum 
-        self.spectrum = []
+
         # wavelength
         self.wavelength = _WAVE_LENGTH
         # wavelength spread (FWHM)
         self.wavelength_spread = _WAVE_SPREAD
-        # mean wavelength
-        self.mean = _WAVE_LENGTH
-        # wavelength = distribution after velocity selector
-        self.peak = []
-        # std of of the spectrum
-        self.std = None
+        # wavelength spectrum 
+        self.spectrum = self.get_default_spectrum()        
+        # intensity in counts/sec
+        self.intensity = numpy.interp(self.wavelength, 
+                                      self.spectrum[0],
+                                      self.spectrum[1],
+                                      0.0,
+                                      0.0)
         # min max range of the spectrum
-        self.min = _MIN
-        self.max = _MAX
-        # x-range of the spectrum
-        self.x_val = X_VAL
-        # default distribution function
-        # ex., 'gaussian', or an array
-        self.func = _LAMBDA_ARRAY
+        self.min = min(self.spectrum[0])
+        self.max = max(self.spectrum[0])
+        # wavelength band
+        self.band = [self.min, self.max]
+
         # default unit of the thickness
         self.wavelength_unit = 'A'
-        
+    
+    def set_full_band(self):
+        """
+        set band to default value
+        """
+        self.band = self.spectrum
+    def set_spectrum(self, spectrum):  
+        """
+        Set spectrum
+        
+        :param spectrum: numpy array
+        """
+        self.spectrum = spectrum 
+        self.setup_spectrum()
+         
+    def setup_spectrum(self):
+        """
+        To set the wavelength spectrum, and intensity, assumes 
+        wavelength is already within the spectrum
+        """
+        spectrum = self.spectrum
+        intensity = numpy.interp(self.wavelength, 
+                                      spectrum[0],
+                                      spectrum[1],
+                                      0.0,
+                                      0.0)
+        self.set_intensity(intensity)
+        # min max range of the spectrum
+        self.min = min(self.spectrum[0])
+        self.max = max(self.spectrum[0])
+        # set default band
+        self.set_band([self.min,self.max])
+        
+    def set_band(self, band=[]):
+        """
+        To set the wavelength band
+        
+        :param band: array of [min, max]
+        """
+        # check if the wavelength is in range
+        if min(band) < self.min or\
+                max(band) > self.max:
+            raise 
+        self.band = band
+
+         
     def set_intensity(self, intensity = 368428):
         """
@@ -198 +239 @@
         Sets the wavelength
         """
+        # check if the wavelength is in range
+        if wavelength < min(self.band) or\
+                wavelength > max(self.band):
+            raise 
         self.wavelength = wavelength
         validate(wavelength)
+        self.intensity = numpy.interp(self.wavelength, 
+                                  self.spectrum[0],
+                                  self.spectrum[1],
+                                  0.0,
+                                  0.0)
+
 
     def set_mass(self, mass = _MASS):
@@ -245 +296 @@
         """
         return self.wavelength
-    
-    def _set_mean(self):
-        """
-        To get mean value of wavelength
-        """
-        mean_value = numpy.mean(self.peak[0]*
-                                self.peak[1])
-        self.mean = mean_value
-
-        
-    def get_mean_peakvalue(self):
-        """
-        To get mean value of wavelength
-        """
-        mean_value = numpy.mean(self.peak[1])
-        return mean_value
             
     def get_spectrum(self):
@@ -267 +302 @@
         """
         return self.spectrum
+    def get_default_spectrum(self):
+        """
+        get default spectrum
+        """
+        return numpy.array(_LAMBDA_ARRAY)
+    def get_band(self):
+        """
+        To get the wavelength band
+        """
+        return self.band 
     
     def plot_spectrum(self):
@@ -275 +320 @@
         try:
             import matplotlib.pyplot as plt
-            plt.plot(self.x_val, self.spectrum, linewidth = 2, color = 'r')
+            plt.plot(self.spectrum[0], self.spectrum[1], linewidth = 2, color = 'r')
             plt.legend(['Spectrum'], loc = 'best')
             plt.show()
@@ -281 +326 @@
             raise RuntimeError, "Can't import matplotlib required to plot..."
         
-    def plot_peak(self):
-        """
-        To plot the wavelength peak spactrum
-        : requirment: matplotlib.pyplot
-        """
-        try:
-            min = self.mean * (1 - self.wavelength_spread)
-            max = self.mean * (1 + self.wavelength_spread)
-            x_val =   numpy.linspace(min, max, 310)
-            import matplotlib.pyplot as plt
-            plt.plot(x_val, self.peak, linewidth = 2, color = 'r')
-            plt.legend(['peak'], loc = 'best')
-            plt.show()
-        except:
-            raise RuntimeError, "Can't import matplotlib required to plot..."
-
+    
+class TOF(Neutron):
+    """
+    TOF: make list of wavelength and wave length spreads
+    """
+    def __init__(self):
+        """
+        Init
+        """
+        Neutron.__init__(self)
+        #self.switch = switch
+        self.wavelength_list = [self.wavelength]
+        self.wavelength_spread_list = [self.wavelength_spread]
+        self.intensity_list = self.get_intensity_list()
+    
+    def get_intensity_list(self):
+        """
+        get list of the intensity wrt wavelength_list
+        """
+        out = numpy.interp(self.wavelength_list, 
+                                      self.spectrum[0],
+                                      self.spectrum[1],
+                                      0.0,
+                                      0.0)
+        return out
+    
+    def get_wave_list(self):
+        """
+        Get wavelength and wavelength_spread list
+        """
+        return self.wavelength_list, self.wavelengthspread_list
+    
+    def set_wave_list(self, wavelength=[]):
+        """
+        Set wavelength list
+        
+        :param wavelength: list of wavelengths
+        """
+        self.wavelength_list = wavelength
+        
+    def set_wave_spread_list(self, wavelength_spread=[]):
+        """
+        Set wavelength_spread list
+        
+        :param wavelength_spread: list of wavelength spreads
+        """
+        self.wavelengthspread_list = wavelength_spread
+        
+        
 def validate(value = None):
     """