001    /*
002     $Id: MapExpression.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.ArrayList;
049    import java.util.List;
050    
051    import org.codehaus.groovy.ast.ClassHelper;
052    import org.codehaus.groovy.ast.GroovyCodeVisitor;
053    
054    /**
055     * Represents a map expression [1 : 2, "a" : "b", x : y] which creates a mutable Map
056     * 
057     * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
058     * @version $Revision: 3419 $
059     */
060    public class MapExpression extends Expression {
061        private List mapEntryExpressions;
062    
063        public MapExpression() {
064            this(new ArrayList());
065        }
066        
067        public MapExpression(List mapEntryExpressions) {
068            this.mapEntryExpressions = mapEntryExpressions;
069            //TODO: get the type's of the expressions to specify the
070            // map type to Map<X> if possible.
071            setType(ClassHelper.MAP_TYPE);
072        }
073        
074        public void addMapEntryExpression(MapEntryExpression expression) {
075            mapEntryExpressions.add(expression);
076        }
077        
078        public List getMapEntryExpressions() {
079            return mapEntryExpressions;
080        }
081    
082        public void visit(GroovyCodeVisitor visitor) {
083            visitor.visitMapExpression(this);
084        }
085    
086        public boolean isDynamic() {
087            return false;  //To change body of implemented methods use File | Settings | File Templates.
088        }
089    
090        public Expression transformExpression(ExpressionTransformer transformer) {
091            Expression ret = new MapExpression(transformExpressions(getMapEntryExpressions(), transformer));
092            ret.setSourcePosition(this);
093            return ret;        
094        }
095        
096        public String toString() {
097            return super.toString() + mapEntryExpressions;
098        }
099    
100        public String getText() {
101            StringBuffer sb = new StringBuffer(32);
102            sb.append("[");
103            int size = mapEntryExpressions.size();
104            MapEntryExpression mapEntryExpression = null;
105            if (size > 0) {
106                mapEntryExpression = (MapEntryExpression) mapEntryExpressions.get(0);
107                sb.append(mapEntryExpression.getKeyExpression().getText() + ":" + mapEntryExpression.getValueExpression().getText());
108                for (int i = 1; i < size; i++) {
109                    mapEntryExpression = (MapEntryExpression) mapEntryExpressions.get(i);
110                    sb.append(", " + mapEntryExpression.getKeyExpression().getText() + ":" + mapEntryExpression.getValueExpression().getText());
111                    if (sb.length() > 120 && i < size - 1) {
112                        sb.append(", ... ");
113                        break;
114                    }
115                }
116            }
117            sb.append("]");
118            return sb.toString();
119        }
120    
121        public void addMapEntryExpression(Expression keyExpression, Expression valueExpression) {
122            addMapEntryExpression(new MapEntryExpression(keyExpression, valueExpression));
123        }
124    
125    }