#Copyright ReportLab Europe Ltd. 2000-2004

#see license.txt for license details

#history http://www.reportlab.co.uk/cgi-bin/viewcvs.cgi/public/reportlab/trunk/reportlab/platypus/tables.py

__version__=''' $Id$ '''

__doc__="""

Tables are created by passing the constructor a tuple of column widths, a tuple of row heights and the data in

row order. Drawing of the table can be controlled by using a TableStyle instance. This allows control of the

color and weight of the lines (if any), and the font, alignment and padding of the text.

None values in the sequence of row heights or column widths, mean that the corresponding rows

or columns should be automatically sized.

All the cell values should be convertible to strings; embedded newline '\\n' characters

cause the value to wrap (ie are like a traditional linefeed).

See the test output from running this module as a script for a discussion of the method for constructing

tables and table styles.

"""

from reportlab.platypus.flowables import Flowable, Preformatted

from reportlab import rl_config

from reportlab.lib.styles import PropertySet, ParagraphStyle

from reportlab.lib import colors

from reportlab.lib.utils import fp_str

from reportlab.pdfbase.pdfmetrics import stringWidth

import operator, string

from types import TupleType, ListType, StringType, FloatType, IntType

class CellStyle(PropertySet):

    defaults = {

        'fontname':'Times-Roman',

        'fontsize':10,

        'leading':12,

        'leftPadding':6,

        'rightPadding':6,

        'topPadding':3,

        'bottomPadding':3,

        'firstLineIndent':0,

        'color':colors.black,

        'alignment': 'LEFT',

        'background': (1,1,1),

        'valign': 'BOTTOM',

        'href': None,

        'destination':None,

        }

LINECAPS={None: None, 'butt':0,'round':1,'projecting':2,'squared':2}

LINEJOINS={None: None, 'miter':0, 'mitre':0, 'round':1,'bevel':2}

# experimental replacement

class CellStyle1(PropertySet):

    fontname = "Times-Roman"

    fontsize = 10

    leading = 12

    leftPadding = 6

    rightPadding = 6

    topPadding = 3

    bottomPadding = 3

    firstLineIndent = 0

    color = colors.black

    alignment = 'LEFT'

    background = (1,1,1)

    valign = "BOTTOM"

    href = None

    destination = None

    def __init__(self, name, parent=None):

        self.name = name

        if parent is not None:

            parent.copy(self)

    def copy(self, result=None):

        if result is None:

            result = CellStyle1()

        for name in dir(self):

            setattr(result, name, getattr(self, name))

        return result

CellStyle = CellStyle1

class TableStyle:

    def __init__(self, cmds=None, parent=None, **kw):

        #handle inheritance from parent first.

        commands = []

        if parent:

            # copy the parents list at construction time

            commands = commands + parent.getCommands()

            self._opts = parent._opts

            for a in ('spaceBefore','spaceAfter'):

                if hasattr(parent,a):

                    setattr(self,a,getattr(parent,a))

        if cmds:

            commands = commands + list(cmds)

        self._cmds = commands

        self._opts={}

        self._opts.update(kw)

    def add(self, *cmd):

        self._cmds.append(cmd)

    def __repr__(self):

        L = map(repr, self._cmds)

        import string

        L = string.join(L, "  \n")

        return "TableStyle(\n%s\n) # end TableStyle" % L

    def getCommands(self):

        return self._cmds

TableStyleType = type(TableStyle())

_SeqTypes = (TupleType, ListType)

def _rowLen(x):

    return type(x) not in _SeqTypes and 1 or len(x)

def _calc_pc(V,avail):

    '''check list V for percentage or * values

    1) absolute values go through unchanged

    2) percentages are used as weights for unconsumed space

    3) if no None values were seen '*' weights are

    set equally with unclaimed space

    otherwise * weights are assigned as None'''

    R = []

    r = R.append

    I = []

    i = I.append

    J = []

    j = J.append

    s = avail

    w = n = 0.

    for v in V:

        if type(v) is type(""):

            v = v.strip()

            if not v:

                v = None

                n += 1

            elif v.endswith('%'):

                v = float(v[:-1])

                w += v

                i(len(R))

            elif v=='*':

                j(len(R))

            else:

                v = float(v)

                s -= v

        elif v is None:

            n += 1

        else:

            s -= v

        r(v)

    s = max(0.,s)

    f = s/max(100.,w)

    for i in I:

        R[i] *= f

        s -= R[i]

    s = max(0.,s)

    m = len(J)

    if m:

        v =  n==0 and s/m or None

        for j in J:

            R[j] = v

    return R

def _hLine(canvLine, scp, ecp, y, hBlocks, FUZZ=rl_config._FUZZ):

    '''

    Draw horizontal lines; do not draw through regions specified in hBlocks

    This also serves for vertical lines with a suitable canvLine

    '''

    if hBlocks: hBlocks = hBlocks.get(y,None)

    if not hBlocks or scp>=hBlocks[-1][1]-FUZZ or ecp<=hBlocks[0][0]+FUZZ:

        canvLine(scp,y,ecp,y)

    else:

        i = 0

        n = len(hBlocks)

        while scp<ecp-FUZZ and i<n:

            x0, x1 = hBlocks[i]

            if x1<=scp+FUZZ or x0>=ecp-FUZZ:

                i += 1

                continue

            i0 = max(scp,x0)

            i1 = min(ecp,x1)

            if i0>scp: canvLine(scp,y,i0,y)

            scp = i1

        if scp<ecp-FUZZ: canvLine(scp,y,ecp,y)

def _multiLine(scp,ecp,y,canvLine,ws,count):

    offset = 0.5*(count-1)*ws

    y += offset

    for idx in xrange(count):

        canvLine(scp, y, ecp, y)

        y -= ws

def _convert2int(value, map, low, high, name, cmd):

    '''private converter tries map(value) low<=int(value)<=high or finally an error'''

    try:

        return map[value]

    except KeyError:

        try:

            ivalue = int(value)

            if low<=ivalue<=high: return ivalue

        except:

            pass

    raise ValueError('Bad %s value %s in %s'%(name,value,str(cmd)))

def _endswith(obj,s):

    try:

        return obj.endswith(s)

    except:

        return 0

def spanFixDim(V0,V,spanCons,FUZZ=rl_config._FUZZ):

    #assign required space to variable rows equally to existing calculated values

    M = {}

    for (x0,x1),v in spanCons.iteritems():

        t = sum([V[x]+M.get(x,0) for x in xrange(x0,x1+1)])

        if t>=v-FUZZ: continue      #already good enough

        X = [x for x in xrange(x0,x1+1) if V0[x] is None]   #variable candidates

        if not X: continue          #something wrong here mate

        v -= t

        v /= float(len(X))

        for x in X:

            M[x] = max(M.get(x,v),v)

    for x,v in M.iteritems():

        V[x] += v

class Table(Flowable):

    def __init__(self, data, colWidths=None, rowHeights=None, style=None,

                repeatRows=0, repeatCols=0, splitByRow=1, emptyTableAction=None, ident=None,

                hAlign=None,vAlign=None):

        self.ident = ident

        self.hAlign = hAlign or 'CENTER'

        self.vAlign = vAlign or 'MIDDLE'

        if type(data) not in _SeqTypes:

        self._nrows = nrows = len(data)

        self._cellvalues = []

        _seqCW = type(colWidths) in _SeqTypes

        _seqRH = type(rowHeights) in _SeqTypes

        if nrows: self._ncols = ncols = max(map(_rowLen,data))

        elif colWidths and _seqCW: ncols = len(colWidths)

        else: ncols = 0

        if not emptyTableAction: emptyTableAction = rl_config.emptyTableAction

        if not (nrows and ncols):

            if emptyTableAction=='error':

            elif emptyTableAction=='indicate':

                self.__class__ = Preformatted

                global _emptyTableStyle

                if '_emptyTableStyle' not in globals().keys():

                    _emptyTableStyle = ParagraphStyle('_emptyTableStyle')

                    _emptyTableStyle.textColor = colors.red

                    _emptyTableStyle.backColor = colors.yellow

                Preformatted.__init__(self,'%s(%d,%d)' % (self.__class__.__name__,nrows,ncols), _emptyTableStyle)

            elif emptyTableAction=='ignore':

                self.__class__ = Spacer

                Spacer.__init__(self,0,0)

            else:

            return

        # we need a cleanup pass to ensure data is strings - non-unicode and non-null

        self._cellvalues = self.normalizeData(data)

        if not _seqCW: colWidths = ncols*[colWidths]

        elif len(colWidths) != ncols:

        if not _seqRH: rowHeights = nrows*[rowHeights]

        elif len(rowHeights) != nrows:

        for i in xrange(nrows):

            if len(data[i]) != ncols:

        self._rowHeights = self._argH = rowHeights

        self._colWidths = self._argW = colWidths

        cellrows = []

        for i in xrange(nrows):

            cellcols = []

            for j in xrange(ncols):

                cellcols.append(CellStyle(`(i,j)`))

            cellrows.append(cellcols)

        self._cellStyles = cellrows

        self._bkgrndcmds = []

        self._linecmds = []

        self._spanCmds = []

        self._nosplitCmds = []

        self.repeatRows = repeatRows

        self.repeatCols = repeatCols

        self.splitByRow = splitByRow

        if style:

            self.setStyle(style)

    def __repr__(self):

        "incomplete, but better than nothing"

        r = getattr(self,'_rowHeights','[unknown]')

        c = getattr(self,'_colWidths','[unknown]')

        cv = getattr(self,'_cellvalues','[unknown]')

        import pprint, string

        cv = pprint.pformat(cv)

        cv = string.replace(cv, "\n", "\n  ")

        return "%s(\n rowHeights=%s,\n colWidths=%s,\n%s\n) # end table" % (self.__class__.__name__,r,c,cv)

    def normalizeData(self, data):

        """Takes a block of input data (list of lists etc.) and

        - coerces unicode strings to non-unicode UTF8

        - coerces nulls to ''

        -

        """

        def normCell(stuff):

            if stuff is None:

                return ''

            elif type(stuff) == type(u''):

                return stuff.encode('utf8')

            else:

                return stuff

        outData = []

        for row in data:

            outRow = [normCell(cell) for cell in row]

            outData.append(outRow)

        from pprint import pprint as pp

        #pp(outData)

        return outData

    def identity(self, maxLen=30):

        '''Identify our selves as well as possible'''

        if self.ident: return self.ident

        vx = None

        nr = getattr(self,'_nrows','unknown')

        nc = getattr(self,'_ncols','unknown')

        cv = getattr(self,'_cellvalues',None)

        if cv and 'unknown' not in (nr,nc):

            b = 0

            for i in xrange(nr):

                for j in xrange(nc):

                    v = cv[i][j]

                    t = type(v)

                    if t in _SeqTypes or isinstance(v,Flowable):

                        if not t in _SeqTypes: v = (v,)

                        r = ''

                        for vij in v:

                            r = vij.identity(maxLen)

                            if r and r[-4:]!='>...':

                                break

                        if r and r[-4:]!='>...':

                            ix, jx, vx, b = i, j, r, 1

                    else:

                        v = v is None and '' or str(v)

                        ix, jx, vx = i, j, v

                        b = (vx and t is StringType) and 1 or 0

                        if maxLen: vx = vx[:maxLen]

                    if b: break

                if b: break

        if vx:

            vx = ' with cell(%d,%d) containing\n%s' % (ix,jx,repr(vx))

        else:

            vx = '...'

        return "<%s at %d %d rows x %s cols>%s" % (self.__class__.__name__, id(self), nr, nc, vx)

    def _listCellGeom(self, V,w,s,W=None,H=None,aH=72000):

        if not V: return 0,0

        aW = w - s.leftPadding - s.rightPadding

        aH = aH - s.topPadding - s.bottomPadding

        t = 0

        w = 0

        canv = getattr(self,'canv',None)

        sb0 = None

        for v in V:

            vw, vh = v.wrapOn(canv, aW, aH)

            sb = v.getSpaceBefore()

            sa = v.getSpaceAfter()

            if W is not None: W.append(vw)

            if H is not None: H.append(vh)

            w = max(w,vw)

            t += vh + sa + sb

            if sb0 is None:

                sb0 = sb

        return w, t - sb0 - sa

    def _listValueWidth(self,V,aH=72000,aW=72000):

        if not V: return 0,0

        t = 0

        w = 0

        canv = getattr(self,'canv',None)

        return max([v.wrapOn(canv,aW,aH)[0] for v in V])

    def _calc_width(self,availWidth,W=None):

        if getattr(self,'_width_calculated_once',None): return

        #comments added by Andy to Robin's slightly terse variable names

        if not W: W = _calc_pc(self._argW,availWidth)   #widths array

        if None in W:  #some column widths are not given

            canv = getattr(self,'canv',None)

            saved = None

            if self._spanCmds:

                colSpanCells = self._colSpanCells

                spanRanges = self._spanRanges

            else:

                colSpanCells = ()

                spanRanges = {}

            spanCons = {}

            if W is self._argW:

                W0 = W

                W = W[:]

            else:

                W0 = W[:]

            while None in W:

                j = W.index(None) #find first unspecified column

                f = lambda x,j=j: operator.getitem(x,j)

                V = map(f,self._cellvalues)  #values for this column

                S = map(f,self._cellStyles)  #styles for this column

                w = 0

                i = 0

                for v, s in map(None, V, S):

                    ji = j,i

                    span = spanRanges.get(ji,None)

                    if ji in colSpanCells and not span: #if the current cell is part of a spanned region,

                        t = 0.0                         #assume a zero size.

                    else:#work out size

                        t = self._elementWidth(v,s)

                        if t is None:

                        t += s.leftPadding+s.rightPadding

                        if span:

                            c0 = span[0]

                            c1 = span[2]

                            if c0!=c1:

                                x = c0,c1

                                spanCons[x] = max(spanCons.get(x,t),t)

                                t = 0

                    if t>w: w = t   #record a new maximum

                    i += 1

                W[j] = w

            if spanCons:

                spanFixDim(W0,W,spanCons)

        self._colWidths = W

        width = 0

        self._colpositions = [0]        #index -1 is right side boundary; we skip when processing cells

        for w in W:

            width = width + w

            self._colpositions.append(width)

        self._width = width

        self._width_calculated_once = 1

    def _elementWidth(self,v,s):

        if isinstance(v,(list,tuple)):

            w = 0

            for e in v:

                ew = self._elementWidth(e,s)

                if ew is None: return None

                w = max(w,ew)

            return w

        elif isinstance(v,Flowable) and v._fixedWidth:

            if hasattr(v, 'width') and type(v.width) in (IntType,FloatType): return v.width

            if hasattr(v, 'drawWidth') and type(v.drawWidth) in (IntType,FloatType): return v.drawWidth

        # Even if something is fixedWidth, the attribute to check is not

        # necessarily consistent (cf. Image.drawWidth).  Therefore, we'll

        # be extra-careful and fall through to this code if necessary.

        if hasattr(v, 'minWidth'):

            try:

                w = v.minWidth() # should be all flowables

                if type(w) in (FloatType,IntType): return w

            except AttributeError:

                pass

        v = (v is not None and str(v) or '').split("\n")

        fontName = s.fontname

        fontSize = s.fontsize

        return max([stringWidth(x,fontName,fontSize) for x in v])

    def _calc_height(self, availHeight, availWidth, H=None, W=None):

        H = self._argH

        if not W: W = _calc_pc(self._argW,availWidth)   #widths array

        hmax = lim = len(H)

        longTable = getattr(self,'_longTableOptimize',rl_config.longTableOptimize)

        if None in H:

            canv = getattr(self,'canv',None)

            saved = None

            #get a handy list of any cells which span rows. should be ignored for sizing

            if self._spanCmds:

                rowSpanCells = self._rowSpanCells

                colSpanCells = self._colSpanCells

                spanRanges = self._spanRanges