+++ /dev/null
-package tim.prune.save;
-
-import java.awt.BorderLayout;
-import java.awt.Color;
-import java.awt.Component;
-import java.awt.FlowLayout;
-import java.awt.Font;
-import java.awt.FontMetrics;
-import java.awt.Graphics;
-import java.awt.event.ActionEvent;
-import java.awt.event.ActionListener;
-import java.awt.event.KeyAdapter;
-import java.awt.event.KeyEvent;
-import java.io.File;
-import java.io.IOException;
-
-import javax.imageio.ImageIO;
-import javax.swing.BorderFactory;
-import javax.swing.BoxLayout;
-import javax.swing.JButton;
-import javax.swing.JCheckBox;
-import javax.swing.JDialog;
-import javax.swing.JFileChooser;
-import javax.swing.JLabel;
-import javax.swing.JOptionPane;
-import javax.swing.JPanel;
-
-import tim.prune.App;
-import tim.prune.GenericFunction;
-import tim.prune.I18nManager;
-import tim.prune.config.ColourScheme;
-import tim.prune.config.Config;
-import tim.prune.data.DataPoint;
-import tim.prune.data.DoubleRange;
-import tim.prune.data.Track;
-import tim.prune.gui.BaseImageDefinitionPanel;
-import tim.prune.gui.GuiGridLayout;
-import tim.prune.gui.WholeNumberField;
-import tim.prune.gui.colour.PointColourer;
-import tim.prune.gui.map.MapSource;
-import tim.prune.gui.map.MapSourceLibrary;
-import tim.prune.gui.map.MapUtils;
-import tim.prune.gui.map.WpIconDefinition;
-import tim.prune.gui.map.WpIconLibrary;
-import tim.prune.load.GenericFileFilter;
-import tim.prune.threedee.ImageDefinition;
-
-/**
- * Class to handle the exporting of map images, optionally with track data drawn on top.
- * This allows images larger than the screen to be generated.
- */
-public class ImageExporter extends GenericFunction implements BaseImageConsumer
-{
- private JDialog _dialog = null;
- private JCheckBox _drawDataCheckbox = null;
- private JCheckBox _drawTrackPointsCheckbox = null;
- private WholeNumberField _textScaleField = null;
- private BaseImageDefinitionPanel _baseImagePanel = null;
- private JFileChooser _fileChooser = null;
- private JButton _okButton = null;
-
- /**
- * Constructor
- * @param inApp App object
- */
- public ImageExporter(App inApp)
- {
- super(inApp);
- }
-
- /** Get the name key */
- public String getNameKey() {
- return "function.exportimage";
- }
-
- /**
- * Begin the function by showing the input dialog
- */
- public void begin()
- {
- // Make dialog window
- if (_dialog == null)
- {
- _dialog = new JDialog(_parentFrame, I18nManager.getText(getNameKey()), true);
- _dialog.setLocationRelativeTo(_parentFrame);
- _dialog.getContentPane().add(makeDialogComponents());
- _dialog.pack();
- _textScaleField.setValue(100);
- }
-
- // Check if there is a cache to use
- if (!BaseImageConfigDialog.isImagePossible())
- {
- _app.showErrorMessage(getNameKey(), "dialog.exportimage.noimagepossible");
- return;
- }
-
- _baseImagePanel.updateBaseImageDetails();
- baseImageChanged();
- // Show dialog
- _dialog.setVisible(true);
- }
-
- /**
- * Make the dialog components to select the export options
- * @return Component holding gui elements
- */
- private Component makeDialogComponents()
- {
- JPanel panel = new JPanel();
- panel.setLayout(new BorderLayout(4, 4));
- // Checkbox for drawing track or not
- _drawDataCheckbox = new JCheckBox(I18nManager.getText("dialog.exportimage.drawtrack"));
- _drawDataCheckbox.setSelected(true); // draw by default
- // Also whether to draw track points or not
- _drawTrackPointsCheckbox = new JCheckBox(I18nManager.getText("dialog.exportimage.drawtrackpoints"));
- _drawTrackPointsCheckbox.setSelected(true);
- // Add listener to en/disable trackpoints checkbox
- _drawDataCheckbox.addActionListener(new ActionListener() {
- public void actionPerformed(ActionEvent arg0) {
- _drawTrackPointsCheckbox.setEnabled(_drawDataCheckbox.isSelected());
- }
- });
-
- // TODO: Maybe have other controls such as line width, symbol scale factor
- JPanel controlsPanel = new JPanel();
- GuiGridLayout grid = new GuiGridLayout(controlsPanel);
- grid.add(new JLabel(I18nManager.getText("dialog.exportimage.textscalepercent") + ": "));
- _textScaleField = new WholeNumberField(3);
- _textScaleField.setText("888");
- grid.add(_textScaleField);
-
- // OK, Cancel buttons
- JPanel buttonPanel = new JPanel();
- buttonPanel.setLayout(new FlowLayout(FlowLayout.RIGHT));
- _okButton = new JButton(I18nManager.getText("button.ok"));
- _okButton.addActionListener(new ActionListener() {
- public void actionPerformed(ActionEvent e)
- {
- doExport();
- _baseImagePanel.getGrouter().clearMapImage();
- _dialog.dispose();
- }
- });
- buttonPanel.add(_okButton);
- JButton cancelButton = new JButton(I18nManager.getText("button.cancel"));
- cancelButton.addActionListener(new ActionListener() {
- public void actionPerformed(ActionEvent e)
- {
- _baseImagePanel.getGrouter().clearMapImage();
- _dialog.dispose();
- }
- });
- buttonPanel.add(cancelButton);
- panel.add(buttonPanel, BorderLayout.SOUTH);
-
- // Listener to close dialog if escape pressed
- KeyAdapter closer = new KeyAdapter() {
- public void keyReleased(KeyEvent e)
- {
- if (e.getKeyCode() == KeyEvent.VK_ESCAPE) {
- _dialog.dispose();
- _baseImagePanel.getGrouter().clearMapImage();
- }
- }
- };
- _drawDataCheckbox.addKeyListener(closer);
-
- // Panel for the base image
- _baseImagePanel = new BaseImageDefinitionPanel(this, _dialog, _app.getTrackInfo().getTrack());
-
- // Panel for the checkboxes at the top
- JPanel checkPanel = new JPanel();
- checkPanel.setLayout(new BoxLayout(checkPanel, BoxLayout.Y_AXIS));
- checkPanel.add(_drawDataCheckbox);
- checkPanel.add(_drawTrackPointsCheckbox);
-
- // add these panels to the holder panel
- JPanel holderPanel = new JPanel();
- holderPanel.setLayout(new BorderLayout(5, 5));
- holderPanel.add(checkPanel, BorderLayout.NORTH);
- holderPanel.add(controlsPanel, BorderLayout.CENTER);
- holderPanel.add(_baseImagePanel, BorderLayout.SOUTH);
- holderPanel.setBorder(BorderFactory.createEmptyBorder(4, 4, 4, 4));
-
- panel.add(holderPanel, BorderLayout.NORTH);
- return panel;
- }
-
-
- /**
- * Select the file and export data to it
- */
- private void doExport()
- {
- _okButton.setEnabled(false);
- // OK pressed, so choose output file
- if (_fileChooser == null)
- {
- _fileChooser = new JFileChooser();
- _fileChooser.setDialogType(JFileChooser.SAVE_DIALOG);
- _fileChooser.setFileFilter(new GenericFileFilter("filetype.png", new String[] {"png"}));
- _fileChooser.setAcceptAllFileFilterUsed(false);
- // start from directory in config which should be set
- final String configDir = Config.getConfigString(Config.KEY_TRACK_DIR);
- if (configDir != null) {_fileChooser.setCurrentDirectory(new File(configDir));}
- }
-
- // Allow choose again if an existing file is selected
- boolean chooseAgain = false;
- do
- {
- chooseAgain = false;
- if (_fileChooser.showSaveDialog(_parentFrame) == JFileChooser.APPROVE_OPTION)
- {
- // OK pressed and file chosen
- File pngFile = _fileChooser.getSelectedFile();
- if (!pngFile.getName().toLowerCase().endsWith(".png"))
- {
- pngFile = new File(pngFile.getAbsolutePath() + ".png");
- }
- // Check if file exists and if necessary prompt for overwrite
- Object[] buttonTexts = {I18nManager.getText("button.overwrite"), I18nManager.getText("button.cancel")};
- if (!pngFile.exists() || JOptionPane.showOptionDialog(_parentFrame,
- I18nManager.getText("dialog.save.overwrite.text"),
- I18nManager.getText("dialog.save.overwrite.title"), JOptionPane.YES_NO_OPTION,
- JOptionPane.WARNING_MESSAGE, null, buttonTexts, buttonTexts[1])
- == JOptionPane.YES_OPTION)
- {
- // Export the file
- if (!exportFile(pngFile))
- {
- // export failed so need to choose again
- chooseAgain = true;
- }
- }
- else
- {
- // overwrite cancelled so need to choose again
- chooseAgain = true;
- }
- }
- } while (chooseAgain);
- }
-
- /**
- * Export the track data to the specified file
- * @param inPngFile File object to save to
- * @return true if successful
- */
- private boolean exportFile(File inPngFile)
- {
- // Get the image file from the grouter
- ImageDefinition imageDef = _baseImagePanel.getImageDefinition();
- MapSource source = MapSourceLibrary.getSource(imageDef.getSourceIndex());
- MapGrouter grouter = _baseImagePanel.getGrouter();
- GroutedImage baseImage = grouter.getMapImage(_app.getTrackInfo().getTrack(), source,
- imageDef.getZoom());
- if (baseImage == null || !baseImage.isValid())
- {
- _app.showErrorMessage(getNameKey(), "dialog.exportpov.cannotmakebaseimage");
- return true;
- }
- try
- {
- if (_drawDataCheckbox.isSelected())
- {
- // Draw the track on top of this image
- drawData(baseImage);
- }
- // Write composite image to file
- if (!ImageIO.write(baseImage.getImage(), "png", inPngFile)) {
- _app.showErrorMessage(getNameKey(), "dialog.exportpov.cannotmakebaseimage");
- return false; // choose again - the image creation worked but the save failed
- }
- }
- catch (IOException ioe) {
- System.err.println("Can't write image: " + ioe.getClass().getName());
- }
- return true;
- }
-
- /**
- * Draw the track and waypoint data from the current Track onto the given image
- * @param inImage GroutedImage from map tiles
- */
- private void drawData(GroutedImage inImage)
- {
- // Work out x, y limits for drawing
- DoubleRange xRange = inImage.getXRange();
- DoubleRange yRange = inImage.getYRange();
- final int zoomFactor = 1 << _baseImagePanel.getImageDefinition().getZoom();
- Graphics g = inImage.getImage().getGraphics();
- // TODO: Set line width, style etc
- final PointColourer pointColourer = _app.getPointColourer();
- final Color defaultPointColour = Config.getColourScheme().getColour(ColourScheme.IDX_POINT);
- g.setColor(defaultPointColour);
-
- // Loop to draw all track points
- final Track track = _app.getTrackInfo().getTrack();
- final int numPoints = track.getNumPoints();
- int prevX = 0, prevY = 0;
- for (int i=0; i<numPoints; i++)
- {
- DataPoint point = track.getPoint(i);
- if (!point.isWaypoint())
- {
- // Determine what colour to use to draw the track point
- if (pointColourer != null)
- {
- Color c = pointColourer.getColour(i);
- g.setColor(c == null ? defaultPointColour : c);
- }
- double x = track.getX(i) - xRange.getMinimum();
- double y = track.getY(i) - yRange.getMinimum();
- // use zoom level to calculate pixel coords on image
- int px = (int) (x * zoomFactor * 256), py = (int) (y * zoomFactor * 256);
- // System.out.println("Point: x=" + x + ", px=" + px + ", y=" + y + ", py=" + py);
- if (!point.getSegmentStart()) {
- // draw from previous point to this one
- g.drawLine(prevX, prevY, px, py);
- }
- // Only draw points if requested
- if (_drawTrackPointsCheckbox.isSelected())
- {
- g.drawRect(px-2, py-2, 3, 3);
- }
- // save coordinates
- prevX = px; prevY = py;
- }
- }
-
- // Now the waypoints
- final Color textColour = Config.getColourScheme().getColour(ColourScheme.IDX_TEXT);
- g.setColor(textColour);
- WpIconDefinition wpIconDefinition = null;
- final int wpType = Config.getConfigInt(Config.KEY_WAYPOINT_ICONS);
- if (wpType != WpIconLibrary.WAYPT_DEFAULT)
- {
- wpIconDefinition = WpIconLibrary.getIconDefinition(wpType, WpIconLibrary.SIZE_MEDIUM);
- }
- // Loop again to draw waypoints
- for (int i=0; i<numPoints; i++)
- {
- DataPoint point = track.getPoint(i);
- if (point.isWaypoint())
- {
- // use zoom level to calculate pixel coords on image
- double x = track.getX(i) - xRange.getMinimum();
- double y = track.getY(i) - yRange.getMinimum();
- int px = (int) (x * zoomFactor * 256), py = (int) (y * zoomFactor * 256);
- // Fill Rect or draw icon image?
- g.fillRect(px-3, py-3, 6, 6);
- if (wpIconDefinition == null)
- {
- g.fillRect(px-3, py-3, 6, 6);
- }
- else
- {
- g.drawImage(wpIconDefinition.getImageIcon().getImage(), px-wpIconDefinition.getXOffset(),
- py-wpIconDefinition.getYOffset(), null);
- }
- }
- }
- // Set text size according to input
- int fontScalePercent = _textScaleField.getValue();
- if (fontScalePercent > 10 && fontScalePercent <= 999)
- {
- Font gFont = g.getFont();
- g.setFont(gFont.deriveFont((float) (gFont.getSize() * 0.01 * fontScalePercent)));
- }
- FontMetrics fm = g.getFontMetrics();
- final int nameHeight = fm.getHeight();
- final int imageSize = inImage.getImageSize();
-
- // Loop over points again, draw photo points
- final Color photoColour = Config.getColourScheme().getColour(ColourScheme.IDX_SECONDARY);
- g.setColor(photoColour);
- for (int i=0; i<numPoints; i++)
- {
- DataPoint point = track.getPoint(i);
- if (point.hasMedia())
- {
- // draw blob for each photo
- double x = track.getX(i) - xRange.getMinimum();
- double y = track.getY(i) - yRange.getMinimum();
- // use zoom level to calculate pixel coords on image
- int px = (int) (x * zoomFactor * 256), py = (int) (y * zoomFactor * 256);
- g.fillRect(px-3, py-3, 6, 6);
- }
- }
-
- // Loop over points again, now draw names for waypoints
- g.setColor(textColour);
- for (int i=0; i<numPoints; i++)
- {
- DataPoint point = track.getPoint(i);
- if (point.isWaypoint())
- {
- double x = track.getX(i) - xRange.getMinimum();
- double y = track.getY(i) - yRange.getMinimum();
- int px = (int) (x * zoomFactor * 256), py = (int) (y * zoomFactor * 256);
-
- // Figure out where to draw waypoint name so it doesn't obscure track
- String waypointName = point.getWaypointName();
- int nameWidth = fm.stringWidth(waypointName);
- boolean drawnName = false;
- // Make arrays for coordinates right left up down
- int[] nameXs = {px + 2, px - nameWidth - 2, px - nameWidth/2, px - nameWidth/2};
- int[] nameYs = {py + (nameHeight/2), py + (nameHeight/2), py - 2, py + nameHeight + 2};
- for (int extraSpace = 4; extraSpace < 13 && !drawnName; extraSpace+=2)
- {
- // Shift arrays for coordinates right left up down
- nameXs[0] += 2; nameXs[1] -= 2;
- nameYs[2] -= 2; nameYs[3] += 2;
- // Check each direction in turn right left up down
- for (int a=0; a<4; a++)
- {
- if (nameXs[a] > 0 && (nameXs[a] + nameWidth) < imageSize
- && nameYs[a] < imageSize && (nameYs[a] - nameHeight) > 0
- && !MapUtils.overlapsPoints(inImage.getImage(), nameXs[a], nameYs[a],
- nameWidth, nameHeight, textColour))
- {
- // Found a rectangle to fit - draw name here and quit
- g.drawString(waypointName, nameXs[a], nameYs[a]);
- drawnName = true;
- break;
- }
- }
- }
- }
- }
-
- // Maybe draw note at the bottom, export from GpsPrune? Filename?
- // Note: Differences from main map: No mapPosition (modifying position and visible points),
- // no selection, no opacities, maybe different scale/text factors
- }
-
- /**
- * Base image has changed, need to enable/disable ok button
- */
- public void baseImageChanged()
- {
- final boolean useImage = _baseImagePanel.getImageDefinition().getUseImage();
- final int zoomLevel = _baseImagePanel.getImageDefinition().getZoom();
- final boolean okEnabled = useImage && _baseImagePanel.getFoundData()
- && MapGrouter.isZoomLevelOk(_app.getTrackInfo().getTrack(), zoomLevel);
- _okButton.setEnabled(okEnabled);
- }
-}
projects / GpsPrune.git / blobdiff