예제 #1
0
  private void customizeComponents() {
    // initialize supported image types
    // TODO use mime-types instead once we have support
    String[] fxSupportedImagesSuffixes = ImageIO.getReaderFileSuffixes();
    imageExtensions = new String[fxSupportedImagesSuffixes.length];
    // logger.log(Level.INFO, "Supported image formats by javafx image viewer: ");
    for (int i = 0; i < fxSupportedImagesSuffixes.length; ++i) {
      String suffix = fxSupportedImagesSuffixes[i];
      // logger.log(Level.INFO, "suffix: " + suffix);
      imageExtensions[i] = "." + suffix;
    }

    add(imagePanel, IMAGE_VIEWER_LAYER);
    add(videoPanel, VIDEO_VIEWER_LAYER);

    switchPanels(false);
  }
예제 #2
0
  public String promptUserForFile() {
    try {
      JFileChooser fileOpen = new JFileChooser("Desktop\\News and Opinion Corpus - Ralph Regalado");
      String[] suffices = ImageIO.getReaderFileSuffixes();
      for (int i = 0; i < suffices.length; i++) {
        FileFilter filter = new FileNameExtensionFilter(suffices[i] + " files", suffices[i]);
        fileOpen.addChoosableFileFilter(filter);
      }
      int ret = fileOpen.showDialog(null, "Open file");
      filepath = fileOpen.getSelectedFile().getAbsolutePath();

    } catch (Exception exc) {
      System.out.println(filepath);
    }
    System.out.println(filepath);
    return filepath;
  }
  /** Open a file and load the image. */
  public void openFile() {
    JFileChooser chooser = new JFileChooser();
    chooser.setCurrentDirectory(new File("."));
    String[] extensions = ImageIO.getReaderFileSuffixes();
    chooser.setFileFilter(new FileNameExtensionFilter("Image files", extensions));
    int r = chooser.showOpenDialog(this);
    if (r != JFileChooser.APPROVE_OPTION) return;

    try {
      Image img = ImageIO.read(chooser.getSelectedFile());
      image =
          new BufferedImage(img.getWidth(null), img.getHeight(null), BufferedImage.TYPE_INT_RGB);
      image.getGraphics().drawImage(img, 0, 0, null);
    } catch (IOException e) {
      JOptionPane.showMessageDialog(this, e);
    }
    repaint();
  }
예제 #4
0
/**
 * Utilities for creating and manipulating thumbnail and icon images.
 *
 * @author jwallace
 */
public class ImageUtils {
  public static final int ICON_SIZE_SMALL = 50;
  public static final int ICON_SIZE_MEDIUM = 100;
  public static final int ICON_SIZE_LARGE = 200;
  private static final Logger logger = Logger.getLogger(ImageUtils.class.getName());
  private static final Image DEFAULT_ICON =
      new ImageIcon("/org/sleuthkit/autopsy/images/file-icon.png").getImage(); // NON-NLS
  private static final List<String> SUPP_EXTENSIONS =
      Arrays.asList(ImageIO.getReaderFileSuffixes());
  private static final List<String> SUPP_MIME_TYPES = Arrays.asList(ImageIO.getReaderMIMETypes());
  /**
   * Get the default Icon, which is the icon for a file.
   *
   * @return
   */
  public static Image getDefaultIcon() {
    return DEFAULT_ICON;
  }

  /**
   * Can a thumbnail be generated for the content?
   *
   * @param content
   * @return
   */
  public static boolean thumbnailSupported(Content content) {
    if (content instanceof AbstractFile == false) {
      return false;
    }

    AbstractFile f = (AbstractFile) content;
    if (f.getSize() == 0) {
      return false;
    }

    // check the blackboard for a file type attribute
    try {
      ArrayList<BlackboardAttribute> attributes =
          f.getGenInfoAttributes(ATTRIBUTE_TYPE.TSK_FILE_TYPE_SIG);
      for (BlackboardAttribute attribute : attributes) {
        if (SUPP_MIME_TYPES.contains(attribute.getValueString())) {
          return true;
        }
      }
    } catch (TskCoreException ex) {
      logger.log(
          Level.WARNING, "Error while getting file signature from blackboard.", ex); // NON-NLS
    }

    final String extension = f.getNameExtension();

    // if we have an extension, check it
    if (extension.equals("") == false) {
      // Note: thumbnail generator only supports JPG, GIF, and PNG for now
      if (SUPP_EXTENSIONS.contains(extension)) {
        return true;
      }
    }

    // if no extension or one that is not for an image, then read the content
    return isJpegFileHeader(f);
  }

  /**
   * Get an icon of a specified size.
   *
   * @param content
   * @param iconSize
   * @return
   */
  public static Image getIcon(Content content, int iconSize) {
    Image icon;
    // If a thumbnail file is already saved locally
    File file = getFile(content.getId());
    if (file.exists()) {
      try {
        BufferedImage bicon = ImageIO.read(file);
        if (bicon == null) {
          icon = DEFAULT_ICON;
        } else if (bicon.getWidth() != iconSize) {
          icon = generateAndSaveIcon(content, iconSize);
        } else {
          icon = bicon;
        }
      } catch (IOException ex) {
        logger.log(Level.WARNING, "Error while reading image.", ex); // NON-NLS
        icon = DEFAULT_ICON;
      }
    } else { // Make a new icon
      icon = generateAndSaveIcon(content, iconSize);
    }
    return icon;
  }

  /**
   * Get the cached file of the icon. Generates the icon and its file if it doesn't already exist,
   * so this method guarantees to return a file that exists.
   *
   * @param content
   * @param iconSize
   * @return
   */
  public static File getIconFile(Content content, int iconSize) {
    if (getIcon(content, iconSize) != null) {
      return getFile(content.getId());
    }
    return null;
  }

  /**
   * Get the cached file of the content object with the given id.
   *
   * <p>The returned file may not exist.
   *
   * @param id
   * @return
   */
  public static File getFile(long id) {
    return new File(Case.getCurrentCase().getCacheDirectory() + File.separator + id + ".png");
  }

  /**
   * Check if is jpeg file based on header
   *
   * @param file
   * @return true if jpeg file, false otherwise
   */
  public static boolean isJpegFileHeader(AbstractFile file) {
    if (file.getSize() < 100) {
      return false;
    }

    byte[] fileHeaderBuffer = new byte[2];
    int bytesRead;
    try {
      bytesRead = file.read(fileHeaderBuffer, 0, 2);
    } catch (TskCoreException ex) {
      // ignore if can't read the first few bytes, not a JPEG
      return false;
    }
    if (bytesRead != 2) {
      return false;
    }
    /*
     * Check for the JPEG header. Since Java bytes are signed, we cast them
     * to an int first.
     */
    return (((fileHeaderBuffer[0] & 0xff) == 0xff) && ((fileHeaderBuffer[1] & 0xff) == 0xd8));
  }

  private static Image generateAndSaveIcon(Content content, int iconSize) {
    Image icon = null;
    try {
      icon = generateIcon(content, iconSize);
      if (icon == null) {
        return DEFAULT_ICON;
      } else {
        File f = getFile(content.getId());
        if (f.exists()) {
          f.delete();
        }
        ImageIO.write((BufferedImage) icon, "png", getFile(content.getId())); // NON-NLS
      }
    } catch (IOException ex) {
      logger.log(Level.WARNING, "Could not write cache thumbnail: " + content, ex); // NON-NLS
    }
    return icon;
  }

  /*
   * Generate a scaled image
   */
  private static BufferedImage generateIcon(Content content, int iconSize) {

    InputStream inputStream = null;
    try {
      inputStream = new ReadContentInputStream(content);
      BufferedImage bi = ImageIO.read(inputStream);
      if (bi == null) {
        logger.log(Level.WARNING, "No image reader for file: " + content.getName()); // NON-NLS
        return null;
      }
      BufferedImage biScaled = ScalrWrapper.resizeFast(bi, iconSize);

      return biScaled;
    } catch (OutOfMemoryError e) {
      logger.log(
          Level.WARNING, "Could not scale image (too large): " + content.getName(), e); // NON-NLS
      return null;
    } catch (Exception e) {
      logger.log(Level.WARNING, "Could not scale image: " + content.getName(), e); // NON-NLS
      return null;
    } finally {
      if (inputStream != null) {
        try {
          inputStream.close();
        } catch (IOException ex) {
          logger.log(
              Level.WARNING,
              "Could not close input stream after resizing thumbnail: " + content.getName(),
              ex); // NON-NLS
        }
      }
    }
  }
}