mpq.MpqUtil.java Source code

Introduction

Here is the source code for mpq.MpqUtil.java

Source

/*
 * Copyright (C) 2011 Thedeath<www.fseek.org>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package mpq;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.logging.Level;
import java.util.logging.Logger;
import mwt.wow.mpq.MpqFile;
import org.apache.commons.io.IOUtils;
import wowimage.BLPFile;
import wowimage.ConversionException;

public class MpqUtil {

    // if rekursivly = true all directorys in the directory also scanned
    public static ExtMpqArchive[] getMPQFiles(File directory, boolean rekursivly) {
        if (!directory.isDirectory()) {
            throw new IllegalArgumentException("The given file isnt a directory !");
        }
        return getMpqFilesFromDir(directory, rekursivly);
    }

    private static ExtMpqArchive[] getMpqFilesFromDir(File directory, boolean rekursivly) {

        ExtMpqArchive[] archives = null;
        if (rekursivly) {
            File[] listFiles = directory.listFiles(new MpqFileFilter(true));
            archives = getMpqFilesFromDirRek(listFiles);
        } else {
            File[] listFiles = directory.listFiles(new MpqFileFilter());
            archives = getMpqFilesFromDirImpl(listFiles);
        }
        return archives;
    }

    private static ExtMpqArchive[] getMpqFilesFromDirImpl(File[] listFiles) {
        ExtMpqArchive[] archives = new ExtMpqArchive[listFiles.length];
        for (int i = 0; i < listFiles.length; i++) {
            try {
                archives[i] = new ExtMpqArchive(listFiles[i]);
            } catch (IOException ex) {
                // file isnt a "real" mpq file (?)
                Logger.getLogger(MpqUtil.class.getName()).log(Level.SEVERE, null, ex);
            }
        }
        return archives;
    }

    private static ExtMpqArchive[] getMpqFilesFromDirRek(File[] listFiles) {
        ArrayList<ExtMpqArchive> archives = new ArrayList<ExtMpqArchive>();
        archives = getMpqFilesFromDirRekImpl(listFiles, archives);
        Object[] toArray = archives.toArray();
        ExtMpqArchive[] archs = new ExtMpqArchive[toArray.length];
        for (int i = 0; i < toArray.length; i++) {
            archs[i] = (ExtMpqArchive) toArray[i];
        }
        return archs;
    }

    private static ArrayList<ExtMpqArchive> getMpqFilesFromDirRekImpl(File[] listFiles,
            ArrayList<ExtMpqArchive> archives) {
        for (int i = 0; i < listFiles.length; i++) {
            try {
                File f = listFiles[i];
                if (f.isDirectory()) {
                    getMpqFilesFromDirRekImpl(f.listFiles(new MpqFileFilter(true)), archives);
                } else {
                    archives.add(new ExtMpqArchive(f));
                }
            } catch (IOException ex) {
                // file isnt a "real" mpq file (?)
                Logger.getLogger(MpqUtil.class.getName()).log(Level.SEVERE, null, ex);
            }
        }
        return archives;
    }

    public static BufferedImage convertMpqFileToImage(MpqFile file) throws ConversionException, IOException {
        //        File createTempFile = File.createTempFile(getMpqFileName(file), null);
        //        file.extractTo(createTempFile);
        BufferedImage convertFileToImage = convertFileToImage(file);
        //        boolean delete = createTempFile.delete();
        //        if(delete == false)
        //        {
        //            createTempFile.deleteOnExit();
        //        }
        return convertFileToImage;
    }

    public static String convertMpqFileToString(MpqFile file) throws IOException {
        if (file == null)
            return null;
        StringWriter writer = new StringWriter();
        IOUtils.copy(file.getInputStream(), writer);
        String theString = writer.toString();
        return theString;
    }

    public static BufferedImage convertFileToImage(File file) throws ConversionException, IOException {
        BLPFile blpFile = new BLPFile(file);
        BufferedImage img = blpFile.getImg();
        return img;
    }

    public static BufferedImage convertFileToImage(MpqFile file) throws ConversionException, IOException {
        BLPFile blpFile = new BLPFile(file.getInputStream(), (int) file.getFileSize());
        BufferedImage img = blpFile.getImg();
        return img;
    }

    public static String getMpqFileName(MpqFile file) {
        String[] split = file.getFilePath().split("\\\\");
        String mpqFileName = split[split.length - 1];
        return mpqFileName;
    }

}