matrix.py

#  m a t r i x . p y
#
#  Support for 2d matrix that renders to an HTML table
#
#  Chris Meyers. 09/25/2013
#

dftTableAttr = 'cellspacing="0" cellpadding="10"'

class Matrix :
    def __init__ (self, nrows=1, ncols=1, data=None,
                  dftFormat="", dftStyle="", title="",
                  tableAttr=dftTableAttr, tableHeaders=None,
                  Expand=True) :
        self.nrows = nrows
        self.ncols = ncols
        self.values = {}
        self.expanded = Expand
        if Expand :
            # get attributes only on the main Matrix
            self.dftFormat    = dftFormat
            self.dftStyle     = dftStyle
            self.title        = title
            self.tableAttr    = tableAttr
            self.tableHeaders = tableHeaders
            self.format = Matrix(nrows, ncols, Expand=False)
            self.style  = Matrix(nrows, ncols, Expand=False)
        if data :
            if type(data) == type({}) : data=dictToLol(data)
            if type(data) == type([]) :
                self.populate(data)

    def __getitem__(self, coords) :
        row, col = coords
        return self.values.get((row,col))

    def __setitem__(self, coords, value) :
        row, col = coords
        self.values[(row,col)] = value
        self.nrows = max(self.nrows,row+1)
        self.ncols = max(self.ncols,col+1)
        if self.expanded :
            self.format.nrows = self.nrows
            self.format.ncols = self.ncols
            self.style.nrows  = self.nrows
            self.style.ncols  = self.ncols
        return value

#===========================================

    def setrowVals(self, row, values) :
        "set each column to a seperate value"
        col = 0
        for col in range(len(values)) :
            self.__setitem__((row,col),values[col])
            col += 1

    def setrowVal(self, row, value) :
        "set all columns to the same value"
        col = 0
        while col < self.ncols :
            self.__setitem__((row,col),value)
            col += 1

    def getrow (self, row) :
        vals = []
        for c in range(self.ncols) :
            vals.append(self.__getitem__( (row,c) ))
        return vals

#===========================================

    def setcolVals(self, col, values) :
        "set each row to a seperate value"
        row = 0
        for row in range(len(values)) :
            self.__setitem__((row,col),values[row])
            row += 1

    def setcolVal(self, col, value) :
        "set all rowumns to the same value"
        row = 0
        while row < self.nrows :
            self.__setitem__((row,col),value)
            row += 1

    def getcol (self, col) :
        vals = []
        for r in range(self.nrows) :
            vals.append(self.__getitem__( (r,col) ))
        return vals

#===========================================

    def populate(self, lists) :
        "Fill self from a list of lists"
        nRows = len(lists)
        nCols = max([len(l) for l in lists])
        for row in range(len(lists)) :
            vals = lists[row]
            if type(vals) != list : vals = [vals] # make sing col
            self.setrowVals(row, vals)

    def renderHtml(self,wrap=None) :
        lins = ["","<table %s>" % self.tableAttr]
        if self.title : lins[0] = "<div>%s</div>" % self.title
        headers = self.tableHeaders
        if headers :
            lins.append("<tr><th>"+"</th><th>".join(map(str,headers))+
                        "</th></tr>")
        for row in range(self.nrows) :
            rowLin = ["  <tr>"]
            vals = self.getrow(row)
            if self.format : formats = self.format.getrow(row)
            else           : formats = ['']*self.ncols
            if self.style  : styles  = self.style.getrow(row)
            else           : styles  = ['']*self.ncols
            for c in range(self.ncols) :
                val = vals[c]; style=styles[c]; format=formats[c]
                if val == None : val = ""
                if not format : format = self.dftFormat
                if format :
                    if type(format)==type("") : val = format % val
                    else                      : val = format(val)
                if not style : style = self.dftStyle
                if style : cell = '<td style="%s">%s</td>' % (style,val)
                else     : cell = '<td>%s</td>' % val
                if wrap and c>0 and c%wrap==0 : cell="</tr><tr>"+cell
                rowLin.append(cell)
            rowLin.append("</tr>")
            lins.append("".join(rowLin))
        lins.append("</table>")
        return "\n".join(lins)

    def __str__ (self) :
        return "Matrix-%dx%d" % (self.nrows,self.ncols)

#===========================================

typeSeq = (type([]), type((1,2)))

def dictToLol(dic) :
    "Convert dict to a list of lists"
    keys = dic.keys(); keys.sort()
    lists = []
    for key in keys :
        val = dic[key]
        if type(val) not in typeSeq : val = [val]
        lists.append([key]+list(val))
    return lists