com.igormaznitsa.jhexed.swing.editor.filecontainer.FileContainer.java Source code

Introduction

Here is the source code for com.igormaznitsa.jhexed.swing.editor.filecontainer.FileContainer.java

Source

/* 
 * Copyright 2014 Igor Maznitsa (http://www.igormaznitsa.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.igormaznitsa.jhexed.swing.editor.filecontainer;

import java.io.*;
import java.util.*;
import org.apache.commons.io.IOUtils;

public class FileContainer implements Iterable<FileContainerSection> {

    private final List<FileContainerSection> sections;

    private static final int MAGIC = 0xCCAA01BB;
    private static final int FORMAT_VERSION = 0x0100;

    public FileContainer(final File file) throws IOException {
        FileInputStream in = null;
        try {
            in = new FileInputStream(file);
            this.sections = loadFromStream(in);
        } finally {
            IOUtils.closeQuietly(in);
        }
    }

    public FileContainer(final InputStream stream) throws IOException {
        this.sections = loadFromStream(stream);
    }

    public FileContainer(final FileContainerSection... sections) {
        this.sections = new ArrayList<FileContainerSection>();
        this.sections.addAll(Arrays.asList(sections));
    }

    public void addSection(final FileContainerSection section) {
        if (section == null) {
            throw new NullPointerException("Section is null");
        }
        for (final FileContainerSection s : this.sections) {
            if (section.getSectionName().equals(s.getSectionName())) {
                throw new IllegalArgumentException("Duplicated section name '" + s.getSectionName() + '\'');
            }
            if (section.getUID().equals(s.getUID())) {
                throw new IllegalArgumentException("Duplicated UUID name '" + s.getUID() + '\'');
            }
        }
        this.sections.add(section);
    }

    public void write(final OutputStream out) throws IOException {
        final DataOutputStream dout = out instanceof DataOutputStream ? (DataOutputStream) out
                : new DataOutputStream(out);
        dout.writeInt(MAGIC);
        dout.writeShort(FORMAT_VERSION);

        dout.writeShort(this.sections.size());
        for (final FileContainerSection s : this.sections) {
            s.write(dout);
        }
        dout.writeInt(MAGIC);

        dout.flush();
    }

    private List<FileContainerSection> loadFromStream(final InputStream in) throws IOException {
        final DataInputStream din = in instanceof DataInputStream ? (DataInputStream) in : new DataInputStream(in);

        if (din.readInt() != MAGIC) {
            throw new IOException("Wrong format, can't find magic");
        }

        final int version = din.readShort() & 0xFFFF;

        if (version > FORMAT_VERSION) {
            throw new IllegalArgumentException("Detected unsupported version [" + version + ']');
        }

        final int sectionNumber = din.readUnsignedShort();

        final List<FileContainerSection> result = new ArrayList<FileContainerSection>(Math.max(5, sectionNumber));

        for (int i = 0; i < sectionNumber; i++) {
            final FileContainerSection s = new FileContainerSection(in);
            result.add(s);
        }
        if (din.readInt() != MAGIC) {
            throw new IOException("Can't detecte the end MAGIC");
        }

        return result;
    }

    public int getSectionsNumber() {
        return this.sections.size();
    }

    public void clear() {
        this.sections.clear();
    }

    public FileContainerSection findSectionForName(final String str) {
        for (final FileContainerSection s : this.sections) {
            if (s.getSectionName().equals(str)) {
                return s;
            }
        }
        return null;
    }

    @Override
    public Iterator<FileContainerSection> iterator() {
        return this.sections.iterator();
    }

}