package test.graph;

import java.awt.BorderLayout;
import java.text.ParseException;
import javax.swing.JFrame;
import javax.swing.JPanel;
import org.das2.datum.Units;
import org.das2.graph.DasAxis;
import org.das2.graph.DasCanvas;
import org.das2.graph.DasColumn;
import org.das2.graph.DasPlot;
import org.das2.graph.DasRow;
import org.das2.graph.GraphUtil;
import org.das2.graph.Renderer;
import org.das2.graph.SeriesRenderer;
import org.das2.qds.AbstractQFunction;
import org.das2.qds.BundleDataSet;
import org.das2.qds.DDataSet;
import org.das2.qds.DRank0DataSet;
import org.das2.qds.DataSetUtil;
import org.das2.qds.JoinDataSet;
import org.das2.qds.QDataSet;
import org.das2.qds.ops.Ops;

/**
 * relatively simple demo for Larry at APL.
 * @author jbf
 */
public class SimpleDemo {
    public static void main( String[] args ) throws ParseException {

        int width = 500;
        int height = 400;

        JPanel panel= new JPanel();

        panel.setLayout(new BorderLayout());

        DasCanvas canvas = new DasCanvas(width, height);
        canvas.setAntiAlias(true);

        panel.add(canvas, BorderLayout.CENTER );

        // read data
        QDataSet yds = Ops.sin( Ops.linspace(0,10,1000) );
        QDataSet tds= Ops.timegen( "2010-01-01T00:00", "1 s", 1000 );

        QDataSet ds= Ops.link( tds, yds );

        // here's some old das2 autoranging, works for this case
        DasAxis xaxis = GraphUtil.guessXAxis(ds);
        DasAxis yaxis = GraphUtil.guessYAxis(ds);

        DasPlot plot = new DasPlot( xaxis, yaxis );

        // here's autoplot as of 2005
        Renderer r= GraphUtil.guessRenderer(ds);
        plot.addRenderer( r );

        // ugh.  I need to make antialiased the default.  Right now it reads the property from $HOME/.dasrc
        if ( r instanceof SeriesRenderer ) {
            ((SeriesRenderer)r).setAntiAliased(true);
        }

        xaxis.setTcaFunction( new AbstractQFunction() {

            public QDataSet value(QDataSet parm) {

                if ( parm.rank()==1 ) {
                    parm= new JoinDataSet(parm);
                } else {
                    throw new IllegalArgumentException("rank should be 1");
                }

                QDataSet result= values( parm );

                return result.slice(0);
            }

            public QDataSet values(QDataSet parms) {

                BundleDataSet outbds1= BundleDataSet.createRank1Bundle();

                //TODO: this example should use bundle descriptor, and then just set BUNDLE_1 property.

                DDataSet sec= DDataSet.createRank1(parms.length());
                sec.putProperty( QDataSet.LABEL, "Sec" );
                DDataSet rand= DDataSet.createRank1(parms.length());
                rand.putProperty( QDataSet.LABEL, "Rand" );
                rand.putProperty( QDataSet.FORMAT, "%5.2f" );
                DDataSet rand2= DDataSet.createRank1(parms.length());
                rand2.putProperty( QDataSet.LABEL, "Rand2" );
                rand2.putProperty( QDataSet.FORMAT, "%5.3f" );

                for ( int i=0; i<parms.length(); i++ ) {
                    QDataSet parm= parms.slice(i);

                    QDataSet time= parm.slice(0);

                    sec.putValue( i, Ops.mod( time, DataSetUtil.asDataSet(3600,Units.seconds) ).value() );
                    rand.putValue(i, Ops.randu(1).slice(0).value() );
                    rand2.putValue( Ops.randu(1).slice(0).value() );

                }

                outbds1.bundle(sec);
                outbds1.bundle(rand);
                outbds1.bundle(rand2);

                return outbds1;
            }

            public QDataSet exampleInput() {
                BundleDataSet inbds= BundleDataSet.createRank0Bundle();
                DRank0DataSet dd= DataSetUtil.asDataSet(0,Units.t2000);
                dd.putProperty(QDataSet.LABEL, "Time") ;
                inbds.bundle( dd );
                return inbds;
            }

            public QDataSet exampleOutput() {
                return value( exampleInput() );
            }
        } );

        xaxis.setDrawTca(true);

        canvas.add( plot, DasRow.create( canvas, null, "0%+2em", "100%-4em" ),
                DasColumn.create( canvas, null, "0%+5em", "100%-7em" ) );

        JFrame frame= new JFrame();
        frame.getContentPane().add( panel );
        frame.pack();

        frame.setVisible(true);
    }

}