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

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

There is a newer version: 20151002
Show newest version
/*
 * ===========================================
 * 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