All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.idrsolutions.pdf.color.shading.AxialShadeContext Maven / Gradle / Ivy

/*
 * ===========================================
 * Java Pdf Extraction Decoding Access Library
 * ===========================================
 *
 * Project Info:  http://www.idrsolutions.com
 * Help section for developers at http://www.idrsolutions.com/support/
 *
 * (C) Copyright 1997-2017 IDRsolutions and Contributors.
 *
 * This file is part of JPedal/JPDF2HTML5
 *
     This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


 *
 * ---------------
 * AxialShadeContext.java
 * ---------------
 */
package com.idrsolutions.pdf.color.shading;

import java.awt.Color;
import java.awt.PaintContext;
import java.awt.geom.AffineTransform;
import java.awt.geom.GeneralPath;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;
import java.awt.image.ColorModel;
import java.awt.image.Raster;
import java.awt.image.WritableRaster;

import org.jpedal.color.GenericColorSpace;
import org.jpedal.function.PDFFunction;
import org.jpedal.objects.raw.PdfDictionary;
import org.jpedal.objects.raw.PdfObject;
import org.jpedal.utils.Matrix;

/**
 * @author suda
 */
public class AxialShadeContext implements PaintContext {

    private final GenericColorSpace shadingColorSpace;
    private final PDFFunction[] function;

    private final float[] background;
    private float[] domain = {0.0f, 1.0f};
    private boolean[] extension;
    final float t0, t1, x0, y0, x1, y1, deltaX, deltaY, multiXY;
    private Rectangle2D bboxRect;

    private final float[][] toUserSpace;
    private final float[][] toShadeSpace;

    public AxialShadeContext(final AffineTransform xform, final GenericColorSpace shadingColorSpace, final float[] background, final PdfObject shadingObject, final float[][] mm, final PDFFunction[] function) {

        this.shadingColorSpace = shadingColorSpace;
        this.function = function;

        final float[] newDomain = shadingObject.getFloatArray(PdfDictionary.Domain);
        if (newDomain != null) {
            domain = newDomain;
        }
        this.background = background;
        extension = shadingObject.getBooleanArray(PdfDictionary.Extend);
        if (extension == null) {
            extension = new boolean[]{false, false};
        }

        t0 = domain[0];
        t1 = domain[1];

        float[] bbox = shadingObject.getFloatArray(PdfDictionary.BBox);
        if (bbox != null) {
            GeneralPath gp = new GeneralPath();
            gp.moveTo(bbox[0], bbox[1]);
            gp.lineTo(bbox[2], bbox[1]);
            gp.lineTo(bbox[2], bbox[3]);
            gp.lineTo(bbox[0], bbox[3]);
            gp.lineTo(bbox[0], bbox[1]);
            gp.closePath();
            bboxRect = gp.getBounds2D();
        }

        float[][] caller = {{1, 0, 0}, {0, 1, 0}, {0, 0, 1}};
        float[][] shadeMatrix = {{1, 0, 0}, {0, 1, 0}, {0, 0, 1}};

        if (mm != null) {
            caller = mm;
        }

        final float[] inputs = shadingObject.getFloatArray(PdfDictionary.Matrix);
        if (inputs != null) {
            shadeMatrix = new float[][]{{inputs[0], inputs[1], 0}, {inputs[2], inputs[3], 0}, {inputs[4], inputs[5], 1}};
        }

        final float[][] shader = Matrix.concatenate(caller, shadeMatrix);

        final float[][] xformMatrix = {
                {(float) xform.getScaleX(), (float) xform.getShearX(), 0},
                {(float) xform.getShearY(), (float) xform.getScaleY(), 0},
                {(float) xform.getTranslateX(), (float) xform.getTranslateY(), 1}
        };
        toUserSpace = Matrix.inverse(xformMatrix);
        toShadeSpace = Matrix.inverse(shader);

        final float[] coords = shadingObject.getFloatArray(PdfDictionary.Coords);

        x0 = coords[0];
        y0 = coords[1];
        x1 = coords[2];
        y1 = coords[3];

        deltaX = (x1 - x0);
        deltaY = (y1 - y0);
        multiXY = deltaX * deltaX + deltaY * deltaY;

    }

    @Override
    public void dispose() {

    }

    @Override
    public ColorModel getColorModel() {
        return ColorModel.getRGBdefault();
    }

    @Override
    public Raster getRaster(final int startX, final int startY, final int w, final int h) {

        final int rastSize = (w * h * 4);
        final int[] data = new int[rastSize];

        if (background != null) {
            shadingColorSpace.setColor(background, 4);
            final Color c = (Color) shadingColorSpace.getColor();
            for (int i = 0; i < h; i++) {
                for (int j = 0; j < w; j++) {
                    final int base = (i * w + j) * 4;
                    data[base] = c.getRed();
                    data[base + 1] = c.getGreen();
                    data[base + 2] = c.getBlue();
                    data[base + 3] = 255;
                }
            }
        }

        float x, y, xp;
        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                boolean render = true;
                float[] src = {startX + j, startY + i};
                src = Matrix.transformPoint(toUserSpace, src[0], src[1]);
                src = Matrix.transformPoint(toShadeSpace, src[0], src[1]);

                x = src[0];
                y = src[1];
                float t = 0;
                xp = (deltaX * (x - x0) + deltaY * (y - y0)) / multiXY;

                if (xp >= 0 && xp <= 1) {
                    t = t0 + (t1 - t0) * xp;
                } else if (xp < 0 && extension[0]) {
                    t = t0;
                } else if (xp > 1 && extension[1]) {
                    t = t1;
                } else {
                    render = false;
                }

                if (bboxRect != null && render) {
                    render = bboxRect.contains(x, y);
                }
                if (render) {
                    final Color c = calculateColor(t);
                    final int base = (i * w + j) * 4;
                    data[base] = c.getRed();
                    data[base + 1] = c.getGreen();
                    data[base + 2] = c.getBlue();
                    data[base + 3] = 255;
                }
            }
        }

        final WritableRaster raster = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB).getRaster();
        raster.setPixels(0, 0, w, h, data);
        return raster;

    }

    private Color calculateColor(final float val) {
        final Color col;
        final float[] colValues = ShadingFactory.applyFunctions(function, new float[]{val});
        shadingColorSpace.setColor(colValues, colValues.length);
        col = (Color) shadingColorSpace.getColor();
        return col;
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy