Library: org.gvsig.sldtools

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.sldconverter.legend.driver;

import java.io.File;

import java.io.IOException;

import org.gvsig.fmap.mapcontext.exceptions.WriteLegendException;

import org.gvsig.fmap.mapcontext.rendering.legend.ILegend;

import org.gvsig.fmap.mapcontext.rendering.legend.driver.ILegendWriter;

import org.gvsig.sldconverter.SLDConverterLocator;

import org.gvsig.sldconverter.SLDConverterManager;

import org.gvsig.sldsupport.SLDSupportLocator;

import org.gvsig.sldsupport.SLDSupportManager;

import org.gvsig.sldsupport.sld.SLDRoot;

import org.gvsig.sldsupport.sld.layer.SLDLayer;

/**

 * Legend writer based on the SLD converter

 */

public class LegendToSLDWriter implements ILegendWriter {

    public void write(ILegend legend, File outFile, String format)

        throws WriteLegendException, IOException {

        SLDSupportManager sldSupMan = null;

        sldSupMan = SLDSupportLocator.getInstance().getSLDSupportManager();

        if (sldSupMan == null) {

            throw new WriteLegendException(legend,

                new Exception("No SLD support manager registered."));

        }

        SLDConverterManager sldConMan = null;

        sldConMan = SLDConverterLocator.getInstance().getSLDConverterManager();        

        if (sldConMan == null) {

            throw new WriteLegendException(legend,

                new Exception("No SLD conversion manager registered."));

        }

        String fext = getFileExtension(format);

        if (fext.compareToIgnoreCase(SLDSupportManager.FILE_EXTENSION) != 0) {

            throw new WriteLegendException(

                legend, new Exception("Unexpected format requested: " + format));

        }

        String fver = this.getFormatVersion(format);

        if (fver == null) {

            fver = "1.0.0";

        }

        SLDLayer lyr = null;

        try {

            lyr = sldConMan.toSLDLayer(legend);

        } catch (Exception e) {

            throw new WriteLegendException(legend, e);

        }

        if (lyr == null) {

            throw new WriteLegendException(legend,

                new Exception("Unable to convert to SLD layer"));

        }

        SLDRoot root = new SLDRoot(fver);

        root.getLayers().add(lyr);

        try {

            sldSupMan.write(root, fver, outFile);

        } catch (Exception e) {

            throw new WriteLegendException(legend, e);

        }        

    }

    // ========================================================

    /**

     * Returns null if mime format is not parsed properly

     * 

     * @param fmt MIME format

     * 

     * @return

     */

    private String getFileExtension(String fmt) {

        // Example: "text/xml; subtype=gml/2.1.2" => "gml"

        if (fmt == null) {

            return null;

        }

        String[] parts = fmt.split(";");

        String aux = "";

        if (parts.length > 1) {

            aux = parts[1].trim();

        } else {

            aux = parts[0].trim();

        }

        parts = aux.split("=");

        if (parts.length > 1) {

            aux = parts[1].trim();

            // Example: aux = "gml/2.1.2"

            parts = aux.split("/");

            return parts[0].length() == 0 ? null : parts[0];

        } else {

            aux = parts[0].trim();

            // Example: "text/xml"

            parts = aux.split("/");

            if (parts.length > 1) {

                return parts[1].length() == 0 ? null : parts[1];

            } else {

                return aux.length() == 0 ? null : aux;

            }

        }

    }

    /**

     * Returns null if mime format is not parsed properly

     * 

     * @param fmt MIME format

     * 

     * @return

     */

    private String getFormatVersion(String fmt) {

        // Example: "text/xml; subtype=gml/2.1.2" => "2.1.2"

        if (fmt == null) {

            return null;

        }

        String[] parts = fmt.split(";");

        String aux = "";

        if (parts.length > 1) {

            aux = parts[1].trim();

        } else {

            return null;

        }

        parts = aux.split("=");

        if (parts.length > 1) {

            aux = parts[1].trim();

            // Example: aux = "gml/2.1.2"

            parts = aux.split("/");

            return parts[1].length() == 0 ? null : parts[1];

        } else {

            return null;

        }

    }   

}