/*

 * 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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.sun.xml.internal.xsom.impl;

import com.sun.xml.internal.xsom.XSAttGroupDecl;

import com.sun.xml.internal.xsom.XSAttributeDecl;

import com.sun.xml.internal.xsom.XSAttributeUse;

import com.sun.xml.internal.xsom.XSComplexType;

import com.sun.xml.internal.xsom.XSContentType;

import com.sun.xml.internal.xsom.XSElementDecl;

import com.sun.xml.internal.xsom.XSSchema;

import com.sun.xml.internal.xsom.XSSchemaSet;

import com.sun.xml.internal.xsom.XSSimpleType;

import com.sun.xml.internal.xsom.XSType;

import com.sun.xml.internal.xsom.XSWildcard;

import com.sun.xml.internal.xsom.impl.parser.DelayedRef;

import com.sun.xml.internal.xsom.impl.parser.SchemaDocumentImpl;

import com.sun.xml.internal.xsom.impl.scd.Iterators;

import com.sun.xml.internal.xsom.visitor.XSFunction;

import com.sun.xml.internal.xsom.visitor.XSVisitor;

import java.util.ArrayList;

import java.util.List;

import org.xml.sax.Locator;

import java.util.Collection;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

public class ComplexTypeImpl extends AttributesHolder implements XSComplexType, Ref.ComplexType

{

    public ComplexTypeImpl( SchemaDocumentImpl _parent,

        AnnotationImpl _annon, Locator _loc, ForeignAttributesImpl _fa,

        String _name, boolean _anonymous,

        boolean _abstract, int _derivationMethod,

        Ref.Type _base, int _final, int _block, boolean _mixed ) {

        super(_parent,_annon,_loc,_fa,_name,_anonymous);

        if(_base==null)

            throw new IllegalArgumentException();

        this._abstract = _abstract;

        this.derivationMethod = _derivationMethod;

        this.baseType = _base;

        this.finalValue = _final;

        this.blockValue = _block;

        this.mixed = _mixed;

    }

    public XSComplexType asComplexType(){ return this; }

    public boolean isDerivedFrom(XSType t) {

        XSType x = this;

        while(true) {

            if(t==x)

                return true;

            XSType s = x.getBaseType();

            if(s==x)

                return false;

            x = s;

        }

    }

    public XSSimpleType  asSimpleType()    { return null; }

    public final boolean isSimpleType() { return false; }

    public final boolean isComplexType(){ return true; }

    private int derivationMethod;

    public int getDerivationMethod() { return derivationMethod; }

    private Ref.Type baseType;

    public XSType getBaseType() { return baseType.getType(); }

    /**

     * Called when this complex type redefines the specified complex type.

     */

    public void redefine( ComplexTypeImpl ct ) {

        if( baseType instanceof DelayedRef )

            ((DelayedRef)baseType).redefine(ct);

        else

            this.baseType = ct;

        ct.redefinedBy = this;

        redefiningCount = (short)(ct.redefiningCount+1);

    }

    /**

     * Number of times this component redefines other components.

     */

    private short redefiningCount = 0;

    private ComplexTypeImpl redefinedBy = null;

    public XSComplexType getRedefinedBy() {

        return redefinedBy;

    }

    public int getRedefinedCount() {

        int i=0;

        for( ComplexTypeImpl ct=this.redefinedBy; ct!=null; ct=ct.redefinedBy)

            i++;

        return i;

    }

    private XSElementDecl scope;

    public XSElementDecl getScope() { return scope; }

    public void setScope( XSElementDecl _scope ) { this.scope=_scope; }

    private final boolean _abstract;

    public boolean isAbstract() { return _abstract; }

    private WildcardImpl localAttWildcard;

    /**

     * Set the local attribute wildcard.

     */

    public void setWildcard( WildcardImpl wc ) {

        this.localAttWildcard = wc;

    }

    public XSWildcard getAttributeWildcard() {

        WildcardImpl complete = localAttWildcard;

        Iterator itr = iterateAttGroups();

        while( itr.hasNext() ) {

            WildcardImpl w = (WildcardImpl)((XSAttGroupDecl)itr.next()).getAttributeWildcard();

            if(w==null)     continue;

            if(complete==null)

                complete = w;

            else

                // TODO: the spec says it's intersection,

                // but I think it has to be union.

                complete = complete.union(ownerDocument,w);

        }

        if( getDerivationMethod()==RESTRICTION )    return complete;

        WildcardImpl base=null;

        XSType baseType = getBaseType();

        if(baseType.asComplexType()!=null)

            base = (WildcardImpl)baseType.asComplexType().getAttributeWildcard();

        if(complete==null)  return base;

        if(base==null)      return complete;

        return complete.union(ownerDocument,base);

    }

    private final int finalValue;

    public boolean isFinal( int derivationMethod ) {

        return (finalValue&derivationMethod)!=0;

    }

    private final int blockValue;

    public boolean isSubstitutionProhibited( int method ) {

        return (blockValue&method)!=0;

    }

    private Ref.ContentType contentType;

    public void setContentType( Ref.ContentType v ) { contentType = v; }

    public XSContentType getContentType() { return contentType.getContentType(); }

    private XSContentType explicitContent;

    public void setExplicitContent( XSContentType v ) {

        this.explicitContent = v;

    }

    public XSContentType getExplicitContent() { return explicitContent; }

    private final boolean mixed;

    public boolean isMixed() { return mixed; }

    public XSAttributeUse getAttributeUse( String nsURI, String localName ) {

        UName name = new UName(nsURI,localName);

        if(prohibitedAtts.contains(name))       return null;

        XSAttributeUse o = attributes.get(name);

        if(o==null) {

            Iterator itr = iterateAttGroups();

            while(itr.hasNext() && o==null)

                o = ((XSAttGroupDecl)itr.next()).getAttributeUse(nsURI,localName);

        }

        if(o==null) {

            XSType base = getBaseType();

            if(base.asComplexType()!=null)

                o = base.asComplexType().getAttributeUse(nsURI,localName);

        }

        return o;

    }

    public Iterator<XSAttributeUse> iterateAttributeUses() {

        XSComplexType baseType = getBaseType().asComplexType();

        if( baseType==null )    return super.iterateAttributeUses();

        return new Iterators.Union<XSAttributeUse>(

            new Iterators.Filter<XSAttributeUse>(baseType.iterateAttributeUses()) {

                protected boolean matches(XSAttributeUse value) {

                    XSAttributeDecl u = value.getDecl();

                    UName n = new UName(u.getTargetNamespace(),u.getName());

                    return !prohibitedAtts.contains(n);

                }

            },

            super.iterateAttributeUses() );

    }

    public Collection<XSAttributeUse> getAttributeUses() {

        XSComplexType baseType = getBaseType().asComplexType();

        if( baseType==null )    return super.getAttributeUses();

        // TODO: this is fairly inefficient

        Map<UName,XSAttributeUse> uses = new HashMap<UName, XSAttributeUse>();

        for( XSAttributeUse a : baseType.getAttributeUses())

            uses.put(new UName(a.getDecl()),a);

        uses.keySet().removeAll(prohibitedAtts);

        for( XSAttributeUse a : super.getAttributeUses())

            uses.put(new UName(a.getDecl()),a);

        return uses.values();

    }

    public XSType[] listSubstitutables() {

        return Util.listSubstitutables(this);

    }

    public void visit( XSVisitor visitor ) {

        visitor.complexType(this);

    }

    public <T> T apply( XSFunction<T> function ) {

        return function.complexType(this);

    }

    // Ref.ComplexType implementation

    public XSComplexType getType() { return this; }

    public List<XSComplexType> getSubtypes() {

        ArrayList subtypeList = new ArrayList();

        Iterator<XSComplexType> cTypes = getRoot().iterateComplexTypes();

        while (cTypes.hasNext()) {

            XSComplexType cType= cTypes.next();

            XSType base = cType.getBaseType();

            if ((base != null) && (base.equals(this))) {

                subtypeList.add(cType);

            }

        }

        return subtypeList;

    }

    public List<XSElementDecl> getElementDecls() {

        ArrayList declList = new ArrayList();

        XSSchemaSet schemaSet = getRoot();

        for (XSSchema sch : schemaSet.getSchemas()) {

            for (XSElementDecl decl : sch.getElementDecls().values()) {

                if (decl.getType().equals(this)) {

                    declList.add(decl);

                }

            }

        }

        return declList;

    }

}