java版GDAL学习之路——创建数据集

创建矢量数据集

完整代码

/**
     * 创建一个空的矢量图层
     * @param driverName:驱动名称
     * @param path:图层保存路径，要和驱动匹配
     * @param layerName：图层名称
     * @param spatialReference：图层空间参考
     * @return 返回创建好的图层
     */
    public static Layer createShp(String driverName, String path, String layerName, SpatialReference spatialReference) {
        Layer result = null;

        Driver driver = ogr.GetDriverByName(driverName);
        if (driver == null) {
            System.out.println(driverName + "不可用");
            return null;
        }
        DataSource dataSource = null;
        //这里需要判断一下path是否已经存在，存在的话先删除再创建(如果是已存在的gdb，则直接打开)
        File file = new File(path);
        if (file.exists()) {
            if (file.isFile()) {
                //删掉同名（不带后缀）文件
                String parentDir = getParentDir(file.getPath());
                File dir = new File(parentDir);
                deleteFilesLikeName(dir,layerName);
                //重新创建
                dataSource = driver.CreateDataSource(path, null);
            } else if (file.isDirectory()) {
                dataSource = driver.Open(path, 1);
                //GDB中存在同名图层则删除
                for (int i = 0; i < dataSource.GetLayerCount(); i++) {
                    Layer layer = dataSource.GetLayer(i);
                    if (layerName.equals(layer.GetName())) {
                        dataSource.DeleteLayer(i);
                        dataSource.FlushCache();
                    }
                }
            }
        } else {
            dataSource = driver.CreateDataSource(path, null);
        }
        if (dataSource == null) {
            System.out.println("数据源创建/打开失败");
            return null;
        }
        result = dataSource.CreateLayer(layerName, spatialReference, ogr.wkbPolygon, null);
        if (result == null) {
            System.out.println("【"+layerName+".shp】"+ "创建失败");
            return null;
        }
        System.out.println("【"+layerName+".shp】" + "创建成功");
        return result;
    }

	/**
     * 获取路径的上一级目录
     * @param path：文件路径
     * @return
     */
    public static String getParentDir(String path){
        String parentDir = "";
        int end = path.lastIndexOf("\\");
        parentDir = path.substring(0,end);
        return parentDir;
    }

	/**
     * 删除文件名包含指定字符串的文件
     * @param file
     * @param likeName
     */
    public static void deleteFilesLikeName(File file,String likeName){
        if(file.exists()){
            if(file.isFile()){
                String temp = file.getName().substring(0,file.getName().lastIndexOf('.'));
                if(temp.indexOf(likeName)!=-1){
                    file.delete();
                }
            }
            else {
                File[] files = file.listFiles();
                for (int i = 0; i < files.length; i++) {
                    deleteFilesLikeName(files[i],likeName);
                }
            }
        }
    }

创建栅格数据集

完整代码(待更新)