Application: gvSIG desktop

/* DielmoOpenLiDAR

 *

 * Copyright (C) 2008 DIELMO 3D S.L. (DIELMO) and 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.

 *

 * For more information, contact:

 *

 * DIELMO 3D S.L.

 * Plaza Vicente Andr?s Estell?s 1 Bajo E

 * 46950 Xirivella, Valencia

 * SPAIN

 *   

 * +34 963137212

 * dielmo@dielmo.com

 * www.dielmo.com

 * 

 * or

 * 

 * Generalitat Valenciana

 * Conselleria d'Infraestructures i Transport

 * Av. Blasco Ib??ez, 50

 * 46010 VALENCIA

 * SPAIN

 *

 * +34 963862235

 * gvsig@gva.es

 * www.gvsig.gva.es

 */

/*

 * AUTHORS (In addition to DIELMO and CIT):

 *  

 */

package com.dielmo.gvsig.lidar.test;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileWriter;

import java.io.IOException;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import org.cresques.cts.IProjection;

import com.dielmo.gvsig.lidar.drivers.LiDARDriver;

import com.hardcode.gdbms.engine.values.DoubleValue;

import com.iver.cit.gvsig.fmap.crs.CRSFactory;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.layers.LayerFactory;

import com.iver.cit.gvsig.fmap.layers.ReadableVectorial;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

/**

 * Crear un XYZ a partir de un fichero Lidar

 * 

 * @author Oscar Garcia Gonzalez

 */

public class LidarToXYZ{

        public static void main(String[] args){

                JFileChooser fileChooser = new JFileChooser();

                //fileChooser.addChoosableFileFilter(new GenericFileFilter("Archivos lidar", "*.las;*.bin"));

                File file;

                int result = fileChooser.showOpenDialog(null);

        if (result == JFileChooser.APPROVE_OPTION && (file = fileChooser.getSelectedFile()) != null) {

                IProjection projection = CRSFactory.getCRS("EPSG:23030");

                //IProjection projection = Project.getDefaultProjection();

                createXYZByLidarFile(file, projection);

        } else {

                // si se eligio fichero correctamente.

                if(result == JFileChooser.APPROVE_OPTION )

                        JOptionPane.showMessageDialog(null, "Selecciona un fichero Lidar en formato LAS o BIN");

        }

        }

        /**

         * escribe en formato XYZ un archivo lidar pasado por entrada.

         * @param file fichero lidar del que extraer la informacion.

         */

        public static void createXYZByLidarFile(File file, IProjection projection){

                DoubleValue valueX, valueY, valueZ;

                int k, posTableX = 0, posTableY=1, posTableZ=2;

                if(file.getAbsolutePath().toUpperCase().endsWith("LAS") || file.getAbsolutePath().toUpperCase().endsWith("BIN")){

                try {

                        LiDARDriver lidarDriver = new LiDARDriver();

                        lidarDriver.open(file);

                        lidarDriver.initialize();

                        FLayer lidarLayer;

                        lidarLayer = LayerFactory.createLayer("lidarLayer", lidarDriver, projection);

                        ReadableVectorial rvLidar = ((FLyrVect)lidarLayer).getSource();

                        SelectableDataSource dsLidar = rvLidar.getRecordset();

                        rvLidar.start();

                        String[] fd = dsLidar.getFieldNames();

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

                                if(fd[k].equals("X"))

                                        posTableX = k;

                                else if(fd[k].equals("Y"))

                                        posTableY = k;

                                else if(fd[k].equals("Z"))

                                        posTableZ = k;

                        }

                        FileWriter out = new FileWriter(file+".xyz");

                                int rvlidarCount = rvLidar.getShapeCount();

                                for(k=0;k<rvlidarCount;k++){

                                        valueX = (DoubleValue)dsLidar.getFieldValue(k, posTableX);

                                        valueY = (DoubleValue)dsLidar.getFieldValue(k, posTableY);

                                        valueZ = (DoubleValue)dsLidar.getFieldValue(k, posTableZ);

                                        out.write(valueX.getValue() + " " + valueY.getValue() + " " + valueZ.getValue() + "\n" );

                                }

                                out.close();

                                rvLidar.stop();

                        } catch (DriverIOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        } catch (FileNotFoundException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        } catch (IOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

        }

        }

}