bundles/core/tools/visualization/org.eclipse.papyrus.moka.datavisualization.ui/src/org/eclipse/papyrus/moka/datavisualization/service/VariableEntryTable.java

/*****************************************************************************
 * Copyright (c) 2016 CEA LIST.
 *
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *  David LOPEZ BETANCUR (CEA LIST)
 *  Sebastien REVOL (CEA LIST)
 *
 *****************************************************************************/
package org.eclipse.papyrus.moka.datavisualization.service;

import java.util.HashMap;
import java.util.Set;

import org.eclipse.papyrus.moka.xygraph.mapping.common.Variable.VariableID;

public class VariableEntryTable<E extends VariableEntry> {
	
	private HashMap<VariableID, E> variables;
	
	public VariableEntryTable() {
		super();
		variables = new HashMap<>();
	}
	
	public void addEntryFor(VariableID variableID, E entry){
		variables.put(variableID, entry);
	}

	public boolean hasVariable(VariableID id){
		return variables.containsKey(id);
	}
	
	public VariableEntry getEntry(VariableID id){		
		return variables.get(id);
	}
	
	public Set<VariableID> getRegisteredVariableIDs(){
		return variables.keySet();
	}

	public void remove(VariableID id) {
		variables.remove(id);
	}
}