/* ====================================================================

   Licensed to the Apache Software Foundation (ASF) under one or more

   contributor license agreements.  See the NOTICE file distributed with

   this work for additional information regarding copyright ownership.

   The ASF licenses this file to You under the Apache License, Version 2.0

   (the "License"); you may not use this file except in compliance with

   the License.  You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software

   distributed under the License is distributed on an "AS IS" BASIS,

   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

   See the License for the specific language governing permissions and

   limitations under the License.

==================================================================== */

package org.apache.poi.xssf.streaming;

import org.apache.poi.ss.usermodel.Chart;

import org.apache.poi.ss.usermodel.ClientAnchor;

import org.apache.poi.ss.usermodel.Comment;

import org.apache.poi.ss.usermodel.Drawing;

import org.apache.poi.xssf.usermodel.XSSFDrawing;

import org.apache.poi.xssf.usermodel.XSSFPicture;

/**

 * Streaming version of Drawing.

 * Delegates most tasks to the non-streaming XSSF code.

 * TODO: Potentially, Comment and Chart need a similar streaming wrapper like Picture.

 * @author: Frank Plößel, Beyond Data GmbH

 */

public class SXSSFDrawing implements Drawing {

    private SXSSFWorkbook _wb;

    private XSSFDrawing _drawing;

    public SXSSFDrawing(SXSSFWorkbook workbook, XSSFDrawing drawing) {

        this._wb = workbook;

        this._drawing = drawing;

    }

    @Override

    public SXSSFPicture createPicture(ClientAnchor anchor, int pictureIndex) {

        XSSFPicture pict = _drawing.createPicture(anchor, pictureIndex);

        return new SXSSFPicture(_wb, pict);

    }

    @Override

    public Comment createCellComment(ClientAnchor anchor) {

        return _drawing.createCellComment(anchor);

    }

    @Override

    public Chart createChart(ClientAnchor anchor) {

        return _drawing.createChart(anchor);

    }

    @Override

    public ClientAnchor createAnchor(int dx1, int dy1, int dx2, int dy2, int col1, int row1, int col2, int row2) {

        return _drawing.createAnchor(dx1, dy1, dx2, dy2, col1, row1, col2, row2);

    }

}

native

/* ====================================================================

   Licensed to the Apache Software Foundation (ASF) under one or more

   contributor license agreements.  See the NOTICE file distributed with

   this work for additional information regarding copyright ownership.

   The ASF licenses this file to You under the Apache License, Version 2.0

   (the "License"); you may not use this file except in compliance with

   the License.  You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software

   distributed under the License is distributed on an "AS IS" BASIS,

   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

   See the License for the specific language governing permissions and

   limitations under the License.

==================================================================== */

package org.apache.poi.xssf.streaming;

import org.apache.poi.openxml4j.opc.PackagePart;

import org.apache.poi.ss.usermodel.Picture;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.usermodel.Workbook;

import org.apache.poi.ss.util.ImageUtils;

import org.apache.poi.util.Internal;

import org.apache.poi.util.POILogFactory;

import org.apache.poi.util.POILogger;

import org.apache.poi.xssf.usermodel.*;

import org.openxmlformats.schemas.drawingml.x2006.main.CTPositiveSize2D;

import org.openxmlformats.schemas.drawingml.x2006.main.CTShapeProperties;

import org.openxmlformats.schemas.drawingml.x2006.spreadsheetDrawing.CTPicture;

import org.openxmlformats.schemas.spreadsheetml.x2006.main.CTCol;

import java.awt.*;

import java.io.IOException;

/**

 * Streaming version of Picture.

 * Most of the code is a copy of the non-streaming XSSFPicture code.

 * This is necessary as a private method getRowHeightInPixels of that class needs to be changed, which is called by a method call chain nested several levels.

 *

 * The main change is to access the rows in the SXSSF sheet, not the always empty rows in the XSSF sheet when checking the row heights.

 *

 * @author: Frank Plößel, Beyond Data GmbH

 */

public final class SXSSFPicture implements Picture {

    private static final POILogger logger = POILogFactory.getLogger(SXSSFPicture.class);

    /**

     * Column width measured as the number of characters of the maximum digit width of the

     * numbers 0, 1, 2, ..., 9 as rendered in the normal style's font. There are 4 pixels of margin

     * padding (two on each side), plus 1 pixel padding for the gridlines.

     *

     * This value is the same for default font in Office 2007 (Calibry) and Office 2003 and earlier (Arial)

     */

    private static float DEFAULT_COLUMN_WIDTH = 9.140625f;

    private final SXSSFWorkbook _wb;

    private final XSSFPicture _picture;

    SXSSFPicture(SXSSFWorkbook _wb, XSSFPicture _picture) {

        this._wb = _wb;

        this._picture = _picture;

    }

    /**

     * Return the underlying CTPicture bean that holds all properties for this picture

     *

     * @return the underlying CTPicture bean

     */

    @Internal

    public CTPicture getCTPicture(){

        return _picture.getCTPicture();

    }

    /**

     * Reset the image to the original size.

     *

     * <p>

     * Please note, that this method works correctly only for workbooks

     * with the default font size (Calibri 11pt for .xlsx).

     * If the default font is changed the resized image can be streched vertically or horizontally.

     * </p>

     */

    public void resize(){

        resize(1.0);

    }

    /**

     * Reset the image to the original size.

     * <p>

     * Please note, that this method works correctly only for workbooks

     * with the default font size (Calibri 11pt for .xlsx).

     * If the default font is changed the resized image can be streched vertically or horizontally.

     * </p>

     *

     * @param scale the amount by which image dimensions are multiplied relative to the original size.

     * <code>resize(1.0)</code> sets the original size, <code>resize(0.5)</code> resize to 50% of the original,

     * <code>resize(2.0)</code> resizes to 200% of the original.

     */

    public void resize(double scale){

        XSSFClientAnchor anchor = getAnchor();

        XSSFClientAnchor pref = getPreferredSize(scale);

        int row2 = anchor.getRow1() + (pref.getRow2() - pref.getRow1());

        int col2 = anchor.getCol1() + (pref.getCol2() - pref.getCol1());

        anchor.setCol2(col2);

        anchor.setDx1(0);

        anchor.setDx2(pref.getDx2());

        anchor.setRow2(row2);

        anchor.setDy1(0);

        anchor.setDy2(pref.getDy2());

    }

    /**

     * Calculate the preferred size for this picture.

     *

     * @return XSSFClientAnchor with the preferred size for this image

     */

    public XSSFClientAnchor getPreferredSize(){

        return getPreferredSize(1);

    }

    /**

     * Calculate the preferred size for this picture.

     *

     * @param scale the amount by which image dimensions are multiplied relative to the original size.

     * @return XSSFClientAnchor with the preferred size for this image

     */

    public XSSFClientAnchor getPreferredSize(double scale){

        XSSFClientAnchor anchor = getAnchor();

        XSSFPictureData data = getPictureData();

        Dimension size = getImageDimension(data.getPackagePart(), data.getPictureType());

        double scaledWidth = size.getWidth() * scale;

        double scaledHeight = size.getHeight() * scale;

        float w = 0;

        int col2 = anchor.getCol1();

        int dx2 = 0;

        for (;;) {

            w += getColumnWidthInPixels(col2);

            if(w > scaledWidth) break;

            col2++;

        }

        if(w > scaledWidth) {

            double cw = getColumnWidthInPixels(col2 );

            double delta = w - scaledWidth;

            dx2 = (int)(XSSFShape.EMU_PER_PIXEL * (cw - delta));

        }

        anchor.setCol2(col2);

        anchor.setDx2(dx2);

        double h = 0;

        int row2 = anchor.getRow1();

        int dy2 = 0;

        for (;;) {

            h += getRowHeightInPixels(row2);

            if(h > scaledHeight) break;

            row2++;

        }

        if(h > scaledHeight) {

            double ch = getRowHeightInPixels(row2);

            double delta = h - scaledHeight;

            dy2 = (int)(XSSFShape.EMU_PER_PIXEL * (ch - delta));

        }

        anchor.setRow2(row2);

        anchor.setDy2(dy2);

        CTPositiveSize2D size2d =  getCTPicture().getSpPr().getXfrm().getExt();

        size2d.setCx((long)(scaledWidth * XSSFShape.EMU_PER_PIXEL));

        size2d.setCy((long)(scaledHeight * XSSFShape.EMU_PER_PIXEL));

        return anchor;

    }

    private float getColumnWidthInPixels(int columnIndex){

        XSSFSheet sheet = getParent();

        CTCol col = sheet.getColumnHelper().getColumn(columnIndex, false);

        double numChars = col == null || !col.isSetWidth() ? DEFAULT_COLUMN_WIDTH : col.getWidth();

        return (float)numChars*XSSFWorkbook.DEFAULT_CHARACTER_WIDTH;

    }

    private float getRowHeightInPixels(int rowIndex) {

        // THE FOLLOWING THREE LINES ARE THE MAIN CHANGE compared to the non-streaming version: use the SXSSF sheet,

		// not the XSSF sheet (which never contais rows when using SXSSF)

        XSSFSheet xssfSheet = getParent();

        SXSSFSheet sheet = _wb.getSXSSFSheet(xssfSheet);

        Row row = sheet.getRow(rowIndex);

        float height = row != null ?  row.getHeightInPoints() : sheet.getDefaultRowHeightInPoints();

        return height * XSSFShape.PIXEL_DPI / XSSFShape.POINT_DPI;

    }

    /**

     * Return the dimension of this image

     *

     * @param part the package part holding raw picture data

     * @param type type of the picture: {@link Workbook#PICTURE_TYPE_JPEG},

     * {@link Workbook#PICTURE_TYPE_PNG} or {@link Workbook#PICTURE_TYPE_DIB}

     *

     * @return image dimension in pixels

     */

    protected static Dimension getImageDimension(PackagePart part, int type){

        try {

            return ImageUtils.getImageDimension(part.getInputStream(), type);

        } catch (IOException e){

            //return a "singulariry" if ImageIO failed to read the image

            logger.log(POILogger.WARN, e);

            return new Dimension();

        }

    }

    /**

     * Return picture data for this shape

     *

     * @return picture data for this shape

     */

    public XSSFPictureData getPictureData() {

        return _picture.getPictureData();

    }

    protected CTShapeProperties getShapeProperties(){

        return getCTPicture().getSpPr();

    }

    private XSSFSheet getParent() {

        return (XSSFSheet)_picture.getDrawing().getParent();

    }

    private XSSFClientAnchor getAnchor() {

        return (XSSFClientAnchor)_picture.getAnchor();

    }

}

native