package org.autoplot.jythonsupport; import java.util.logging.Logger; import org.das2.datum.Datum; import org.das2.datum.Units; import org.das2.datum.UnitsConverter; import org.python.core.Py; import org.python.core.PyFloat; import org.python.core.PyInteger; import org.python.core.PyJavaInstance; import org.python.core.PyLong; import org.python.core.PyObject; import org.python.core.PyString; import org.das2.qds.DataSetUtil; import org.das2.qds.QDataSet; /** * Wrap a das2 Datum, implementing operators. * @author jbf */ public class PyDatum extends PyJavaInstance { private static final Logger logger= Logger.getLogger("jython"); Datum datum; public PyDatum( Datum d ) { super(d); this.datum= d; } /* plus, minus, multiply, divide */ @Override public PyObject __add__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( datum.add( ((PyDatum)arg0).datum ) ); } else if ( arg0 instanceof PyQDataSet ) { return arg0.__radd__(this); } else if ( arg0.isNumberType() ) { return new PyDatum( datum.add( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__add__(arg0); } } @Override public PyObject __radd__(PyObject arg0) { return __add__(arg0); } @Override public PyObject __sub__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( datum.subtract(((PyDatum)arg0).datum ) ); } else if ( arg0 instanceof PyQDataSet ) { return arg0.__rsub__(this); } else if ( arg0.isNumberType() ) { return new PyDatum( datum.subtract( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__sub__(arg0); } } @Override public PyObject __rsub__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( (((PyDatum)arg0).datum ).subtract(datum) ); } else if ( arg0.isNumberType() ) { return new PyDatum( datum.getUnits().createDatum( arg0.__float__().getValue() ).subtract( datum ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__rsub__(arg0); } } @Override public PyObject __mul__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( datum.multiply( ((PyDatum)arg0).datum ) ); } else if ( arg0 instanceof PyQDataSet ) { return arg0.__rmul__( this ); } else if ( arg0.isNumberType() ) { return new PyDatum( datum.multiply( Units.dimensionless.createDatum( arg0.__float__().getValue() ) ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__mul__(arg0); } } @Override public PyObject __rmul__(PyObject arg0) { return __mul__(arg0); } @Override public PyObject __div__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( datum.divide( ((PyDatum)arg0).datum ) ); } else if ( arg0 instanceof PyQDataSet ) { return arg0.__rdiv__( this ); } else if ( arg0.isNumberType() ) { return new PyDatum( datum.divide( Units.dimensionless.createDatum( arg0.__float__().getValue() ) ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__div__(arg0); } } @Override public PyObject __rdiv__( PyObject arg0 ) { if ( arg0 instanceof PyDatum ) { return new PyDatum( ((PyDatum)arg0).datum.divide(datum) ); } else if ( arg0.isNumberType() ) { return new PyDatum( Units.dimensionless.createDatum( arg0.__float__().getValue() ).divide( datum ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__rdiv__(arg0); } } @Override public PyObject __floordiv__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { double d= this.datum.doubleValue( datum.getUnits() ); double divisor= ((PyDatum)arg0).datum.doubleValue(datum.getUnits()); d= Math.floor( d/divisor ); return new PyDatum( Units.dimensionless.createDatum(d) ); } else if ( arg0.isNumberType() ) { double d= this.datum.doubleValue( datum.getUnits() ); double divisor= arg0.__float__().getValue(); d= Math.floor( d/divisor ); return new PyDatum( this.datum.getUnits().createDatum(d) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__floordiv__(arg0); } } @Override public PyObject __rfloordiv__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { double divisor= this.datum.doubleValue( datum.getUnits() ); double d= ((PyDatum)arg0).datum.doubleValue(datum.getUnits()); d= Math.floor( d/divisor ); return new PyDatum( Units.dimensionless.createDatum(d) ); } else if ( arg0.isNumberType() ) { double divisor= this.datum.doubleValue( datum.getUnits() ); double d= arg0.__float__().getValue(); d= Math.floor( d/divisor ); return new PyDatum( this.datum.getUnits().createDatum(d) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__rfloordiv__(arg0); } } @Override public PyObject __mod__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { Datum d1= this.datum; Units u1= d1.getUnits().getOffsetUnits(); Datum d2= ((PyDatum)arg0).datum; Units u2= d2.getUnits(); final UnitsConverter uc= u1.getConverter(u2); final double base= 0; double d= uc.convert( d1.value()-base ) % (d2.value()); return new PyDatum( u2.createDatum(d) ); } else if ( arg0.isNumberType() ) { Datum d1= this.datum; final double base= 0; double d= ( d1.value()-base ) % (arg0.__float__().getValue() ); return new PyDatum( Units.dimensionless.createDatum(d) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__mod__(arg0); } } @Override public PyObject __rmod__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { Datum d2= this.datum; Units u2= d2.getUnits().getOffsetUnits(); Datum d1= ((PyDatum)arg0).datum; Units u1= d1.getUnits(); final UnitsConverter uc= u1.getConverter(u2); final double base= 0; double d= uc.convert( d1.value()-base ) % (d2.value()); return new PyDatum( u2.createDatum(d) ); } else if ( arg0.isNumberType() ) { Datum d1= this.datum; double divisor= d1.value(); double d= ( arg0.__float__().getValue() ) % ( divisor ); return new PyDatum( Units.dimensionless.createDatum(d) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__rmod__(arg0); } } /* unary negate and plus operator */ @Override public PyObject __pos__() { return this; } @Override public PyObject __neg__() { return new PyDatum( this.datum.negative() ); } @Override public PyObject __abs__() { return new PyDatum( this.datum.abs() ); } /* pow operator (**) */ @Override public PyObject __pow__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( datum.power( ((PyDatum)arg0).datum ) ); } else if ( arg0.isNumberType() ) { double darg= ( arg0.__float__().getValue() ); return new PyDatum( datum.power( darg ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__pow__(arg0); } } @Override public PyObject __rpow__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return new PyDatum( ((PyDatum)arg0).datum.power( this.datum ) ); } else if ( arg0.isNumberType() ) { double darg= ( arg0.__float__().getValue() ); return new PyDatum( Units.dimensionless.createDatum( darg ).power( datum ) ); } else { return new PyQDataSet( DataSetUtil.asDataSet(datum) ).__rpow__(arg0); } } @Override public boolean __nonzero__() { Units u= datum.getUnits(); if ( u!=null && u.getOffsetUnits()!=u ) { throw new IllegalArgumentException("data must be dimensionless or a ratiometric datum."); } else { return datum.value()!=0; } } @Override public PyObject __int__() { return Py.newInteger((int)datum.value()); } @Override public PyFloat __float__() { return Py.newFloat((double)datum.value()); } @Override public PyLong __long__() { //TODO: check on this. return Py.newLong( ((int)datum.value()) ); } private static PyObject toPyBoolean( boolean r ) { return new PyInteger( r ? 1 : 0 ); } @Override public PyObject __ge__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( datum.ge(((PyDatum)arg0).datum) ); } else if ( arg0.isNumberType() ) { return toPyBoolean( datum.ge( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return toPyBoolean( datum.ge(JythonOps.datum(arg0)) ); } } @Override public PyObject __gt__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( datum.gt(((PyDatum)arg0).datum) ); } else if ( arg0.isNumberType() ) { return toPyBoolean( datum.gt( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return toPyBoolean( datum.gt(JythonOps.datum(arg0)) ); } } @Override public PyObject __le__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( datum.le(((PyDatum)arg0).datum) ); } else if ( arg0.isNumberType() ) { return toPyBoolean( datum.le( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return toPyBoolean( datum.le(JythonOps.datum(arg0)) ); } } @Override public PyObject __lt__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( datum.lt(((PyDatum)arg0).datum) ); } else if ( arg0.isNumberType() ) { return toPyBoolean( datum.lt( datum.getUnits().createDatum( arg0.__float__().getValue() ) ) ); } else { return toPyBoolean( datum.lt(JythonOps.datum(arg0)) ); } } @Override public PyObject __eq__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( datum.equals(((PyDatum)arg0).datum) ); } else { return toPyBoolean( datum.equals(JythonOps.datum(arg0)) ); } } @Override public PyObject __ne__(PyObject arg0) { if ( arg0 instanceof PyDatum ) { return toPyBoolean( !datum.equals(((PyDatum)arg0).datum) ); } else { return toPyBoolean( !datum.equals(JythonOps.datum(arg0)) ); } } @Override public PyObject __and__(PyObject o) { PyObject r= BinaryInfixOps.and( this, o ); if ( r instanceof PyQDataSet ) { throw new IllegalArgumentException("use .and operator"); } else { return r; } } @Override public PyObject __or__(PyObject o) { PyObject r= BinaryInfixOps.or( this, o ); if ( r instanceof PyQDataSet ) { throw new IllegalArgumentException("use .or operator"); } else { return r; } } /** * This is what does the magical coersion, see https://sourceforge.net/p/autoplot/bugs/1861/ * @param c the class needed. * @return instance of the class if available. */ @Override public Object __tojava__(Class c) { logger.fine("this is not supported because the double version would be used where a dataset would work."); // See sftp://nudnik.physics.uiowa.edu/home/jbf/project/autoplot/tests/1861/demo1861UsesDoubleNotQDataSet.jy if ( false && datum.getUnits()==Units.dimensionless ) { if ( c==double.class ) { return datum.value(); } else if ( c==Double.class ) { return datum.value(); } else if ( c==float.class ) { return (float)datum.value(); } else if ( c==Float.class ) { return (float)datum.value(); } else if ( c==Number.class ) { return datum.value(); } } return super.__tojava__(c); } @Override public PyString __str__( ) { return new PyString( datum.toString() ); } }