001    /*
002     $Id: ArrayExpression.java 3419 2006-01-19 00:07:02Z blackdrag $
003    
004     Copyright 2003 (C) James Strachan and Bob Mcwhirter. All Rights Reserved.
005    
006     Redistribution and use of this software and associated documentation
007     ("Software"), with or without modification, are permitted provided
008     that the following conditions are met:
009    
010     1. Redistributions of source code must retain copyright
011        statements and notices.  Redistributions must also contain a
012        copy of this document.
013    
014     2. Redistributions in binary form must reproduce the
015        above copyright notice, this list of conditions and the
016        following disclaimer in the documentation and/or other
017        materials provided with the distribution.
018    
019     3. The name "groovy" must not be used to endorse or promote
020        products derived from this Software without prior written
021        permission of The Codehaus.  For written permission,
022        please contact info@codehaus.org.
023    
024     4. Products derived from this Software may not be called "groovy"
025        nor may "groovy" appear in their names without prior written
026        permission of The Codehaus. "groovy" is a registered
027        trademark of The Codehaus.
028    
029     5. Due credit should be given to The Codehaus -
030        http://groovy.codehaus.org/
031    
032     THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS
033     ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
034     NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
035     FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
036     THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
037     INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
038     (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
039     SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
040     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
041     STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
042     ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
043     OF THE POSSIBILITY OF SUCH DAMAGE.
044    
045     */
046    package org.codehaus.groovy.ast.expr;
047    
048    import java.util.Collections;
049    import java.util.Iterator;
050    import java.util.List;
051    
052    import org.codehaus.groovy.ast.ClassNode;
053    import org.codehaus.groovy.ast.GroovyCodeVisitor;
054    
055    /**
056     * Represents an array object construction either using a fixed size
057     * or an initializer expression
058     * 
059     * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
060     * @version $Revision: 3419 $
061     */
062    public class ArrayExpression extends Expression {
063        private List expressions;
064        private List sizeExpression;
065    
066        private ClassNode elementType;
067        
068        private static ClassNode makeArray(ClassNode base, List sizeExpression) {
069            ClassNode ret = base.makeArray();
070            if (sizeExpression==null) return ret;
071            int size = sizeExpression.size();
072            for (int i=1; i<size; i++) {
073                    ret = ret.makeArray();
074            }
075            return ret;
076        }
077        
078        public ArrayExpression(ClassNode elementType, List expressions, List sizeExpression) {
079            //expect to get the elementType
080            super.setType(makeArray(elementType,sizeExpression));
081            if (expressions==null) expressions=Collections.EMPTY_LIST;
082            this.elementType = elementType;
083            this.expressions = expressions;
084            this.sizeExpression = sizeExpression;
085            
086            for (Iterator iter = expressions.iterator(); iter.hasNext();) {
087                Object item = iter.next();
088                if (item!=null && !(item instanceof Expression)) {
089                    throw new ClassCastException("Item: " + item + " is not an Expression");
090                }
091            }
092            if (sizeExpression!=null) {
093                    for (Iterator iter = sizeExpression.iterator(); iter.hasNext();) {
094                        Object item = iter.next();
095                        if (!(item instanceof Expression)) {
096                            throw new ClassCastException("Item: " + item + " is not an Expression");
097                        }
098                    }
099            }
100        }
101        
102        
103        /**
104         * Creates an array using an initializer expression
105         */
106        public ArrayExpression(ClassNode elementType, List expressions) {
107            this(elementType,expressions,null);
108        }
109    
110        public void addExpression(Expression expression) {
111            expressions.add(expression);
112        }
113    
114        public List getExpressions() {
115            return expressions;
116        }
117    
118        public void visit(GroovyCodeVisitor visitor) {
119            visitor.visitArrayExpression(this);
120        }
121    
122        public boolean isDynamic() {
123            return false;
124        }
125    
126        public Expression transformExpression(ExpressionTransformer transformer) {
127            List exprList = transformExpressions(expressions, transformer);
128            List sizes = null;
129            if (sizeExpression!=null) sizes = transformExpressions(sizeExpression,transformer);
130            Expression ret = new ArrayExpression(elementType, exprList, sizes);
131            ret.setSourcePosition(this);
132            return ret;
133        }
134    
135        public Expression getExpression(int i) {
136            Object object = expressions.get(i);
137            return (Expression) object;
138        }
139    
140        public ClassNode getElementType() {
141            return elementType;
142        }
143        
144        public String getText() {
145            StringBuffer buffer = new StringBuffer("[");
146            boolean first = true;
147            for (Iterator iter = expressions.iterator(); iter.hasNext();) {
148                if (first) {
149                    first = false;
150                }
151                else {
152                    buffer.append(", ");
153                }
154    
155                buffer.append(((Expression) iter.next()).getText());
156            }
157            buffer.append("]");
158            return buffer.toString();
159        }
160    
161        public List getSizeExpression() {
162            return sizeExpression;
163        }
164    
165        public String toString() {
166            return super.toString() + expressions;
167        }
168    }