/*
* Copyright 2016 Igor Maznitsa.
*
* 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.
*/
package com.igormaznitsa.mindmap.plugins.attributes.images;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.Image;
import java.awt.Toolkit;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.util.Locale;
import java.util.ResourceBundle;
import java.util.concurrent.atomic.AtomicInteger;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import javax.swing.ButtonGroup;
import javax.swing.Icon;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JRadioButton;
import javax.swing.filechooser.FileFilter;
import com.igormaznitsa.meta.annotation.MustNotContainNull;
import com.igormaznitsa.mindmap.model.Topic;
import com.igormaznitsa.mindmap.model.logger.Logger;
import com.igormaznitsa.mindmap.model.logger.LoggerFactory;
import com.igormaznitsa.mindmap.plugins.PopUpSection;
import com.igormaznitsa.mindmap.plugins.api.AbstractPopupMenuItem;
import com.igormaznitsa.mindmap.plugins.api.CustomJob;
import com.igormaznitsa.mindmap.swing.panel.DialogProvider;
import com.igormaznitsa.mindmap.swing.panel.MindMapPanel;
import com.igormaznitsa.mindmap.swing.panel.utils.Utils;
import com.igormaznitsa.mindmap.swing.services.IconID;
import com.igormaznitsa.mindmap.swing.services.ImageIconServiceProvider;
public class ImagePopUpMenuPlugin extends AbstractPopupMenuItem {
private static final Logger LOGGER = LoggerFactory.getLogger(ImagePopUpMenuPlugin.class);
private static final ResourceBundle BUNDLE = java.util.ResourceBundle.getBundle("com/igormaznitsa/mindmap/swing/panel/Bundle");//NOI18N
private static final Icon ICON = ImageIconServiceProvider.findInstance().getIconForId(IconID.ICON_IMAGES);
private static File lastSelectedFile = null;
private static int lastSelectedImportIndex = 0;
private static final FileFilter IMAGE_FILE_FILTER = new FileFilter() {
@Override
public boolean accept(@Nonnull final File f) {
final String text = f.getName().toLowerCase(Locale.ENGLISH);
return f.isDirectory() || text.endsWith(".png") || text.endsWith(".jpg") || text.endsWith(".gif"); //NOI18N
}
@Override
@Nonnull
public String getDescription() {
return BUNDLE.getString("Images.Plugin.FilterDescription");
}
};
@Override
@Nullable
public JMenuItem makeMenuItem(@Nonnull final MindMapPanel panel, @Nonnull final DialogProvider dialogProvider, @Nullable final Topic topic, @Nonnull @MustNotContainNull final Topic[] selectedTopics, @Nullable final CustomJob customProcessor) {
final boolean hasAttribute = containAttribute(topic, selectedTopics);
final JMenuItem result;
if (hasAttribute) {
result = new JMenuItem(BUNDLE.getString("Images.Plugin.MenuTitle.Remove"), ICON);//NOI18N
result.setToolTipText(BUNDLE.getString("Images.Plugin.MenuTitle.Remove.Tooltip"));//NOI18N
result.addActionListener(new ActionListener() {
@Override
public void actionPerformed(@Nonnull final ActionEvent e) {
if (dialogProvider.msgConfirmYesNo(BUNDLE.getString("Images.Plugin.Remove.Dialog.Title"), BUNDLE.getString("Images.Plugin.Remove.Dialog.Text"))) {//NOI18N
setAttribute(null, topic, selectedTopics);
ImageVisualAttributePlugin.clearCachedImages();
panel.notifyModelChanged();
}
}
});
} else {
result = new JMenuItem(BUNDLE.getString("Images.Plugin.MenuTitle.Add"), ICON);//NOI18N
result.setToolTipText(BUNDLE.getString("Images.Plugin.MenuTitle.Add.Tooltip"));//NOI18N
result.addActionListener(new ActionListener() {
@Override
public void actionPerformed(@Nonnull final ActionEvent e) {
final Transferable transferable = Toolkit.getDefaultToolkit().getSystemClipboard().getContents(null);
boolean loadFromFile = true;
if (transferable != null && transferable.isDataFlavorSupported(DataFlavor.imageFlavor)) {
final AtomicInteger selectedItem = new AtomicInteger(-1);
if (dialogProvider.msgOkCancel(BUNDLE.getString("Images.Plugin.Select.DialogTitle"), makeSelectPanel(new String[]{BUNDLE.getString("Images.Plugin.Select.FromClipboard"), BUNDLE.getString("Images.Plugin.Select.FromFile")}, selectedItem))) {
lastSelectedImportIndex = selectedItem.get();
if (selectedItem.get() == 0) {
try {
setAttribute(Utils.rescaleImageAndEncodeAsBase64((Image) transferable.getTransferData(DataFlavor.imageFlavor), Utils.getMaxImageSize()), topic, selectedTopics);
panel.notifyModelChanged();
}
catch (final IllegalArgumentException ex) {
dialogProvider.msgError(BUNDLE.getString("Images.Plugin.Error"));
LOGGER.error("Can't import from clipboard image", ex); //NOI18N
}
catch (final Exception ex) {
dialogProvider.msgError(BUNDLE.getString("Images.Plugin.Error"));
LOGGER.error("Unexpected error during image import from clipboard", ex); //NOI18N
}
loadFromFile = false;
}
} else {
loadFromFile = false;
}
}
if (loadFromFile) {
final File selected = dialogProvider.msgOpenFileDialog("select-image-file", BUNDLE.getString("Images.Plugin.Load.DialogTitle"), lastSelectedFile, true, IMAGE_FILE_FILTER, BUNDLE.getString("Images.Plugin.Load.Dialog.Button.Open")); //NOI18N
if (selected != null) {
lastSelectedFile = selected;
try {
setAttribute(Utils.rescaleImageAndEncodeAsBase64(selected, Utils.getMaxImageSize()), topic, selectedTopics);
panel.notifyModelChanged();
}
catch (final IllegalArgumentException ex) {
dialogProvider.msgError(BUNDLE.getString("Images.Plugin.Error"));
LOGGER.warn("Can't load image file : " + selected); //NOI18N
}
catch (final Exception ex) {
dialogProvider.msgError(BUNDLE.getString("Images.Plugin.Error"));
LOGGER.error("Unexpected error during loading of image file : " + selected, ex); //NOI18N
}
}
}
}
});
}
return result;
}
@Nonnull
private JPanel makeSelectPanel(@Nonnull @MustNotContainNull final String[] options, @Nonnull final AtomicInteger selected) {
final JPanel panel = new JPanel(new GridBagLayout());
final GridBagConstraints constraint = new GridBagConstraints();
constraint.gridx = 0;
constraint.fill = GridBagConstraints.HORIZONTAL;
constraint.anchor = GridBagConstraints.WEST;
final ButtonGroup group = new ButtonGroup();
int selectedIndex = lastSelectedImportIndex;
for (int i = 0; i < options.length; i++) {
final JRadioButton button = new JRadioButton(options[i]);
if (selectedIndex == i) {
button.setSelected(true);
selected.set(i);
}
group.add(button);
final int currentIndex = i;
button.addActionListener(new ActionListener() {
@Override
public void actionPerformed(@Nonnull final ActionEvent e) {
if (button.isSelected()) {
selected.set(currentIndex);
}
}
});
panel.add(button, constraint);
}
return panel;
}
private boolean containAttribute(@Nullable final Topic topic, @Nonnull @MustNotContainNull final Topic[] topics) {
boolean result = false;
if (topic != null) {
result |= topic.getAttribute(ImageVisualAttributePlugin.ATTR_KEY) != null;
}
if (!result) {
for (final Topic t : topics) {
result |= t.getAttribute(ImageVisualAttributePlugin.ATTR_KEY) != null;
if (result) {
break;
}
}
}
return result;
}
private void setAttribute(@Nullable final String value, @Nullable final Topic topic, @Nonnull @MustNotContainNull final Topic[] topics) {
if (topic != null) {
topic.setAttribute(ImageVisualAttributePlugin.ATTR_KEY, value);
}
for (final Topic t : topics) {
t.setAttribute(ImageVisualAttributePlugin.ATTR_KEY, value);
}
}
@Override
@Nonnull
public PopUpSection getSection() {
return PopUpSection.EXTRAS;
}
@Override
public boolean needsTopicUnderMouse() {
return true;
}
@Override
public boolean needsSelectedTopics() {
return true;
}
@Override
public int getOrder() {
return CUSTOM_PLUGIN_START - 2;
}
}