/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you 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.
 */
#include "ElemForEach.hpp"


#include <algorithm>
#include <cassert>


#include <xercesc/sax/AttributeList.hpp>


#include <xalanc/Include/STLHelper.hpp>


#include <xalanc/PlatformSupport/DOMStringHelper.hpp>
#include <xalanc/PlatformSupport/XalanMessageLoader.hpp>


#include <xalanc/XPath/XPath.hpp>


#include "ElemSort.hpp"
#include "NodeSorter.hpp"
#include "SelectionEvent.hpp"
#include "StylesheetConstructionContext.hpp"
#include "StylesheetExecutionContext.hpp"


namespace XALAN_CPP_NAMESPACE {


ElemForEach::ElemForEach(
            StylesheetConstructionContext&  constructionContext,
            Stylesheet&                     stylesheetTree,
            const AttributeListType&        atts,
            XalanFileLoc                    lineNumber,
            XalanFileLoc                    columnNumber) :
    ElemTemplateElement(constructionContext,
                        stylesheetTree,
                        lineNumber,
                        columnNumber,
                        StylesheetConstructionContext::ELEMNAME_FOR_EACH),
    m_selectPattern(0),
    m_sortElems(constructionContext.getMemoryManager()),
    m_sortElemsCount(0)
{
    const XalanSize_t  nAttrs = atts.getLength();
        
    for (XalanSize_t i = 0; i < nAttrs; i++)
    {
        const XalanDOMChar* const   aname = atts.getName(i);

        if (equals(aname, Constants::ATTRNAME_SELECT))
        {
            m_selectPattern = constructionContext.createXPath(getLocator(), atts.getValue(i), *this);
        }
        else if (isAttrOK(
                    aname,
                    atts,
                    i,
                    constructionContext) == false &&
                processSpaceAttr(
                    Constants::ELEMNAME_FOREACH_WITH_PREFIX_STRING.c_str(),
                    aname,
                    atts,
                    i,
                    constructionContext) == false)
        {
            error(
                constructionContext,
                XalanMessages::ElementHasIllegalAttribute_2Param,
                Constants::ELEMNAME_FOREACH_WITH_PREFIX_STRING.c_str(),
                aname);
        }
    }

    if (0 == m_selectPattern)
    {
        error(
            constructionContext,
            XalanMessages::ElementMustHaveAttribute_2Param,
            Constants::ELEMNAME_FOREACH_WITH_PREFIX_STRING,
            Constants::ATTRNAME_SELECT);
    }
}


ElemForEach::ElemForEach(
            StylesheetConstructionContext&  constructionContext,
            Stylesheet&                     stylesheetTree,
            XalanFileLoc                    lineNumber,
            XalanFileLoc                    columnNumber,
            int                             xslToken) :
    ElemTemplateElement(constructionContext,
                        stylesheetTree,
                        lineNumber,
                        columnNumber,
                        xslToken),
    m_selectPattern(0),
    m_sortElems(constructionContext.getMemoryManager()),
    m_sortElemsCount(0)
{
}


ElemForEach::~ElemForEach()
{
    using std::for_each;

     MemoryManager& theManager = m_sortElems.getMemoryManager();

    for_each(m_sortElems.begin(),
             m_sortElems.end(),
             DeleteFunctor<ElemSort>(theManager));
}


void
ElemForEach::processSortElement(
            StylesheetConstructionContext&  constructionContext,
            Stylesheet&                     theStylesheet,
            const AttributeListType&        atts,
            const Locator*                  locator)
{
    const XalanFileLoc  lineNumber = XalanLocator::getLineNumber(locator);
    const XalanFileLoc  columnNumber = XalanLocator::getColumnNumber(locator);

    m_sortElems.reserve(m_sortElems.size() + 1);

    ElemSort* sortElem = ElemSort::create(
        constructionContext.getMemoryManager(),
        constructionContext,
        theStylesheet,
        atts,
        lineNumber,
        columnNumber);

    m_sortElems.push_back(sortElem);
}


const XalanDOMString&
ElemForEach::getElementName() const
{
    return Constants::ELEMNAME_FOREACH_WITH_PREFIX_STRING;
}


void
ElemForEach::postConstruction(
            StylesheetConstructionContext&  constructionContext,
            const NamespacesHandler&        theParentHandler)
{
    ElemTemplateElement::postConstruction(constructionContext, theParentHandler);

    m_sortElemsCount = m_sortElems.size();
}


#if !defined(XALAN_RECURSIVE_STYLESHEET_EXECUTION)
const ElemTemplateElement*
ElemForEach::startElement(StylesheetExecutionContext&       executionContext) const
{
    assert(executionContext.getCurrentNode() != 0);

    if (hasChildren() == true)
    {
        executionContext.pushCurrentTemplate(0);
        const NodeRefListBase * nodeList = createSelectedAndSortedNodeList(
                executionContext);
        executionContext.createAndPushNodesToTransformList(nodeList);
        executionContext.pushContextNodeList(*nodeList);
        
        XalanNode* currentNode = executionContext.getNextNodeToTransform();
        if (currentNode == 0)
        {
            return 0;
        }
        executionContext.pushCurrentNode(currentNode);

        return beginExecuteChildren(executionContext);  
    }
    
    return 0;
}


const ElemTemplateElement*
ElemForEach::getNextChildElemToExecute(
                   StylesheetExecutionContext& executionContext,
                   const ElemTemplateElement* currentElem) const
{
    if (hasDirectTemplate() != true)
    {
        ElemTemplateElement* nextElement = currentElem->getNextSiblingElem();

        if (nextElement != 0)
        {
            return nextElement;
        }
    }

    executionContext.popCurrentNode();


    XalanNode * nextNode = executionContext.getNextNodeToTransform();

    if (nextNode == 0)
    {
        return 0;
    }

    executionContext.pushCurrentNode(nextNode);
    endExecuteChildren(executionContext);
    return beginExecuteChildren(executionContext);
}


void
ElemForEach::endElement(StylesheetExecutionContext&     executionContext) const
{
    if (hasChildren() == true)
    {
        // Children only executed if there were selected nodes
        if(executionContext.getContextNodeList().getLength() > 0)
        {
            endExecuteChildren(executionContext);
        }

        executionContext.popNodesToTransformList();
        executionContext.popContextNodeList();
        releaseSelectedAndSortedNodeList(executionContext);
        executionContext.popCurrentTemplate();
    }   
}


const NodeRefListBase*
ElemForEach::createSelectedAndSortedNodeList(
            StylesheetExecutionContext&     executionContext) const 
{
    assert(m_selectPattern != 0);

    typedef StylesheetExecutionContext::SetAndRestoreCurrentStackFrameIndex     SetAndRestoreCurrentStackFrameIndex;

    MutableNodeRefList& selectedNodeList = executionContext.createAndPushMutableNodeRefList();  

    XObjectPtr xobjectResult;
        
    const NodeRefListBase*  nodesToTransform = 0;

    {
        xobjectResult = m_selectPattern->execute(
                        *this,
                        executionContext,
                        selectedNodeList);

        if (xobjectResult.null() == true)
        {
            nodesToTransform = &selectedNodeList;
        }
        else
        {
            nodesToTransform = &xobjectResult->nodeset();
        }
    }
    executionContext.pushXObjectPtr(xobjectResult);

    if(0 != executionContext.getTraceListeners())
    {
        executionContext.fireSelectEvent(
                SelectionEvent(
                    executionContext, 
                    executionContext.getCurrentNode(),
                    *this,
                    XalanDOMString("select", executionContext.getMemoryManager()),
                    *m_selectPattern,
                    *nodesToTransform));
    }

    if (m_sortElemsCount > 0) 
    {
        MutableNodeRefList& sortedNodeList = executionContext.createAndPushMutableNodeRefList();

        if (nodesToTransform->getLength() > 1)
        {
            nodesToTransform = sortChildren(
                    executionContext,
                    *nodesToTransform,
                    sortedNodeList);
        }
    }

    return nodesToTransform;
}


void
ElemForEach::releaseSelectedAndSortedNodeList(
        StylesheetExecutionContext& executionContext) const
{
    executionContext.popXObjectPtr();
    executionContext.releaseAndPopMutableNodeRefList();
    if (m_sortElemsCount > 0)
    {
        executionContext.releaseAndPopMutableNodeRefList();
    }
}


const NodeRefListBase*
ElemForEach::sortChildren(  
            StylesheetExecutionContext&     executionContext,
            const NodeRefListBase&          selectedNodeList,
            MutableNodeRefList&             sortedNodeList) const
{
    typedef NodeSorter::NodeSortKeyVectorType                   NodeSortKeyVectorType;
    typedef StylesheetExecutionContext::SetAndRestoreCurrentStackFrameIndex     SetAndRestoreCurrentStackFrameIndex;
    typedef StylesheetExecutionContext::ContextNodeListPushAndPop               ContextNodeListPushAndPop;
    
    NodeSorter* sorter = executionContext.getNodeSorter();

    NodeSortKeyVectorType&  keys = sorter->getSortKeys();
    assert(keys.empty() == true);

    CollectionClearGuard<NodeSortKeyVectorType>     guard(keys);

    // Reserve the space now...
    keys.reserve(m_sortElemsCount);

    // Get some temporary strings to use for evaluting the AVTs...
    const StylesheetExecutionContext::GetCachedString   theTemp1(executionContext);

    XalanDOMString&     langString = theTemp1.get();

    const StylesheetExecutionContext::GetCachedString   theTemp2(executionContext);

    XalanDOMString&     scratchString = theTemp2.get();

    // March backwards, performing a sort on each xsl:sort child.
    // Probably not the most efficient method.
    for(SortElemsVectorType::size_type  i = 0; i < m_sortElemsCount; i++)
    {
        const ElemSort* const   sort = m_sortElems[i];
        assert(sort != 0);

        const AVT* avt = sort->getLangAVT();

        if(0 != avt)
        {
            avt->evaluate(langString, *this, executionContext);
        }

        avt = sort->getDataTypeAVT();

        if(0 != avt)
        {
            avt->evaluate(scratchString, *this, executionContext);
        }           

        bool    treatAsNumbers = false;

        if (scratchString.empty() == false)
        {
            if (equals(scratchString, Constants::ATTRVAL_DATATYPE_NUMBER) == true)
            {
                treatAsNumbers = true;
            }
            else if (equals(scratchString, Constants::ATTRVAL_DATATYPE_TEXT) == false)
            {
                const XalanQNameByValue     theQName(scratchString, executionContext.getMemoryManager(), this);

                if (theQName.getNamespace().length() == 0)
                {
                    error(
                        executionContext,
                        XalanMessages::SortDataTypeMustBe,
                        sort->getLocator());
                }
                else
                {
                    warn(
                        executionContext,
                        XalanMessages::SortHasUnknownDataType,
                        sort->getLocator());
                }
            }
        }

        scratchString.clear();

        avt = sort->getOrderAVT();

        if(0 != avt)
        {
            avt->evaluate(scratchString, *this, executionContext);
        }           

        bool    descending = false;
        
        if (scratchString.empty() == false)
        {
            if (equals(scratchString, Constants::ATTRVAL_ORDER_DESCENDING) == true)
            {
                descending = true;
            }
            else if (equals(scratchString, Constants::ATTRVAL_ORDER_ASCENDING) == false)
            {
                error(
                    executionContext,
                    XalanMessages::SortMustBeAscendOrDescend,
                    sort->getLocator());
            }
        }

        scratchString.clear();

        avt = sort->getCaseOrderAVT();

        if(0 != avt)
        {
            avt->evaluate(scratchString, *this, executionContext);
        }           

        XalanCollationServices::eCaseOrder  caseOrder = XalanCollationServices::eDefault;

        if (scratchString.empty() == false)
        {
            if (equals(scratchString, Constants::ATTRVAL_CASEORDER_UPPER) == true)
            {
                caseOrder = XalanCollationServices::eUpperFirst;
            }
            else if (equals(scratchString, Constants::ATTRVAL_CASEORDER_LOWER) == true)
            {
                caseOrder = XalanCollationServices::eLowerFirst;
            }
            else
            {
                error(
                    executionContext,
                    XalanMessages::SortCaseOrderMustBe,
                    sort->getLocator());
            }
        }

        scratchString.clear();

        keys.push_back(
                NodeSortKey(
                    executionContext,
                    sort->getSelectPattern(),
                    treatAsNumbers,
                    descending,
                    caseOrder,
                    langString,
                    *this));
    }

    sortedNodeList = selectedNodeList;

    {
        ContextNodeListPushAndPop   theContextNodeListPushAndPop(
                executionContext,
                selectedNodeList);

        sorter->sort(executionContext, sortedNodeList);
    }

    return &sortedNodeList;
}
#endif


#if defined(XALAN_RECURSIVE_STYLESHEET_EXECUTION)
void
ElemForEach::execute(StylesheetExecutionContext&    executionContext) const
{
    assert(m_selectPattern != 0);
    assert(executionContext.getCurrentNode() != 0);

    StylesheetExecutionContext::PushAndPopCurrentTemplate   thePushAndPop(executionContext, 0);

    if (hasChildren() == true)
    {
        transformSelectedChildren(
            executionContext,
            this);
    }
}


void
ElemForEach::transformSelectedChildren(
            StylesheetExecutionContext&     executionContext,
            const ElemTemplateElement*      theTemplate) const
{
    assert(m_selectPattern != 0);
    assert(m_sortElemsCount == m_sortElems.size());

    if (m_sortElemsCount == 0)
    {
        selectAndSortChildren(
                    executionContext,
                    theTemplate,
                    0,
                    executionContext.getCurrentStackFrameIndex());
    }
    else
    {
        typedef NodeSorter::NodeSortKeyVectorType                   NodeSortKeyVectorType;
        typedef StylesheetExecutionContext::BorrowReturnNodeSorter  BorrowReturnNodeSorter;

        BorrowReturnNodeSorter  sorter(executionContext);

        NodeSortKeyVectorType&  keys = sorter->getSortKeys();
        assert(keys.empty() == true);

        CollectionClearGuard<NodeSortKeyVectorType>     guard(keys);

        // Reserve the space now...
        keys.reserve(m_sortElemsCount);

        // Get some temporary strings to use for evaluting the AVTs...
        StylesheetExecutionContext::GetCachedString     theTemp1(executionContext);

        XalanDOMString&     langString = theTemp1.get();

        const StylesheetExecutionContext::GetCachedString   theTemp2(executionContext);

        XalanDOMString&     scratchString = theTemp2.get();

        // March backwards, performing a sort on each xsl:sort child.
        // Probably not the most efficient method.
        for(SortElemsVectorType::size_type  i = 0; i < m_sortElemsCount; i++)
        {
            const ElemSort* const   sort = m_sortElems[i];
            assert(sort != 0);

            const AVT* avt = sort->getLangAVT();

            if(0 != avt)
            {
                avt->evaluate(langString, *this, executionContext);
            }

            avt = sort->getDataTypeAVT();

            if(0 != avt)
            {
                avt->evaluate(scratchString, *this, executionContext);
            }           

            bool    treatAsNumbers = false;

            if (scratchString.empty() == false)
            {
                if (equals(scratchString, Constants::ATTRVAL_DATATYPE_NUMBER) == true)
                {
                    treatAsNumbers = true;
                }
                else if (equals(scratchString, Constants::ATTRVAL_DATATYPE_TEXT) == false)
                {
                    const XalanQNameByValue     theQName(scratchString, executionContext.getMemoryManager(), this);

                    if (theQName.getNamespace().length() == 0)
                    {
                        error(
                            executionContext,
                            XalanMessages::SortDataTypeMustBe);
                    }
                    else
                    {
                        warn(
                            executionContext,
                            XalanMessages::SortHasUnknownDataType);
                    }
                }
            }

            clear(scratchString);

            avt = sort->getOrderAVT();

            if(0 != avt)
            {
                avt->evaluate(scratchString, *this, executionContext);
            }           

            bool    descending = false;
            
            if (scratchString.empty() == false)
            {
                if (equals(scratchString, Constants::ATTRVAL_ORDER_DESCENDING) == true)
                {
                    descending = true;
                }
                else if (equals(scratchString, Constants::ATTRVAL_ORDER_ASCENDING) == false)
                {
                    error(
                        executionContext,
                        XalanMessages::SortMustBeAscendOrDescend);
                }
            }

            clear(scratchString);

            avt = sort->getCaseOrderAVT();

            if(0 != avt)
            {
                avt->evaluate(scratchString, *this, executionContext);
            }           

            XalanCollationServices::eCaseOrder  caseOrder = XalanCollationServices::eDefault;

            if (scratchString.empty() == false)
            {
                if (equals(scratchString, Constants::ATTRVAL_CASEORDER_UPPER) == true)
                {
                    caseOrder = XalanCollationServices::eUpperFirst;
                }
                else if (equals(scratchString, Constants::ATTRVAL_CASEORDER_LOWER) == true)
                {
                    caseOrder = XalanCollationServices::eLowerFirst;
                }
                else
                {
                    error(
                        executionContext,
                        XalanMessages::SortCaseOrderMustBe);
                }
            }

            clear(scratchString);

            keys.push_back(
                    NodeSortKey(
                        executionContext,
                        sort->getSelectPattern(),
                        treatAsNumbers,
                        descending,
                        caseOrder,
                        langString,
                        *this));
        }

        selectAndSortChildren(
                    executionContext,
                    theTemplate,
                    sorter.get(),
                    executionContext.getCurrentStackFrameIndex());
    }
}


void
ElemForEach::selectAndSortChildren(
            StylesheetExecutionContext&     executionContext,
            const ElemTemplateElement*      theTemplate,
            NodeSorter*                     sorter,
            int                             selectStackFrameIndex) const
{
    typedef StylesheetExecutionContext::SetAndRestoreCurrentStackFrameIndex     SetAndRestoreCurrentStackFrameIndex;

    assert(m_selectPattern != 0);

    typedef XPathExecutionContext::BorrowReturnMutableNodeRefList   BorrowReturnMutableNodeRefList;

    BorrowReturnMutableNodeRefList  theGuard(executionContext);

    const NodeRefListBase*  sourceNodes = 0;

    XObjectPtr              xobjectResult;

    {
        SetAndRestoreCurrentStackFrameIndex     theSetAndRestore(
                    executionContext,
                    selectStackFrameIndex);

        xobjectResult = m_selectPattern->execute(
                        *this,
                        executionContext,
                        *theGuard);

        if (xobjectResult.null() == true)
        {
            sourceNodes = &*theGuard;
        }
        else
        {
            theGuard.release();

            sourceNodes = &xobjectResult->nodeset();
        }
    }

    if(0 != executionContext.getTraceListeners())
    {
        executionContext.fireSelectEvent(
                SelectionEvent(
                    executionContext, 
                    executionContext.getCurrentNode(),
                    *this,
                    XalanDOMString("select", executionContext.getMemoryManager()),
                    *m_selectPattern,
                    *sourceNodes));
    }

    const NodeRefListBase::size_type    nNodes = sourceNodes->getLength();

    if (nNodes > 0)
    {
        // If there's not NodeSorter, or we've only selected one node,
        // then just do the transform...
        if (sorter == 0 || nNodes == 1)
        {
            transformSelectedChildren(
                executionContext,
                theTemplate,
                *sourceNodes,
                nNodes);
        }
        else
        {
            typedef StylesheetExecutionContext::SetAndRestoreCurrentStackFrameIndex     SetAndRestoreCurrentStackFrameIndex;
            typedef StylesheetExecutionContext::ContextNodeListPushAndPop               ContextNodeListPushAndPop;
            typedef StylesheetExecutionContext::BorrowReturnMutableNodeRefList          BorrowReturnMutableNodeRefList;

            BorrowReturnMutableNodeRefList  sortedSourceNodes(executionContext);

            *sortedSourceNodes = *sourceNodes;

            {
                SetAndRestoreCurrentStackFrameIndex     theStackFrameSetAndRestore(
                        executionContext,
                        selectStackFrameIndex);

                ContextNodeListPushAndPop   theContextNodeListPushAndPop(
                        executionContext,
                        *sourceNodes);

                sorter->sort(executionContext, *sortedSourceNodes);
            }

            transformSelectedChildren(
                executionContext,
                theTemplate,
                *sortedSourceNodes,
                nNodes);
        }
    }
}


void
ElemForEach::transformSelectedChildren(
            StylesheetExecutionContext&     executionContext,
            const ElemTemplateElement*      theTemplate,
            const NodeRefListBase&          sourceNodes,
            NodeRefListBase::size_type      sourceNodesCount) const
{
    if(executionContext.getTraceSelects() == true)
    {
        executionContext.traceSelect(
            *this,
            sourceNodes,
            m_selectPattern);
    }

    // Create an object to set and restore the context node list...
    const StylesheetExecutionContext::ContextNodeListPushAndPop     theContextNodeLisPushAndPop(
                executionContext,
                sourceNodes);

    for(NodeRefListBase::size_type i = 0; i < sourceNodesCount; i++) 
    {
        XalanNode* const        childNode = sourceNodes.item(i);
        assert(childNode != 0);

        transformChild(
                executionContext,
                *this,
                theTemplate,
                childNode);
    }
}
#endif


const XPath*
ElemForEach::getXPath(XalanSize_t   index) const
{
    return index == 0 ? m_selectPattern : 0;
}


}