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 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;

    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.
     */
    public void setFilename(String name) {
        logger.debug("setFilename()");
        filename = name;
    }


    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 == "") {
            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 {
                        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 :