ArchetypeTypeSet.java

Go to the documentation of this file.

/*
 * Gridarta MMORPG map editor for Crossfire, Daimonin and similar games.
 * Copyright (C) 2000-2015 The Gridarta Developers.
 *
 * 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 2 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package net.sf.gridarta.model.archetypetype;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import net.sf.gridarta.model.baseobject.BaseObject;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class ArchetypeTypeSet implements Iterable<ArchetypeType> {

    @NotNull
    private ArchetypeType fallbackArchetypeType = new ArchetypeType("", 0, "", true, null, false, null, null, new ArchetypeAttributesDefinition());

    @NotNull
    private final List<ArchetypeType> archetypeTypeList = new ArrayList<>();

    @NotNull
    private final Map<String, ArchetypeType> archetypeTypeNames = new HashMap<>();

    @NotNull
    private final Map<String, ArchetypeTypeList> listTable = new HashMap<>();

    @NotNull
    private final Map<String, AttributeBitmask> bitmaskTable = new HashMap<>();

    @NotNull
    private final Map<String, Integer> attributeOrder = new LinkedHashMap<>();

    @NotNull
    private final Comparator<String> attributeOrderComparator = new Comparator<String>() {

        @Override
        public int compare(@NotNull final String o1, @NotNull final String o2) {
            final Integer index1 = attributeOrder.get(o1);
            final Integer index2 = attributeOrder.get(o2);
            if (index1 == null) {
                if (index2 == null) {
                    return o1.compareTo(o2);
                }
                return +1;
            }
            if (index2 == null) {
                return -1;
            }
            if (index1 < index2) {
                return -1;
            }
            if (index1 > index2) {
                return +1;
            }
            return 0;
        }

    };

    public void addArchetypeType(@NotNull final ArchetypeType archetypeType) {
        archetypeTypeList.add(archetypeType);
        final String typeName = archetypeType.getTypeName();
        archetypeTypeNames.put(typeName, archetypeType);
        if (typeName.equals("Misc")) {
            fallbackArchetypeType = archetypeType;
        }
    }

    @Nullable
    public ArchetypeType getArchetypeType(@NotNull final String typeName) {
        return archetypeTypeNames.get(typeName);
    }

    @NotNull
    public ArchetypeType getArchetypeTypeByName(@NotNull final String typeName) {
        final ArchetypeType type = archetypeTypeNames.get(typeName.trim());
        if (type != null) {
            return type;
        }

        return fallbackArchetypeType;
    }

    @NotNull
    public ArchetypeType getArchetypeTypeByBaseObject(@NotNull final BaseObject<?, ?, ?, ?> baseObject) {
        for (final ArchetypeType archetypeType : archetypeTypeList) {
            if (archetypeType.matches(baseObject)) {
                return archetypeType;
            }
        }

        return fallbackArchetypeType;
    }

    @NotNull
    @Override
    public Iterator<ArchetypeType> iterator() {
        return Collections.unmodifiableList(archetypeTypeList).iterator();
    }

    public int getArchetypeTypeIndex(@NotNull final ArchetypeType archetypeType) {
        return archetypeTypeList.indexOf(archetypeType);
    }

    public int getArchetypeTypeCount() {
        return archetypeTypeList.size();
    }

    public boolean isFallbackArchetypeType(@NotNull final ArchetypeType archetypeType) {
        return archetypeType == fallbackArchetypeType;
    }

    public void addList(@NotNull final String attribute, @NotNull final ArchetypeTypeList list) {
        listTable.put(attribute, list);
    }

    @Nullable
    public ArchetypeTypeList getList(@NotNull final String listName) {
        return listTable.get(listName);
    }

    public void addBitmask(@NotNull final String attribute, @NotNull final AttributeBitmask attributeBitmask) {
        bitmaskTable.put(attribute, attributeBitmask);
    }

    @Nullable
    public AttributeBitmask getBitmask(@NotNull final String bitmaskName) {
        return bitmaskTable.get(bitmaskName);
    }

    public boolean addAttributeOrder(@NotNull final String attributeName) {
        if (attributeOrder.containsKey(attributeName)) {
            return true;
        }

        attributeOrder.put(attributeName, attributeOrder.size());
        return false;
    }

    @NotNull
    public Comparator<String> getAttributeOrderComparator() {
        return attributeOrderComparator;
    }

    @NotNull
    public String getDisplayName(@NotNull final BaseObject<?, ?, ?, ?> baseObject) {
        return getArchetypeTypeByBaseObject(baseObject).getDisplayName(baseObject);
    }

    @NotNull
    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder();
        final Map<String, ArchetypeType> sortedArchetypeTypes = new TreeMap<>(archetypeTypeNames);
        for (final ArchetypeType archetypeType : sortedArchetypeTypes.values()) {
            archetypeType.toString(sb);
        }
        return sb.toString();
    }

}