/*
 * Copyright (c) 2011 by Intevation GmbH, Germany <info@intevation.de>
 *
 * This file is part of MXD2map.
 *
 * This program is free software under the LGPL (>=v2.1)
 * Read the file LICENCE.txt coming with the software for details
 * or visit http://www.gnu.org/licenses/ if it does not exist.
 *
 * MXD2map has been developed on behalf of the
 * Bundesamt fuer Seeschifffahrt und Hydrographie (BSH) in Hamburg
 * by Intevation GmbH.
 *
 * Authors:
 * Raimund Renkert <raimund.renkert@intevation.de>
 * Bjoern Schilberg <bjoern.schilberg@intevation.de>
 * Stephan Holl <stephan.holl@intevation.de>
 */

package de.intevation.mxd.reader;

import java.io.IOException;

import org.apache.log4j.Logger;

import com.esri.arcgis.carto.ILayer;
import com.esri.arcgis.carto.IMap;
import com.esri.arcgis.carto.MapDocument;
import com.esri.arcgis.carto.FeatureLayer;
import com.esri.arcgis.carto.GdbRasterCatalogLayer;
import com.esri.arcgis.carto.IFeatureRenderer;
import com.esri.arcgis.carto.SimpleRenderer;
import com.esri.arcgis.carto.ClassBreaksRenderer;
import com.esri.arcgis.carto.UniqueValueRenderer;
import com.esri.arcgis.carto.GroupLayer;
import com.esri.arcgis.carto.WMSMapLayer;
import com.esri.arcgis.carto.WMSGroupLayer;
import com.esri.arcgis.carto.WMSLayer;
import com.esri.arcgis.carto.RasterLayer;

import org.w3c.dom.Document;
import org.w3c.dom.Element;

import de.intevation.mxd.ArcGISInitializer;
import de.intevation.mxd.utils.MapToXMLUtils;

/**
 * The MXD file reader.
 *
 * @author <a href="mailto:raimund.renkert@intevation.de">Raimund Renkert</a>
 */
public class MXDReader
implements IReader {
    /**
     * The Logger.
     */
    private static final Logger logger = Logger.getLogger(MXDReader.class);

    /**
     * Private member.
     */
    private String filename = "";
    private ArcGISInitializer initializer = null;
    private MapToXMLUtils util;
    private IMap map;
    private int invalidLayerCount;

    /**
     * Default constructor.
     */
    public MXDReader()
    throws IOException {
        logger.debug("constructor()");
        initializer = new ArcGISInitializer();
        util = new MapToXMLUtils();
        invalidLayerCount = 0;
    }


    /**
     * Initialize the ArcGIS Objects.
     */
    public boolean init()
    throws IOException {
        logger.debug("init()");
        if(!initializer.initArcGIS()) {
            return false;
        }
        if(!initializer.initArcGISLicenses()) {
            return false;
        }
        return true;
    }

    /**
     * Shutdown the ArcGIS Objects.
     */
    public boolean shutdown()
    throws IOException {
        logger.debug("shutdown()");
        if(initializer == null) {
            throw new IOException("Faild to shutdown!");
        }
        else {
            initializer.shutdownArcGIS();
            return true;
        }
    }

    /**
     * Set the path and filename to the MXD-File.
     *
     * @param name The MXD filename.
     */
    public void setFilename(String name) {
        logger.debug("setFilename()");
        filename = name;
    }

    /**
     * Get the map document from MXD.
     */
    private void openMapDocument()
    throws IOException {
        MapDocument m = new MapDocument();
        if(!m.isMapDocument(filename)) {
            throw new IOException(filename + " is not a map!");
        }
        m.open(filename, null);
        map = m.getMap(0);
    }

    /**
     * Read the MXD file content.
     */
    public boolean read()
    throws Exception {
        logger.debug("read()");
        if(filename.isEmpty()) {
            throw new IOException("Please set filename!");
        }
        else {
            util.addFilename(filename);
            openMapDocument();
            MapReader mreader;
            try {
                mreader = new MapReader (map);
                mreader.setUtil(util);
                mreader.read();

            }
            catch(Exception e) {
                throw e;
            }

            for(int i = 0; i < map.getLayerCount();i++) {
                ILayer layer = map.getLayer(i);
                try {
                    if (layer instanceof FeatureLayer) {
                        FeatureLayerReader lr = new FeatureLayerReader(layer);
                        lr.setUtil(util);
                        Element layerElement = lr.read();
                        if (layerElement == null) {
                            continue;
                        }
                        IFeatureRenderer renderer =
                            ((FeatureLayer)layer).getRenderer();
                        IRendererReader rreader;
                        if(renderer instanceof SimpleRenderer) {
                            rreader = new SimpleRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof ClassBreaksRenderer) {
                            rreader = new ClassBreakRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof UniqueValueRenderer) {
                            rreader = new UniqueValueRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else{
                            logger.debug("Not a known renderer type " +
                                         renderer.getClass().toString());
                        }
                    }
                    else if(layer instanceof GdbRasterCatalogLayer) {
                        GdbRasterCatalogLayerReader gclr =
                            new GdbRasterCatalogLayerReader(layer);
                        gclr.setUtil(util);
                        Element layerElement = gclr.read();
                        IFeatureRenderer renderer =
                            ((GdbRasterCatalogLayer)layer).getRenderer();
                        IRendererReader rreader;
                        if(renderer instanceof SimpleRenderer) {
                            rreader = new SimpleRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof ClassBreaksRenderer) {
                            rreader = new ClassBreakRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof UniqueValueRenderer) {
                            rreader = new UniqueValueRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else {
                            logger.debug("Not a known renderer type " +
                                         renderer.getClass().toString());
                        }
                    }
                    else if(layer instanceof GroupLayer) {
                        GroupLayerReader gl = new GroupLayerReader(layer);
                        gl.setUtil(util);
                        Element layerElement = gl.read("");
                        if(layerElement == null) {
                            throw new IOException ("Error reading grouplayer.");
                        }
                    }
                    else if(layer instanceof WMSMapLayer) {
                        WMSMapLayerReader lr = new WMSMapLayerReader(layer);
                        lr.setUtil(util);
                        Element layerElement = lr.read("");
                        if(layerElement == null) {
                            throw new IOException ("Error reading WMSMapLayer.");
                        }
                    }
                    else if(layer instanceof WMSGroupLayer) {
                        WMSGroupLayerReader lr = new WMSGroupLayerReader(layer);
                        lr.setUtil(util);
                        Element layerElement = lr.read("");
                        if(layerElement == null) {
                            throw new IOException ("Error reading WMSGrouplayer.");
                        }
                    }
                    else if(layer instanceof WMSLayer) {
                        WMSLayerReader lr = new WMSLayerReader(layer);
                        lr.setUtil(util);
                        Element layerElement = lr.read();
                        if(layerElement == null) {
                            throw new IOException ("Error reading WMSlayer.");
                        }
                    }
                    else if(layer instanceof RasterLayer) {
                        RasterLayerReader lr =
                            new RasterLayerReader(layer);
                        lr.setUtil(util);
                        Element layerElement = lr.read();
                        if(layerElement == null) {
                            throw new IOException ("Error reading RasterLayer.");
                        }
/* TODO implement support for classes in Raster Layers
                        IRasterRenderer renderer =
                            ((RasterLayer)layer).getRenderer();
                        IRendererReader rreader;
                        if(renderer instanceof SimpleRenderer) {
                            rreader = new SimpleRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof ClassBreaksRenderer) {
                            rreader = new ClassBreakRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else if(renderer instanceof UniqueValueRenderer) {
                            rreader = new UniqueValueRendererReader(renderer);
                            rreader.setParent(layerElement);
                            rreader.setUtil(util);
                            rreader.read();
                        }
                        else {
                            logger.debug("Not a known renderer type " +
                                         renderer.getClass().toString());
                        }
*/
                    }
                    else {
                        logger.info(
                            "Layertype not known: " +
                            layer.getClass().toString());
                    }
                }
                catch(Exception e) {
                    invalidLayerCount++;
                    if(i == 0 && (i + 1) == map.getLayerCount() ||
                       invalidLayerCount == map.getLayerCount()) {
                        throw new IOException(
                            "No valid layer in MXD file." +
                            " Aborting.");
                    }
                }
            }
        }
        return true;
    }

    /**
     * Get the root DOM document.
     */
    public Document getMapDocument() {
        logger.debug("getMapDocument()");
        return util.getDocument();
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :