AccFlowThread.java

package heyingzhe;

import java.io.IOException;
import java.nio.file.Paths;
import java.sql.SQLException;

public class AccFlowThread extends Thread {
    private int[][] dem;
    private int[][] filledDem;
    private int[][] flowDirectionArray;
    private double[][] idwInterpolationResult;
    private String outPutDirectory;
    private String outputAccFlowPath;
    private String outputWaterAreaPath;
    private String dbUserName;
    private String dbPassword;

    public AccFlowThread(int[][] dem, int[][] filledDem, int[][] flowDirectionArray, double[][] idwInterpolationResult, String outPutDirectory, String outputAccFlowPath, String outputWaterAreaPath, String dbUserName, String dbPassword) {
        this.dem = dem;
        this.filledDem = filledDem;
        this.flowDirectionArray = flowDirectionArray;
        this.idwInterpolationResult = idwInterpolationResult;
        this.outPutDirectory = outPutDirectory;
        this.outputAccFlowPath = outputAccFlowPath;
        this.outputWaterAreaPath = outputWaterAreaPath;
        this.dbUserName = dbUserName;
        this.dbPassword = dbPassword;
    }

    public void run() {
        //计算累积流数组
        System.out.println("正在进行累积流计算……");
        CalculateAccumulatedFlow accCalculation = new CalculateAccumulatedFlow(dem); //计算累计流
        double[][] flowAccumulation = accCalculation.calculateFlow(flowDirectionArray, idwInterpolationResult, filledDem);
        System.out.println("累积流计算完成");

        //将累积流数组输出为图片
        ImageVisualizer imageVision = new ImageVisualizer();
        imageVision.convertFlowAccumulationToImage(flowAccumulation, 10);
        try {
            imageVision.saveImage(Paths.get(outPutDirectory, outputAccFlowPath).toString(), "png");
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        //根据海拔高度，输出水域图片
        imageVision.createWaterAreaImage(filledDem, 250);
        try {
            imageVision.saveImage(Paths.get(outPutDirectory, outputWaterAreaPath).toString(), "png");
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        SaveResult saveProcessing = new SaveResult();
        // 配置数据库
        saveProcessing.setMySQLProperty(dbUserName, dbPassword);
        try {
            saveProcessing.createDatabase("result");
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
        System.out.println("累积流计算：数据库连接成功！");

        System.out.println("正在上传累积流数据");
        try {
            saveProcessing.createTable("result", "accFlow"); //创建累积流表格
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        try {
            saveProcessing.saveArrayToAscFile(flowAccumulation, outPutDirectory, "accFlow.asc"); //保存为二进制文件
            //saveProcessing.saveArrayToMySQL(flowAccumulation, "accFlow"); //上传至MySQL数据库
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        /*
        catch (SQLException e) {
            throw new RuntimeException(e);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }*/
        System.out.println("累积流数据已经添加到数据库里面");
    }
}