trace_reader.py

import csv
from bioreq import BIOReq
import conf
import logger

class TraceReader:
    def __init__(self):
        self.contexts = {}
        self.reader = None
        self.__load()

    def __load(self):
        try:
            f = open(conf.path, "r")
        except IOError:
            logger.error("trace file not found: {}".format(conf.path))
            exit(1)

        self.reader = csv.reader(f, delimiter = ',')
        return True

    def __iter__(self):
        return self

    def __next__(self):
        while True:
            row = next(self.reader)
            if row[3] != 'I':
                continue
            req = BIOReq(row)
            if req.lba == 0:
                continue
            if req.ts < conf.ts_start or req.ts > conf.ts_end:
                continue
            if req.is_read is None:
                continue
            return req