Dépôt officiel du code source de l'ERP OpenConcerto

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 * 
 * Copyright 2011 OpenConcerto, by ILM Informatique. All rights reserved.
 * 
 * The contents of this file are subject to the terms of the GNU General Public License Version 3
 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a
 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific
 * language governing permissions and limitations under the License.
 * 
 * When distributing the software, include this License Header Notice in each file.
 */
 
 package org.openconcerto.erp.generationDoc;

import org.openconcerto.erp.core.edm.Attachment;
import org.openconcerto.erp.core.edm.AttachmentUtils;
import org.openconcerto.sql.model.SQLRow;
import org.openconcerto.sql.model.SQLRowAccessor;
import org.openconcerto.sql.model.SQLRowValues;
import org.openconcerto.sql.model.SQLRowValuesListFetcher;
import org.openconcerto.sql.model.SQLTable;
import org.openconcerto.sql.model.Where;
import org.openconcerto.sql.model.graph.Path;
import org.openconcerto.utils.FileUtils;
import org.openconcerto.utils.FillMode;
import org.openconcerto.utils.ImageUtils;
import org.openconcerto.utils.StringUtils;

import java.awt.Color;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.List;

import javax.imageio.ImageIO;

import org.jdom2.Element;

public class OOXMLTableImage {

    private float x = 0;
    private float y = 0;
    private final float width, height;
    private int dpi = 300;

    private final int rowCount;
    // #ffffff
    private String bgColor = "#ffffff";

    public final static String EDM_URL = "edm://miniature.png";
    // C://Users/Toto.jpg ou edm://miniature.png (prendre nom miniature)
    private final String url;
    // ID_AFFAIRE.ID_CLIENT
    private String fieldPathEDM;
    private String edmProvider;
    // top,center,bottom
    private String vAlign = "center";
    // left, center, right
    private String hAlign = "center";

    private OOXMLElement xmlField;
    private final SQLRowAccessor row;

    public OOXMLTableImage(OOXMLElement xmlField, Element e, SQLRowAccessor row) {
        this.xmlField = xmlField;
        this.row = row;
        this.url = e.getAttributeValue("url");
        this.fieldPathEDM = e.getAttributeValue("fieldPathEDM");
        this.edmProvider = e.getAttributeValue("edmProvider");
        this.width = Float.valueOf(e.getAttributeValue("width"));
        this.height = Float.valueOf(e.getAttributeValue("height"));
        this.rowCount = Integer.valueOf(e.getAttributeValue("rowCount"));
        if (e.getAttributeValue("x") != null && !e.getAttributeValue("x").isEmpty()) {
            this.x = Integer.valueOf(e.getAttributeValue("x"));
        }
        if (e.getAttributeValue("y") != null && !e.getAttributeValue("y").isEmpty()) {
            this.y = Integer.valueOf(e.getAttributeValue("y"));
        }
    }

    private byte[] img = null;
    private boolean cachedImg = false;

    public byte[] getImgBytes() throws IOException {

        if (this.cachedImg) {
            return this.img;
        } else {

            if (this.url.startsWith("edm://")) {
                Path p = new Path(this.row.getTable());
                final SQLRow r = this.row.asRow();
                final SQLRow rowToEdm;
                String sourceTableName = r.getTable().getName();
                int sourceID = r.getID();
                if (this.fieldPathEDM == null || this.fieldPathEDM.isEmpty() || this.fieldPathEDM.equalsIgnoreCase(".")) {
                    rowToEdm = r;
                } else if (this.fieldPathEDM != null && this.fieldPathEDM.equalsIgnoreCase("EDM_PROVIDER")) {
                    SpreadSheetCellValueProvider provider = SpreadSheetCellValueProviderManager.get(this.edmProvider);
                    rowToEdm = (SQLRow) provider.getValue(new SpreadSheetCellValueContext(r));
                    if (rowToEdm == null) {
                        return null;
                    } else {
                        sourceTableName = rowToEdm.getTable().getName();
                        sourceID = rowToEdm.getID();
                    }

                } else {
                    List<String> fields = StringUtils.fastSplit(this.fieldPathEDM, '.');
                    if (fields.size() == 1) {
                        sourceTableName = r.getTable().getForeignTable(fields.get(0)).getName();
                        sourceID = r.getForeignID(fields.get(0));
                    } else {
                        for (String f : fields) {
                            p = p.add(p.getLast().getField(f));
                        }
                        rowToEdm = r.getDistantRow(p);
                        sourceTableName = rowToEdm.getTable().getName();
                        sourceID = rowToEdm.getID();
                    }
                }
                final SQLTable tableAttachment = this.row.getTable().getTable("ATTACHMENT");
                SQLRowValues rowVals = new SQLRowValues(tableAttachment);
                rowVals.putNulls(tableAttachment.getFieldsName());

                SQLRowValuesListFetcher fetcher = SQLRowValuesListFetcher.create(rowVals);
                Where where = new Where(tableAttachment.getField("SOURCE_TABLE"), "=", sourceTableName);
                where = where.and(new Where(tableAttachment.getField("SOURCE_ID"), "=", sourceID));
                where = where.and(new Where(tableAttachment.getField("NAME"), "=", this.url.replaceAll("edm://", "")));
                final List<SQLRowValues> rAttachments = fetcher.fetch(where);
                if (rAttachments.isEmpty()) {
                    this.cachedImg = true;
                    return null;
                } else {
                    final SQLRowValues rowValsAttachment = rAttachments.get(0);
                    final Attachment a = new Attachment(rowValsAttachment);
                    final AttachmentUtils u = new AttachmentUtils();
                    final File f = u.getFile(a);
                    BufferedImage unscaledImage = ImageIO.read(f);

                    // 1 pouce = 25.4 mmm
                    float scale = this.dpi / 25.4f;
                    int maxWidth = (int) (this.width * scale);
                    int maxHeight = (int) (this.height * scale);
                    final BufferedImage im = ImageUtils.createQualityResizedImage(unscaledImage, maxWidth, maxHeight, false, new FillMode.ZoomOut(Color.WHITE), true);
                    ByteArrayOutputStream output = new ByteArrayOutputStream();
                    ImageIO.write(im, "PNG", output);
                    output.close();
                    this.img = output.toByteArray();
                    this.cachedImg = true;
                    return this.img;
                }

            } else

            {
                this.img = FileUtils.readBytes(new File(this.url));
                this.cachedImg = true;
                return this.img;
            }
        }
    }

    public String getBgColor() {
        return this.bgColor;
    }

    public int getDpi() {
        return this.dpi;
    }

    public float getX() {
        return this.x;
    }

    public float getY() {
        return this.y;
    }

    public float getWidth() {
        return this.width;
    }

    public float getHeight() {
        return this.height;
    }

    public int getRowCount() {
        return this.rowCount;
    }
}