EMFJvmTypesBuilder.java

/*******************************************************************************
 * Copyright (c) 2010-2012, Mark Czotter, Istvan Rath and Daniel Varro
 * 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:
 *   Mark Czotter - initial API and implementation
 *******************************************************************************/

package org.eclipse.incquery.tooling.core.generator.util;

import org.apache.log4j.Logger;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.common.types.JvmLowerBound;
import org.eclipse.xtext.common.types.JvmParameterizedTypeReference;
import org.eclipse.xtext.common.types.JvmType;
import org.eclipse.xtext.common.types.JvmTypeReference;
import org.eclipse.xtext.common.types.JvmWildcardTypeReference;
import org.eclipse.xtext.common.types.TypesFactory;
import org.eclipse.xtext.common.types.util.TypeReferences;
import org.eclipse.xtext.xbase.jvmmodel.JvmTypesBuilder;
import org.eclipse.xtext.xbase.lib.Procedures.Procedure1;

import com.google.common.base.Preconditions;
import com.google.inject.Inject;

/**
 * Custom {@link JvmTypesBuilder} for EMFPatternLanguage.
 * 
 * @author Mark Czotter
 * 
 */
@SuppressWarnings("restriction")
public class EMFJvmTypesBuilder extends JvmTypesBuilder {

    @Inject
    private TypesFactory factory = TypesFactory.eINSTANCE;

    @Inject
    private TypeReferences typeReferences;
    @Inject
    private Logger logger;

    /**
     * Creates a {@link JvmWildcardTypeReference} with a {@link JvmLowerBound} constraint to 'clone' parameter.
     * 
     * @param clone
     * @return {@link JvmWildcardTypeReference} with a {@link JvmLowerBound} contraint.
     */
    public JvmWildcardTypeReference wildCardSuper(JvmTypeReference clone) {
        JvmWildcardTypeReference result = factory.createJvmWildcardTypeReference();
        JvmLowerBound lowerBound = factory.createJvmLowerBound();
        lowerBound.setTypeReference(clone);
        result.getConstraints().add(lowerBound);
        return result;
    }

    /**
     * Creates a JvmTypeReference, that does not have any type parameter (serialized as a raw type).
     * 
     * @return
     */
    public JvmTypeReference newRawTypeRef(EObject ctx, Class<?> clazz, JvmTypeReference... typeArgs) {
        Preconditions.checkNotNull(clazz, "clazz");

        JvmType declaredType = typeReferences.findDeclaredType(clazz, ctx);
        if (declaredType == null) {
            return null;
        }
        return createTypeRef(declaredType);
    }

    /**
     * Creates a JvmTypeReference, that does not have any type parameter (serialized as a raw type).
     * 
     * @return
     */
    public JvmTypeReference newRawTypeRef(EObject ctx, String typeName, JvmTypeReference... typeArgs) {
        Preconditions.checkNotNull(typeName, "typeName");
        Preconditions.checkNotNull(ctx, "context");

        JvmType declaredType = typeReferences.findDeclaredType(typeName, ctx);
        if (declaredType == null) {
            return null;
        }
        return createTypeRef(declaredType);
    }

    private JvmTypeReference createTypeRef(JvmType type) {
        JvmParameterizedTypeReference reference = factory.createJvmParameterizedTypeReference();
        reference.setType(type);
        return reference;
    }

    /**
     * Overriding parent method to replace logging {@inheritDoc}
     */
    protected <T extends EObject> T initializeSafely(T targetElement, Procedure1<? super T> initializer) {
        if (targetElement != null && initializer != null) {
            try {
                initializer.apply(targetElement);
            } catch (Exception e) {
                logger.error("Error initializing JvmElement", e);
            }
        }
        return targetElement;
    }
}