/*
 * 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
 */
package com.ibm.wala.analysis.pointers;

import com.ibm.wala.ipa.callgraph.propagation.HeapModel;
import com.ibm.wala.ipa.callgraph.propagation.InstanceKey;
import com.ibm.wala.ipa.callgraph.propagation.PointerAnalysis;
import com.ibm.wala.ipa.callgraph.propagation.PointerKey;
import com.ibm.wala.util.graph.Graph;
import com.ibm.wala.util.graph.impl.NumberedNodeIterator;
import com.ibm.wala.util.graph.traverse.DFS;
import com.ibm.wala.util.intset.IntSet;
import java.util.Collection;
import java.util.Iterator;
import java.util.Set;

/**
 * A {@link Graph} view of a pointer analysis solution.
 *
 * <p>Nodes in the Graph are {@link PointerKey}s and {@link InstanceKey}s.
 *
 * <p>There is an edge from a PointerKey P to an InstanceKey I iff the PointerAnalysis indicates
 * that P may point to I.
 *
 * <p>There is an edge from an InstanceKey I to a PointerKey P iff - P represents a field of an
 * object instance modeled by I, or - P represents the array contents of array instance I.
 */
public abstract class HeapGraphImpl<T extends InstanceKey> implements HeapGraph<T> {

  private final PointerAnalysis<T> pa;

  protected HeapGraphImpl(PointerAnalysis<T> pa) {
    if (pa == null) {
      throw new IllegalArgumentException("null pa ");
    }
    this.pa = pa;
  }

  @Override
  public Iterator<Object> iterateNodes(IntSet s) {
    return new NumberedNodeIterator<>(s, this);
  }

  @Override
  public Collection<Object> getReachableInstances(Set<Object> roots) {
    return DFS.getReachableNodes(this, roots, InstanceKey.class::isInstance);
  }

  @Override
  public void removeNodeAndEdges(Object N) throws UnsupportedOperationException {
    throw new UnsupportedOperationException();
  }

  /** @return the heap model used in this pointer analysis. */
  @Override
  public HeapModel getHeapModel() {
    return pa.getHeapModel();
  }

  @Override
  public PointerAnalysis<T> getPointerAnalysis() {
    return pa;
  }
}
