SITools2

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package fr.ias.sitools.resources.geojson;

import fr.cnes.sitools.common.exception.SitoolsException;

import fr.cnes.sitools.common.resource.SitoolsParameterizedResource;

import fr.cnes.sitools.dataset.DataSetApplication;

import fr.cnes.sitools.dataset.database.DatabaseRequest;

import fr.cnes.sitools.dataset.database.DatabaseRequestFactory;

import fr.cnes.sitools.dataset.database.DatabaseRequestParameters;

import fr.cnes.sitools.dataset.database.common.DataSetExplorerUtil;

import fr.cnes.sitools.dataset.dto.ColumnConceptMappingDTO;

import fr.cnes.sitools.dataset.dto.DictionaryMappingDTO;

import fr.cnes.sitools.datasource.jdbc.model.AttributeValue;

import fr.cnes.sitools.datasource.jdbc.model.Record;

import fr.cnes.sitools.dictionary.model.Concept;

import java.io.IOException;

import java.io.Writer;

import java.util.HashMap;

import java.util.List;

import java.util.logging.Level;

import java.util.logging.Logger;

import org.restlet.data.MediaType;

import org.restlet.ext.wadl.MethodInfo;

import org.restlet.representation.Representation;

import org.restlet.representation.Variant;

import org.restlet.resource.Get;

// IMPORT POUR LE JSON

import org.restlet.representation.WriterRepresentation;

//-------------------------

/**

 *

 * @author marc

 */

public class DatasetToJsonORIG extends SitoolsParameterizedResource {

    private static final Logger LOG = Logger.getLogger(DatasetToJsonORIG.class.getName());

    private String dicoNameString = new String();

    final private String mimeTypeFits = "application/fits";

    final private int maxResultsSend = 999999999;

    final private double ratioSpolyToRaDec = 57.295779513;

    @Override

    public void sitoolsDescribe() {

        setName("DatasetToJson");

        setDescription("retrieves Dataset to Json file");

    }

    @Override

    public void doInit() {

        super.doInit();

    }

    /**

    * Get HTML

    * 

    * @return Representation the HTML result

    */

    @Get

    public Representation get() {

        return execute();

    }

    @Override

    protected void describeGet(MethodInfo info) {

        this.addInfo(info);

        info.setIdentifier("retrieve records and ");

        info.setDocumentation("Method to get Json file from a dataset");

        addStandardGetRequestInfo(info);

        DataSetExplorerUtil.addDatasetExplorerGetRequestInfo(info);

        DataSetApplication application = (DataSetApplication) getApplication();

        DataSetExplorerUtil.addDatasetExplorerGetFilterInfo(info, application.getFilterChained());

        addStandardResponseInfo(info);

        addStandardInternalServerErrorInfo(info);

    }

    @Override

    protected Representation head(Variant variant) {

        Representation repr = super.head();

        repr.setMediaType(MediaType.APPLICATION_JSON);

        return repr;

    }

    private Representation execute() {

        dicoNameString = DatasetToJsonModel.DICO_PARAM_NAME;

        Representation repr = new WriterRepresentation(MediaType.APPLICATION_JSON) {

            @Override

            public void write(Writer writer) throws IOException {

                // generate the DatabaseRequest

                DataSetApplication datasetApp = (DataSetApplication) getApplication();

                DataSetExplorerUtil dsExplorerUtil = new DataSetExplorerUtil(datasetApp, getRequest(), getContext());

                DictionaryMappingDTO dico =null;

                String dicoName = getParameterValue(dicoNameString);

                // Get the HashMap with as key the concept and value the columnAlias

                final HashMap<Concept,String> conceptsColumns = getcolumnAliasFromDico(dicoName, dico, datasetApp);

                // Get request parameters

                if (datasetApp.getConverterChained() != null) {

                    datasetApp.getConverterChained().getContext().getAttributes().put("REQUEST", getRequest());

                }

                // Get DatabaseRequestParameters

                final DatabaseRequestParameters params = dsExplorerUtil.getDatabaseParams();

                params.setPaginationExtend(maxResultsSend);

                //params.setPaginationExtend(datasetApp.getDataSet().getNbRecords());

                DatabaseRequest databaseRequest = DatabaseRequestFactory.getDatabaseRequest(params);

                try {

                    databaseRequest.createRequest();

                }catch (SitoolsException e) {

                    e.printStackTrace();

                }

                writer.write("{");

                writer.write("\"type\":\"FeatureCollection\",");

                // start features

                writer.write("\"totalResults\":" + databaseRequest.getTotalCount() + ",");

                writer.write("\"features\":[");

                // Next for reading first record

                try {

                    boolean first = true;

                    while (databaseRequest.nextResult()) {

                        Record rec = databaseRequest.getRecord();

                        if (!first) {

                          writer.write(",");

                        }

                        else {

                          first = false;

                        }

                        // creates a geometry and a properties string

                        String geometry = "";

                        String services = "";

                        String properties = "";

                        boolean firstProp = true;

                        String coords[] = new String[2];

                        String coordinateReference = "";

                        String urlDownloadFits = "";    

                        String spoly = "";

                        for(Concept concept : conceptsColumns.keySet()){

                            concept.getPropertyFromName("category");                            

                            String colAlias = conceptsColumns.get(concept);

                            for(AttributeValue attr : rec.getAttributeValues()){

                                if(attr.getName().equals(colAlias) && attr.getValue() != null && !attr.getValue().equals("")){

                                        if(concept.getPropertyFromName("category").getValue().contains("properties")){

                                            if (!firstProp) {

                                                properties += ",";

                                            }

                                            else {

                                                firstProp = false;

                                            }

                                            properties += "\"" + concept.getName().toString() + "\":\"" + attr.getValue() + "\"";

                                        }

                                        if(concept.getPropertyFromName("category").getValue().contains("geometry")){

                                            if(concept.getName().equals("ra")){

                                                coords[0]= attr.getValue().toString();

                                            }else if(concept.getName().equals("dec")){

                                                coords[1]= attr.getValue().toString();

                                            }

                                            if(concept.getName().equals("coordref")){

                                                coordinateReference = attr.getValue().toString();

                                            }

                                            if(concept.getName().equals("spoly")){

                                                spoly = attr.getValue().toString();

                                            }

                                        }

                                        if(concept.getPropertyFromName("category").getValue().contains("services")){

                                            if(concept.getName().equals("download")){

                                                 urlDownloadFits = attr.getValue().toString();

                                            }

                                        }

                                }

                            }  

                        }

                        // Set the geometry

                        geometry = setGeometry(coords, coordinateReference, spoly);

                        // Set The services

                        if(!urlDownloadFits.isEmpty()){

                            services = setServices(urlDownloadFits);

                        }

                        // start feature

                        writer.write("{");

                        writer.write("\"type\":\"feature\",");

                        // start geometry

                        writer.write("\"geometry\":{");

                        writer.write(geometry);

                        writer.write("}");

                        // end geometry

                        writer.write(",");

                        // start properties

                        writer.write("\"properties\":{");

                        writer.write(properties);

                        // end properties

                        writer.write("}");

                        // start services

                        if(!services.equals("")){

                            writer.write(",");

                            writer.write("\"services\":{");

                            writer.write(services);

                             // end services

                            writer.write("}");

                        }

                        // end feature

                        writer.write("}");

                    }

                    // end features

                    writer.write("]");

                }catch (SitoolsException ex) {

                    Logger.getLogger(DatasetToJsonORIG.class.getName()).log(Level.SEVERE, null, ex);

                }finally {

                    writer.write("}");

                    if (databaseRequest != null) {

                      try {

                        databaseRequest.close();

                      }catch (SitoolsException e) {

                        e.printStackTrace();

                      }

                    }

                    if (writer != null) {

                    writer.flush();

                    }

                }

            }

        };

        return repr;

    }

    private HashMap<Concept,String> getcolumnAliasFromDico(String dicoName, DictionaryMappingDTO dico, DataSetApplication datasetApp){

        dico = datasetApp.getColumnConceptMappingDTO(dicoName);

        final List<ColumnConceptMappingDTO> colConceptMappingDTOList = dico.getMapping();

        final HashMap<Concept,String> conceptColumn = new HashMap<Concept,String>();

        for(ColumnConceptMappingDTO concepts : colConceptMappingDTOList){

            conceptColumn.put(concepts.getConcept(), concepts.getColumnAlias());

        }

        return conceptColumn;

    }

    private String setGeometry(String[] coords, String coordRef, String spoly){

        String geometry = new String();

        String[] spolyStringTmp = new String[4];

        String[] spolyConverted = new String[4];

        if(spoly.equals("")){

            geometry = "\"coordinates\": ["+coords[0]+","+coords[1]+"],";

            geometry += "\"referencesystem\": \""+coordRef+"\",\"type\": \"Point\"";

        }else{

            String[] tmpString = spoly.split("\\)");

            //LOG.log(Level.INFO, " SPOLY : "+spoly);

            tmpString[0] = tmpString[0].substring(2, tmpString[0].length());

            spolyStringTmp[0] = tmpString[0];

            //LOG.log(Level.INFO, " i : 0 = "+spolyStringTmp[0]);

            for(int i=1;i<4;i++){

                spolyStringTmp[i] = tmpString[i].substring(2, tmpString[i].length());

                //LOG.log(Level.INFO, "spolyStringTmp  i :"+i+ " = "+spolyStringTmp[i]);

            }

            for(int k=0;k<spolyStringTmp.length;k++){

                spolyConverted[k] = "["+String.valueOf((Double.parseDouble(spolyStringTmp[k].split(",")[0])*ratioSpolyToRaDec))

                        +" , "+String.valueOf((Double.parseDouble(spolyStringTmp[k].split(",")[1])*ratioSpolyToRaDec))+"]";

                //LOG.log(Level.INFO, " spolyConverted k :"+k+ " = "+spolyConverted[k]);

            }

            geometry = "\"coordinates\":[["+spolyConverted[3]+","+spolyConverted[2]+","+spolyConverted[1]+","+spolyConverted[0]+","+spolyConverted[3]+"]],";

            geometry += "\"referencesystem\": \""+coordRef+"\",\"type\": \"Polygon\"";

        }

        return  geometry;

    }

    private String setServices(String urlDownFits){

        String services = new String();

        services += "\"download\":{ \"mimetype\":\""+mimeTypeFits+"\",";

        services += "\"url\":\""+urlDownFits+"\"}";

        return services;

    }

}