aboutsummaryrefslogtreecommitdiff
path: root/project/QBarInterpreter/src/com/modulus/qbar/lang/QBIterator.java
blob: 251e6f3afc2899a9bf82bf4e47fee1c0ee7451c1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package com.modulus.qbar.lang;

import java.util.Iterator;

import com.modulus.qbar.core.QBFunction;
import com.modulus.qbar.core.QBNamespace;
import com.modulus.qbar.core.QBObject;
import com.modulus.qbar.core.QBStruct;
import com.modulus.qbar.core.primitive.QBInt;

public abstract class QBIterator<T extends QBObject> extends QBObject implements Iterator<QBObject>{
	/**
	 * 
	 */
	private static final long serialVersionUID = -8496833881500665535L;
	private String variableName;
	private T iterable;
	
	public static final QBStruct ITERATOR = new QBStruct("Iterator");
	
	public static final QBFunction hasNext = new QBFunction(1){

		/**
		 * 
		 */
		private static final long serialVersionUID = -7059204966992996830L;

		@Override
		public QBObject execute(QBObject[] args) {
			QBObject obj = args[0];
			QBIterator iter = (QBIterator) obj;
			
			return iter.hasNext() ? new QBInt(1) : new QBInt(0);
		}
		
	};
	
	public static final QBFunction next = new QBFunction(1){

		/**
		 * 
		 */
		private static final long serialVersionUID = 9165087060843674645L;

		@Override
		public QBObject execute(QBObject[] args) {
			QBObject obj = args[0];
			
			@SuppressWarnings("unchecked")
			QBIterator<QBObject> iter = (QBIterator<QBObject>) obj;
			
			return iter.next();
		}
		
	};
	
	static{
		ITERATOR.set("hasNext", hasNext);
		ITERATOR.set("next", next);
	}
	
	private QBNamespace variableNmspce;
	
	public QBIterator(T iterable) {
		super(ITERATOR);
		this.iterable = iterable;
	}

	@Override
	public QBObject get(String str){
		QBNamespace sup = this.getSuper();
		
		if(sup == null)
			return null;
		
		return sup.get(str);
	}
	
	@Override
	public void set(String str, QBObject obj){
		QBNamespace sup = this.getSuper();
		
		if( this.variableName.equals(str) && sup != null)
			variableNmspce.set(str, obj);
		
		else
			super.set(str, obj);
	}

	@Override
	public void remove() {
		throw new RuntimeException("Not Implemented");
	}

	public void setVariableName(String variableName) {
		this.variableName = variableName;
	}

	public String getVariableName() {
		return variableName;
	}

	public T getIterable() {
		return iterable;
	}
	
	protected abstract QBObject nextObject();
	
	public QBObject next(){
		
		QBObject ret = nextObject();
		this.set(this.getVariableName(), ret);
		
		return ret;
	}

	public void setVariableNmspce(QBNamespace variableNmspce) {
		this.variableNmspce = variableNmspce;
	}

	public QBNamespace getVariableNmspce() {
		return variableNmspce;
	}
}