1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.utils.ui.color;
\r
14 import java.util.ArrayList;
\r
15 import java.util.Collection;
\r
16 import java.util.Collections;
\r
17 import java.util.List;
\r
19 import org.eclipse.swt.SWT;
\r
20 import org.eclipse.swt.graphics.Color;
\r
21 import org.eclipse.swt.graphics.GC;
\r
22 import org.eclipse.swt.graphics.Image;
\r
23 import org.eclipse.swt.widgets.Display;
\r
28 * Color gradient between multiple colors
\r
31 * Supports RGB and HSV linear interpolation between colors
\r
34 * @author Marko Luukkainen
\r
37 public class ColorGradient {
\r
38 public static final int RGB = 0;
\r
40 public static final int HSV = 1;
\r
42 protected ArrayList<ColorValue> values;
\r
46 public ColorGradient() {
\r
47 this.values = new ArrayList<ColorValue>();
\r
51 public ColorGradient(ColorGradient copyFrom) {
\r
52 this.values = new ArrayList<ColorValue>(copyFrom.values);
\r
53 this.type = copyFrom.type;
\r
56 public ColorGradient(Collection<ColorValue> values) {
\r
57 this.values = new ArrayList<ColorValue>(values);
\r
58 Collections.sort(this.values, new ColorValueComparator());
\r
62 public ColorGradient(ColorValue array[]) {
\r
63 this.values = new ArrayList<ColorValue>(array.length);
\r
64 for (ColorValue c : array) {
\r
67 Collections.sort(this.values, new ColorValueComparator());
\r
71 public ColorGradient(Collection<ColorValue> values, int type) {
\r
72 this.values = new ArrayList<ColorValue>(values);
\r
73 Collections.sort(this.values, new ColorValueComparator());
\r
77 public ColorGradient(ColorValue array[], int type) {
\r
78 this.values = new ArrayList<ColorValue>();
\r
79 for (ColorValue c : array) {
\r
82 Collections.sort(this.values, new ColorValueComparator());
\r
87 * Interpolates color in RGB space
\r
92 private byte[] getRGBColor(double value) {
\r
94 while (values.get(index).getValue() <= value && index < values.size()-1)
\r
97 value -= values.get(index - 1).getValue();
\r
98 value /= (values.get(index).getValue() - values.get(index - 1).getValue());
\r
99 double valuei = 1.0 - value;
\r
100 byte color[] = new byte[] {
\r
101 (byte) Math.min(255.0, Math.floor(value * values.get(index).getColor().getR() + valuei * values.get(index - 1).getColor().getR())),
\r
102 (byte) Math.min(255.0, Math.floor(value * values.get(index).getColor().getG() + valuei * values.get(index - 1).getColor().getG())),
\r
103 (byte) Math.min(255.0, Math.floor(value * values.get(index).getColor().getB() + valuei * values.get(index - 1).getColor().getB())) };
\r
109 * Interpolates color in HSV space
\r
114 private byte[] getHSVColor(double value) {
\r
116 while (values.get(index).getValue() <= value && index < values.size()-1)
\r
119 value -= values.get(index - 1).getValue();
\r
120 value /= (values.get(index).getValue() - values.get(index - 1).getValue());
\r
121 double valuei = 1.0 - value;
\r
123 if (Float.isNaN(values.get(index).getColor().getH())) {
\r
124 h = values.get(index-1).getColor().getH();
\r
125 } else if (Float.isNaN(values.get(index-1).getColor().getH())) {
\r
126 h = values.get(index).getColor().getH();
\r
128 // selecting shortest direction between hues
\r
129 float angle = values.get(index).getColor().getH() - values.get(index - 1).getColor().getH();
\r
132 else if (angle < -180.f)
\r
134 h = values.get(index - 1).getColor().getH() + value * angle;
\r
140 org.simantics.utils.ui.color.Color interpolated = new org.simantics.utils.ui.color.Color(h, value * values.get(index).getColor().getS() + valuei * values.get(index - 1).getColor().getS(),
\r
141 value * values.get(index).getColor().getV() + valuei * values.get(index - 1).getColor().getV());
\r
142 byte color[] = new byte[] { (byte) interpolated.getR(), (byte) interpolated.getG(), (byte) interpolated.getB() };
\r
150 * Returns gradient in array of bytes. Array is RGB order and int contains 3 * requested size of bytes.
\r
153 * If gradient contains only one color array is filled with that color
\r
156 * if gradient has no colors array is filled with white
\r
158 * @param size number of pixels
\r
159 * @return gradient in array of bytes
\r
161 public byte[] getGradientArray(int size) {
\r
162 byte array[] = new byte[size * 3];
\r
163 if (values.size() > 1) {
\r
164 for (int i = 0; i < size; i++) {
\r
166 double value = values.get(0).getValue() + (values.get(values.size() - 1).getValue() - values.get(0).getValue()) * (double) i / (double) size;
\r
169 color = getRGBColor(value);
\r
171 color = getHSVColor(value);
\r
172 array[index] = color[0];
\r
173 array[index + 1] = color[1];
\r
174 array[index + 2] = color[2];
\r
176 } else if (values.size() == 1) {
\r
177 byte color[] = new byte[3];
\r
178 color[0] = (byte)values.get(0).getColor().getR();
\r
179 color[1] = (byte)values.get(0).getColor().getG();
\r
180 color[2] = (byte)values.get(0).getColor().getB();
\r
181 for (int i = 0; i < size; i++) {
\r
183 array[index] = color[0];
\r
184 array[index + 1] = color[1];
\r
185 array[index + 2] = color[2];
\r
188 for (int i = 0; i < size; i++) {
\r
190 array[index] = (byte)255;
\r
191 array[index + 1] = (byte)255;
\r
192 array[index + 2] = (byte)255;
\r
200 * Returns gradient in image.
\r
203 * If gradient contains only one color image is filled with that color
\r
206 * if gradient has no colors image is filled with white
\r
209 * Style must be set to <code>SWT.HORIZONTAL</code> or <code>SWT.VERTICAL</code>
\r
211 * @param size number of pixels
\r
212 * @return gradient in array of bytes
\r
215 public Image getGradientImage(int width, int height, int style) {
\r
216 Image image = new Image(Display.getCurrent(), width, height);
\r
217 GC gc = new GC(image);
\r
218 gc.setBackground(Display.getCurrent().getSystemColor(SWT.COLOR_WHITE));
\r
219 gc.fillRectangle(0, 0, width, height);
\r
220 if (values.size() > 1) {
\r
221 if (SWT.HORIZONTAL == (style | SWT.HORIZONTAL)) {
\r
222 for (int x = 0; x < width; x++) {
\r
223 double value = values.get(0).getValue() + (values.get(values.size() - 1).getValue() - values.get(0).getValue()) * (double) x
\r
224 / (double) (width - 1);
\r
227 byteColor = getRGBColor(value);
\r
229 byteColor = getHSVColor(value);
\r
230 Color color = new Color(Display.getCurrent(), byteColor[0] & 0xff, byteColor[1] & 0xff, byteColor[2] & 0xff);
\r
231 gc.setForeground(color);
\r
232 gc.drawLine(x, 0, x, height);
\r
235 } else if (SWT.VERTICAL == (style | SWT.VERTICAL)){
\r
236 for (int y = 0; y < height; y++) {
\r
237 double value = values.get(0).getValue() + (values.get(values.size() - 1).getValue() - values.get(0).getValue()) * (double) y
\r
238 / (double) (height - 1);
\r
241 byteColor = getRGBColor(value);
\r
243 byteColor = getHSVColor(value);
\r
244 Color color = new Color(Display.getCurrent(), byteColor[0] & 0xff, byteColor[1] & 0xff, byteColor[2] & 0xff);
\r
245 gc.setForeground(color);
\r
246 gc.drawLine(0, y, width, y);
\r
252 SWT.error(SWT.ERROR_INVALID_ARGUMENT);
\r
254 } else if (values.size() == 1) {
\r
255 Color color = new Color(Display.getCurrent(), values.get(0).getColor().getR(), values.get(0).getColor().getG(), values.get(0).getColor().getB());
\r
256 gc.setBackground(color);
\r
257 gc.fillRectangle(0, 0, width, height);
\r
260 gc.fillRectangle(0, 0, width, height);
\r
266 public int getType() {
\r
270 public List<ColorValue> getColorValues() {
\r
274 public ColorValue[] getColorValueArray() {
\r
275 return values.toArray(new ColorValue[values.size()]);
\r
279 public void addCastorColorValue(ColorValue value) {
\r
281 Collections.sort(this.values, new ColorValueComparator());
\r
285 public void setCastorType(int type) {
\r
290 public int hashCode() {
\r
291 int hash = 0x58fb3;
\r
292 for (ColorValue cv : values)
\r
293 hash ^= cv.hashCode();
\r
298 public boolean equals(Object obj) {
\r
301 if (obj.getClass() != getClass())
\r
303 ColorGradient cg = (ColorGradient) obj;
\r
304 if (cg.type != type) return false;
\r
305 if (values.size()!=cg.values.size()) return false;
\r
306 if (!values.containsAll(cg.values)) return false;
\r