org.pentaho.di.ui.util.ImageUtil.java Source code

Java tutorial

Introduction

Here is the source code for org.pentaho.di.ui.util.ImageUtil.java

Source

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2013 by Pentaho : http://www.pentaho.com
 *
 *******************************************************************************
 *
 * Licensed 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.
 *
 ******************************************************************************/

/*******************************************************************************
 * Copyright (c) 2000, 2004 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/

package org.pentaho.di.ui.util;

import java.awt.image.BufferedImage;
import java.awt.image.DirectColorModel;
import java.awt.image.IndexColorModel;
import java.awt.image.WritableRaster;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;

import org.apache.commons.vfs.FileObject;
import org.apache.commons.vfs.FileSystemException;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.graphics.ImageData;
import org.eclipse.swt.graphics.PaletteData;
import org.eclipse.swt.graphics.RGB;
import org.eclipse.swt.widgets.Display;
import org.pentaho.di.core.vfs.KettleVFS;

public class ImageUtil {
    private static FileObject base;

    static {
        try {
            base = KettleVFS.getInstance().getFileSystemManager().resolveFile(System.getProperty("user.dir"));
        } catch (FileSystemException e) {
            e.printStackTrace();
            base = null;
        }
    }

    public static Image makeImageTransparent(Display display, Image tempImage, RGB transparentColor) {
        ImageData imageData = tempImage.getImageData();
        int pixelIndex = imageData.palette.getPixel(transparentColor);
        imageData.transparentPixel = pixelIndex;
        Image image = new Image(display, imageData);
        tempImage.dispose();
        return image;
    }

    public static Image getImageAsResource(Display display, String location) {
        // assume the classloader for the active thread
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        URL res = cl.getResource(location);
        if (res != null) {
            try {
                java.io.InputStream s = res.openStream();
                if (s != null) {
                    return new Image(display, s);
                }
            } catch (IOException e) {
                // do nothing. just move on to trying to load via file system
            }
        }
        try {
            FileObject imageFileObject = KettleVFS.getInstance().getFileSystemManager().resolveFile(base, location);
            return new Image(display, KettleVFS.getInputStream(imageFileObject));
        } catch (FileSystemException e) {
            throw new RuntimeException("Unable to load image with name [" + location + "]", e);
        }
    }

    public static Image getImage(Display display, Class<?> resourceClass, String filename) {
        try {
            return new Image(display, resourceClass.getResourceAsStream(filename));
        } catch (Exception e) {
            try {
                return new Image(display, resourceClass.getResourceAsStream("/" + filename));
            } catch (Exception e2) {
                return getImage(display, filename);
            }
        }
    }

    public static Image getImage(Display display, ClassLoader classLoader, String filename) {
        try {
            return new Image(display, classLoader.getResourceAsStream(filename));
        } catch (Exception e) {
            try {
                return new Image(display, classLoader.getResourceAsStream("/" + filename));
            } catch (Exception e2) {
                return getImage(display, filename);
            }
        }
    }

    public static Image getImage(Display display, String location) {
        // TODO: find other instances of getImage (plugin, steps) and transition them to new model through an laf manager
        try {
            InputStream is = KettleVFS.getInputStream(location);
            Image im = new Image(display, is);
            is.close();
            return im;
        } catch (Exception e) {
            try {
                return new Image(display, ImageUtil.class.getClassLoader().getResourceAsStream(location));
            } catch (Exception npe) {
                throw new RuntimeException("Unable to load image with name [" + location + "]", e);
            }
        }

    }

    public static ImageData convertToSWT(BufferedImage bufferedImage) {
        if (bufferedImage.getColorModel() instanceof DirectColorModel) {
            DirectColorModel colorModel = (DirectColorModel) bufferedImage.getColorModel();
            PaletteData palette = new PaletteData(colorModel.getRedMask(), colorModel.getGreenMask(),
                    colorModel.getBlueMask());
            ImageData data = new ImageData(bufferedImage.getWidth(), bufferedImage.getHeight(),
                    colorModel.getPixelSize(), palette);
            WritableRaster raster = bufferedImage.getRaster();
            int[] pixelArray = new int[3];
            for (int y = 0; y < data.height; y++) {
                for (int x = 0; x < data.width; x++) {
                    raster.getPixel(x, y, pixelArray);
                    int pixel = palette.getPixel(new RGB(pixelArray[0], pixelArray[1], pixelArray[2]));
                    data.setPixel(x, y, pixel);
                }
            }
            return data;
        } else if (bufferedImage.getColorModel() instanceof IndexColorModel) {
            IndexColorModel colorModel = (IndexColorModel) bufferedImage.getColorModel();
            int size = colorModel.getMapSize();
            byte[] reds = new byte[size];
            byte[] greens = new byte[size];
            byte[] blues = new byte[size];
            colorModel.getReds(reds);
            colorModel.getGreens(greens);
            colorModel.getBlues(blues);
            RGB[] rgbs = new RGB[size];
            for (int i = 0; i < rgbs.length; i++) {
                rgbs[i] = new RGB(reds[i] & 0xFF, greens[i] & 0xFF, blues[i] & 0xFF);
            }
            PaletteData palette = new PaletteData(rgbs);
            ImageData data = new ImageData(bufferedImage.getWidth(), bufferedImage.getHeight(),
                    colorModel.getPixelSize(), palette);
            data.transparentPixel = colorModel.getTransparentPixel();
            WritableRaster raster = bufferedImage.getRaster();
            int[] pixelArray = new int[1];
            for (int y = 0; y < data.height; y++) {
                for (int x = 0; x < data.width; x++) {
                    raster.getPixel(x, y, pixelArray);
                    data.setPixel(x, y, pixelArray[0]);
                }
            }
            return data;
        }
        return null;
    }

}