python.TurnDataReader.TurnDataReader Class Reference

Collaboration diagram for python.TurnDataReader.TurnDataReader:

Public Member Functions
	__init__ (self, infile=None)
	readData (self)

Public Attributes
bool	verbose = False
	infile = infile
	data = None

Detailed Description

Definition at line 15 of file TurnDataReader.py.

Constructor & Destructor Documentation

__init__()

python.TurnDataReader.TurnDataReader.__init__	(		self,
			infile = None )

Definition at line 17 of file TurnDataReader.py.

    def __init__(self, infile=None):
        
        self.verbose = False
        
        # File to read data from
        self.infile = infile
        
        # dict keyed by pseudoLB number containing dict by algorithm with data Tuple
        # (turns, counts) = self.data[lb][algName] 
        self.data = None
 

Member Function Documentation

readData()

python.TurnDataReader.TurnDataReader.readData

(

self

)

Definition at line 28 of file TurnDataReader.py.

    def readData(self):
 
        self.data = None
        if self.infile is None:
            print('TurnDataReader.readData() called with infile == None!')
            return self.data
 
        # Open input file
        try:
            f = open(self.infile, 'r')
        except IOError as e:
            print('Error opening file', self.infile)
            print(e)
            return self.data
 
        # Initialize our new dictionary
        self.data = dict()
        
        # Must parse lines looking for either LB records or algorithm type
        
        # pseudoLB record (number followed by two text records
        matchlb = re.compile(r'^([0-9]+) (\S+) (\S+)$')
        
        # algorithm name (only thing on the line)
        matchalg = re.compile(r'^(\S+)$')
 
        # Current algorithm read in file
        algId = None
        
        for line in f.readlines():
 
            if self.verbose: print(line, end=' ')
 
            m = matchlb.search(line)
            if m:
                if self.verbose: print('Found data record match:', m.group(1), m.group(2), m.group(3))
                lb = int(m.group(1))
                # Hack to realign these properly.  This will probably bite us in the ass someday
                lb -= 1
                
                if m.group(2) == 'nan':
                    counts = 0.
                else:
                    counts = float(m.group(2))
                    
                if m.group(3) == 'nan':
                    turns = 0.
                else:
                    turns = float(m.group(3))
 
                # Do we have a valid algorithm?
                if algId is None:
                    print('TurnDataReader.readData() found data record with no algorithm defined in', self.infile)
                    print(line)
                    sys.exit()
                
                # save data
                if lb not in self.data:
                    self.data[lb] = dict()
 
                self.data[lb][algId] = (turns, counts)
 
                if self.verbose:
                    print('Alg:', algId, 'LB:', lb, 'Turns:', turns, 'Counts:', counts)
                continue
            
            m = matchalg.search(line)
            if m:
                if self.verbose: print('Found new algorithm:', m.group(1))
 
                # Decode Nitesh's into my values, just hardcode here
                algId = m.group(1)
                if algId is None:
                    print('TurnDataReader.readData() found unrecognized algorithm name in', self.infile)
                    print(line)
                    sys.exit()
 
        f.close()
        return self.data
    
# Test by running from command line

Member Data Documentation

data

python.TurnDataReader.TurnDataReader.data = None

Definition at line 26 of file TurnDataReader.py.

infile

python.TurnDataReader.TurnDataReader.infile = infile

Definition at line 22 of file TurnDataReader.py.

verbose

bool python.TurnDataReader.TurnDataReader.verbose = False

Definition at line 19 of file TurnDataReader.py.

---

The documentation for this class was generated from the following file:

• TurnDataReader.py