/*   Copyright 2004 The Apache Software Foundation
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package org.apache.xmlbeans.impl.values;

import org.apache.xmlbeans.SchemaType;
import org.apache.xmlbeans.XmlObject;
import org.apache.xmlbeans.impl.common.ValidationContext;
import org.apache.xmlbeans.impl.common.QNameHelper;



public abstract class JavaFloatHolderEx extends JavaFloatHolder
{
    public JavaFloatHolderEx(SchemaType type, boolean complex)
        { _schemaType = type; initComplexType(complex, false); }
    
    private SchemaType _schemaType;

    public SchemaType schemaType()
        { return _schemaType; }

    protected void set_float(float v)
    {
        if (_validateOnSet())
            validateValue(v, _schemaType, _voorVc);
        super.set_float(v);
    }
    
    public static float validateLexical(String v, SchemaType sType, ValidationContext context)
    {
        float f = JavaFloatHolder.validateLexical(v, context);

        if (!sType.matchPatternFacet(v))
            context.invalid("Float (" + v + ") does not match pattern for " + QNameHelper.readable(sType));
        
        return f;
    }
    
    public static void validateValue(float v, SchemaType sType, ValidationContext context)
    {
        XmlObject x;
        float f;

        if ((x = sType.getFacet(SchemaType.FACET_MIN_EXCLUSIVE)) != null)
        {
            if (compare(v, f = ((XmlObjectBase)x).floatValue()) <= 0)
            {
                context.invalid(
                    "Float (" + v + ") is less than or equal to " +
                        "min exclusive facet (" + f + ") for " + QNameHelper.readable(sType));
            }
        }

        if ((x = sType.getFacet(SchemaType.FACET_MIN_INCLUSIVE)) != null)
        {
            if (compare(v, f = ((XmlObjectBase)x).floatValue()) < 0)
            {
                context.invalid(
                    "Float (" + v + ") is less than " +
                        "min inclusive facet (" + f + ") for " + QNameHelper.readable(sType));
            }
        }
        
        if ((x = sType.getFacet(SchemaType.FACET_MAX_INCLUSIVE)) != null)
        {
            if (compare(v, f = ((XmlObjectBase)x).floatValue()) > 0)
            {
                context.invalid(
                    "Float (" + v + ") is greater than " +
                        "max inclusive facet (" + f + ") for " + QNameHelper.readable(sType));
            }
        }
        
        if ((x = sType.getFacet(SchemaType.FACET_MAX_EXCLUSIVE)) != null)
        {
            if (compare(v, f = ((XmlObjectBase)x).floatValue()) >= 0)
            {
                context.invalid(
                    "Float (" + v + ") is greater than or equal to " +
                        "max exclusive facet (" + f + ") for " + QNameHelper.readable(sType));
            }
        }
        
        XmlObject[] vals = sType.getEnumerationValues();
        if (vals != null)
        {
            for (int i = 0; i < vals.length; i++)
                if (compare(v, ((XmlObjectBase)vals[i]).floatValue()) == 0)
                    return;
            context.invalid("Float (" + v + ") is not a valid enumerated value for " + QNameHelper.readable(sType));
        }
    }
    
    protected void validate_simpleval(String lexical, ValidationContext ctx)
    {
        validateLexical(lexical, schemaType(), ctx);
        validateValue(floatValue(), schemaType(), ctx);
    }

}
