summaryrefslogtreecommitdiff
path: root/ooxml/source/framework/SchemaParser/src/org/apache/openoffice/ooxml/schema/iterator/AttributeNodeIterator.java
blob: 0a5bc410f073a733f20e5b3189c70c095eac0334 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/**************************************************************
*
* 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.
*
*************************************************************/

package org.apache.openoffice.ooxml.schema.iterator;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Vector;

import org.apache.openoffice.ooxml.schema.model.attribute.Attribute;
import org.apache.openoffice.ooxml.schema.model.attribute.AttributeGroup;
import org.apache.openoffice.ooxml.schema.model.attribute.AttributeGroupReference;
import org.apache.openoffice.ooxml.schema.model.attribute.AttributeReference;
import org.apache.openoffice.ooxml.schema.model.base.INode;
import org.apache.openoffice.ooxml.schema.model.base.INodeVisitor;
import org.apache.openoffice.ooxml.schema.model.base.NodeVisitorAdapter;
import org.apache.openoffice.ooxml.schema.model.schema.SchemaBase;

/** This iterator is very similar to the AttributeIterator but it
 *  a) returns the attributes as INode objects and
 *  b) includes referencing nodes and group nodes in the iteration and
 *  c) if there are multiple references to the same attribute then each
 *     of them is visited.
 */
public class AttributeNodeIterator
    implements Iterable<INode>
{
    public AttributeNodeIterator (
        final INode aNode,
        final SchemaBase aSchemaBase)
    {
        maAttributes = new Vector<INode>();
        CollectAttributes(aNode, aSchemaBase);
    }




    public Iterator<INode> iterator ()
    {
        return maAttributes.iterator();
    }




    private void CollectAttributes (
        final INode aType,
        final SchemaBase aSchemaBase)
    {
        final Queue<INode> aTodo = new LinkedList<>();
        for (final INode aAttribute : aType.GetAttributes())
            aTodo.add(aAttribute);
        final INodeVisitor aVisitor = new NodeVisitorAdapter()
        {
            @Override public void Visit (final Attribute aAttribute)
            {
                maAttributes.add(aAttribute);
            }
            @Override public void Visit (final AttributeReference aAttributeReference)
            {
                maAttributes.add(aAttributeReference);
                aTodo.add(aAttributeReference.GetReferencedAttribute(aSchemaBase));
            }
            @Override public void Visit (final AttributeGroup aAttributeGroup)
            {
                maAttributes.add(aAttributeGroup);
                for (final INode aGroupAttribute : aAttributeGroup.GetAttributes())
                    aTodo.add(aGroupAttribute);
            }
            @Override public void Visit (final AttributeGroupReference aAttributeGroupReference)
            {
                maAttributes.add(aAttributeGroupReference);
                aTodo.add(aAttributeGroupReference.GetReferencedAttributeGroup(aSchemaBase));
            }
            @Override public void Default (final INode aNode)
            {
                throw new RuntimeException("can not handle attribute of type "+aNode.GetNodeType());
            }
        };
        while ( ! aTodo.isEmpty())
        {
            final INode aAttribute = aTodo.poll();
            if (aAttribute != null)
                aAttribute.AcceptVisitor(aVisitor);
        }
    }




    private final Vector<INode> maAttributes;
}