MeteoInfo/MeteoInfoLib/src/main/java/org/meteoinfo/data/StationTableData.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.meteoinfo.data;

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import javax.swing.JOptionPane;

import org.meteoinfo.common.util.GlobalUtil;
import org.meteoinfo.ndarray.DataType;
import org.meteoinfo.projection.KnownCoordinateSystems;
import org.meteoinfo.projection.info.ProjectionInfo;

/**
 *
 * @author yaqiang
 */
public class StationTableData extends TableData{
    // <editor-fold desc="Variables">    
    private ProjectionInfo projInfo;
    private int stIdx;
    private int lonIdx;
    private int latIdx;
    // </editor-fold>
    // <editor-fold desc="Constructor">
    /**
     * Constructor
     */
    public StationTableData(){
        this.projInfo = KnownCoordinateSystems.geographic.world.WGS1984;
        stIdx = 0;
        lonIdx = 1;
        latIdx = 2;
    }
    // </editor-fold>
    // <editor-fold desc="Get Set Methods">
    /**
     * Get projection info
     * @return Projection info
     */
    public ProjectionInfo getProjectionInfo(){
        return this.projInfo;
    }
    
    /**
     * Set projection info
     * @param value Projection info
     */
    public void setProjectionInfo(ProjectionInfo value){
        this.projInfo = value;
    }
    
    /**
     * Get station column index
     * @return Station column index
     */
    public int getStationIndex(){
        return this.stIdx;
    }
    
    /**
     * Set station column index
     * @param value Statin column index
     */
    public void setStationIndex(int value){
        this.stIdx = value;
    }
    
    /**
     * Get longitude column index
     * @return Longitude column index
     */
    public int getLonIndex(){
        return this.lonIdx;
    }
    
    /**
     * Set longitude column index
     * @param value Longitude column index
     */
    public void setLonIndex(int value){
        this.latIdx = value;
    }
    
    /**
     * Get latitude column index
     * @return Latitude column index
     */
    public int getLatIndex(){
        return this.latIdx;
    }
    
    /**
     * Set Latitude column index
     * @param value Latitude column index
     */
    public void setLatIndex(int value){
        this.lonIdx = value;
    }
    // </editor-fold>
    // <editor-fold desc="Methods">
    /**
     * Read data table from ASCII file
     *
     * @param fileName File name
     * @param lonIdx Longitude index
     * @param latIdx Latitude index
     * @throws java.io.FileNotFoundException
     */
    public void readASCIIFile(String fileName, int lonIdx, int latIdx) throws FileNotFoundException, IOException, Exception {
        this.readASCIIFile(fileName, 0, lonIdx, latIdx);
    }      
    
    /**
     * Read data table from ASCII file
     *
     * @param fileName File name
     * @param stIdx Station column index
     * @param lonIdx Longitude column index
     * @param latIdx Latitude column index
     * @throws java.io.FileNotFoundException
     */
    public void readASCIIFile(String fileName, int stIdx, int lonIdx, int latIdx) throws FileNotFoundException, IOException, Exception {
        this.lonIdx = lonIdx;
        this.latIdx = latIdx;
        //DataTable dTable = new DataTable();

        BufferedReader sr = new BufferedReader(new InputStreamReader(new FileInputStream(fileName), "utf-8"));
        String title = sr.readLine().trim();
        //Determine separator
        String separator = GlobalUtil.getDelimiter(title);
        String[] titleArray = GlobalUtil.split(title, separator);
        if (titleArray.length < 2) {
            JOptionPane.showMessageDialog(null, "File Format Error!");
            sr.close();
        } else {
            //Get fields
            List<Integer> dataIdxs = new ArrayList<>();
            String fieldName;
            for (int i = 0; i < titleArray.length; i++) {
                fieldName = titleArray[i];
                if (i == lonIdx || i == latIdx)
                    this.addColumn(fieldName, DataType.FLOAT);
                else
                    this.addColumn(fieldName, DataType.STRING);
                dataIdxs.add(i);
            }

            String[] dataArray;
            int rn = 0;
            String line = sr.readLine();
            while (line != null) {
                line = line.trim();
                if (line.isEmpty()) {
                    continue;
                }
                dataArray = GlobalUtil.split(line, separator);
                this.addRow();
                int cn = 0;
                for (int idx : dataIdxs) {
                    if (idx == lonIdx || idx == latIdx)
                        this.setValue(rn, cn, Float.parseFloat(dataArray[idx]));
                    else
                        this.setValue(rn, cn, dataArray[idx]);
                    cn++;
                }

                rn += 1;
                line = sr.readLine();
            }

            //dataTable = dTable;
            sr.close();
        }
    }      
    
    /**
     * Clone
     * @return Cloned StationTableData object
     */
    @Override
    public Object clone(){
        StationTableData std = new StationTableData();
        std = (StationTableData)super.clone();
        //std.dataTable = (DataTable)this.dataTable.clone();
        std.missingValue = this.missingValue;
        std.projInfo = this.projInfo;
        std.stIdx = this.stIdx;
        std.lonIdx = this.lonIdx;
        std.latIdx = this.latIdx;
        
        return std;
    }
    // </editor-fold>
}