Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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 2

 * 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.

 *

 */

package org.gvsig.raster.impl.store.rmf;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileWriter;

import java.io.IOException;

import java.io.InputStreamReader;

import java.util.ArrayList;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.exception.ParsingException;

/**

 * Gestor para la escritura de bloques XML en el fichero RMF. Cada cliente que quiere

 * escribir en el se registrar? a traves de ClientRegister y esta clase ser? la encargada

 * de gestionar la lectura y escritura de bloques.

 *

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class RmfBlocksManager extends ClientRegister implements IRmfBlock {

        private final String MAIN_TAG = "RasterMetaFile";

        private String       version  = "1.0";

        private String path = null;

        /**

         * Constructor. Asigna la ruta del fichero.

         * @param path

         */

        public RmfBlocksManager(String path) {

                setPath(path);

        }

        /**

         * Asigna la ruta del fichero

         * @param path

         */

        public void setPath(String path) {

                this.path = path;

        }

        /**

         * Obtiene la ruta del fichero

         * @return path

         */

        public String getPath() {

                return path;

        }

        /**

         * Genera una copia de seguridad del fichero del rmf

         * @throws IOException

         */

        public void fileBackup() throws IOException {

                RasterLocator.getManager().getFileUtils().copyFile(getPath(), getPath() + "~");

        }

        public void read(String xml) throws ParsingException {

                File file = new File(getPath());

                ArrayList<String> lines = new ArrayList<String>();

                BufferedReader inGrf = null;

                try {

                        inGrf = new BufferedReader(new InputStreamReader(new FileInputStream(file)));

                        String str = inGrf.readLine();

                        while (str != null) {

                                for (int i = 0; i < clients.size(); i++) {

                                        IRmfBlock block = ((IRmfBlock) clients.get(i));

                                        String main = block.getMainTag();

                                        if (str.startsWith("<" + main)) {

                                                lines.clear();

                                                while (str.compareTo("</" + main + ">") != 0) {

                                                        lines.add(str);

                                                        str = inGrf.readLine();

                                                        if(str == null) {

                                                                inGrf.close();

                                                                return;

                                                        }

                                                }

                                                lines.add(str);

                                                StringBuffer buf = new StringBuffer();

                                                for (int j = 0; j < lines.size(); j++)

                                                        buf.append((String) lines.get(j));

                                                block.read(buf.toString());

                                        }

                                }

                                str = inGrf.readLine();

                        }

                        inGrf.close();

                } catch (FileNotFoundException e) {

                        throw new ParsingException("File Input error: creating BufferedReader");

                } catch (IOException ex) {

                        throw new ParsingException("File Input error: reading lines");

                }

        }

        /**

         * Creaci?n de nuevo fichero RMF. A?ade la cabecera y vuelca el contenido de

         * todos los IRmfBlock.

         * @param file Fichero

         * @return true si el fichero no existia y se ha creado nuevo

         * @throws IOException 

         * @throws IOException

         */

        private void create(File file) throws IOException {

                file.createNewFile();

                FileWriter writer = new FileWriter(file);

                writer.write("<?xml version=\"1.0\" encoding=\"ISO-8859-15\"?>\n");

                writer.write("<" + getMainTag() + " version=\"" + version + "\">\n");

                for (int i = 0; i < clients.size(); i++) {

                        IRmfBlock block = ((IRmfBlock) clients.get(i));

                        String xmlBlock = block.write();

                        if(xmlBlock != null)

                                writer.write(xmlBlock);

                }

                writer.write("</" + getMainTag() + ">\n");

                writer.close();

        }

        public String write() throws IOException {

                File file = new File(getPath());

                ArrayList<String> lines = new ArrayList<String>();

                // Si no existe a?n el rmf se crea, se a?ade las cabeceras y se vuelca el

                // contenido de los bloques

                if (!file.exists())

                        create(file);

                // A?adir bloques al fichero.

                BufferedReader inGrf = new BufferedReader(new InputStreamReader(new FileInputStream(file)));

                String str = inGrf.readLine();

                while (str != null) {

                        lines.add(str);

                        str = inGrf.readLine();

                }

                // Obtenemos el primer tag de un bloque. Para cada uno recorremos todo el

                // rmf actual buscando ese tag. Si existe se a?ade el nuevo bloque y se

                // borra el viejo. Si no existe lo a?ade al final antes de la etiqueta de

                // cierre </RasterMetaFile>

                for (int i = 0; i < clients.size(); i++) {

                        IRmfBlock block = ((IRmfBlock) clients.get(i));

                        String tag = block.getMainTag();

                        for (int nLine = 0; nLine < lines.size(); nLine++) {

                                String line = (String) lines.get(nLine);

                                if (line != null && (line.compareTo("</" + getMainTag() + ">") == 0 || line.startsWith("<" + tag))) {

                                        String xmlBlock = block.write();

                                        if(xmlBlock != null) { //Sustituci?n de bloque

                                                if (nLine != 0)

                                                        lines.add(nLine, xmlBlock);

                                                if (line.startsWith("<" + tag)) {

                                                        while (((String) lines.get(nLine + 1)).compareTo("</" + tag + ">") != 0)

                                                                lines.remove(nLine + 1);

                                                        lines.remove(nLine + 1);

                                                }

                                        } else { //Eliminado de bloque

                                                if (line.startsWith("<" + tag)) {

                                                        while (((String) lines.get(nLine)).compareTo("</" + tag + ">") != 0)

                                                                lines.remove(nLine);

                                                        lines.remove(nLine);

                                                }

                                        }

                                        break;

                                }

                        }

                }

                inGrf.close();

                // Escribir fichero de salida.

                file.delete();

                file.createNewFile();

                FileWriter writer = new FileWriter(file);

                for (int i = 0; i < lines.size(); i++) {

                        if (((String) lines.get(i)).length() == 0)

                                continue;

                        if (!((String) lines.get(i)).endsWith("\n"))

                                writer.write((String) lines.get(i) + "\n");

                        else

                                writer.write((String) lines.get(i));

                }

                writer.close();

                return null;

        }

        /**

         * Obtiene un bloque XML que representa a las propiedades del objeto a

         * serializar. Antes de la operaci?n hace una copia de seguridad del fichero

         * RMF.

         * @param rmfBackup Especifica si debe hacer la copia de seguridad.

         * @return Texto XML que representa el objeto.

         */

        public String write(boolean rmfBackup) throws IOException, FileNotFoundException {

                if (rmfBackup && new File(getPath()).exists())

                        RasterLocator.getManager().getFileUtils().copyFile(getPath(), getPath() + "~");

                return write();

        }

        /**

         * M?todo que checkea si el fichero pasado es valido para ser rmf. Si existe

         * el fichero ser? valido si se puede escribir en el y no est? vacio y no es

         * un directorio y contiene los caracteres de cabecera de comienzo de XML (<?xml

         * .... >). En caso de que el fichero no exista tambi?n es valido ya que se

         * crear? de cero.

         * @return true si es un rmf valido y false si no lo es

         */

        public boolean checkRmf() {

                File f = new File(getPath());

                if (f.exists()) {

                        if (!f.canWrite() || f.length() == 0 || f.isDirectory())

                                return false;

                        BufferedReader inGrf;

                        try {

                                inGrf = new BufferedReader(new InputStreamReader(new FileInputStream(f)));

                                String str = inGrf.readLine();

                                if (!str.startsWith("<?xml")) {

                                        inGrf.close();

                                        return false;

                                }

                                while (str != null) {

                                        if (str.startsWith("<" + getMainTag())) {

                                                inGrf.close();

                                                if(str.contains("version=\"")) {

                                                        String versionNbr = str.substring(str.indexOf("version=\"") + 9);

                                                        versionNbr = versionNbr.substring(0, versionNbr.indexOf("\""));

                                                        if(!versionNbr.equals(version)) {

                                                                return false;

                                                        }

                                                } 

                                                return true;

                                        }

                                        str = inGrf.readLine();

                                }

                                inGrf.close();

                        } catch (FileNotFoundException e) {

                                return false;

                        } catch (IOException e) {

                                return false;

                        }

                }

                return true;

        }

        public String getMainTag() {

                return MAIN_TAG;

        }

        public Object getResult() {

                return null;

        }

}