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.replacements.test; 024 025import jdk.internal.jvmci.code.*; 026import jdk.internal.jvmci.meta.*; 027import jdk.internal.jvmci.meta.JavaTypeProfile.*; 028 029import com.oracle.graal.compiler.test.*; 030import com.oracle.graal.nodes.*; 031 032/** 033 * Base class for checkcast and instanceof test classes. 034 */ 035public abstract class TypeCheckTest extends GraalCompilerTest { 036 037 protected abstract void replaceProfile(StructuredGraph graph, JavaTypeProfile profile); 038 039 protected JavaTypeProfile currentProfile; 040 041 @Override 042 protected StructuredGraph parseForCompile(ResolvedJavaMethod method) { 043 StructuredGraph graph = super.parseForCompile(method); 044 if (currentProfile != null) { 045 replaceProfile(graph, currentProfile); 046 } 047 return graph; 048 } 049 050 @Override 051 protected InstalledCode getCode(final ResolvedJavaMethod method, final StructuredGraph graph) { 052 return getCode(method, graph, currentProfile != null); 053 } 054 055 protected JavaTypeProfile profile(Class<?>... types) { 056 return profile(TriState.FALSE, types); 057 } 058 059 protected JavaTypeProfile profile(TriState nullSeen, Class<?>... types) { 060 if (types.length == 0) { 061 return null; 062 } 063 ProfiledType[] ptypes = new ProfiledType[types.length]; 064 for (int i = 0; i < types.length; i++) { 065 ptypes[i] = new ProfiledType(getMetaAccess().lookupJavaType(types[i]), 1.0D / types.length); 066 } 067 return new JavaTypeProfile(nullSeen, 0.0D, ptypes); 068 } 069 070 protected void test(String name, JavaTypeProfile profile, Object... args) { 071 assert currentProfile == null; 072 currentProfile = profile; 073 try { 074 super.test(name, args); 075 } finally { 076 currentProfile = null; 077 } 078 } 079}