All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.antlr.stringtemplate.language.ASTExpr Maven / Gradle / Ivy

There is a newer version: 7.4.3.112-ga112
Show newest version
/*
 [The "BSD licence"]
 Copyright (c) 2003-2005 Terence Parr
 All rights reserved.

 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions
 are met:
 1. Redistributions of source code must retain the above copyright
    notice, this list of conditions and the following disclaimer.
 2. Redistributions in binary form must reproduce the above copyright
    notice, this list of conditions and the following disclaimer in the
    documentation and/or other materials provided with the distribution.
 3. The name of the author may not be used to endorse or promote products
    derived from this software without specific prior written permission.

 THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.antlr.stringtemplate.language;

import antlr.RecognitionException;
import antlr.collections.AST;
import org.antlr.stringtemplate.AttributeRenderer;
import org.antlr.stringtemplate.StringTemplate;
import org.antlr.stringtemplate.StringTemplateGroup;
import org.antlr.stringtemplate.StringTemplateWriter;

import java.io.IOException;
import java.io.StringWriter;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;

/** A single string template expression enclosed in $...; separator=...$
 *  parsed into an AST chunk to be evaluated.
 */
public class ASTExpr extends Expr {
    public static final int MISSING = -1; // writing -1 char means missing not empty
    public static final String DEFAULT_ATTRIBUTE_NAME = "it";
    public static final String DEFAULT_ATTRIBUTE_NAME_DEPRECATED = "attr";
	public static final String DEFAULT_INDEX_VARIABLE_NAME = "i";
	public static final String DEFAULT_INDEX0_VARIABLE_NAME = "i0";
	public static final String DEFAULT_MAP_VALUE_NAME = "_default_";
	public static final String DEFAULT_MAP_KEY_NAME = "key";

	/** Used to indicate "default:key" in maps within groups */
	public static final StringTemplate MAP_KEY_VALUE = new StringTemplate();

	/** Using an expr option w/o value, makes options table hold EMPTY_OPTION
	 *  value for that key.
	 */
	public static final String EMPTY_OPTION = "empty expr option";

	public static final Map defaultOptionValues = new HashMap() {
		{
			put("anchor", new StringTemplateAST(ActionEvaluator.STRING,"true"));
			put("wrap", new StringTemplateAST(ActionEvaluator.STRING,"\n"));
		}
	};

	/** John Snyders gave me an example implementation for this checking */
	public static final Set supportedOptions = new HashSet() {
		{
			add("anchor");
			add("format");
			add("null");
			add("separator");
			add("wrap");
		}
	};

	AST exprTree = null;

    /** store separator etc... */
    Map options = null;

	/** A cached value of wrap=expr from the <...> expression.
	 *  Computed in write(StringTemplate, StringTemplateWriter) and used
	 *  in writeAttribute.
	 */
	String wrapString = null;

	/** For null values in iterated attributes and single attributes that
	 *  are null, use this value instead of skipping.  For single valued
	 *  attributes like  it's a shorthand for
	 *  n/a
	 *  For iterated values , you get 0 for
	 *  for null list values.  Works for template application like:
	 *  }; null="0"> also.
	 */
	String nullValue = null;

	/** A cached value of separator=expr from the <...> expression.
	 *  Computed in write(StringTemplate, StringTemplateWriter) and used
	 *  in writeAttribute.
	 */
	String separatorString = null;

	/** A cached value of option format=expr */
	String formatString = null;

	public ASTExpr(StringTemplate enclosingTemplate, AST exprTree, Map options) {
		super(enclosingTemplate);
        this.exprTree = exprTree;
        this.options = options;
    }

	/** Return the tree interpreted when this template is written out. */
	public AST getAST() {
		return exprTree;
	}

    /** To write out the value of an ASTExpr, invoke the evaluator in eval.g
     *  to walk the tree writing out the values.  For efficiency, don't
     *  compute a bunch of strings and then pack them together.  Write out directly.
	 *
	 *  Compute separator and wrap expressions, save as strings so we don't
	 *  recompute for each value in a multi-valued attribute or expression.
	 *
	 *  If they set anchor option, then inform the writer to push current
	 *  char position.
     */
    public int write(StringTemplate self, StringTemplateWriter out) throws IOException {
        if ( exprTree==null || self==null || out==null ) {
            return 0;
        }
		// handle options, anchor, wrap, separator...
		StringTemplateAST anchorAST = (StringTemplateAST)getOption("anchor");
		if ( anchorAST!=null ) { // any non-empty expr means true; check presence
			out.pushAnchorPoint();
		}
        out.pushIndentation(getIndentation());
		handleExprOptions(self);
		//System.out.println("evaluating tree: "+exprTree.toStringList());
        ActionEvaluator eval =
                new ActionEvaluator(self,this,out);
		int n = 0;
        try {
            n = eval.action(exprTree); // eval and write out tree
        }
        catch (RecognitionException re) {
            self.error("can't evaluate tree: "+exprTree.toStringList(), re);
        }
        out.popIndentation();
		if ( anchorAST!=null ) {
			out.popAnchorPoint();
		}
		return n;
    }

	/** Grab and cache options; verify options are valid */
	protected void handleExprOptions(StringTemplate self) {
		// make sure options don't use format / renderer.  They are usually
		// strings which might invoke a string renderer etc...
		formatString = null;
		StringTemplateAST wrapAST = (StringTemplateAST)getOption("wrap");
		if ( wrapAST!=null ) {
			wrapString = evaluateExpression(self,wrapAST);
		}
		StringTemplateAST nullValueAST = (StringTemplateAST)getOption("null");
		if ( nullValueAST!=null ) {
			nullValue = evaluateExpression(self,nullValueAST);
		}
		StringTemplateAST separatorAST = (StringTemplateAST)getOption("separator");
		if ( separatorAST!=null ) {
			separatorString = evaluateExpression(self, separatorAST);
		}
		// following addition inspired by John Snyders
		StringTemplateAST formatAST =
			(StringTemplateAST)getOption("format");
		if ( formatAST!=null ) {
			formatString = evaluateExpression(self, formatAST);
		}

		// Check that option is valid
		if ( options != null ) {
			Iterator it = options.keySet().iterator();
			while ( it.hasNext() ) {
				String option = (String)it.next();
				if (!supportedOptions.contains(option)) {
					self.warning("ignoring unsupported option: "+option);
				}
			}
		}
	}

	// HELP ROUTINES CALLED BY EVALUATOR TREE WALKER

	/** For  treat the names, phones as lists
	 *  to be walked in lock step as n=names[i], p=phones[i].
	 */
	public Object applyTemplateToListOfAttributes(StringTemplate self,
												  List attributes,
												  StringTemplate templateToApply)
	{
		if ( attributes==null || templateToApply==null || attributes.size()==0 ) {
			return null; // do not apply if missing templates or empty values
		}
		Map argumentContext = null;
		// indicate it's an ST-created list
		List results = new StringTemplate.STAttributeList();

		// convert all attributes to iterators even if just one value
		for (int a = 0; a < attributes.size(); a++) {
			Object o = (Object) attributes.get(a);
			if ( o!=null ) {
				o = convertAnythingToIterator(o);
				attributes.set(a, o); // alter the list in place
			}
		}

		int numAttributes = attributes.size();

		// ensure arguments line up
		Map formalArguments = templateToApply.getFormalArguments();
		if ( formalArguments==null || formalArguments.size()==0 ) {
			self.error("missing arguments in anonymous"+
					   " template in context "+self.getEnclosingInstanceStackString());
			return null;
		}
		Object[] formalArgumentNames = formalArguments.keySet().toArray();
		if ( formalArgumentNames.length!=numAttributes ) {
			self.error("number of arguments "+formalArguments.keySet()+
					   " mismatch between attribute list and anonymous"+
					   " template in context "+self.getEnclosingInstanceStackString());
			// truncate arg list to match smaller size
			int shorterSize = Math.min(formalArgumentNames.length, numAttributes);
			numAttributes = shorterSize;
			Object[] newFormalArgumentNames = new Object[shorterSize];
			System.arraycopy(formalArgumentNames, 0,
							 newFormalArgumentNames, 0,
							 shorterSize);
			formalArgumentNames = newFormalArgumentNames;
		}

		// keep walking while at least one attribute has values
		int i = 0; // iteration number from 0
		while ( true ) {
			argumentContext = new HashMap();
			// get a value for each attribute in list; put into arg context
			// to simulate template invocation of anonymous template
			int numEmpty = 0;
			for (int a = 0; a < numAttributes; a++) {
				Iterator it = (Iterator) attributes.get(a);
				if ( it!=null && it.hasNext() ) {
					String argName = (String)formalArgumentNames[a];
					Object iteratedValue = it.next();
					argumentContext.put(argName, iteratedValue);
				}
				else {
					numEmpty++;
				}
			}
			if ( numEmpty==numAttributes ) {
				break;
			}
			argumentContext.put(DEFAULT_INDEX_VARIABLE_NAME, new Integer(i+1));
			argumentContext.put(DEFAULT_INDEX0_VARIABLE_NAME, new Integer(i));
			StringTemplate embedded = templateToApply.getInstanceOf();
			embedded.setEnclosingInstance(self);
			embedded.setArgumentContext(argumentContext);
			results.add(embedded);
			i++;
		}

		return results;
	}

	public Object applyListOfAlternatingTemplates(StringTemplate self,
                                                  Object attributeValue,
                                                  List templatesToApply)
    {
        if ( attributeValue==null || templatesToApply==null || templatesToApply.size()==0 ) {
            return null; // do not apply if missing templates or empty value
        }
        StringTemplate embedded = null;
        Map argumentContext = null;

        // normalize collections and such to use iterators
        // anything iteratable can be used for "APPLY"
		attributeValue = convertArrayToList(attributeValue);
		attributeValue = convertAnythingIteratableToIterator(attributeValue);

        if ( attributeValue instanceof Iterator ) {
			// results can be treated list an attribute, indicate ST created list
			List resultVector = new StringTemplate.STAttributeList();
            Iterator iter = (Iterator)attributeValue;
            int i = 0;
            while ( iter.hasNext() ) {
                Object ithValue = iter.next();
                if ( ithValue==null ) {
					if ( nullValue==null ) {
						continue;
					}
					ithValue = nullValue;
				}
                int templateIndex = i % templatesToApply.size(); // rotate through
                embedded = (StringTemplate)templatesToApply.get(templateIndex);
                // template to apply is an actual StringTemplate (created in
                // eval.g), but that is used as the examplar.  We must create
                // a new instance of the embedded template to apply each time
                // to get new attribute sets etc...
                StringTemplateAST args = embedded.getArgumentsAST();
                embedded = embedded.getInstanceOf(); // make new instance
                embedded.setEnclosingInstance(self);
                embedded.setArgumentsAST(args);
                argumentContext = new HashMap();
				Map formalArgs = embedded.getFormalArguments();
				boolean isAnonymous =
					embedded.getName() == StringTemplate.ANONYMOUS_ST_NAME;
				setSoleFormalArgumentToIthValue(embedded, argumentContext, ithValue);
				// if it's an anonymous template with a formal arg, don't set it/attr
				if ( !(isAnonymous && formalArgs!=null && formalArgs.size()>0) ) {
					argumentContext.put(DEFAULT_ATTRIBUTE_NAME, ithValue);
	                argumentContext.put(DEFAULT_ATTRIBUTE_NAME_DEPRECATED, ithValue);
				}
				argumentContext.put(DEFAULT_INDEX_VARIABLE_NAME, new Integer(i+1));
				argumentContext.put(DEFAULT_INDEX0_VARIABLE_NAME, new Integer(i));
                embedded.setArgumentContext(argumentContext);
                evaluateArguments(embedded);
                /*
				System.err.println("i="+i+": applyTemplate("+embedded.getName()+
                        ", args="+argumentContext+
                        " to attribute value "+ithValue);
                */
				resultVector.add(embedded);
                i++;
            }
			if ( resultVector.size()==0 ) {
				resultVector = null;
			}
			return resultVector;
        }
        else {
            /*
            System.out.println("setting attribute "+DEFAULT_ATTRIBUTE_NAME+" in arg context of "+
            embedded.getName()+
            " to "+attributeValue);
            */
            embedded = (StringTemplate)templatesToApply.get(0);
            argumentContext = new HashMap();
			Map formalArgs = embedded.getFormalArguments();
			StringTemplateAST args = embedded.getArgumentsAST();
			setSoleFormalArgumentToIthValue(embedded, argumentContext, attributeValue);
			boolean isAnonymous =
				embedded.getName() == StringTemplate.ANONYMOUS_ST_NAME;
			// if it's an anonymous template with a formal arg, don't set it/attr
			if ( !(isAnonymous && formalArgs!=null && formalArgs.size()>0) ) {
				argumentContext.put(DEFAULT_ATTRIBUTE_NAME, attributeValue);
				argumentContext.put(DEFAULT_ATTRIBUTE_NAME_DEPRECATED, attributeValue);
			}
			argumentContext.put(DEFAULT_INDEX_VARIABLE_NAME, new Integer(1));
			argumentContext.put(DEFAULT_INDEX0_VARIABLE_NAME, new Integer(0));
            embedded.setArgumentContext(argumentContext);
            evaluateArguments(embedded);
            return embedded;
        }
    }

	protected void setSoleFormalArgumentToIthValue(StringTemplate embedded, Map argumentContext, Object ithValue) {
		Map formalArgs = embedded.getFormalArguments();
		if ( formalArgs!=null ) {
			String soleArgName = null;
			boolean isAnonymous =
				embedded.getName() == StringTemplate.ANONYMOUS_ST_NAME;
			if ( formalArgs.size()==1 || (isAnonymous&&formalArgs.size()>0) ) {
				if ( isAnonymous && formalArgs.size()>1 ) {
					embedded.error("too many arguments on {...} template: "+formalArgs);
				}
				// if exactly 1 arg or anonymous, give that the value of
				// "it" as a convenience like they said
				// $list:template(arg=it)$
				Set argNames = formalArgs.keySet();
				soleArgName = (String)argNames.toArray()[0];
				argumentContext.put(soleArgName, ithValue);
			}
		}
	}

	/** Return o.getPropertyName() given o and propertyName.  If o is
     *  a stringtemplate then access it's attributes looking for propertyName
     *  instead (don't check any of the enclosing scopes; look directly into
     *  that object).  Also try isXXX() for booleans.  Allow Map
	 *  as special case (grab value for key).
	 *
	 *  Cache repeated requests for obj.prop within same group.
     */
	public Object getObjectProperty(StringTemplate self,
									final Object o,
									final Object propertyName) {
		if ( o==null || propertyName==null ) {
			return null;
		}
		/*
		// see if property is cached in group's cache
		Object cachedValue =
			self.getGroup().getCachedObjectProperty(o,propertyName);
		if ( cachedValue!=null ) {
			return cachedValue;
		}
		Object value = rawGetObjectProperty(self, o, propertyName);
		// take care of array properties...convert to a List so we can
		// apply templates to the elements etc...
		value = convertArrayToList(value);
		self.getGroup().cacheObjectProperty(o,propertyName,value);
		*/
		Object value = rawGetObjectProperty(self, o, propertyName);
		// take care of array properties...convert to a List so we can
		// apply templates to the elements etc...
		value = convertArrayToList(value);
		return value;
	}

	protected Object rawGetObjectProperty(StringTemplate self, Object o, Object property) {
		Class c = o.getClass();
        Object value = null;

		// Special case: our automatically created Aggregates via
        // attribute name: "{obj.{prop1,prop2}}"
        if ( c==StringTemplate.Aggregate.class ) {
			String propertyName = (String)property;
            value = ((StringTemplate.Aggregate)o).get((String)propertyName);
			return value;
        }

        // Special case: if it's a template, pull property from
        // it's attribute table.
        // TODO: TJP just asked himself why we can't do inherited attr here?
        else if ( c==StringTemplate.class ) {
            Map attributes = ((StringTemplate)o).getAttributes();
            if ( attributes!=null ) {
				String propertyName = (String)property;
                value = attributes.get(propertyName);
				return value;
            }
        }

        // Special case: if it's a Map then pull using
        // key not the property method.
        if ( o instanceof Map ) {
            Map map = (Map)o;
			if ( property.equals("keys") ) {
				value = map.keySet();
			}
			else if ( property.equals("values") ) {
				value = map.values();
			}
			else if ( map.containsKey(property) ) {
				value = map.get(property);
			}
			else if ( map.containsKey(property.toString()) ) {
				// if we can't find the key, toString it
				value = map.get(property.toString());				
			}
			else {
				if ( map.containsKey(DEFAULT_MAP_VALUE_NAME) ) {
					value = map.get(DEFAULT_MAP_VALUE_NAME);
				}
			}
			if ( value == MAP_KEY_VALUE ) {
				value = property;
			}
			return value;
        }

		// try getXXX and isXXX properties

		// check cache
		Method m = null;
		/*
		Member cachedMember =
			self.getGroup().getCachedClassProperty(c,propertyName);
		if ( cachedMember!=null ) {
			try {
				if ( cachedMember instanceof Method ) {
					m = (Method)cachedMember;
					value = invokeMethod(m, o, value);
				}
				else {
					// must be a field
					Field f = (Field)cachedMember;
					value = accessField(f, o, value);
				}
			}
			catch (Exception e) {
				self.error("Can't get property "+propertyName+
					" from "+c.getName()+" instance", e);
			}
			return value;
		}
		*/

		// must look up using reflection
		String propertyName = (String)property;
		String methodSuffix = Character.toUpperCase(propertyName.charAt(0))+
			propertyName.substring(1,propertyName.length());
		m = getMethod(c,"get"+methodSuffix);
		if ( m==null ) {
			m = getMethod(c, "is"+methodSuffix);
		}
		if ( m != null ) {
			// save to avoid lookup later
			//self.getGroup().cacheClassProperty(c,propertyName,m);
			try {
				value = invokeMethod(m, o, value);
			}
			catch (Exception e) {
				self.error("Can't get property "+propertyName+" using method get/is"+methodSuffix+
					" from "+c.getName()+" instance", e);
			}
		}
		else {
			// try for a visible field
			try {
				Field f = c.getField(propertyName);
				//self.getGroup().cacheClassProperty(c,propertyName,f);
				try {
					value = accessField(f, o, value);
				}
				catch (IllegalAccessException iae) {
					self.error("Can't access property "+propertyName+" using method get/is"+methodSuffix+
						" or direct field access from "+c.getName()+" instance", iae);
				}
			}
			catch (NoSuchFieldException nsfe) {
				self.error("Class "+c.getName()+" has no such attribute: "+propertyName+
					" in template context "+self.getEnclosingInstanceStackString(), nsfe);
			}
		}

        return value;
    }

	protected Object accessField(Field f, Object o, Object value) throws IllegalAccessException {
		try {
			// make sure it's accessible (stupid java)
			f.setAccessible(true);
		}
		catch (SecurityException se) {
			; // oh well; security won't let us
		}
		value = f.get(o);
		return value;
	}

	protected Object invokeMethod(Method m, Object o, Object value) throws IllegalAccessException, InvocationTargetException {
		try {
			// make sure it's accessible (stupid java)
			m.setAccessible(true);
		}
		catch (SecurityException se) {
			; // oh well; security won't let us
		}
		value = m.invoke(o,(Object[])null);
		return value;
	}

	protected Method getMethod(Class c, String methodName) {
		Method m;
		try {
			m = c.getMethod(methodName, (Class[])null);
		}
		catch (NoSuchMethodException nsme) {
			m = null;
		}
		return m;
	}

	/** Normally StringTemplate tests presence or absence of attributes
	 *  for adherence to my principles of separation, but some people
	 *  disagree and want to change.
	 *
	 *  For 2.0, if the object is a boolean, do something special. $if(boolean)$
	 *  will actually test the value.  Now, this breaks my rules of entanglement
	 *  listed in my paper, but it truly surprises programmers to have booleans
	 *  always true.  Further, the key to isolating logic in the model is avoiding
	 *  operators (for which you need attribute values).  But, no operator is
	 *  needed to use boolean values.  Well, actually I guess "!" (not) is
	 *  an operator.  Regardless, for practical reasons, I'm going to technically
	 *  violate my rules as I currently have them defined.  Perhaps for a future
	 *  version of the paper I will refine the rules.
	 *
	 *  Post 2.1, I added a check for non-null Iterators, Collections, ...
	 *  with size==0 to return false. TJP 5/1/2005
	 */
	public boolean testAttributeTrue(Object a) {
		if ( a==null ) {
			return false;
		}
		if ( a instanceof Boolean ) {
			return ((Boolean)a).booleanValue();
		}
		if ( a instanceof Collection ) {
			return ((Collection)a).size()>0;
		}
		if ( a instanceof Map ) {
			return ((Map)a).size()>0;
		}
		if ( a instanceof Iterator ) {
			return ((Iterator)a).hasNext();
		}
		return true; // any other non-null object, return true--it's present
	}

    /** For now, we can only add two objects as strings; convert objects to
	 *  Strings then cat.
     */
    public Object add(Object a, Object b) {
        if ( a==null ) { // a null value means don't do cat, just return other value
            return b;
        }
        else if ( b==null ) {
            return a;
        }
		return a.toString() + b.toString();
    }

    /** Call a string template with args and return result.  Do not convert
     *  to a string yet.  It may need attributes that will be available after
     *  this is inserted into another template.
     */
	public StringTemplate getTemplateInclude(StringTemplate enclosing,
											 String templateName,
											 StringTemplateAST argumentsAST)
    {
		//System.out.println("getTemplateInclude: look up "+enclosing.getGroup().getName()+"::"+templateName);
        StringTemplateGroup group = enclosing.getGroup();
        StringTemplate embedded = group.getEmbeddedInstanceOf(enclosing, templateName);
        if ( embedded==null ) {
            enclosing.error("cannot make embedded instance of "+templateName+
                    " in template "+enclosing.getName());
            return null;
        }
        embedded.setArgumentsAST(argumentsAST);
        evaluateArguments(embedded);
        return embedded;
    }

    /** How to spit out an object.  If it's not a StringTemplate nor a
     *  List, just do o.toString().  If it's a StringTemplate,
     *  do o.write(out).  If it's a Vector, do a write(out,
     *  o.elementAt(i)) for all elements.  Note that if you do
     *  something weird like set the values of a multivalued tag
     *  to be vectors, it will effectively flatten it.
     *
     *  If self is an embedded template, you might have specified
     *  a separator arg; used when is a vector.
	 */
    public int writeAttribute(StringTemplate self, Object o, StringTemplateWriter out) {
        return write(self,o,out);
    }

	/*	Write o relative to self to out.
	 *
	 *  John Snyders fixes here for formatString.  Basically, any time
	 *  you are about to write a value, check formatting.
	 */
	protected int write(StringTemplate self,
						Object o,
						StringTemplateWriter out)
    {
        if ( o==null ) {
			if ( nullValue==null ) {
				return MISSING;
			}
			o = nullValue; // continue with null option if specified
        }
        int n = 0;
        try {
            if ( o instanceof StringTemplate ) return writeTemplate(self, o, out);
            o = convertAnythingIteratableToIterator(o); // normalize
            if ( o instanceof Iterator ) return writeIterableValue(self, o, out);
            return writePOJO(self, o, out);
        }
        catch (IOException io) {
            self.error("problem writing object: "+o, io);
        }
		return n;
    }

    protected int writePOJO(StringTemplate self, Object o, StringTemplateWriter out) throws IOException {
        int n = 0;
        AttributeRenderer renderer =
            self.getAttributeRenderer(o.getClass());
        String v = null;
        if ( renderer!=null ) {
            if ( formatString != null ) v = renderer.toString(o, formatString);
            else v = renderer.toString(o);
        }
        else v = o.toString();
        if ( wrapString!=null ) n = out.write(v, wrapString);
        else n = out.write( v );
        return n;
    }

    protected int writeTemplate(StringTemplate self, Object o, StringTemplateWriter out) throws IOException {
        int n = 0;
        StringTemplate stToWrite = (StringTemplate)o;
        // failsafe: perhaps enclosing instance not set
        // Or, it could be set to another context!  This occurs
        // when you store a template instance as an attribute of more
        // than one template (like both a header file and C file when
        // generating C code).  It must execute within the context of
        // the enclosing template.
        stToWrite.setEnclosingInstance(self);
        // if self is found up the enclosing instance chain, then
        // infinite recursion
        if ( StringTemplate.inLintMode() &&
             StringTemplate.isRecursiveEnclosingInstance(stToWrite) )
        {
            // throw exception since sometimes eval keeps going
            // even after I ignore this write of o.
            throw new IllegalStateException("infinite recursion to "+
                    stToWrite.getTemplateDeclaratorString()+" referenced in "+
                    stToWrite.getEnclosingInstance().getTemplateDeclaratorString()+
                    "; stack trace:\n"+stToWrite.getEnclosingInstanceStackTrace());
        }
        else {
            // if we have a wrap string, then inform writer it
            // might need to wrap
            if ( wrapString!=null ) {
                n = out.writeWrapSeparator(wrapString);
            }
            // check if formatting needs to be applied to the stToWrite
            if ( formatString != null ) {
                AttributeRenderer renderer =
                    self.getAttributeRenderer(String.class);
                if ( renderer != null ) {
                    // you pay a penalty for applying format option to a template
                    // because the template must be written to a temp StringWriter so it can
                    // be formatted before being written to the real output.
                    StringWriter buf = new StringWriter();
                    StringTemplateWriter sw =
                        self.getGroup().getStringTemplateWriter(buf);
                    stToWrite.write(sw);
                    n = out.write(renderer.toString(buf.toString(), formatString));
                    return n;
                }
            }
            n = stToWrite.write(out);
        }
        return n;
    }

    protected int writeIterableValue(StringTemplate self,
                                     Object o,
                                     StringTemplateWriter out)
        throws IOException
    {
        int n = 0;
        Iterator iter = (Iterator)o;
        boolean seenAValue = false;
        while ( iter.hasNext() ) {
            Object iterValue = iter.next();
            if ( iterValue==null ) iterValue = nullValue;
            if ( iterValue!=null ) {
                // if no separator or separator but iterValue isn't
                // a single IF condition template
                if ( separatorString==null ) {
                    // if no separator, don't waste time writing to
                    // temp buffer
                    int nw = write(self, iterValue, out);
                    if ( nw!=MISSING ) n += nw;
                    continue;
                }
                // if value to emit is a template, only buffer its
                // value if it's nullable (can eval to missing).
                // Only a sequence of IF can eval to missing.
                if ( iterValue instanceof StringTemplate ) {
                    StringTemplate st = (StringTemplate)iterValue;
                    int nchunks = st.getChunks()!=null ? st.getChunks().size() : 0;
                    boolean nullable = true;
                    for (int i=0; i");
		argContextST.setEnclosingInstance(enclosing);
		argContextST.setArgumentContext(self.getArgumentContext());

        ActionEvaluator eval =
                new ActionEvaluator(argContextST,this,null);
		/*
		System.out.println("eval args: "+argumentsAST.toStringList());
		System.out.println("ctx is "+self.getArgumentContext());
		*/
        try {
            // using any initial argument context (such as when obj is set),
            // evaluate the arg list like bold(item=obj).  Since we pass
            // in any existing arg context, that context gets filled with
            // new values.  With bold(item=obj), context becomes:
            // {[obj=...],[item=...]}.
            Map ac = eval.argList(argumentsAST, self, self.getArgumentContext());
            self.setArgumentContext(ac);
        }
        catch (RecognitionException re) {
            self.error("can't evaluate tree: "+argumentsAST.toStringList(), re);
        }
	}

	/*
	public static final Set arraysConvertibleToList = new HashSet() {
		{
			add(int[].class);
			add(long[].class);
			add(float[].class);
			add(double[].class);
		}
	};
	*/

	/** Do a standard conversion of array attributes to a List.  Wrap the
	 *  array instead of copying like old version.  Make an
	 *  ArrayWrappedInList that knows to create an ArrayIterator.
	 */
	public static Object convertArrayToList(Object value) {
		if ( value==null ) {
			return null;
		}
		if ( value.getClass().isArray() ) {
			if ( value.getClass().getComponentType().isPrimitive() ) {
				return new ArrayWrappedInList(value);
			}
			return Arrays.asList((Object[])value);
		}
		return value;
	}

	protected static Object convertAnythingIteratableToIterator(Object o) {
		Iterator iter = null;
		if ( o instanceof Collection ) {
			iter = ((Collection)o).iterator();
		}
		else if ( o instanceof Map ) {
			iter = ((Map)o).values().iterator();
		}
		else if ( o instanceof Iterator ) {
			iter = (Iterator)o;
		}
		if ( iter==null ) {
			return o;
		}
		return iter;
	}

	protected static Iterator convertAnythingToIterator(Object o) {
		Iterator iter = null;
		if ( o instanceof Collection ) {
			iter = ((Collection)o).iterator();
		}
		else if ( o instanceof Map ) {
			iter = ((Map)o).values().iterator();
		}
		else if ( o instanceof Iterator ) {
			iter = (Iterator)o;
		}
		if ( iter==null ) {
			List singleton = new StringTemplate.STAttributeList(1);
			singleton.add(o);
			return singleton.iterator();
		}
		return iter;
	}

	/** Return the first attribute if multiple valued or the attribute
	 *  itself if single-valued.  Used in 
	 */
	public Object first(Object attribute) {
		if ( attribute==null ) {
			return null;
		}
		Object f = attribute;
		attribute = convertAnythingIteratableToIterator(attribute);
		if ( attribute instanceof Iterator ) {
			Iterator it = (Iterator)attribute;
			if ( it.hasNext() ) {
				f = it.next();
			}
		}

		return f;
	}

	/** Return the everything but the first attribute if multiple valued
	 *  or null if single-valued.  Used in .
	 */
	public Object rest(Object attribute) {
		if ( attribute==null ) {
			return null;
		}
		Object theRest = attribute;
		attribute = convertAnythingIteratableToIterator(attribute);
		if ( attribute instanceof Iterator ) {
			List a = new ArrayList();
			Iterator it = (Iterator)attribute;
            if ( !it.hasNext() ) {
                return null; // if not even one value return null
            }
            it.next(); // ignore first value
			while (it.hasNext()) {
				Object o = (Object) it.next();
				if ( o!=null ) a.add(o);
			}
			return a;
		}
		else {
			theRest = null;  // rest of single-valued attribute is null
		}

		return theRest;
	}

	/** Return the last attribute if multiple valued or the attribute
	 *  itself if single-valued.  Used in .  This is pretty
	 *  slow as it iterates until the last element.  Ultimately, I could
	 *  make a special case for a List or Vector.
	 */
	public Object last(Object attribute) {
		if ( attribute==null ) {
			return null;
		}
		Object last = attribute;
		attribute = convertAnythingIteratableToIterator(attribute);
		if ( attribute instanceof Iterator ) {
			Iterator it = (Iterator)attribute;
			while ( it.hasNext() ) {
				last = it.next();
			}
		}

		return last;
	}

	/** Return a new list w/o null values. */
	public Object strip(Object attribute) {
		if ( attribute==null ) {
			return null;
		}
		attribute = convertAnythingIteratableToIterator(attribute);
		if ( attribute instanceof Iterator ) {
			List a = new ArrayList();
			Iterator it = (Iterator)attribute;
			while (it.hasNext()) {
				Object o = (Object) it.next();
				if ( o!=null ) a.add(o);
			}
			return a;
		}
		return attribute; // strip(x)==x when x single-valued attribute
	}

	/** Return all but the last element.  trunc(x)=null if x is single-valued. */
	public Object trunc(Object attribute) {
		if ( attribute==null ) {
			return null;
		}
		attribute = convertAnythingIteratableToIterator(attribute);
		if ( attribute instanceof Iterator ) {
			List a = new ArrayList();
			Iterator it = (Iterator)attribute;
			while (it.hasNext()) {
				Object o = (Object) it.next();
				if ( it.hasNext() ) a.add(o); // only add if not last one
			}
			return a;
		}
		return null; // trunc(x)==null when x single-valued attribute
	}

	/** Return the length of a multiple valued attribute or 1 if it is a
	 *  single attribute. If attribute is null return 0.
	 *  Special case several common collections and primitive arrays for
	 *  speed.  This method by Kay Roepke.
	 */
	public Object length(Object attribute) {
		if ( attribute == null) {
			return new Integer(0);
		}
		int i = 1;		// we have at least one of something. Iterator and arrays might be empty.
		if (attribute instanceof Map) {
			i = ((Map)attribute).size();
		}
		else if (attribute instanceof Collection) {
			i = ((Collection)attribute).size();
		}
		else if ( attribute instanceof Object[] ) {
			Object[] list = (Object[])attribute;
			i = list.length;
		}
		else if ( attribute instanceof int[] ) {
			int[] list = (int[])attribute;
			i = list.length;
		}
		else if ( attribute instanceof long[] ) {
			long[] list = (long[])attribute;
			i = list.length;
		}
		else if ( attribute instanceof float[] ) {
			float[] list = (float[])attribute;
			i = list.length;
		}
		else if ( attribute instanceof double[] ) {
			double[] list = (double[])attribute;
			i = list.length;
		}
		else if ( attribute instanceof Iterator) {
			Iterator it = (Iterator)attribute;
			i = 0;
			while ( it.hasNext() ) {
				it.next();
				i++;
			}
		}
		return new Integer(i);
	}

	public Object getOption(String name) {
		Object value = null;
		if ( options!=null ) {
			value = options.get(name);
			if ( value==EMPTY_OPTION ) {
				return defaultOptionValues.get(name);
			}
		}
		return value;
	}

	public String toString() {
		return exprTree.toStringList();
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy