001: /*
002: * Copyright (c) 2003 The Visigoth Software Society. All rights
003: * reserved.
004: *
005: * Redistribution and use in source and binary forms, with or without
006: * modification, are permitted provided that the following conditions
007: * are met:
008: *
009: * 1. Redistributions of source code must retain the above copyright
010: * notice, this list of conditions and the following disclaimer.
011: *
012: * 2. Redistributions in binary form must reproduce the above copyright
013: * notice, this list of conditions and the following disclaimer in
014: * the documentation and/or other materials provided with the
015: * distribution.
016: *
017: * 3. The end-user documentation included with the redistribution, if
018: * any, must include the following acknowledgement:
019: * "This product includes software developed by the
020: * Visigoth Software Society (http://www.visigoths.org/)."
021: * Alternately, this acknowledgement may appear in the software itself,
022: * if and wherever such third-party acknowledgements normally appear.
023: *
024: * 4. Neither the name "FreeMarker", "Visigoth", nor any of the names of the
025: * project contributors may be used to endorse or promote products derived
026: * from this software without prior written permission. For written
027: * permission, please contact visigoths@visigoths.org.
028: *
029: * 5. Products derived from this software may not be called "FreeMarker" or "Visigoth"
030: * nor may "FreeMarker" or "Visigoth" appear in their names
031: * without prior written permission of the Visigoth Software Society.
032: *
033: * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
034: * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
035: * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
036: * DISCLAIMED. IN NO EVENT SHALL THE VISIGOTH SOFTWARE SOCIETY OR
037: * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
038: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
039: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
040: * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
041: * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
042: * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
043: * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
044: * SUCH DAMAGE.
045: * ====================================================================
046: *
047: * This software consists of voluntary contributions made by many
048: * individuals on behalf of the Visigoth Software Society. For more
049: * information on the Visigoth Software Society, please see
050: * http://www.visigoths.org/
051: */
052:
053: package freemarker.core;
054:
055: import freemarker.template.*;
056: import java.util.*;
057: import java.io.Serializable;
058:
059: /**
060: * Add sequence capabilities to an existing collection, or
061: * vice versa. Used by ?keys and ?values built-ins.
062: */
063: final public class CollectionAndSequence implements
064: TemplateCollectionModel, TemplateSequenceModel, Serializable {
065: private TemplateCollectionModel collection;
066: private TemplateSequenceModel sequence;
067: private ArrayList data;
068:
069: public CollectionAndSequence(TemplateCollectionModel collection) {
070: this .collection = collection;
071: }
072:
073: public CollectionAndSequence(TemplateSequenceModel sequence) {
074: this .sequence = sequence;
075: }
076:
077: public TemplateModelIterator iterator()
078: throws TemplateModelException {
079: if (collection != null) {
080: return collection.iterator();
081: } else {
082: return new SequenceIterator(sequence);
083: }
084: }
085:
086: public TemplateModel get(int i) throws TemplateModelException {
087: if (sequence != null) {
088: return sequence.get(i);
089: } else {
090: initSequence();
091: return (TemplateModel) data.get(i);
092: }
093: }
094:
095: public int size() throws TemplateModelException {
096: if (sequence != null) {
097: return sequence.size();
098: } else {
099: initSequence();
100: return data.size();
101: }
102: }
103:
104: private void initSequence() throws TemplateModelException {
105: if (data == null) {
106: data = new ArrayList();
107: TemplateModelIterator it = collection.iterator();
108: while (it.hasNext()) {
109: data.add(it.next());
110: }
111: }
112: }
113:
114: private static class SequenceIterator implements
115: TemplateModelIterator {
116: private final TemplateSequenceModel sequence;
117: private final int size;
118: private int index = 0;
119:
120: SequenceIterator(TemplateSequenceModel sequence)
121: throws TemplateModelException {
122: this .sequence = sequence;
123: this .size = sequence.size();
124:
125: }
126:
127: public TemplateModel next() throws TemplateModelException {
128: return sequence.get(index++);
129: }
130:
131: public boolean hasNext() {
132: return index < size;
133: }
134: }
135: }
|