/* * Forge Mod Loader * Copyright (c) 2012-2013 cpw. * All rights reserved. This program and the accompanying materials * are made available under the terms of the GNU Lesser Public License v2.1 * which accompanies this distribution, and is available at * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html * * Contributors: * cpw - implementation */ package net.minecraftforge.fml.common.discovery; import java.util.List; import java.util.Map; import java.util.Set; import net.minecraftforge.fml.common.ModContainer; import com.google.common.base.Predicate; import com.google.common.collect.HashMultimap; import com.google.common.collect.ImmutableMap; import com.google.common.collect.ImmutableSetMultimap; import com.google.common.collect.Lists; import com.google.common.collect.Multimap; import com.google.common.collect.Multimaps; import com.google.common.collect.SetMultimap; public class ASMDataTable { public final static class ASMData implements Cloneable { private ModCandidate candidate; private String annotationName; private String className; private String objectName; private Map<String,Object> annotationInfo; public ASMData(ModCandidate candidate, String annotationName, String className, String objectName, Map<String,Object> info) { this.candidate = candidate; this.annotationName = annotationName; this.className = className; this.objectName = objectName; this.annotationInfo = info; } public ModCandidate getCandidate() { return candidate; } public String getAnnotationName() { return annotationName; } public String getClassName() { return className; } public String getObjectName() { return objectName; } public Map<String, Object> getAnnotationInfo() { return annotationInfo; } public ASMData copy(Map<String,Object> newAnnotationInfo) { try { ASMData clone = (ASMData) this.clone(); clone.annotationInfo = newAnnotationInfo; return clone; } catch (CloneNotSupportedException e) { throw new RuntimeException("Unpossible", e); } } } private static class ModContainerPredicate implements Predicate<ASMData> { private ModContainer container; public ModContainerPredicate(ModContainer container) { this.container = container; } @Override public boolean apply(ASMData data) { return container.getSource().equals(data.candidate.getModContainer()); } } private SetMultimap<String, ASMData> globalAnnotationData = HashMultimap.create(); private Map<ModContainer, SetMultimap<String,ASMData>> containerAnnotationData; private List<ModContainer> containers = Lists.newArrayList(); private SetMultimap<String,ModCandidate> packageMap = HashMultimap.create(); public SetMultimap<String,ASMData> getAnnotationsFor(ModContainer container) { if (containerAnnotationData == null) { ImmutableMap.Builder<ModContainer, SetMultimap<String, ASMData>> mapBuilder = ImmutableMap.<ModContainer, SetMultimap<String,ASMData>>builder(); for (ModContainer cont : containers) { Multimap<String, ASMData> values = Multimaps.filterValues(globalAnnotationData, new ModContainerPredicate(cont)); mapBuilder.put(cont, ImmutableSetMultimap.copyOf(values)); } containerAnnotationData = mapBuilder.build(); } return containerAnnotationData.get(container); } public Set<ASMData> getAll(String annotation) { return globalAnnotationData.get(annotation); } public void addASMData(ModCandidate candidate, String annotation, String className, String objectName, Map<String,Object> annotationInfo) { globalAnnotationData.put(annotation, new ASMData(candidate, annotation, className, objectName, annotationInfo)); } public void addContainer(ModContainer container) { this.containers.add(container); } public void registerPackage(ModCandidate modCandidate, String pkg) { this.packageMap.put(pkg,modCandidate); } public Set<ModCandidate> getCandidatesFor(String pkg) { return this.packageMap.get(pkg); } }