src/main/java/com/seaofnodes/simple/node/SarNode.java


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

package com.seaofnodes.simple.node;

import com.seaofnodes.simple.Parser;
import com.seaofnodes.simple.type.Type;
import com.seaofnodes.simple.type.TypeInteger;

public class SarNode extends ArithNode {
    public SarNode(Parser.Lexer loc, Node lhs, Node rhs) { super(loc, lhs, rhs, (byte)1); }
    @Override public Tag serialTag() { return Tag.Sar; }

    @Override public String op() { return ">>"; }
    @Override long doOp( long x, long y ) { return x >> y; }
    @Override TypeInteger doOp( TypeInteger x, TypeInteger y ) {
        if( y.isConstant() ) {
            int log = (int)y.value();
            return TypeInteger.make(-1L<<(63-log),(1L<<(63-log))-1);
        }
        return TypeInteger.BOT;
    }

    @Override
    public Node idealize() {
        Node lhs = in(1);
        Node rhs = in(2);

        Type t2 = rhs._type;

        // Sar of 0.
        if( t2.isConstant() && t2 instanceof TypeInteger i && (i.value()&63)==0 && forceInt(lhs) && lhs._type instanceof TypeInteger )
            return lhs;

        // TODO: x >> 3 >> (y ? 1 : 2) ==> x >> (y ? 4 : 5)

        return super.idealize();
    }
    @Override Node copy(Node lhs, Node rhs) { return new SarNode(_loc,lhs,rhs); }
}