source: ether_megapoli/trunk/web/src/com/ether/ControllerPlot.java @ 130

package com.ether;

import com.medias.Context;
import gov.noaa.pmel.sgt.dm.SGTMetaData;
import gov.noaa.pmel.sgt.dm.SimpleLine;
import gov.noaa.pmel.util.GeoDateArray;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.springframework.context.ApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

import javax.imageio.ImageIO;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.List;

/**
 * @author vmipsl
 * @date 17 june 2011
 */
public class ControllerPlot
        extends HttpServlet
{
    public void init( final ServletConfig servletConfig )
            throws ServletException
    {
        try
        {
            final ApplicationContext appContext = WebApplicationContextUtils.getRequiredWebApplicationContext( servletConfig.getServletContext() );
            _etherService = (EtherService) appContext.getBean( "etherService", EtherService.class );
            _etherPlotService = (EtherPlotService) appContext.getBean( "etherPlotService", EtherPlotService.class );
        }
        catch( Throwable tx )
        {
            LOGGER.error( "Error initializing EtherService.", tx );
            throw new ServletException( "Error initializing EtherService.", tx );
        }
    }

    /**
     * Creates an image with the plot and send it to the response
     * Call by a jsp with <img src="visualization/plotEther?....">
     *
     * @param request
     * @param response
     * @throws ServletException
     */
    public void doGet( final HttpServletRequest request, final HttpServletResponse response )
            throws ServletException
    {
        boolean flagException = false;

        try
        {
            //Parameters
            final Integer plateformId = Integer.valueOf( request.getParameter( "plateformId" ) );
            final Integer parameterId = Integer.valueOf( request.getParameter( "parameterId" ) );
            final String title = request.getParameter( "title" );
            final String axeType = request.getParameter( "axeType" );

            final String dateBegin = request.getParameter( "dateBegin" );
            final String dateEnd = request.getParameter( "dateEnd" );
            final Calendar calendar = Calendar.getInstance();
            final Date formatedDateBegin;
            final Date formatedDateEnd;
            try
            {
                calendar.setTimeInMillis( Long.valueOf( dateBegin ) );
                formatedDateBegin = calendar.getTime();
                calendar.setTimeInMillis( Long.valueOf( dateEnd ) );
                formatedDateEnd = calendar.getTime();
            }
            catch( Exception e )
            {
                createErrorPane( response, request, WebException.WebCode.INVALID_DATE.toString(), e );
                flagException = true;
                throw new ServletException( WebException.WebCode.INVALID_DATE.toString(), e );
            }

            // Create plot
            //** ******************************************************************** **//
            // TODO : replace List<Data> by List<value> and List<double>
            //** ******************************************************************** **//
//            final List<Pair<Double, Date>> values = _etherService.getValuesByPlateformByParameterByPeriod( plateformId, parameterId, formatedDateBegin, formatedDateEnd );
            final List<Pair<Double, Date>> values = new ArrayList<Pair<Double, Date>>();
            final Pair<Double, Date> p1 = new Pair<Double, Date>( Double.valueOf( 23 ), new Date() );
            final Pair<Double, Date> p2 = new Pair<Double, Date>( Double.valueOf( 24 ), new Date() );
            final Pair<Double, Date> p3 = new Pair<Double, Date>( Double.valueOf( 25 ), new Date() );
            final Pair<Double, Date> p4 = new Pair<Double, Date>( Double.valueOf( 26 ), new Date() );
            values.add( p1 );
            values.add( p2 );
            values.add( p3 );
            values.add( p4 );
            final double[] dataArray = extractDoubles( values );
            final Date[] dateValues = extractDates( values );

            final GeoDateArray dateArray = new GeoDateArray( dateValues );
            //** ******************************************************************** **//
            //** ******************************************************************** **//


            final SimpleLine data = new SimpleLine( dateArray, dataArray, "legend" );
            SGTMetaData meta = new SGTMetaData( "Longitude", "degrees East", false, false );
            data.setXMetaData( meta );

            meta = new SGTMetaData( "parameterName", "parameterUnit", false, false );
            data.setYMetaData( meta );

            final MegapoliPlot megapoliPlot = new MegapoliPlot();
            megapoliPlot.setTitle( title );
            megapoliPlot.setData( data );

            final BufferedImage bufferedImage = _etherPlotService.createMainPane( megapoliPlot, Context.getLocale( request ) );
            ImageIO.write( bufferedImage, "png", response.getOutputStream() );
        }
        catch( Exception e1 )
        {
            try
            {
                if( !flagException )
                    createErrorPane( response, request, "", e1 );
            }
            catch( Exception e2 )
            {
                throw new ServletException( "Error : no possibity to write image in response", e2 );
            }
        }
    }

    private void createErrorPane( final HttpServletResponse response, final HttpServletRequest request, @NotNull final String mainError, @NotNull final Exception e )
            throws IOException
    {
        final BufferedImage errorImage = _etherPlotService.createErrorPane( mainError, e.toString(), Context.getLocale( request ) );
        ImageIO.write( errorImage, "png", response.getOutputStream() );
    }

    // TODO : remove this code !! ultracrados !
    private double[] extractDoubles( @NotNull final List<Pair<Double, Date>> datas )
    {
        final double[] doubles = new double[datas.size()];
        int i = 0;
        for( final Pair<Double, Date> data : datas )
        {
            doubles[i] = data.getFirstValue().doubleValue();
            i++;
        }
        return doubles;
    }

    // TODO : remove this code !! ultracrados !
    private Date[] extractDates( @NotNull final List<Pair<Double, Date>> datas )
    {
        final Date[] dates = new Date[datas.size()];
        int i = 0;
        for( final Pair<Double, Date> data : datas )
        {
            dates[i] = data.getSecondValue();
            i++;
        }
        return dates;
    }

    private static final Log LOGGER = LogFactory.getLog( ControllerPlot.class );

    private EtherService _etherService;
    private EtherPlotService _etherPlotService;

}