| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * 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 com.intellij.debugger.engine.evaluation.expression; |
| |
| import com.intellij.debugger.engine.evaluation.EvaluateException; |
| import com.intellij.debugger.engine.evaluation.EvaluationContextImpl; |
| import com.sun.jdi.ArrayReference; |
| import com.sun.jdi.ObjectReference; |
| import com.sun.jdi.Value; |
| |
| /** |
| * @author egor |
| */ |
| public class ForeachStatementEvaluator extends ForStatementEvaluatorBase { |
| private final Evaluator myIterationParameterEvaluator; |
| private final Evaluator myIterableEvaluator; |
| private final Evaluator myBodyEvaluator; |
| |
| private Evaluator myConditionEvaluator; |
| private Evaluator myNextEvaluator; |
| |
| private int myArrayLength = -1; |
| private int myCurrentIndex = 0; |
| |
| private Modifier myModifier; |
| |
| public ForeachStatementEvaluator(Evaluator iterationParameterEvaluator, |
| Evaluator iterableEvaluator, |
| Evaluator bodyEvaluator, |
| String labelName) { |
| super(labelName); |
| myIterationParameterEvaluator = iterationParameterEvaluator; |
| myIterableEvaluator = new DisableGC(iterableEvaluator); |
| myBodyEvaluator = bodyEvaluator != null ? new DisableGC(bodyEvaluator) : null; |
| } |
| |
| public Modifier getModifier() { |
| return myModifier; |
| } |
| |
| @Override |
| protected Object evaluateInitialization(EvaluationContextImpl context, Object value) throws EvaluateException { |
| final Object iterable = myIterableEvaluator.evaluate(context); |
| if (!(iterable instanceof ObjectReference)) { |
| throw new EvaluateException("Unable to do foreach for" + iterable); |
| } |
| IdentityEvaluator iterableEvaluator = new IdentityEvaluator((Value)iterable); |
| if (iterable instanceof ArrayReference) { |
| myArrayLength = ((ArrayReference)iterable).length(); |
| myNextEvaluator = new AssignmentEvaluator(myIterationParameterEvaluator, |
| new Evaluator() { |
| @Override |
| public Object evaluate(EvaluationContextImpl context) throws EvaluateException { |
| return ((ArrayReference)iterable).getValue(myCurrentIndex++); |
| } |
| |
| @Override |
| public Modifier getModifier() { |
| return null; |
| } |
| }); |
| } |
| else { |
| Object iterator = new MethodEvaluator(iterableEvaluator, null, "iterator", null, new Evaluator[0]).evaluate(context); |
| IdentityEvaluator iteratorEvaluator = new IdentityEvaluator((Value)iterator); |
| myConditionEvaluator = new MethodEvaluator(iteratorEvaluator, null, "hasNext", null, new Evaluator[0]); |
| myNextEvaluator = new AssignmentEvaluator(myIterationParameterEvaluator, |
| new MethodEvaluator(iteratorEvaluator, null, "next", null, new Evaluator[0])); |
| } |
| return value; |
| } |
| |
| private boolean isArray() { |
| return myArrayLength > -1; |
| } |
| |
| @Override |
| protected Object evaluateCondition(EvaluationContextImpl context) throws EvaluateException { |
| if (isArray()) { |
| return myCurrentIndex < myArrayLength; |
| } |
| else { |
| Object res = myConditionEvaluator.evaluate(context); |
| myModifier = myConditionEvaluator.getModifier(); |
| return res; |
| } |
| } |
| |
| @Override |
| protected void evaluateBody(EvaluationContextImpl context) throws EvaluateException { |
| myNextEvaluator.evaluate(context); |
| if (myBodyEvaluator != null) { |
| myBodyEvaluator.evaluate(context); |
| } |
| } |
| } |