com.igormaznitsa.mindmap.plugins.attributes.images.ImagePopUpMenuPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mind-map-swing-panel Show documentation
Swing based panel to show and interact with mind map
The newest version!
/*
 * Copyright (C) 2015-2022 Igor A. 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 com.igormaznitsa.mindmap.model.MMapURI;
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.PluginContext;
import com.igormaznitsa.mindmap.swing.ide.IDEBridgeFactory;
import com.igormaznitsa.mindmap.swing.panel.utils.PathStore;
import com.igormaznitsa.mindmap.swing.panel.utils.Utils;
import com.igormaznitsa.mindmap.swing.services.IconID;
import com.igormaznitsa.mindmap.swing.services.ImageIconServiceProvider;
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.io.File;
import java.util.Locale;
import java.util.concurrent.atomic.AtomicInteger;
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 org.apache.commons.io.FilenameUtils;

public class ImagePopUpMenuPlugin extends AbstractPopupMenuItem {

  private static final Logger LOGGER = LoggerFactory.getLogger(ImagePopUpMenuPlugin.class);
  private static final Icon ICON = ImageIconServiceProvider.findInstance().getIconForId(IconID.ICON_IMAGES);
  private final FileFilter imageFileFilter = new FileFilter() {
    @Override
    public boolean accept(final File f) {
      final String text = f.getName().toLowerCase(Locale.ENGLISH);
      return f.isDirectory() || text.endsWith(".png") || text.endsWith(".jpg") ||
          text.endsWith(".gif");
    }

    @Override
    public String getDescription() {
      return getResourceBundle().getString("Images.Plugin.FilterDescription");
    }

  };
  private static final PathStore PATH_STORE = new PathStore();
  private static int lastSelectedImportIndex = 0;

  @Override
  public JMenuItem makeMenuItem(final PluginContext context, final Topic activeTopic) {
    final boolean hasAttribute = containAttribute(context, activeTopic);

    final JMenuItem result;
    if (hasAttribute) {
      result = UI_COMPO_FACTORY.makeMenuItem(
          this.getResourceBundle().getString("Images.Plugin.MenuTitle.Remove"),
          ICON);
      result.setToolTipText(
          this.getResourceBundle().getString("Images.Plugin.MenuTitle.Remove.Tooltip"));
      result.addActionListener(e -> {
        if (context.getDialogProvider()
            .msgConfirmYesNo(IDEBridgeFactory.findInstance().findApplicationComponent(),
                this.getResourceBundle().getString("Images.Plugin.Remove.Dialog.Title"),
                this.getResourceBundle().getString("Images.Plugin.Remove.Dialog.Text"))) {
          setAttribute(context, activeTopic, null, null, null);
          ImageVisualAttributePlugin.clearCachedImages();
          context.getPanel().doNotifyModelChanged(true);
        }
      });
    } else {
      result = UI_COMPO_FACTORY.makeMenuItem(
          this.getResourceBundle().getString("Images.Plugin.MenuTitle.Add"), ICON);
      result.setToolTipText(
          this.getResourceBundle().getString("Images.Plugin.MenuTitle.Add.Tooltip"));

      result.addActionListener(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 (context.getDialogProvider()
              .msgOkCancel(IDEBridgeFactory.findInstance().findApplicationComponent(),
                  this.getResourceBundle().getString("Images.Plugin.Select.DialogTitle"),
                  makeSelectPanel(
                      new String[] {
                          this.getResourceBundle().getString("Images.Plugin.Select.FromClipboard"),
                          this.getResourceBundle().getString("Images.Plugin.Select.FromFile")},
                      selectedItem))) {
            lastSelectedImportIndex = selectedItem.get();

            if (selectedItem.get() == 0) {
              try {
                final String rescaledImageAsBase64 = Utils.rescaleImageAndEncodeAsBase64(
                    (Image) transferable.getTransferData(DataFlavor.imageFlavor),
                    Utils.getMaxImageSize());
                final String filePath = null;
                setAttribute(context, activeTopic, rescaledImageAsBase64, filePath, null);
                context.getPanel().doNotifyModelChanged(true);
              } catch (final IllegalArgumentException ex) {
                context.getDialogProvider()
                    .msgError(IDEBridgeFactory.findInstance().findApplicationComponent(),
                        this.getResourceBundle().getString("Images.Plugin.Error"));
                LOGGER.error("Can't import from clipboard image", ex);
              } catch (final Exception ex) {
                context.getDialogProvider()
                    .msgError(IDEBridgeFactory.findInstance().findApplicationComponent(),
                        this.getResourceBundle().getString("Images.Plugin.Error"));
                LOGGER.error("Unexpected error during image import from clipboard", ex);
              }
              loadFromFile = false;
            }
          } else {
            loadFromFile = false;
          }
        }

        if (loadFromFile) {
          final File selected = PATH_STORE.put(context.getPanel().getUuid().toString(),
              context.getDialogProvider()
                  .msgOpenFileDialog(IDEBridgeFactory.findInstance().findApplicationComponent(),
                      context,
                      ImagePopUpMenuPlugin.class.getName(),
                      this.getResourceBundle().getString("Images.Plugin.Load.DialogTitle"),
                      PATH_STORE.find(context, context.getPanel().getUuid().toString()), true,
                      new FileFilter[] {imageFileFilter},
                      this.getResourceBundle().getString("Images.Plugin.Load.Dialog.Button.Open")));
          if (selected != null) {
            try {
              final String rescaledImageAsBase64 =
                  Utils.rescaleImageAndEncodeAsBase64(selected, Utils.getMaxImageSize());
              final String fileName = FilenameUtils.getBaseName(selected.getName());
              final String filePath;
              if (context.getDialogProvider()
                  .msgConfirmYesNo(IDEBridgeFactory.findInstance().findApplicationComponent(),
                      this.getResourceBundle()
                          .getString("Images.Plugin.Question.AddFilePath.Title"),
                      this.getResourceBundle().getString("Images.Plugin.Question.AddFilePath"))) {
                filePath =
                    MMapURI.makeFromFilePath(context.getProjectFolder(), selected.getAbsolutePath(),
                        null).toString();
              } else {
                filePath = null;
              }
              setAttribute(context, activeTopic, rescaledImageAsBase64, filePath, fileName);
              context.getPanel().doNotifyModelChanged(true);
            } catch (final IllegalArgumentException ex) {
              context.getDialogProvider()
                  .msgError(IDEBridgeFactory.findInstance().findApplicationComponent(),
                      this.getResourceBundle().getString("Images.Plugin.Error"));
              LOGGER.warn("Can't load image file : " + selected);
            } catch (final Exception ex) {
              context.getDialogProvider()
                  .msgError(IDEBridgeFactory.findInstance().findApplicationComponent(),
                      this.getResourceBundle().getString("Images.Plugin.Error"));
              LOGGER.error("Unexpected error during loading of image file : " + selected,
                  ex);
            }
          }
        }
      });
    }

    return result;
  }

  private JPanel makeSelectPanel(final String[] options, 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 = UI_COMPO_FACTORY.makeButtonGroup();

    int selectedIndex = lastSelectedImportIndex;

    for (int i = 0; i < options.length; i++) {
      final JRadioButton button = UI_COMPO_FACTORY.makeRadioButton();
      button.setText(options[i]);
      if (selectedIndex == i) {
        button.setSelected(true);
        selected.set(i);
      }
      group.add(button);

      final int currentIndex = i;

      button.addActionListener(e -> {
        if (button.isSelected()) {
          selected.set(currentIndex);
        }
      });

      panel.add(button, constraint);
    }

    return panel;
  }

  private boolean containAttribute(PluginContext context, final Topic activeTopic) {
    boolean result = false;
    if (activeTopic != null) {
      result =
          activeTopic.getAttribute(ImageVisualAttributePlugin.MMD_TOPIC_ATTRIBUTE_IMAGE_DATA) !=
              null;
    }
    if (!result) {
      for (final Topic t : context.getSelectedTopics()) {
        result = t.getAttribute(ImageVisualAttributePlugin.MMD_TOPIC_ATTRIBUTE_IMAGE_DATA) != null;
        if (result) {
          break;
        }
      }
    }
    return result;
  }

  private void setAttributeToTopic(
      final Topic topic,
      final String packedImage,
      final String imageFilePath,
      final String imageName
  ) {
    topic.putAttribute(ImageVisualAttributePlugin.MMD_TOPIC_ATTRIBUTE_IMAGE_DATA, packedImage);
    topic.putAttribute(ImageVisualAttributePlugin.MMD_TOPIC_ATTRIBUTE_IMAGE_NAME, imageName);
    topic.putAttribute(ImageVisualAttributePlugin.MMD_TOPIC_ATTRIBUTE_IMAGE_URI, imageFilePath);
  }

  private void setAttribute(final PluginContext context,
                            final Topic activeTopic,
                            final String packedImage,
                            final String imageFilePath,
                            final String imageName
  ) {
    if (activeTopic != null) {
      setAttributeToTopic(activeTopic, packedImage, imageFilePath, imageName);
    }
    for (final Topic t : context.getSelectedTopics()) {
      this.setAttributeToTopic(t, packedImage, imageFilePath, imageName);
    }
  }

  @Override
  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;
  }

}