UnknownTypeCollector.java

/*
 * Copyright Red Hat Inc. and/or its affiliates and other contributors
 * as indicated by the authors tag. All rights reserved.
 *
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU General Public License version 2.
 * 
 * This particular file is subject to the "Classpath" exception as provided in the 
 * LICENSE file that accompanied this code.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT A
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.  See the GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License,
 * along with this distribution; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301, USA.
 */

package com.redhat.ceylon.compiler.java.loader;

import java.util.IdentityHashMap;
import java.util.List;
import java.util.Map;

import com.redhat.ceylon.compiler.typechecker.tree.Tree;
import com.redhat.ceylon.compiler.typechecker.tree.Visitor;
import com.redhat.ceylon.model.typechecker.model.ClassOrInterface;
import com.redhat.ceylon.model.typechecker.model.Declaration;
import com.redhat.ceylon.model.typechecker.model.Functional;
import com.redhat.ceylon.model.typechecker.model.Parameter;
import com.redhat.ceylon.model.typechecker.model.ParameterList;
import com.redhat.ceylon.model.typechecker.model.Type;
import com.redhat.ceylon.model.typechecker.model.TypeDeclaration;
import com.redhat.ceylon.model.typechecker.model.UnknownType;
import com.redhat.ceylon.model.typechecker.model.Value;

/**
 *
 * @author Stéphane Épardaud <stef@epardaud.fr>
 */
public class UnknownTypeCollector extends Visitor {
    public void visit(Tree.BaseMemberOrTypeExpression that) {
        super.visit(that);
        Declaration declaration = that.getDeclaration();
        if(declaration == null)
            return;
        if(declaration instanceof Functional){
            Functional m = (Functional)declaration;
            collectUnknownTypes(m.getType());
            for(ParameterList pl : m.getParameterLists()){
                for(Parameter p : pl.getParameters()){
                    collectUnknownTypes(p.getType());
                }
            }
        }else if(declaration instanceof Value){
            Value v = (Value)declaration;
            collectUnknownTypes(v.getType());
        }
    }

    private void collectUnknownTypes(Type type) {
        Map<Declaration, Declaration> visited = new IdentityHashMap<Declaration, Declaration>(0); // expect the best case: no error
        collectUnknownTypes(type, visited);
    }

    private void collectUnknownTypesResolved(Type type, Map<Declaration, Declaration> visited) {
        if(type != null){
            collectUnknownTypes(type, visited);
            List<Type> typeArguments = type.getTypeArgumentList();
            // cheaper c-for than foreach
            for (int i=0,l=typeArguments.size();i<l;i++) {
                Type tl = typeArguments.get(i);
                collectUnknownTypesResolved(tl, visited);
            }
        }
    }

    private void collectUnknownTypes(Type type, Map<Declaration, Declaration> visited) {
        if (type!=null) {
            type = type.resolveAliases();
            if(type.isUnknown()){
                UnknownType ut = (UnknownType) type.getDeclaration();
                ut.reportErrors();
                // don't report it twice
                ut.setErrorReporter(null);
            }else if(type.isUnion()){
                for(Type t : type.getCaseTypes()){
                    collectUnknownTypesResolved(t, visited);
                }
            }else if(type.isIntersection()){
                for(Type t : type.getSatisfiedTypes()){
                    collectUnknownTypesResolved(t, visited);
                }
            }else if(type.isUnknown() || type.isTypeParameter()){
                // do nothing
            }
            else {
                TypeDeclaration declaration = type.getDeclaration();
                if(visited.put(declaration, declaration) != null)
                    return;
                if(type.isClassOrInterface()){
                    // these are not resolved
                    if(type.getExtendedType() != null)
                        collectUnknownTypes(type.getExtendedType(), visited);
                    for(Type t : type.getSatisfiedTypes())
                        collectUnknownTypes(t, visited);

                }
            }
        }
    }
}