/*
* Copyright 2014 Attila Szegedi, Daniel Dekany, Jonathan Revusky
*
* Licensed 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 freemarker.core;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.ListIterator;
import freemarker.template.SimpleSequence;
import freemarker.template.TemplateCollectionModel;
import freemarker.template.TemplateException;
import freemarker.template.TemplateHashModelEx;
import freemarker.template.TemplateModel;
import freemarker.template._TemplateAPI;
final class HashLiteral extends Expression {
private final ArrayList keys, values;
private final int size;
HashLiteral(ArrayList/*<Expression>*/ keys, ArrayList/*<Expression>*/ values) {
this.keys = keys;
this.values = values;
this.size = keys.size();
keys.trimToSize();
values.trimToSize();
}
TemplateModel _eval(Environment env) throws TemplateException {
return new SequenceHash(env);
}
public String getCanonicalForm() {
StringBuffer buf = new StringBuffer("{");
for (int i = 0; i < size; i++) {
Expression key = (Expression) keys.get(i);
Expression value = (Expression) values.get(i);
buf.append(key.getCanonicalForm());
buf.append(" : ");
buf.append(value.getCanonicalForm());
if (i != size-1) {
buf.append(",");
}
}
buf.append("}");
return buf.toString();
}
String getNodeTypeSymbol() {
return "{...}";
}
boolean isLiteral() {
if (constantValue != null) {
return true;
}
for (int i = 0; i < size; i++) {
Expression key = (Expression) keys.get(i);
Expression value = (Expression) values.get(i);
if (!key.isLiteral() || !value.isLiteral()) {
return false;
}
}
return true;
}
protected Expression deepCloneWithIdentifierReplaced_inner(
String replacedIdentifier, Expression replacement, ReplacemenetState replacementState) {
ArrayList clonedKeys = (ArrayList)keys.clone();
for (ListIterator iter = clonedKeys.listIterator(); iter.hasNext();) {
iter.set(((Expression)iter.next()).deepCloneWithIdentifierReplaced(
replacedIdentifier, replacement, replacementState));
}
ArrayList clonedValues = (ArrayList)values.clone();
for (ListIterator iter = clonedValues.listIterator(); iter.hasNext();) {
iter.set(((Expression)iter.next()).deepCloneWithIdentifierReplaced(
replacedIdentifier, replacement, replacementState));
}
return new HashLiteral(clonedKeys, clonedValues);
}
private class SequenceHash implements TemplateHashModelEx {
private HashMap map; // maps keys to integer offset
private TemplateCollectionModel keyCollection, valueCollection; // ordered lists of keys and values
SequenceHash(Environment env) throws TemplateException {
if (_TemplateAPI.getTemplateLanguageVersionAsInt(HashLiteral.this) >= _TemplateAPI.VERSION_INT_2_3_21) {
map = new LinkedHashMap();
for (int i = 0; i < size; i++) {
Expression keyExp = (Expression) keys.get(i);
Expression valExp = (Expression) values.get(i);
String key = keyExp.evalAndCoerceToString(env);
TemplateModel value = valExp.eval(env);
if (env == null || !env.isClassicCompatible()) {
valExp.assertNonNull(value, env);
}
map.put(key, value);
}
} else {
// Legacy hash literal, where repeated keys were kept when doing ?values or ?keys, yet overwritten when
// doing hash[key].
map = new HashMap();
ArrayList keyList = new ArrayList(size);
ArrayList valueList = new ArrayList(size);
for (int i = 0; i< size; i++) {
Expression keyExp = (Expression) keys.get(i);
Expression valExp = (Expression) values.get(i);
String key = keyExp.evalAndCoerceToString(env);
TemplateModel value = valExp.eval(env);
if (env == null || !env.isClassicCompatible()) {
valExp.assertNonNull(value, env);
}
map.put(key, value);
keyList.add(key);
valueList.add(value);
}
keyCollection = new CollectionAndSequence(new SimpleSequence(keyList));
valueCollection = new CollectionAndSequence(new SimpleSequence(valueList));
}
}
public int size() {
return size;
}
public TemplateCollectionModel keys() {
if (keyCollection == null) {
// This can only happen when IcI >= 2.3.21, an the map is a LinkedHashMap.
keyCollection = new CollectionAndSequence(new SimpleSequence(map.keySet()));
}
return keyCollection;
}
public TemplateCollectionModel values() {
if (valueCollection == null) {
// This can only happen when IcI >= 2.3.21, an the map is a LinkedHashMap.
valueCollection = new CollectionAndSequence(new SimpleSequence(map.values()));
}
return valueCollection;
}
public TemplateModel get(String key) {
return (TemplateModel) map.get(key);
}
public boolean isEmpty() {
return size == 0;
}
}
int getParameterCount() {
return size * 2;
}
Object getParameterValue(int idx) {
checkIndex(idx);
return idx % 2 == 0 ? keys.get(idx / 2) : values.get(idx / 2);
}
ParameterRole getParameterRole(int idx) {
checkIndex(idx);
return idx % 2 == 0 ? ParameterRole.ITEM_KEY : ParameterRole.ITEM_VALUE;
}
private void checkIndex(int idx) {
if (idx >= size * 2) {
throw new IndexOutOfBoundsException();
}
}
}