mirror of
https://github.com/overcuriousity/autopsy-flatpak.git
synced 2025-07-14 08:56:15 +00:00
video thumbs IG temp
This commit is contained in:
parent
faa0c886cf
commit
733c4ee9c2
@ -195,7 +195,7 @@ public class ImageUtils {
|
|||||||
BufferedImage bicon = ImageIO.read(iconFile);
|
BufferedImage bicon = ImageIO.read(iconFile);
|
||||||
if (bicon == null) {
|
if (bicon == null) {
|
||||||
icon = DEFAULT_ICON;
|
icon = DEFAULT_ICON;
|
||||||
} else if (bicon.getWidth() != iconSize) {
|
} else if (bicon.getWidth() < iconSize) {
|
||||||
icon = generateAndSaveIcon(content, iconSize, iconFile);
|
icon = generateAndSaveIcon(content, iconSize, iconFile);
|
||||||
} else {
|
} else {
|
||||||
icon = bicon;
|
icon = bicon;
|
||||||
|
@ -23,31 +23,22 @@ import com.google.common.cache.CacheBuilder;
|
|||||||
import com.google.common.cache.CacheLoader;
|
import com.google.common.cache.CacheLoader;
|
||||||
import com.google.common.util.concurrent.UncheckedExecutionException;
|
import com.google.common.util.concurrent.UncheckedExecutionException;
|
||||||
import java.awt.image.BufferedImage;
|
import java.awt.image.BufferedImage;
|
||||||
import java.io.BufferedInputStream;
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.net.MalformedURLException;
|
import java.net.MalformedURLException;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
import java.util.concurrent.ExecutionException;
|
import java.util.concurrent.ExecutionException;
|
||||||
import java.util.concurrent.Executor;
|
|
||||||
import java.util.concurrent.Executors;
|
|
||||||
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.TimeUnit;
|
||||||
import java.util.logging.Level;
|
import java.util.logging.Level;
|
||||||
import javafx.beans.property.SimpleIntegerProperty;
|
import javafx.beans.property.SimpleIntegerProperty;
|
||||||
import javafx.embed.swing.SwingFXUtils;
|
import javafx.embed.swing.SwingFXUtils;
|
||||||
import javafx.scene.image.Image;
|
import javafx.scene.image.Image;
|
||||||
import javafx.scene.image.WritableImage;
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
import javax.imageio.ImageIO;
|
|
||||||
import org.apache.commons.lang3.concurrent.BasicThreadFactory;
|
|
||||||
import org.openide.util.Utilities;
|
import org.openide.util.Utilities;
|
||||||
import org.sleuthkit.autopsy.casemodule.Case;
|
import org.sleuthkit.autopsy.casemodule.Case;
|
||||||
import org.sleuthkit.autopsy.corelibs.ScalrWrapper;
|
import org.sleuthkit.autopsy.coreutils.ImageUtils;
|
||||||
import org.sleuthkit.autopsy.coreutils.Logger;
|
import org.sleuthkit.autopsy.coreutils.Logger;
|
||||||
import org.sleuthkit.autopsy.imagegallery.datamodel.DrawableFile;
|
import org.sleuthkit.autopsy.imagegallery.datamodel.DrawableFile;
|
||||||
import org.sleuthkit.autopsy.imagegallery.gui.Toolbar;
|
import org.sleuthkit.autopsy.imagegallery.gui.Toolbar;
|
||||||
import org.sleuthkit.datamodel.ReadContentInputStream;
|
|
||||||
import org.sleuthkit.datamodel.TskCoreException;
|
import org.sleuthkit.datamodel.TskCoreException;
|
||||||
|
|
||||||
/** Singleton to manage creation and access of icons. Keeps a cache in memory of
|
/** Singleton to manage creation and access of icons. Keeps a cache in memory of
|
||||||
@ -81,9 +72,7 @@ public enum ThumbnailCache {
|
|||||||
/** currently desired icon size. is bound in {@link Toolbar} */
|
/** currently desired icon size. is bound in {@link Toolbar} */
|
||||||
public final SimpleIntegerProperty iconSize = new SimpleIntegerProperty(200);
|
public final SimpleIntegerProperty iconSize = new SimpleIntegerProperty(200);
|
||||||
|
|
||||||
/** thread that saves generated thumbnails to disk for use later */
|
|
||||||
private final Executor imageSaver = Executors.newSingleThreadExecutor(new BasicThreadFactory.Builder().namingPattern("icon saver-%d").build());
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Clear out the cache between cases
|
* Clear out the cache between cases
|
||||||
*/
|
*/
|
||||||
@ -185,104 +174,9 @@ public enum ThumbnailCache {
|
|||||||
*/
|
*/
|
||||||
@Nullable
|
@Nullable
|
||||||
private Image generateAndSaveThumbnail(final DrawableFile<?> file) {
|
private Image generateAndSaveThumbnail(final DrawableFile<?> file) {
|
||||||
//create a buffered input stream for the underlying Abstractfile
|
|
||||||
try (InputStream inputStream = new BufferedInputStream(new ReadContentInputStream(file.getAbstractFile()))) {
|
return SwingFXUtils.toFXImage((BufferedImage) ImageUtils.getIcon(file.getAbstractFile(), MAX_ICON_SIZE), null);
|
||||||
final Image thumbnail = new Image(inputStream, MAX_ICON_SIZE, MAX_ICON_SIZE, true, true);
|
|
||||||
if (thumbnail.isError()) { //if there was an error loading the image via JFX, fall back on Swing
|
|
||||||
LOGGER.log(Level.WARNING, "problem loading thumbnail for image: " + file.getName() + " .");
|
|
||||||
// Doing it this way puts the whole stack trace in the console output, which is probably not
|
|
||||||
// needed. There are a significant number of cases where this is expected to fail (bitmaps,
|
|
||||||
// empty files, etc.)
|
|
||||||
//LOGGER.log(Level.WARNING, "problem loading image: " + file.getName() + " .", thumbnail.getException());
|
|
||||||
return fallbackToSwingImage(file);
|
|
||||||
} else { //if the load went successfully, save the thumbnail to disk on a background thread
|
|
||||||
imageSaver.execute(() -> {
|
|
||||||
saveIcon(file, thumbnail);
|
|
||||||
});
|
|
||||||
return thumbnail;
|
|
||||||
}
|
|
||||||
} catch (IOException ex) {
|
|
||||||
//if the JX load throws an exception fall back to Swing
|
|
||||||
return fallbackToSwingImage(file);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* use Swing to generate and save a thumbnail for the given file
|
|
||||||
*
|
|
||||||
* @param file
|
|
||||||
*
|
|
||||||
* @return a thumbnail generated for the given file, or {@code null} if a
|
|
||||||
* thumbnail could not be generated
|
|
||||||
*/
|
|
||||||
@Nullable
|
|
||||||
private Image fallbackToSwingImage(final DrawableFile<?> file) {
|
|
||||||
final BufferedImage generateSwingIcon = generateSwingThumbnail(file);
|
|
||||||
if (generateSwingIcon == null) { //if swing failed,
|
|
||||||
return null; //propagate failure up cal stack.
|
|
||||||
} else {//Swing load succeeded, convert to JFX Image
|
|
||||||
final WritableImage toFXImage = SwingFXUtils.toFXImage(generateSwingIcon, null);
|
|
||||||
if (toFXImage != null) { //if conversion succeeded save to disk cache
|
|
||||||
imageSaver.execute(() -> {
|
|
||||||
saveIcon(file, toFXImage);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
return toFXImage; //could be null
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* use Swing/ImageIO to generate a thumbnail for the given file
|
|
||||||
*
|
|
||||||
* @param file
|
|
||||||
*
|
|
||||||
* @return a BufferedImage thumbail for the given file, or {@code null} if a
|
|
||||||
* thumbnail could not be generated
|
|
||||||
*/
|
|
||||||
private BufferedImage generateSwingThumbnail(DrawableFile<?> file) {
|
|
||||||
//create a buffered input stream for the underlying Abstractfile
|
|
||||||
try (InputStream inputStream = new BufferedInputStream(new ReadContentInputStream(file.getAbstractFile()))) {
|
|
||||||
BufferedImage bi = ImageIO.read(inputStream);
|
|
||||||
if (bi != null) {
|
|
||||||
try { // resize (shrink) the buffered image if needed
|
|
||||||
if (Math.max(bi.getWidth(), bi.getHeight()) > MAX_ICON_SIZE) {
|
|
||||||
bi = ScalrWrapper.resizeFast(bi, iconSize.get());
|
|
||||||
}
|
|
||||||
} catch (IllegalArgumentException e) {
|
|
||||||
//if scalr failed, just use unscaled image
|
|
||||||
LOGGER.log(Level.WARNING, "scalr could not scale image to 0: {0}", file.getName());
|
|
||||||
} catch (OutOfMemoryError e) {
|
|
||||||
LOGGER.log(Level.WARNING, "scalr could not scale image (too large): {0}", file.getName());
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
} else { //ImageIO failed to read the image
|
|
||||||
LOGGER.log(Level.WARNING, "No image reader for file: {0}", file.getName());
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
return bi;
|
|
||||||
} catch (IOException ex) {
|
|
||||||
LOGGER.log(Level.WARNING, "Could not read image: " + file.getName());
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* save the generated thumbnail to disk in the cache folder with
|
|
||||||
* the obj_id as the name.
|
|
||||||
*
|
|
||||||
* @param file the file the given image is a thumbnail for
|
|
||||||
* @param bi the thumbnail to save for the given DrawableFile
|
|
||||||
*/
|
|
||||||
private void saveIcon(final DrawableFile<?> file, final Image bi) {
|
|
||||||
if (bi != null) {
|
|
||||||
getCacheFile(file.getId()).ifPresent((File cacheFile) -> {
|
|
||||||
try {
|
|
||||||
//convert back to swing to save
|
|
||||||
ImageIO.write(SwingFXUtils.fromFXImage(bi, null), FORMAT, cacheFile);
|
|
||||||
} catch (IllegalArgumentException | IOException ex) {
|
|
||||||
LOGGER.log(Level.WARNING, "failed to save generated icon");
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -21,6 +21,7 @@ package org.sleuthkit.autopsy.imagegallery.gui.drawableviews;
|
|||||||
import java.util.Objects;
|
import java.util.Objects;
|
||||||
import java.util.logging.Level;
|
import java.util.logging.Level;
|
||||||
import javafx.application.Platform;
|
import javafx.application.Platform;
|
||||||
|
import javafx.concurrent.Task;
|
||||||
import javafx.fxml.FXML;
|
import javafx.fxml.FXML;
|
||||||
import javafx.scene.CacheHint;
|
import javafx.scene.CacheHint;
|
||||||
import javafx.scene.control.Control;
|
import javafx.scene.control.Control;
|
||||||
@ -113,9 +114,30 @@ public class DrawableTile extends DrawableTileBase {
|
|||||||
@Override
|
@Override
|
||||||
protected Runnable getContentUpdateRunnable() {
|
protected Runnable getContentUpdateRunnable() {
|
||||||
if (getFile().isPresent()) {
|
if (getFile().isPresent()) {
|
||||||
Image image = getFile().get().getThumbnail();
|
|
||||||
|
|
||||||
return () -> {
|
return () -> {
|
||||||
|
new Task<Image>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Image call() throws Exception {
|
||||||
|
Image image = getFile().get().getThumbnail();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void failed() {
|
||||||
|
super.failed();
|
||||||
|
LOGGER.log(null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void succeeded() {
|
||||||
|
super.succeeded(); //To change body of generated methods, choose Tools | Templates.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
imageView.setImage(image);
|
imageView.setImage(image);
|
||||||
};
|
};
|
||||||
} else {
|
} else {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user