dotaSoundEditor.Helpers.PortraitFinder.java Source code

Introduction

Here is the source code for dotaSoundEditor.Helpers.PortraitFinder.java

Source

/* 
 * The MIT License
 *
 * Copyright 2015 Neil McAlister.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package dotaSoundEditor.Helpers;

import info.ata4.vpk.VPKArchive;
import info.ata4.vpk.VPKEntry;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.nio.channels.FileChannel;
import java.util.TreeMap;
import javax.imageio.ImageIO;
import org.apache.commons.io.FileUtils;

/**
 *
 * @author
 * Image
 * 17
 */
public class PortraitFinder {

    private String fileName;
    private TreeMap<String, BufferedImage> portraitMap = new TreeMap<>();

    public PortraitFinder(String _fileName) {
        fileName = _fileName;
    }

    public void buildPortraits() {
        File file = new File(fileName);
        VPKArchive vpk = new VPKArchive();
        try {
            vpk.load(file);
        } catch (Exception ex) {
            System.err.println("Can't open archive: " + ex.getMessage());
            return;
        }
        buildHeroPortraits(vpk);
        buildItemPortraits(vpk);
        buildAnnouncerPortraits(vpk);
    }

    private void buildHeroPortraits(VPKArchive vpk) {

        BufferedImage image = null;
        for (VPKEntry entry : vpk.getEntriesForDir("resource/flash3/images/heroes/")) {
            if (entry.getType().equals("png") && !(entry.getPath().contains("selection"))) {
                File imageFile = new File(entry.getPath());

                try (FileChannel fc = FileUtils.openOutputStream(imageFile).getChannel()) {
                    fc.write(entry.getData());
                    image = ImageIO.read(imageFile);
                    portraitMap.put(entry.getName(), image);
                } catch (IOException ex) {
                    System.err.println("Can't write " + entry.getPath() + ": " + ex.getMessage());
                }
            }
        }
    }

    private void buildItemPortraits(VPKArchive vpk) {

        BufferedImage image = null;
        for (VPKEntry entry : vpk.getEntriesForDir("resource/flash3/images/items/")) {
            if (entry.getType().equals("png")) {
                File imageFile = new File(entry.getPath());

                try (FileChannel fc = FileUtils.openOutputStream(imageFile).getChannel()) {
                    fc.write(entry.getData());
                    image = ImageIO.read(imageFile);
                    String item = entry.getName();
                    portraitMap.put(item, image);
                } catch (IOException ex) {
                    System.err.println("Can't write " + entry.getPath() + ": " + ex.getMessage());
                }
            }
        }
    }

    private void buildAnnouncerPortraits(VPKArchive vpk) {

        BufferedImage image = null;
        for (VPKEntry entry : vpk.getEntriesForDir("resource/flash3/images/econ/announcer/")) {
            if (entry.getName().contains("large")) {
                continue; //don't grab the huge images, we won't use them.
            }

            File imageFile = new File(entry.getPath());

            try (FileChannel fc = FileUtils.openOutputStream(imageFile).getChannel()) {
                fc.write(entry.getData());
                image = ImageIO.read(imageFile);
                String announcer = entry.getName();
                portraitMap.put(announcer, image);
            } catch (IOException ex) {
                System.err.println("Can't write " + entry.getPath() + ": " + ex.getMessage());
            }
        }
    }

    public BufferedImage getPortrait(String portraitName) {
        if (portraitMap.containsKey(portraitName)) {
            return portraitMap.get(portraitName);
        } else {
            return portraitMap.get("default");
        }
    }
}