WALA/com.ibm.wala.cast/source/java/com/ibm/wala/cast/tree/impl/CAstNodeTypeMapRecorder.java

49 lines
1.3 KiB
Java

/******************************************************************************
* Copyright (c) 2002 - 2006 IBM Corporation.
* 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:
* IBM Corporation - initial API and implementation
*****************************************************************************/
/*
* Created on Oct 10, 2005
*/
package com.ibm.wala.cast.tree.impl;
import java.util.Collection;
import java.util.HashMap;
import com.ibm.wala.cast.tree.CAstNode;
import com.ibm.wala.cast.tree.CAstNodeTypeMap;
import com.ibm.wala.cast.tree.CAstType;
public class CAstNodeTypeMapRecorder
extends HashMap <CAstNode,CAstType>
implements CAstNodeTypeMap
{
private static final long serialVersionUID= 7812144102027916961L;
@Override
public CAstType getNodeType(CAstNode node) {
return get(node);
}
public void add(CAstNode node, CAstType type) {
put(node, type);
}
@Override
public Collection<CAstNode> getMappedNodes() {
return keySet();
}
public void addAll(CAstNodeTypeMap other) {
for(CAstNode o : other.getMappedNodes()) {
put(o, other.getNodeType(o));
}
}
}