#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 import rl_config

from reportlab.lib.styles import PropertySet, ParagraphStyle, _baseFontName

from reportlab.lib import colors

from reportlab.lib.utils import fp_str, annotateException, IdentStr

from reportlab.lib.abag import ABag as CellFrame

from reportlab.pdfbase.pdfmetrics import stringWidth

from reportlab.platypus.doctemplate import Indenter

from reportlab.platypus.flowables import LIIndenter

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

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

class CellStyle(PropertySet):

    fontname = _baseFontName

    fontsize = 10

    leading = 12

    leftPadding = 6

    rightPadding = 6

    topPadding = 3

    bottomPadding = 3

    firstLineIndent = 0

    color = 'black'

    alignment = 'LEFT'

    background = 'white'

    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 = CellStyle()

        for name in dir(self):

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

        return result

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):

        return "TableStyle(\n%s\n) # end TableStyle" % "  \n".join(map(repr, self._cmds))

    def getCommands(self):

        return self._cmds

def _rowLen(x):

    return not isinstance(x,(tuple,list)) 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 isinstance(v,basestring):

            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,lim=None,FUZZ=rl_config._FUZZ):

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

    M = {}

    if not lim: lim = len(V0)   #in longtables the row calcs may be truncated

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

        if x0>=lim: continue

        x1 += 1

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

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

        X = [x for x in xrange(x0,x1) 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] = M.get(x,0)+v

    for x,v in M.iteritems():

        V[x] += v

class _ExpandedCellTuple(tuple):

    pass

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, normalizedData=0, cellStyles=None):

        self.ident = ident

        self.hAlign = hAlign or 'CENTER'

        self.vAlign = vAlign or 'MIDDLE'

        if not isinstance(data,(tuple,list)):

            raise ValueError("%s invalid data type" % self.identity())

        self._nrows = nrows = len(data)

        self._cellvalues = []

        _seqCW = isinstance(colWidths,(tuple,list))

        _seqRH = isinstance(rowHeights,(tuple,list))

        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

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

        if not (nrows and ncols):

            if emptyTableAction=='error':

                raise ValueError("%s must have at least a row and column" % self.identity())

            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:

                raise ValueError('%s bad emptyTableAction: "%s"' % (self.identity(),emptyTableAction))

            return

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

        if normalizedData:

            self._cellvalues = data

        else:

            self._cellvalues = data = self.normalizeData(data)

        if not _seqCW: colWidths = ncols*[colWidths]

        elif len(colWidths)!=ncols:

            if rl_config.allowShortTableRows and isinstance(colWidths,list):

                n = len(colWidths)

                if n<ncols:

                    colWidths[n:] = (ncols-n)*[colWidths[-1]]

                else:

                    colWidths = colWidths[:ncols]

            else:

                raise ValueError("%s data error - %d columns in data but %d in column widths" % (self.identity(),ncols, len(colWidths)))

        if not _seqRH: rowHeights = nrows*[rowHeights]

        elif len(rowHeights) != nrows:

            raise ValueError("%s data error - %d rows in data but %d in row heights" % (self.identity(),nrows, len(rowHeights)))

        for i,d in enumerate(data):

            n = len(d)

            if n!=ncols:

                if rl_config.allowShortTableRows and isinstance(d,list):

                    d[n:] = (ncols-n)*['']

                else:

                    raise ValueError("%s expected %d not %d columns in row %d!" % (self.identity(),ncols,n,i))

        self._rowHeights = self._argH = rowHeights

        self._colWidths = self._argW = colWidths

        if cellStyles is None:

            cellrows = []

            for i in xrange(nrows):

                cellcols = []

                for j in xrange(ncols):

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

                cellrows.append(cellcols)

            self._cellStyles = cellrows

        else:

            self._cellStyles = cellStyles

        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

        cv = pprint.pformat(cv)

        cv = cv.replace("\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 isinstance(stuff,unicode):

                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)

        rh = getattr(self, '_rowHeights', 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]

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

                        if not isinstance(v,(tuple,list)): 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 isinstance(v,basestring)) and 1 or 0

                        if maxLen: vx = vx[:maxLen]

                    if b: break

                if b: break

        if rh:  #find tallest row, it's of great interest'

            tallest = '(tallest row %d)' % int(max(rh))

        else:

            tallest = ''

        if vx:

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

        else:

            vx = '...'

        return "<%s@0x%8.8X %s rows x %s cols%s>%s" % (self.__class__.__name__, id(self), nr, nc, tallest, vx)

    def _cellListIter(self,C,aW,aH):

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

        for c in C:

            if getattr(c,'__split_only__',None):

                for d in c.splitOn(canv,aW,aH):

                    yield d

            else:

                yield c

    def _cellListProcess(self,C,aW,aH):

        if not isinstance(C,_ExpandedCellTuple):

            frame = None

            R = [].append

            for c in self._cellListIter(C,aW,aH):

                if isinstance(c,Indenter):

                    if not frame:

                        frame = CellFrame(_leftExtraIndent=0,_rightExtraIndent=0)

                    c.frameAction(frame)

                    if frame._leftExtraIndent<1e-8 and frame._rightExtraIndent<1e-8:

                        frame = None

                    continue

                if frame:

                    R(LIIndenter(c,leftIndent=frame._leftExtraIndent,rightIndent=frame._rightExtraIndent))

                else:

                    R(c)

            C = _ExpandedCellTuple(R.__self__)

        return C

    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[:]

            V = self._cellvalues

            S = self._cellStyles

            while None in W:

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

                w = 0

                for i,Vi in enumerate(V):

                    v = Vi[j]

                    s = S[i][j]

                    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:

                            raise ValueError("Flowable %s in cell(%d,%d) can't have auto width\n%s" % (v.identity(30),i,j,self.identity(30)))

                        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

                W[j] = w

            if spanCons:

                try:

                    spanFixDim(W0,W,spanCons)

                except:

                    annotateException('\nspanning problem in %s\nW0=%r W=%r\nspanCons=%r' % (self.identity(),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