com.igormaznitsa.mindmap.plugins.attributes.images.ImageVisualAttributePlugin 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.Extra;
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.api.MindMapPlugin;
import com.igormaznitsa.mindmap.plugins.api.PluginContext;
import com.igormaznitsa.mindmap.plugins.api.Renderable;
import com.igormaznitsa.mindmap.plugins.api.VisualAttributePlugin;
import com.igormaznitsa.mindmap.swing.i18n.MmdI18n;
import com.igormaznitsa.mindmap.swing.ide.IDEBridgeFactory;
import com.igormaznitsa.mindmap.swing.panel.MindMapPanelConfig;
import com.igormaznitsa.mindmap.swing.panel.utils.Utils;
import java.awt.Image;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.net.URISyntaxException;
import java.util.Map;
import java.util.ResourceBundle;
import java.util.Set;
import java.util.WeakHashMap;
import java.util.regex.Pattern;
import javax.imageio.ImageIO;

public class ImageVisualAttributePlugin implements VisualAttributePlugin {

  private static final Logger LOGGER = LoggerFactory.getLogger(ImageVisualAttributePlugin.class);
  private static final Map CACHED_IMAGES = new WeakHashMap<>();
  private final ResourceBundle resourceBundle = MmdI18n.getInstance().findBundle();

  public static void clearCachedImages() {
    CACHED_IMAGES.clear();
  }

  @Override
  public boolean doesTopicContentMatches(
      final Topic topic,
      final File baseFolder,
      final Pattern pattern,
      final Set extraTypes
  ) {
    boolean result = false;
    if (extraTypes != null && topic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_DATA) != null) {
      if (extraTypes.contains(Extra.ExtraType.NOTE)) {
        final String text = topic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_NAME);
        if (text != null) {
          result = pattern.matcher(text).find();
        }
      }
      if (!result &&
          (extraTypes.contains(Extra.ExtraType.LINK) ||
              extraTypes.contains(Extra.ExtraType.FILE))) {
        final String text = topic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_URI);
        if (text != null) {
          try {
            result =
                pattern.matcher(MMapURI.makeFromFilePath(baseFolder, text, null).toString()).find();
          } catch (URISyntaxException ex) {
            // ignore
          }
        }
      }
    }
    return result;
  }

  @Override
  public Renderable getScaledImage(final MindMapPanelConfig config,
                                   final Topic activeTopic) {
    Renderable result = CACHED_IMAGES.get(activeTopic);
    if (result == null) {
      result = new ScalableRenderableImage(extractImage(activeTopic));
      CACHED_IMAGES.put(activeTopic, result);
    }
    return result;
  }

  private Image extractImage(final Topic topic) {
    Image result = null;
    final String encoded = topic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_DATA);
    if (encoded != null) {
      try {
        result = ImageIO.read(new ByteArrayInputStream(Utils.base64decode(encoded)));
      } catch (Exception ex) {
        LOGGER.error("Can't extract image", ex);
      }
    }
    return result;
  }

  @Override
  public boolean onClick(final PluginContext context, final Topic topic,
                         final boolean activeGroupModifier, final int clickCount) {
    if (clickCount > 1) {
      final String imageFilePathUri = topic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_URI);
      if (imageFilePathUri != null) {
        try {
          context.openFile(new MMapURI(imageFilePathUri).asFile(context.getProjectFolder()), false);
        } catch (URISyntaxException ex) {
          context.getDialogProvider()
              .msgWarn(IDEBridgeFactory.findInstance().findApplicationComponent(), String.format(
                  this.resourceBundle.getString("pluginImageVisualAttr.warn.uriSyntaxError"),
                  imageFilePathUri));
        }
      }
    } else {
      if (!activeGroupModifier) {
        context.getPanel().removeAllSelection();
      }
      context.getPanel().select(topic, false);
    }
    return false;
  }

  @Override
  public String getToolTip(final PluginContext context, final Topic activeTopic) {
    String result = activeTopic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_URI);
    if (result == null) {
      result = activeTopic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_NAME);
    }
    return result;
  }

  @Override
  public boolean isClickable(final PluginContext context,
                             final Topic activeTopic) {
    final String imageFilePath = activeTopic.getAttribute(MMD_TOPIC_ATTRIBUTE_IMAGE_URI);
    return imageFilePath != null;
  }

  @Override
  public String getAttributeKey() {
    return MMD_TOPIC_ATTRIBUTE_IMAGE_DATA;
  }

  @Override
  public int getOrder() {
    return Integer.MIN_VALUE + 100;
  }

  @Override
  public int compareTo(final MindMapPlugin o) {
    return Integer.compare(this.getOrder(), o.getOrder());
  }

}