1 /* 2 * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package com.sun.codemodel.internal; 27 28 /** 29 * ForEach Statement 30 * This will generate the code for statement based on the new 31 * j2se 1.5 j.l.s. 32 * 33 * @author Bhakti 34 */ 35 public final class JForEach implements JStatement { 36 37 private final JType type; 38 private final String var; 39 private JBlock body = null; // lazily created 40 private final JExpression collection; 41 private final JVar loopVar; 42 JForEach(JType vartype, String variable, JExpression collection)43 public JForEach(JType vartype, String variable, JExpression collection) { 44 45 this.type = vartype; 46 this.var = variable; 47 this.collection = collection; 48 loopVar = new JVar(JMods.forVar(JMod.NONE), type, var, collection); 49 } 50 51 52 /** 53 * Returns a reference to the loop variable. 54 */ var()55 public JVar var() { 56 return loopVar; 57 } 58 body()59 public JBlock body() { 60 if (body == null) 61 body = new JBlock(); 62 return body; 63 } 64 state(JFormatter f)65 public void state(JFormatter f) { 66 f.p("for ("); 67 f.g(type).id(var).p(": ").g(collection); 68 f.p(')'); 69 if (body != null) 70 f.g(body).nl(); 71 else 72 f.p(';').nl(); 73 } 74 75 } 76