Mercurial > hg > graal-compiler
view graal/com.oracle.truffle.codegen.processor/src/com/oracle/truffle/codegen/processor/ProcessorContext.java @ 8504:86b4965f0c9a
-Cleanup code
author | Christos Kotselidis <christos.kotselidis@oracle.com> |
---|---|
date | Wed, 20 Mar 2013 21:39:24 +0100 |
parents | 5e3d1a68664e |
children | 6d92fdf1c999 |
line wrap: on
line source
/* * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code 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 General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package com.oracle.truffle.codegen.processor; import static com.oracle.truffle.codegen.processor.Utils.*; import java.util.*; import javax.annotation.processing.*; import javax.lang.model.element.*; import javax.lang.model.type.*; import com.oracle.truffle.codegen.processor.ast.CodeTypeMirror.ArrayCodeTypeMirror; import com.oracle.truffle.codegen.processor.template.*; /** * THIS IS NOT PUBLIC API. */ public class ProcessorContext { private final ProcessingEnvironment environment; private final Map<String, Template> models = new HashMap<>(); private final Map<String, Map<String, TypeMirror>> generatedClasses = new HashMap<>(); private final ProcessCallback callback; private final Log log; private final TruffleTypes truffleTypes; public ProcessorContext(ProcessingEnvironment env, ProcessCallback callback) { this.environment = env; this.callback = callback; this.log = new Log(environment); this.truffleTypes = new TruffleTypes(this); } public TruffleTypes getTruffleTypes() { return truffleTypes; } public Log getLog() { return log; } public ProcessingEnvironment getEnvironment() { return environment; } public boolean containsTemplate(TypeElement element) { return models.containsKey(Utils.getQualifiedName(element)); } public void registerTemplate(TypeElement element, Template model) { models.put(Utils.getQualifiedName(element), model); } public void registerType(TypeElement templateType, TypeMirror generatedTypeMirror) { String templateQualifiedName = getQualifiedName(templateType); Map<String, TypeMirror> simpleNameToType = generatedClasses.get(templateQualifiedName); if (simpleNameToType == null) { simpleNameToType = new HashMap<>(); generatedClasses.put(templateQualifiedName, simpleNameToType); } String generatedSimpleName = getSimpleName(generatedTypeMirror); simpleNameToType.put(generatedSimpleName, generatedTypeMirror); } public Template getTemplate(TypeMirror templateTypeMirror, boolean invokeCallback) { String qualifiedName = Utils.getQualifiedName(templateTypeMirror); Template model = models.get(qualifiedName); if (model == null && invokeCallback) { callback.callback(Utils.fromTypeMirror(templateTypeMirror)); model = models.get(qualifiedName); } return model; } public TypeMirror resolveNotYetCompiledType(TypeMirror mirror, Template templateHint) { TypeMirror resolvedType = null; if (mirror.getKind() == TypeKind.ARRAY) { TypeMirror originalComponentType = ((ArrayType) mirror).getComponentType(); TypeMirror resolvedComponent = resolveNotYetCompiledType(originalComponentType, templateHint); if (resolvedComponent != originalComponentType) { return new ArrayCodeTypeMirror(resolvedComponent); } } if (mirror.getKind() == TypeKind.ERROR) { Element element = ((ErrorType) mirror).asElement(); ElementKind kind = element.getKind(); if (kind == ElementKind.CLASS || kind == ElementKind.PARAMETER || kind == ElementKind.ENUM) { String simpleName = element.getSimpleName().toString(); resolvedType = findGeneratedClassBySimpleName(simpleName, templateHint); } } else { resolvedType = mirror; } return resolvedType; } public TypeMirror findGeneratedClassBySimpleName(String simpleName, Template templateHint) { if (templateHint == null) { // search all for (String qualifiedTemplateName : generatedClasses.keySet()) { Map<String, TypeMirror> mirrors = generatedClasses.get(qualifiedTemplateName); if (mirrors.get(simpleName) != null) { return mirrors.get(simpleName); } } return null; } else { String templateQualifiedName = getQualifiedName(templateHint.getTemplateType()); Map<String, TypeMirror> simpleNameToType = generatedClasses.get(templateQualifiedName); if (simpleNameToType == null) { return null; } return simpleNameToType.get(simpleName); } } public TypeMirror getType(String className) { TypeElement element = environment.getElementUtils().getTypeElement(className); if (element != null) { return element.asType(); } return null; } public TypeMirror getType(Class<?> element) { TypeMirror mirror; if (element.isPrimitive()) { if (element == boolean.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.BOOLEAN); } else if (element == byte.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.BYTE); } else if (element == short.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.SHORT); } else if (element == char.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.CHAR); } else if (element == int.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.INT); } else if (element == long.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.LONG); } else if (element == float.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.FLOAT); } else if (element == double.class) { mirror = environment.getTypeUtils().getPrimitiveType(TypeKind.DOUBLE); } else if (element == void.class) { mirror = environment.getTypeUtils().getNoType(TypeKind.VOID); } else { assert false; return null; } } else { TypeElement typeElement = environment.getElementUtils().getTypeElement(element.getCanonicalName()); mirror = typeElement.asType(); } return mirror; } public interface ProcessCallback { void callback(TypeElement template); } }