org.bimserver.plugins.objectidms.FieldIgnoreMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of PluginBase Show documentation
Base project for BIMserver plugin development. Some plugins mights also need the Shared library
There is a newer version: 2.0.0
package org.bimserver.plugins.objectidms;

/******************************************************************************
 * Copyright (C) 2009-2016  BIMserver.org
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see {@literal}.
 *****************************************************************************/

import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EStructuralFeature;

public class FieldIgnoreMap implements ObjectIDM {
	private final Set generalIgnoreSet = new HashSet();
	private final Map> specificIncludeMap = new HashMap>();
	private final Set packages;

	public FieldIgnoreMap(Set ePackages) {
		packages = ePackages;
	}
	
	public void addToGeneralIgnoreSet(StructuralFeatureIdentifier structuralFeatureIdentifier) {
		generalIgnoreSet.add(structuralFeatureIdentifier);
	}

	public void addToSpecificIncludeMap(EClass eClass, StructuralFeatureIdentifier structuralFeatureIdentifier) {
		if (specificIncludeMap.containsKey(eClass)) {
			specificIncludeMap.get(eClass).add(structuralFeatureIdentifier);
		} else {
			HashSet hashSet = new HashSet();
			specificIncludeMap.put(eClass, hashSet);
			specificIncludeMap.put(eClass, hashSet);
		}
	}
	
	protected EClass getEClass(String className) {
		for (EPackage ePackage : packages) {
			if (ePackage.getEClassifier(className) != null) {
				return (EClass)ePackage.getEClassifier(className);
			}
		}
		return null;
	}
	
	public void removeFromGeneralIgnoreSet(EStructuralFeature eStructuralFeature) {
		Iterator iterator = generalIgnoreSet.iterator();
		while (iterator.hasNext()) {
			StructuralFeatureIdentifier next = iterator.next();
			if (next.geteStructuralFeature().equals(eStructuralFeature.getName())) {
				iterator.remove();
			}
		}
	}
	
	public boolean removeFromGeneralIgnoreSet(StructuralFeatureIdentifier structuralFeatureIdentifier) {
		return generalIgnoreSet.remove(structuralFeatureIdentifier);
	}
	
	protected EStructuralFeature getEStructuralFeature(String className, String fieldName) {
		for (EPackage ePackage : packages) {
			if (ePackage.getEClassifier(className) != null) {
				EClass eClass = (EClass)ePackage.getEClassifier(className);
				return eClass.getEStructuralFeature(fieldName);
			}
		}
		return null;
	}

	public boolean shouldFollowReference(EClass originalQueryClass, EClass eClass, EStructuralFeature eStructuralFeature) {
		StructuralFeatureIdentifier o = new StructuralFeatureIdentifier(eClass.getName(), eStructuralFeature.getName());
		boolean generalShouldIgnoreAnswer = generalIgnoreSet.contains(o);
		if (generalShouldIgnoreAnswer) {
			// If the general ignore file says ignore, we should check for a specific override
			if (specificIncludeMap.containsKey(originalQueryClass)) {
				return specificIncludeMap.get(originalQueryClass).contains(o);
			}
			return false;
		}
		return true;
	}

	@Override
	public boolean shouldIncludeClass(EClass originalClass, EClass eClass) {
		return true;
	}
}