Wrote a handy little tool for fiddling with the image recoloring routines.
git-svn-id: svn+ssh://src.earth.threerings.net/narya/trunk@1071 542714f4-19e9-0310-aa3c-eee0fc999fb1
This commit is contained in:
Michael Bayne
@@ -0,0 +1,263 @@
|
||||
//
|
||||
// $Id: RecolorImage.java,v 1.1 2002/02/25 07:28:34 mdb Exp $
|
||||
|
||||
package com.threerings.media.tools;
|
||||
|
||||
import java.awt.BorderLayout;
|
||||
import java.awt.Color;
|
||||
import java.awt.Rectangle;
|
||||
import java.awt.event.ActionEvent;
|
||||
import java.awt.event.ActionListener;
|
||||
import java.awt.event.MouseAdapter;
|
||||
import java.awt.event.MouseEvent;
|
||||
import java.awt.image.BufferedImage;
|
||||
|
||||
import javax.imageio.ImageIO;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
||||
import javax.swing.BorderFactory;
|
||||
import javax.swing.ImageIcon;
|
||||
import javax.swing.JButton;
|
||||
import javax.swing.JFileChooser;
|
||||
import javax.swing.JFrame;
|
||||
import javax.swing.JLabel;
|
||||
import javax.swing.JPanel;
|
||||
import javax.swing.JScrollPane;
|
||||
import javax.swing.JTextField;
|
||||
|
||||
import com.samskivert.swing.HGroupLayout;
|
||||
import com.samskivert.swing.VGroupLayout;
|
||||
import com.samskivert.swing.util.SwingUtil;
|
||||
|
||||
import com.threerings.media.Log;
|
||||
import com.threerings.media.util.ImageUtil;
|
||||
|
||||
/**
|
||||
* Tests the image recoloring code.
|
||||
*/
|
||||
public class RecolorImage extends JPanel
|
||||
implements ActionListener
|
||||
{
|
||||
public RecolorImage ()
|
||||
{
|
||||
setBorder(BorderFactory.createEmptyBorder(5, 5, 5, 5));
|
||||
|
||||
VGroupLayout vlay = new VGroupLayout(VGroupLayout.STRETCH);
|
||||
vlay.setOffAxisPolicy(VGroupLayout.STRETCH);
|
||||
setLayout(vlay);
|
||||
|
||||
JPanel images = new JPanel(new HGroupLayout());
|
||||
images.add(_oldImage = new JLabel());
|
||||
images.add(_newImage = new JLabel());
|
||||
add(new JScrollPane(images));
|
||||
|
||||
JPanel file = new JPanel(new HGroupLayout(HGroupLayout.STRETCH));
|
||||
file.add(new JLabel("Image file:"), HGroupLayout.FIXED);
|
||||
file.add(_imagePath = new JTextField());
|
||||
_imagePath.setEditable(false);
|
||||
JButton browse = new JButton("Browse...");
|
||||
browse.setActionCommand("browse");
|
||||
browse.addActionListener(this);
|
||||
file.add(browse, HGroupLayout.FIXED);
|
||||
add(file, VGroupLayout.FIXED);
|
||||
|
||||
JPanel controls = new JPanel(new HGroupLayout(HGroupLayout.STRETCH));
|
||||
controls.add(new JLabel("Source color:"), HGroupLayout.FIXED);
|
||||
controls.add(_source = new JTextField());
|
||||
_colorLabel = new JPanel();
|
||||
_colorLabel.setSize(48, 48);
|
||||
_colorLabel.setOpaque(true);
|
||||
controls.add(_colorLabel, HGroupLayout.FIXED);
|
||||
controls.add(new JLabel("Target color:"), HGroupLayout.FIXED);
|
||||
controls.add(_target = new JTextField());
|
||||
JButton update = new JButton("Update");
|
||||
update.setActionCommand("update");
|
||||
update.addActionListener(this);
|
||||
controls.add(update, HGroupLayout.FIXED);
|
||||
add(controls, VGroupLayout.FIXED);
|
||||
|
||||
HGroupLayout hlay = new HGroupLayout(HGroupLayout.STRETCH);
|
||||
JPanel dists = new JPanel(hlay);
|
||||
dists.add(new JLabel("HSV distances:"), HGroupLayout.FIXED);
|
||||
dists.add(_hueD = new JTextField("0.05"));
|
||||
dists.add(_saturationD = new JTextField("0.8"));
|
||||
dists.add(_valueD = new JTextField("0.6"));
|
||||
add(dists, VGroupLayout.FIXED);
|
||||
|
||||
hlay = new HGroupLayout(HGroupLayout.STRETCH);
|
||||
JPanel offsets = new JPanel(hlay);
|
||||
offsets.add(new JLabel("HSV offsets:"), HGroupLayout.FIXED);
|
||||
offsets.add(_hueO = new JTextField("0.1"));
|
||||
offsets.add(_saturationO = new JTextField("0.0"));
|
||||
offsets.add(_valueO = new JTextField("0.0"));
|
||||
add(offsets, VGroupLayout.FIXED);
|
||||
|
||||
add(_status = new JTextField(), VGroupLayout.FIXED);
|
||||
_status.setEditable(false);
|
||||
|
||||
hlay = new HGroupLayout();
|
||||
hlay.setJustification(HGroupLayout.CENTER);
|
||||
JPanel buttons = new JPanel(hlay);
|
||||
JButton convert = new JButton("Convert");
|
||||
convert.setActionCommand("convert");
|
||||
convert.addActionListener(this);
|
||||
buttons.add(convert);
|
||||
add(buttons, VGroupLayout.FIXED);
|
||||
|
||||
// listen for mouse clicks
|
||||
images.addMouseListener(new MouseAdapter() {
|
||||
public void mousePressed (MouseEvent event) {
|
||||
RecolorImage.this.mousePressed(event);
|
||||
}
|
||||
});
|
||||
|
||||
// we'll be using a file chooser
|
||||
String cwd = System.getProperty("user.dir");
|
||||
if (cwd == null) {
|
||||
_chooser = new JFileChooser();
|
||||
} else {
|
||||
_chooser = new JFileChooser(cwd);
|
||||
}
|
||||
}
|
||||
|
||||
public void actionPerformed (ActionEvent event)
|
||||
{
|
||||
String cmd = event.getActionCommand();
|
||||
|
||||
if (cmd.equals("convert")) {
|
||||
// obtain the target color and offset
|
||||
try {
|
||||
int color = Integer.parseInt(_source.getText(), 16);
|
||||
|
||||
float hueD = Float.parseFloat(_hueD.getText());
|
||||
float satD = Float.parseFloat(_saturationD.getText());
|
||||
float valD = Float.parseFloat(_valueD.getText());
|
||||
float[] dists = new float[] { hueD, satD, valD };
|
||||
|
||||
float hueO = Float.parseFloat(_hueO.getText());
|
||||
float satO = Float.parseFloat(_saturationO.getText());
|
||||
float valO = Float.parseFloat(_valueO.getText());
|
||||
float[] offsets = new float[] { hueO, satO, valO };
|
||||
|
||||
BufferedImage image =
|
||||
ImageUtil.recolorImage(_image, color, dists, offsets);
|
||||
_newImage.setIcon(new ImageIcon(image));
|
||||
_status.setText("Recolored image.");
|
||||
repaint();
|
||||
|
||||
} catch (NumberFormatException nfe) {
|
||||
_status.setText("Invalid value: " + nfe.getMessage());
|
||||
}
|
||||
|
||||
} else if (cmd.equals("update")) {
|
||||
// obtain the target color and offset
|
||||
try {
|
||||
int source = Integer.parseInt(_source.getText(), 16);
|
||||
int target = Integer.parseInt(_target.getText(), 16);
|
||||
float[] shsv = rgbToHSV(source);
|
||||
float[] thsv = rgbToHSV(target);
|
||||
|
||||
// set the offsets based on the differences
|
||||
_hueO.setText("" + (thsv[0] - shsv[0]));
|
||||
_saturationO.setText("" + (thsv[1] - shsv[1]));
|
||||
_valueO.setText("" + (thsv[2] - shsv[2]));
|
||||
|
||||
} catch (NumberFormatException nfe) {
|
||||
_status.setText("Invalid value: " + nfe.getMessage());
|
||||
}
|
||||
|
||||
} else if (cmd.equals("browse")) {
|
||||
int result = _chooser.showOpenDialog(this);
|
||||
if (result == JFileChooser.APPROVE_OPTION) {
|
||||
setImage(_chooser.getSelectedFile());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void setImage (File path)
|
||||
{
|
||||
try {
|
||||
_image = ImageIO.read(path);
|
||||
_imagePath.setText(path.getPath());
|
||||
_oldImage.setIcon(new ImageIcon(_image));
|
||||
|
||||
} catch (IOException ioe) {
|
||||
_status.setText("Error opening image file: " + ioe);
|
||||
}
|
||||
}
|
||||
|
||||
protected static float[] rgbToHSV (int rgb)
|
||||
{
|
||||
float[] hsv = new float[3];
|
||||
Color color = new Color(rgb);
|
||||
Color.RGBtoHSB(color.getRed(), color.getGreen(),
|
||||
color.getBlue(), hsv);
|
||||
return hsv;
|
||||
}
|
||||
|
||||
public void mousePressed (MouseEvent event)
|
||||
{
|
||||
// if the click was in the bounds of the source image, grab the
|
||||
// pixel color and use that to set the "source" color
|
||||
int x = event.getX(), y = event.getY();
|
||||
Rectangle ibounds = _oldImage.getBounds();
|
||||
if (ibounds.contains(x, y)) {
|
||||
int argb = _image.getRGB(x - ibounds.x, y - ibounds.y);
|
||||
String cstr = Integer.toString(argb & 0xFFFFFF, 16);
|
||||
_source.setText(cstr.toUpperCase());
|
||||
_colorLabel.setBackground(new Color(argb));
|
||||
_colorLabel.repaint();
|
||||
}
|
||||
}
|
||||
|
||||
public static void main (String[] args)
|
||||
{
|
||||
try {
|
||||
JFrame frame = new JFrame("Image recoloring test");
|
||||
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
|
||||
|
||||
RecolorImage panel = new RecolorImage();
|
||||
|
||||
// load up the image from the command line if one was
|
||||
// specified
|
||||
if (args.length > 0) {
|
||||
panel.setImage(new File(args[0]));
|
||||
}
|
||||
|
||||
frame.getContentPane().add(panel, BorderLayout.CENTER);
|
||||
frame.setSize(600, 600);
|
||||
SwingUtil.centerWindow(frame);
|
||||
frame.show();
|
||||
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace(System.err);
|
||||
}
|
||||
}
|
||||
|
||||
protected BufferedImage _image;
|
||||
protected JFileChooser _chooser;
|
||||
protected JTextField _imagePath;
|
||||
|
||||
protected JLabel _oldImage;
|
||||
protected JLabel _newImage;
|
||||
protected JPanel _colorLabel;
|
||||
|
||||
protected JTextField _source;
|
||||
protected JTextField _target;
|
||||
|
||||
protected JTextField _hueO;
|
||||
protected JTextField _saturationO;
|
||||
protected JTextField _valueO;
|
||||
|
||||
protected JTextField _hueD;
|
||||
protected JTextField _saturationD;
|
||||
protected JTextField _valueD;
|
||||
|
||||
protected JTextField _status;
|
||||
|
||||
protected static final String IMAGE_PATH =
|
||||
// "bundles/components/pirate/torso/regular/standing.png";
|
||||
"bundles/components/pirate/head/regular/standing.png";
|
||||
}
|
||||
Reference in New Issue
Block a user