/**
 * ************************************************************************
 * Copyright (C) 2010 Atlas of Living Australia All Rights Reserved.
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with the
 * License. You may obtain a copy of the License at http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for
 * the specific language governing rights and limitations under the License.
 * *************************************************************************
 */
package org.ala.spatial.web.services;

import java.io.File;
import java.net.URLDecoder;
import java.util.HashMap;
import javax.servlet.http.HttpServletRequest;
import org.ala.layers.intersect.Grid;
import org.ala.layers.intersect.SimpleRegion;
import org.ala.layers.intersect.SimpleShapeFile;
import org.ala.layers.util.SpatialUtil;
import org.ala.spatial.analysis.index.LayerFilter;
import org.ala.spatial.util.AlaspatialProperties;
import org.ala.spatial.util.GridCutter;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * 
 * @author ajay
 */
@Controller
public class SamplingWSController {

    /**
     * chart basis: occurrence or area filter: [lsid] + [area] series: [taxon
     * level] or [string attribute] or [boolean attribute] or [shape file layer]
     * xaxis: [taxon level] or [attribute] or [layer] Optional ',' + min + ' '
     * max Optional ',' + categories delimited by '\t' yaxis: [taxon level] or
     * [attribute] or [layer] or [count type=countspecies, countoccurrence, size
     * of area] Optional ',' + min + ' ' max Optional ',' + categories delimited
     * by '\t' zaxis: [count type=species, occurrence, size of area]
     *
     * Types of chart basis: bA. Occurrence locations. bB. Layer intersections.
     *
     * Types of variables: tA. catagorical (e.g. STRING ATTRIBUTES[bA], SHAPE
     * FILE INTERSECTION, TAXON NAMES[bA]) tB. continous (e.g. INT or DOUBLE
     * ATTRIBUTES[bA] and GRID FILE INTERSECTION) tC. presence (e.g. BOOLEAN
     * ATTRIBUTES[bA], BOOLEAN=TRUE COUNT[bA], SPECIES COUNT[bA] and OCCURRENCE
     * COUNT[bA], INTERSECTION AREA SIZE[bB])
     *
     * Valid variables for Filtering: fZ. none. fA. catagorical (e.g. STRING
     * ATTRIBUTES[bA], SHAPE FILE INTERSECTION, TAXON NAMES[bA]) fB. continous
     * (e.g. INT or DOUBLE ATTRIBUTES[bA] and GRID FILE INTERSECTION) fC.
     * presence (e.g. BOOLEAN ATTRIBUTES[bA]) fD. Active Area
     *
     * Valid variables for Series: sZ. none sA. catagorical (e.g. STRING
     * ATTRIBUTES[bA], SHAPE FILE INTERSECTION, TAXON NAMES[bA]) sB. presence
     * (e.g. BOOLEAN ATTRIBUTES[bA])
     *
     * Valid variables for Y-Axis: yA. catagorical (e.g. STRING ATTRIBUTES[bA],
     * SHAPE FILE INTERSECTION, TAXON NAMES[bA]) yB. continous (e.g. INT or
     * DOUBLE ATTRIBUTES[bA] and GRID FILE INTERSECTION) yC. presence (e.g.
     * BOOLEAN=TRUE COUNT[bA], SPECIES COUNT[bA] and OCCURRENCE COUNT[bA],
     * INTERSECTION AREA SIZE[bB])
     *
     * Valid variables for X-Axis: tA. catagorical (e.g. STRING ATTRIBUTES[bA],
     * SHAPE FILE INTERSECTION, TAXON NAMES[bA]) tB. continous (e.g. INT or
     * DOUBLE ATTRIBUTES[bA] and GRID FILE INTERSECTION)
     *
     * Valid variables for Z-Axis: zZ. none. zA. presence (e.g. BOOLEAN=TRUE
     * COUNT[bA], SPECIES COUNT[bA] and OCCURRENCE COUNT[bA], INTERSECTION AREA
     * SIZE[bB])
     *
     *
     * Axis Combinations
     *
     * table for bA yA	yB	yC tA	(1) XYBlockChart	Box & Whisker	Histogram tB	(2)
     * Scatterplot or (1) XYBlockChart	Histogram
     *
     * table for bB yA	yB	yC tA	(1) XYBlockChart	Box & Whisker	(2) tB	(2)	(1)
     * XYBlockChart	(2)
     *
     * (1) requires Z-Axis variable. (2) not allowed
     *
     *
     *
     * @param req
     * @return
     */
    @RequestMapping(value = "/ws/chart", method = RequestMethod.GET)
    public @ResponseBody
    String chart(HttpServletRequest req) {
        try {
            String wkt = (req.getParameter("wkt") == null) ? null : URLDecoder.decode(req.getParameter("wkt"), "UTF-8");
            String xaxis = URLDecoder.decode(req.getParameter("xaxis"), "UTF-8");
            String yaxis = URLDecoder.decode(req.getParameter("yaxis"), "UTF-8");
            String sDivisions = (req.getParameter("divisions") == null) ? null : URLDecoder.decode(req.getParameter("divisions"), "UTF-8");
            int divisions = 20;
            if (sDivisions != null) {
                divisions = Integer.parseInt(sDivisions);
            }

            LayerFilter[] envelope = null;
            SimpleRegion region = null;
            if (wkt != null && wkt.startsWith("ENVELOPE")) {
                envelope = LayerFilter.parseLayerFilters(wkt);
            } else {
                region = SimpleShapeFile.parseWKT(wkt);
            }

            //bB

            //TODO: more than env layers
            String[] layers = new String[2];
            double[][] extents = new double[2][2];
            String[] s = xaxis.split(",");
            layers[0] = s[0];
            extents[0][0] = Double.parseDouble(s[1]);
            extents[0][1] = Double.parseDouble(s[2]);
            s = yaxis.split(",");
            layers[1] = s[0];
            extents[1][0] = Double.parseDouble(s[1]);
            extents[1][1] = Double.parseDouble(s[2]);

            float[][] cutoffs = new float[2][divisions];

            //get linear cutoffs
            for (int i = 0; i < layers.length; i++) {
                for (int j = 0; j < divisions; j++) {
                    cutoffs[i][j] = (float) (extents[i][0] + (extents[i][1] - extents[i][0]) * ((j + 1) / (float) divisions));
                }
                cutoffs[i][divisions - 1] = (float) extents[i][1];  //set max
            }

            //get grid data
            float[][] data = new float[2][];
            String cutDataPath = GridCutter.cut2(layers, AlaspatialProperties.getLayerResolutionDefault(), region, envelope, null);
            Grid g = new Grid(cutDataPath + File.separator + layers[0]);
            data[0] = g.getGrid();
            data[1] = new Grid(cutDataPath + File.separator + layers[1]).getGrid();

            //TODO: delete cut grid files

            int len = data[0].length;
            int divs = 20;  //same as number of cutpoints in Legend
            double[][] area = new double[divs][divs];

            for (int i = 0; i < len; i++) {
                if (Float.isNaN(data[0][i]) || Float.isNaN(data[1][i])) {
                    continue;
                }
                int x = getPos(data[0][i], cutoffs[0]);
                int y = getPos(data[1][i], cutoffs[1]);

                if (x >= 0 && x < area.length
                        && y >= 0 && y < area[x].length) {
                    area[x][y] += SpatialUtil.cellArea(Double.parseDouble(AlaspatialProperties.getLayerResolutionDefault()), g.ymin + (i / g.ncols) * g.yres);
                }
            }

            //to csv
            StringBuilder sb = new StringBuilder();
            sb.append(",").append(layers[1]).append("\n").append(layers[0]);
            for (int j = 0; j < divs; j++) {
                sb.append(",").append(cutoffs[1][j]);
            }
            for (int i = 0; i < divs; i++) {
                sb.append("\n").append(cutoffs[0][i]);
                for (int j = 0; j < divs; j++) {
                    sb.append(",").append(area[i][j]);
                }
            }

            return sb.toString();

        } catch (Exception e) {
            e.printStackTrace();
        }

        return null;
    }

    int getPos(float d, float[] cutoffs) {
        int pos = java.util.Arrays.binarySearch(cutoffs, d);
        if (pos < 0) {
            pos = (pos * -1) - 1;
        }

        return pos;
    }
}