net.rptools.maptool.util.AssetExtractor.java Source code

Introduction

Here is the source code for net.rptools.maptool.util.AssetExtractor.java

Source

/*
 * This software copyright by various authors including the RPTools.net
 * development team, and licensed under the LGPL Version 3 or, at your option,
 * any later version.
 *
 * Portions of this software were originally covered under the Apache Software
 * License, Version 1.1 or Version 2.0.
 *
 * See the file LICENSE elsewhere in this distribution for license details.
 */

package net.rptools.maptool.util;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.io.Reader;
import java.util.Set;

import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;

import net.rptools.lib.FileUtil;
import net.rptools.lib.io.PackedFile;
import net.rptools.lib.swing.SwingUtil;
import net.rptools.maptool.client.MapTool;
import net.rptools.maptool.model.Asset;

import org.apache.commons.io.IOUtils;

import com.thoughtworks.xstream.XStream;

/**
 * Appears to be unused within MapTool.  What was its original purpose?  It appears
 * to be some way to extract individual images from a campaign, but it has multiple
 * problems:  always names output images with <b>.jpg</b> extensions, doesn't
 * allow a choice of which images are extracted, doesn't turn on annotation processing
 * for {@link Asset} objects (needed for XStream processing), and doesn't used
 * buffered I/O classes.
 * 
 * @author ??
 */
public class AssetExtractor {
    public static void extract() throws Exception {
        new Thread() {
            @Override
            public void run() {
                JFileChooser chooser = new JFileChooser();
                chooser.setMultiSelectionEnabled(false);
                chooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
                if (chooser.showOpenDialog(null) != JFileChooser.APPROVE_OPTION) {
                    return;
                }
                File file = chooser.getSelectedFile();
                File newDir = new File(file.getParentFile(),
                        file.getName().substring(0, file.getName().lastIndexOf('.')) + "_images");

                JLabel label = new JLabel("", JLabel.CENTER);
                JFrame frame = new JFrame();
                frame.setTitle("Campaign Image Extractor");
                frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
                frame.setSize(400, 75);
                frame.add(label);
                SwingUtil.centerOnScreen(frame);
                frame.setVisible(true);
                Reader r = null;
                OutputStream out = null;
                PackedFile pakfile = null;
                try {
                    newDir.mkdirs();

                    label.setText("Loading campaign ...");
                    pakfile = new PackedFile(file);

                    Set<String> files = pakfile.getPaths();
                    XStream xstream = new XStream();
                    int count = 0;
                    for (String filename : files) {
                        count++;
                        if (filename.indexOf("assets") < 0) {
                            continue;
                        }
                        r = pakfile.getFileAsReader(filename);
                        Asset asset = (Asset) xstream.fromXML(r);
                        IOUtils.closeQuietly(r);

                        File newFile = new File(newDir, asset.getName() + ".jpg");
                        label.setText("Extracting image " + count + " of " + files.size() + ": " + newFile);
                        if (newFile.exists()) {
                            newFile.delete();
                        }
                        newFile.createNewFile();
                        out = new FileOutputStream(newFile);
                        FileUtil.copyWithClose(new ByteArrayInputStream(asset.getImage()), out);
                    }
                    label.setText("Done.");
                } catch (Exception ioe) {
                    MapTool.showInformation("AssetExtractor failure", ioe);
                } finally {
                    if (pakfile != null)
                        pakfile.close();
                    IOUtils.closeQuietly(r);
                    IOUtils.closeQuietly(out);
                }
            }
        }.start();
    }
}