/thirdparties-extension/org.apache.poi.xwpf.converter.core/src/main/java/org/apache/poi/xwpf/converter/core/styles/TableInfo.java

/**

 * Copyright (C) 2011-2012 The XDocReport Team <xdocreport@googlegroups.com>

 *

 * All rights reserved.

 *

 * Permission is hereby granted, free  of charge, to any person obtaining

 * a  copy  of this  software  and  associated  documentation files  (the

 * "Software"), to  deal in  the Software without  restriction, including

 * without limitation  the rights to  use, copy, modify,  merge, publish,

 * distribute,  sublicense, and/or sell  copies of  the Software,  and to

 * permit persons to whom the Software  is furnished to do so, subject to

 * the following conditions:

 *

 * The  above  copyright  notice  and  this permission  notice  shall  be

 * included in all copies or substantial portions of the Software.

 *

 * THE  SOFTWARE IS  PROVIDED  "AS  IS", WITHOUT  WARRANTY  OF ANY  KIND,

 * EXPRESS OR  IMPLIED, INCLUDING  BUT NOT LIMITED  TO THE  WARRANTIES OF

 * MERCHANTABILITY,    FITNESS    FOR    A   PARTICULAR    PURPOSE    AND

 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE

 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION

 * OF CONTRACT, TORT OR OTHERWISE,  ARISING FROM, OUT OF OR IN CONNECTION

 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

 */

package org.apache.poi.xwpf.converter.core.styles;



import java.math.BigInteger;

import java.util.HashMap;

import java.util.List;

import java.util.Map;



import org.apache.poi.xwpf.converter.core.utils.XWPFTableUtil;

import org.apache.poi.xwpf.usermodel.XWPFTable;

import org.apache.poi.xwpf.usermodel.XWPFTableCell;

import org.apache.poi.xwpf.usermodel.XWPFTableRow;

import org.apache.xmlbeans.XmlCursor;

import org.apache.xmlbeans.XmlObject;

import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTRow;

import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTSdtCell;

import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTc;



public class TableInfo

{



    private final XWPFTable table;



    private final XWPFStylesDocument stylesDocument;



    private final Map<XWPFTableCell, TableCellInfo> cellInfos;



    private Integer nbColumns;



    private final boolean canApplyFirstRow;



    private final boolean canApplyLastRow;



    private boolean canApplyFirstCol;



    private final boolean canApplyLastCol;



    public TableInfo( XWPFTable table, XWPFStylesDocument stylesDocument )

    {

        this.table = table;

        this.cellInfos = new HashMap<XWPFTableCell, TableCellInfo>();

        this.stylesDocument = stylesDocument;

        this.nbColumns = null;



        // Compute value apply lastRow etc from the tblLook

        int tblLookVal = XWPFTableUtil.getTblLookVal( table );

        this.canApplyFirstRow = XWPFTableUtil.canApplyFirstRow( tblLookVal );

        this.canApplyLastRow = XWPFTableUtil.canApplyLastRow( tblLookVal );

        this.canApplyFirstCol = XWPFTableUtil.canApplyFirstCol( tblLookVal );

        this.canApplyLastCol = XWPFTableUtil.canApplyLastCol( tblLookVal );



    }



    public TableCellInfo getCellInfo( XWPFTableCell cell )

    {

        TableCellInfo cellInfo = cellInfos.get( cell );

        if ( cellInfo != null )

        {

            return cellInfo;

        }

        // Compute it

        computeCellInfos( cell.getTableRow() );

        return cellInfos.get( cell );



    }



    private void computeCellInfos( XWPFTableRow row )

    {

        if ( nbColumns == null )

        {

            nbColumns = XWPFTableUtil.computeColWidths( row.getTable() ).length;

        }



        int rowIndex = table.getRows().indexOf( row );

        boolean firstRow = rowIndex == 0;

        boolean lastRow = rowIndex == ( table.getRows().size() - 1 );



        boolean firstCol = true;

        boolean lastCol = false;

        int cellIndex = 0;

        CTRow ctRow = row.getCtRow();

        XmlCursor c = ctRow.newCursor();

        c.selectPath( "./*" );

        while ( c.toNextSelection() )

        {

            XmlObject o = c.getObject();

            if ( o instanceof CTTc )

            {

                CTTc tc = (CTTc) o;

                XWPFTableCell cell = row.getTableCell( tc );

                cellIndex = getCellIndex( cellIndex, cell );

                lastCol = ( cellIndex == nbColumns - 1 );

                addCellInfo( cell, firstRow, lastRow, firstCol, lastCol );

                firstCol = false;

            }

            else if ( o instanceof CTSdtCell )

            {

                // Fix bug of POI

                CTSdtCell sdtCell = (CTSdtCell) o;

                List<CTTc> tcList = sdtCell.getSdtContent().getTcList();

                for ( CTTc ctTc : tcList )

                {

                    XWPFTableCell cell = new XWPFTableCell( ctTc, row, row.getTable().getBody() );

                    cellIndex = getCellIndex( cellIndex, cell );

                    lastCol = ( cellIndex == nbColumns - 1 );

                    addCellInfo( cell, firstRow, lastRow, firstCol, lastCol );

                    firstCol = false;

                }

            }

        }

        c.dispose();

    }



    public void addCellInfo( XWPFTableCell cell, boolean firstRow, boolean lastRow, boolean firstCol, boolean lastCol )

    {

        TableCellInfo cellInfo = new TableCellInfo( this, firstRow, lastRow, firstCol, lastCol );

        cellInfos.put( cell, cellInfo );

    }



    private int getCellIndex( int cellIndex, XWPFTableCell cell )

    {

        BigInteger gridSpan = stylesDocument.getTableCellGridSpan( cell.getCTTc().getTcPr() );

        if ( gridSpan != null )

        {

            cellIndex = cellIndex + gridSpan.intValue();

        }

        else

        {

            cellIndex++;

        }

        return cellIndex;

    }



    public boolean canApplyFirstRow()

    {

        return canApplyFirstRow;

    }



    public boolean canApplyLastRow()

    {

        return canApplyLastRow;

    }



    public boolean canApplyFirstCol()

    {

        return canApplyFirstCol;

    }



    public boolean canApplyLastCol()

    {

        return canApplyLastCol;

    }

}