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

com.idrsolutions.pdf.color.shading.FunctionShadeContext 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


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

import java.awt.Color;
import java.awt.PaintContext;
import java.awt.geom.AffineTransform;
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 FunctionShadeContext implements PaintContext {

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

    private final float[] background;
    private float[] domain = {0f, 1f, 0f, 1f};

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


    public FunctionShadeContext(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;

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

//		BBox = shadingObject.getFloatArray(PdfDictionary.BBox);

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

    }

    @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;
        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                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];

                if (x < domain[0] || x > domain[1] || y < domain[2] || y > domain[3]) {
                    continue;
                }

                final Color c = calculateColor(new float[]{x, y});

                if (c != null) {
                    final int base = (i * w + j) * 4;
                    data[base] = c.getRed();
                    data[base + 1] = c.getGreen();
                    data[base + 2] = c.getBlue();
                    data[base + 3] = c.getAlpha();
                }

            }
        }

        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, val);
        shadingColorSpace.setColor(colValues, colValues.length);
        col = (Color) shadingColorSpace.getColor();
        return col;
    }

//	public static GeneralPath transformBBox(float [][] matrix, float[] bbox){
//		
//		float[] p1 = {bbox[0],bbox[1]};
//		float[] p2 = {bbox[2],bbox[1]};
//		float[] p3 = {bbox[2],bbox[3]};
//		float[] p4 = {bbox[0],bbox[3]};
//		
//		p1 = Matrix.transformPoint(matrix, p1[0], p1[1]);
//		p2 = Matrix.transformPoint(matrix, p2[0], p2[1]);
//		p3 = Matrix.transformPoint(matrix, p3[0], p3[1]);
//		p4 = Matrix.transformPoint(matrix, p4[0], p4[1]);
//		
//		GeneralPath gp = new GeneralPath();
//		gp.moveTo(p1[0], p1[1]);
//		gp.lineTo(p2[0], p2[1]);
//		gp.lineTo(p3[0], p3[1]);
//		gp.lineTo(p4[0], p4[1]);
//		gp.closePath();
//		
//		return gp;
//	}


}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy