CmpopAdapter.java

package org.python.antlr.adapter;

import java.util.ArrayList;
import java.util.List;

import org.python.antlr.ast.cmpopType;
import org.python.antlr.op.Eq;
import org.python.antlr.op.Gt;
import org.python.antlr.op.GtE;
import org.python.antlr.op.In;
import org.python.antlr.op.Is;
import org.python.antlr.op.IsNot;
import org.python.antlr.op.Lt;
import org.python.antlr.op.LtE;
import org.python.antlr.op.NotEq;
import org.python.antlr.op.NotIn;
import org.python.core.Py;
import org.python.core.PyObject;

public class CmpopAdapter implements AstAdapter {

    public Object py2ast(PyObject o) {
        if (o != Py.None) {
            switch ((o).asInt()) {
                case 1:
                    return cmpopType.Eq;
                case 2:
                    return cmpopType.NotEq;
                case 3:
                    return cmpopType.Lt;
                case 4:
                    return cmpopType.LtE;
                case 5:
                    return cmpopType.Gt;
                case 6:
                    return cmpopType.GtE;
                case 7:
                    return cmpopType.Is;
                case 8:
                    return cmpopType.IsNot;
                case 9:
                    return cmpopType.In;
                case 10:
                    return cmpopType.NotIn;
                default:
                    return cmpopType.UNDEFINED;
            }
        }
        return cmpopType.UNDEFINED;
    }

    public PyObject ast2py(Object o) {
        if (o == null) {
            return Py.None;
        }
        switch ((cmpopType)o) {
            case Eq:
                return new Eq();
            case NotEq:
                return new NotEq();
            case Lt:
                return new Lt();
            case LtE:
                return new LtE();
            case Gt:
                return new Gt();
            case GtE:
                return new GtE();
            case Is:
                return new Is();
            case IsNot:
                return new IsNot();
            case In:
                return new In();
            case NotIn:
                return new NotIn();
            default:
                return Py.None;
        }
    }

    public List iter2ast(PyObject iter) {
        List<cmpopType> cmpops = new ArrayList<cmpopType>();
        if (iter != Py.None) {
            for(Object o : (Iterable)iter) {
                cmpops.add((cmpopType)py2ast((PyObject)o));
            }
        }
        return cmpops;
    }
}