/* Soot - a J*va Optimization Framework * Copyright (C) 2004 Jennifer Lhotak * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ package soot.javaToJimple; import java.util.*; import polyglot.types.Type; public class TypeListBuilder extends polyglot.visit.NodeVisitor { private final HashSet<Type> list; public HashSet<Type> getList() { return list; } public TypeListBuilder(){ list = new HashSet<Type>(); } public polyglot.ast.Node leave(polyglot.ast.Node old, polyglot.ast.Node n, polyglot.visit.NodeVisitor visitor) { if (n instanceof polyglot.ast.Typed) { polyglot.ast.Typed typedNode = (polyglot.ast.Typed)n; if (typedNode.type() instanceof polyglot.types.ClassType) { list.add(typedNode.type()); } else { } } if (n instanceof polyglot.ast.ClassDecl){ polyglot.ast.ClassDecl cd = (polyglot.ast.ClassDecl)n; list.add(cd.type()); } return n; } }