python.TurnDataReader.TurnDataReader Class Reference

Collaboration diagram for python.TurnDataReader.TurnDataReader:

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

Public Attributes
	verbose
	infile
	data

Detailed Description

Definition at line 16 of file TurnDataReader.py.

Constructor & Destructor Documentation

__init__()

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

Definition at line 18 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()

def python.TurnDataReader.TurnDataReader.readData

(

self

)

Definition at line 29 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

Definition at line 27 of file TurnDataReader.py.

infile

python.TurnDataReader.TurnDataReader.infile

Definition at line 23 of file TurnDataReader.py.

verbose

python.TurnDataReader.TurnDataReader.verbose

Definition at line 20 of file TurnDataReader.py.

---

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

• TurnDataReader.py