001/* 002 * Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved. 003 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 004 * 005 * This code is free software; you can redistribute it and/or modify it 006 * under the terms of the GNU General Public License version 2 only, as 007 * published by the Free Software Foundation. 008 * 009 * This code is distributed in the hope that it will be useful, but WITHOUT 010 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 011 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 012 * version 2 for more details (a copy is included in the LICENSE file that 013 * accompanied this code). 014 * 015 * You should have received a copy of the GNU General Public License version 016 * 2 along with this work; if not, write to the Free Software Foundation, 017 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 018 * 019 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 020 * or visit www.oracle.com if you need additional information or have any 021 * questions. 022 */ 023package com.oracle.graal.graph.iterators; 024 025import java.util.*; 026 027import com.oracle.graal.graph.*; 028 029public class FilteredNodeIterable<T extends Node> implements NodeIterable<T> { 030 031 protected final NodeIterable<T> nodeIterable; 032 protected NodePredicate predicate = NodePredicates.alwaysTrue(); 033 034 public FilteredNodeIterable(NodeIterable<T> nodeIterable) { 035 this.nodeIterable = nodeIterable; 036 } 037 038 public FilteredNodeIterable<T> and(NodePredicate nodePredicate) { 039 this.predicate = this.predicate.and(nodePredicate); 040 return this; 041 } 042 043 public FilteredNodeIterable<T> or(NodePredicate nodePredicate) { 044 this.predicate = this.predicate.or(nodePredicate); 045 return this; 046 } 047 048 @Override 049 public FilteredNodeIterable<T> nonNull() { 050 this.predicate = this.predicate.and(NodePredicates.isNotNull()); 051 return this; 052 } 053 054 @Override 055 public DistinctFilteredNodeIterable<T> distinct() { 056 DistinctFilteredNodeIterable<T> distinct = new DistinctFilteredNodeIterable<>(nodeIterable); 057 distinct.predicate = predicate; 058 return distinct; 059 } 060 061 @Override 062 public Iterator<T> iterator() { 063 return new PredicatedProxyNodeIterator<>(nodeIterable.iterator(), predicate); 064 } 065 066 @SuppressWarnings("unchecked") 067 @Override 068 public <F extends T> FilteredNodeIterable<F> filter(Class<F> clazz) { 069 return (FilteredNodeIterable<F>) this.and(NodePredicates.isA(clazz)); 070 } 071 072 @Override 073 public FilteredNodeIterable<T> filter(NodePredicate p) { 074 return this.and(p); 075 } 076 077 @Override 078 public FilteredNodeIterable<T> filterInterface(Class<?> iface) { 079 return this.and(NodePredicates.isAInterface(iface)); 080 } 081}