source: sasview/src/sas/fit/Loader.py @ dca6188

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since dca6188 was 79492222, checked in by krzywon, 10 years ago

Changed the file and folder names to remove all SANS references.

  • Property mode set to 100644
File size: 2.6 KB
RevLine 
[7705306]1# class Loader  to load any king of file
[89f3b66]2#import wx
3#import string
4import numpy
[c4d6900]5
[7705306]6class Load:
7    """
[aa36f96]8    This class is loading values from given file or value giving by the user
[7705306]9    """
[c4d6900]10    def __init__(self, x=None, y=None, dx=None, dy=None):
[6fe5100]11        raise NotImplementedError("a code search shows that this code is not active, and you are not seeing this message")
[7705306]12        # variable to store loaded values
13        self.x = x
14        self.y = y
15        self.dx = dx
16        self.dy = dy
17        self.filename = None
18       
[89f3b66]19    def set_filename(self, path=None):
[7705306]20        """
[c4d6900]21        Store path into a variable.If the user doesn't give
22        a path as a parameter a pop-up
[aa36f96]23        window appears to select the file.
24       
25        :param path: the path given by the user
26       
[7705306]27        """
28        self.filename = path
29       
30    def get_filename(self):
31        """ return the file's path"""
32        return self.filename
33   
34    def set_values(self):
[aa36f96]35        """ Store the values loaded from file in local variables"""
[7705306]36        if not self.filename == None:
[89f3b66]37            input_f =  open(self.filename, 'r')
[7705306]38            buff = input_f.read()
39            lines = buff.split('\n')
[89f3b66]40            self.x = []
41            self.y = []
[7705306]42            self.dx = [] 
[89f3b66]43            self.dy = []
[7705306]44            for line in lines:
45                try:
46                    toks = line.split()
47                    x = float(toks[0])
48                    y = float(toks[1])
49                    dy = float(toks[2])
50                   
51                    self.x.append(x)
52                    self.y.append(y)
53                    self.dy.append(dy)
54                    self.dx = numpy.zeros(len(self.x))
55                except:
56                    print "READ ERROR", line
57            # Sanity check
58            if not len(self.x) == len(self.dx):
59                raise ValueError, "x and dx have different length"
60            if not len(self.y) == len(self.dy):
61                raise ValueError, "y and dy have different length"
62           
63           
64    def get_values(self):
[aa36f96]65        """ Return x, y, dx, dy"""
[89f3b66]66        return self.x, self.y, self.dx, self.dy
[7705306]67   
[89f3b66]68    def load_data(self, data):
[aa36f96]69        """ Return plottable"""
[7705306]70        #load data
71        data.x = self.x
72        data.y = self.y
73        data.dx = self.dx
[89f3b66]74        data.dy = self.dy
[7705306]75        #Load its View class
76        #plottable.reset_view()
77       
78   
79if __name__ == "__main__": 
[89f3b66]80    load = Load()
[7705306]81    load.set_filename("testdata_line.txt")
82    print load.get_filename() 
83    load.set_values()
84    print load.get_values()
85   
86           
Note: See TracBrowser for help on using the repository browser.