-
Notifications
You must be signed in to change notification settings - Fork 227
Expand file tree
/
Copy pathNBinOp.java
More file actions
75 lines (62 loc) · 2.12 KB
/
NBinOp.java
File metadata and controls
75 lines (62 loc) · 2.12 KB
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
/**
* Copyright 2009, Google Inc. All rights reserved.
* Licensed to PSF under a Contributor Agreement.
*/
package org.python.indexer.ast;
import org.python.indexer.Indexer;
import org.python.indexer.Scope;
import org.python.indexer.types.NType;
import org.python.indexer.types.NUnknownType;
import org.python.indexer.types.NUnionType;
public class NBinOp extends NNode {
static final long serialVersionUID = -8961832251782335108L;
public NNode left;
public NNode right;
public String op;
public NBinOp(NNode target, NNode value, String op) {
this(target, value, op, 0, 1);
}
public NBinOp(NNode target, NNode value, String op, int start, int end) {
super(start, end);
this.left = target;
this.right = value;
this.op = op;
addChildren(target, value);
}
@Override
public NType resolve(Scope s) throws Exception {
NType ltype = null, rtype = null;
if (left != null) {
ltype = resolveExpr(left, s).follow();
}
if (right != null) {
rtype = resolveExpr(right, s).follow();
}
// If either non-null operand is a string, assume the result is a string.
if (ltype == Indexer.idx.builtins.BaseStr || rtype == Indexer.idx.builtins.BaseStr) {
return setType(Indexer.idx.builtins.BaseStr);
}
// If either non-null operand is a number, assume the result is a number.
if (ltype == Indexer.idx.builtins.BaseNum || rtype == Indexer.idx.builtins.BaseNum) {
return setType(Indexer.idx.builtins.BaseNum);
}
if (ltype == null) {
return setType(rtype == null ? new NUnknownType() : rtype);
}
if (rtype == null) {
return setType(new NUnknownType());
}
return setType(NUnionType.union(ltype, rtype));
}
@Override
public String toString() {
return "<BinOp:" + left + " " + op + " " + right + ">";
}
@Override
public void visit(NNodeVisitor v) {
if (v.visit(this)) {
visitNode(left, v);
visitNode(right, v);
}
}
}