基于geotools实现geotif切瓦片

1. 前言

本篇文章主要提供了geotools实现瓦片的切割。程序默认的瓦片大小是256✖256。另外,提供了保存png格式和webp格式的瓦片文件。

2. pom依赖

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>cn.surpass.geotools</groupId>
    <artifactId>geotools-tile</artifactId>
    <version>1.0-SNAPSHOT</version>

    <name>geotools-tile</name>
    <url>http://www.example.com</url>

    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <maven.compiler.source>1.8</maven.compiler.source>
        <maven.compiler.target>1.8</maven.compiler.target>
        <geotools.version>19.2</geotools.version>
    </properties>

    <dependencies>

        <!--日志依赖包-->
        <dependency>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-log4j12</artifactId>
            <version>1.6.1</version>
        </dependency>
        <dependency>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-log4j12</artifactId>
            <version>1.6.1</version>
        </dependency>

        <!--webp依赖包-->
        <dependency>
            <groupId>org.sejda.imageio</groupId>
            <artifactId>webp-imageio</artifactId>
            <version>0.1.6</version>
        </dependency>

        <!--geotools依赖包-->
        <dependency>
            <groupId>org.geotools</groupId>
            <artifactId>gt-geotiff</artifactId>
            <version>${geotools.version}</version>
        </dependency>
        <dependency>
            <groupId>org.geotools</groupId>
            <artifactId>gt-image</artifactId>
            <version>${geotools.version}</version>
        </dependency>
        <dependency>
            <groupId>org.geotools</groupId>
            <artifactId>gt-imagemosaic</artifactId>
            <version>${geotools.version}</version>
        </dependency>
        <dependency>
            <groupId>org.geotools</groupId>
            <artifactId>gt-render</artifactId>
            <version>${geotools.version}</version>
        </dependency>
        <dependency>
            <groupId>org.geotools</groupId>
            <artifactId>gt-imageio-ext-gdal</artifactId>
            <version>${geotools.version}</version>
        </dependency>

        <!--单元测试包-->
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.11</version>
            <scope>test</scope>
        </dependency>
    </dependencies>

    <repositories>
        <repository>
            <id>osgeo</id>
            <name>OSGeo Release Repository</name>
            <url>https://repo.osgeo.org/repository/release/</url>
            <snapshots>
                <enabled>false</enabled>
            </snapshots>
            <releases>
                <enabled>false</enabled>
            </releases>
        </repository>
        <repository>
            <id>osgeo-snapshot</id>
            <name>OSGeo Snapshot Repository</name>
            <url>https://repo.osgeo.org/repository/snapshot/</url>
            <snapshots>
                <enabled>false</enabled>
            </snapshots>
            <releases>
                <enabled>false</enabled>
            </releases>
        </repository>
    </repositories>

    <build>
        <pluginManagement>
            <plugins>
                <plugin>
                    <artifactId>maven-clean-plugin</artifactId>
                    <version>3.1.0</version>
                </plugin>
                <plugin>
                    <artifactId>maven-resources-plugin</artifactId>
                    <version>3.0.2</version>
                </plugin>
                <plugin>
                    <artifactId>maven-compiler-plugin</artifactId>
                    <version>3.8.0</version>
                </plugin>
                <plugin>
                    <artifactId>maven-surefire-plugin</artifactId>
                    <version>2.22.1</version>
                </plugin>
                <plugin>
                    <artifactId>maven-jar-plugin</artifactId>
                    <version>3.0.2</version>
                </plugin>
                <plugin>
                    <artifactId>maven-install-plugin</artifactId>
                    <version>2.5.2</version>
                </plugin>
                <plugin>
                    <artifactId>maven-deploy-plugin</artifactId>
                    <version>2.8.2</version>
                </plugin>
                <plugin>
                    <artifactId>maven-site-plugin</artifactId>
                    <version>3.7.1</version>
                </plugin>
                <plugin>
                    <artifactId>maven-project-info-reports-plugin</artifactId>
                    <version>3.0.0</version>
                </plugin>
            </plugins>
        </pluginManagement>
    </build>
</project>
  • 这里需要有几个地方说明一下,引用以下的以来实现webp的输出:
      <dependency>
        <groupId>org.sejda.imageio</groupId>
        <artifactId>webp-imageio</artifactId>
        <version>0.1.6</version>
      </dependency>
    
  • maven中心库没有geotools的依赖,所以要加以下的仓库配置。
    <repositories>
      <repository>
        <id>osgeo</id>
        <name>OSGeo Release Repository</name>
        <url>https://repo.osgeo.org/repository/release/</url>
        <snapshots>
          <enabled>false</enabled>
        </snapshots>
        <releases>
           <enabled>false</enabled>
        </releases>
      </repository>
      <repository>
        <id>osgeo-snapshot</id>
        <name>OSGeo Snapshot Repository</name>
        <url>https://repo.osgeo.org/repository/snapshot/</url>
        <snapshots>
          <enabled>false</enabled>
        </snapshots>
        <releases>
          <enabled>false</enabled>
        </releases>
      </repository>
    </repositories>
    

3.代码

package cn.surpass.geotools;

import org.geotools.coverage.grid.GridCoverage2D;
import org.geotools.coverage.grid.io.AbstractGridCoverage2DReader;
import org.geotools.coverage.grid.io.AbstractGridFormat;
import org.geotools.coverage.grid.io.GridFormatFinder;
import org.geotools.coverage.processing.CoverageProcessor;
import org.geotools.factory.Hints;
import org.geotools.gce.geotiff.GeoTiffFormat;
import org.geotools.geometry.Envelope2D;
import org.geotools.geometry.jts.ReferencedEnvelope;
import org.opengis.geometry.Envelope;
import org.opengis.parameter.ParameterValueGroup;
import org.opengis.referencing.crs.CoordinateReferenceSystem;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.imageio.ImageIO;
import javax.media.jai.PlanarImage;
import java.awt.image.BufferedImage;
import java.awt.image.RenderedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * @author surpassliang
 */
public class ImageTiles {

    private final static Logger LOGGER = LoggerFactory.getLogger(ImageTiles.class);

    private static final Integer TILE_SIZE = 256;

    public static void handlerImage2Tiles(String geoTifPath, String saveDir, ImageFormat imageFormat) {
        LOGGER.info(String.format("传入参数==>geoTifPath:%s;saveDir:%s;imageFormat:%s", geoTifPath, saveDir, imageFormat.name()));
        //获取GridCoverage2D 对象
        GridCoverage2D gridCoverage = GetGridCoverage2D(geoTifPath);

        //创建目标文件夹
        File saveFolder = new File(saveDir);
        if (saveFolder.exists()) {
            saveFolder.delete();
        }
        saveFolder.mkdirs();
        //影像切片
        handlerTile(gridCoverage, saveFolder, imageFormat);
        LOGGER.info("切片完成");
    }

    private static void handlerTile(GridCoverage2D gridCoverage, File saveFolder, ImageFormat imageFormat) {
        //获取gridCoverage边界
        Envelope2D coverageEnvelope = gridCoverage.getEnvelope2D();
        double minX = coverageEnvelope.getMinX();
        double maxX = coverageEnvelope.getMaxX();
        double minY = coverageEnvelope.getMinY();
        double maxY = coverageEnvelope.getMaxY();
        LOGGER.info(String.format("图像的四至范围:(%f,%f)~(%f,%f)",minX,minY,maxX,maxY));

        //获取像素大小
        RenderedImage sourceImage = gridCoverage.getRenderableImage(0, 1).createDefaultRendering();
        int rasterXSize = sourceImage.getWidth();
        int rasterYSize = sourceImage.getHeight();
        LOGGER.info("图像像素大小:X = {};Y = {}",rasterXSize,rasterYSize);

        //获取geoTif坐标系统
        CoordinateReferenceSystem targetCRS = gridCoverage.getCoordinateReferenceSystem();
        LOGGER.info("图像的坐标系统为:{}",targetCRS.toString());

        //计算瓦片数量
        int htc = (int) Math.ceil(rasterXSize * 1.0 / TILE_SIZE);
        int vtc = (int) Math.ceil(rasterYSize * 1.0 / TILE_SIZE);

        //地理瓦片长宽
        double geoTileWidth = (maxX - minX) / (double) htc;
        double geoTileHeight = (maxY - minY) / (double) vtc;

        for (int hIndex = 0; hIndex < htc; hIndex++) {
            for (int vIndex = 0; vIndex < vtc; vIndex++) {
                LOGGER.info("Processing tile at indices i: {} and j: {}", vIndex, hIndex);

                //创建瓦片envelope
                double startX = (hIndex * geoTileWidth) + minX;
                double startY = (vIndex * geoTileHeight) + minY;
                Envelope envelope = new ReferencedEnvelope(startX, startX + geoTileWidth, startY, startY + geoTileHeight, targetCRS);

                //根据范围切割瓦片
                GridCoverage2D finalCoverage = cropCoverage(gridCoverage, envelope);

                //保存文件
                File tileFile = new File(saveFolder,String.format("%d_%d.%s",vIndex,hIndex,imageFormat.getSuffixName()));

                //保存文件
                BufferedImage tempImage = PlanarImage.wrapRenderedImage(finalCoverage.getRenderedImage()).getAsBufferedImage();
                saveTile(tempImage, tileFile, imageFormat);
            }
        }
    }

    private static void saveTile(BufferedImage image, File tileFile, ImageFormat imageFormat){
        try (FileOutputStream fos = new FileOutputStream(tileFile)) {
            ImageIO.write(image, imageFormat.getFormatName(), fos);
            fos.flush();
        } catch (IOException e) {
            throw new RuntimeException(e.getMessage(),e);
        }
    }

    private static GridCoverage2D cropCoverage(GridCoverage2D gridCoverage, Envelope envelope) {
        CoverageProcessor processor = CoverageProcessor.getInstance();
        ParameterValueGroup coverageCrop = processor.getOperation("CoverageCrop").getParameters();
        coverageCrop.parameter("Source").setValue(gridCoverage);
        coverageCrop.parameter("Envelope").setValue(envelope);
        return (GridCoverage2D) processor.doOperation(coverageCrop);
    }

    private static GridCoverage2D GetGridCoverage2D(String geoTifPath) {
        File geoTifFile = new File(geoTifPath);
        if (!geoTifFile.exists()) {
            throw new RuntimeException("geoTif文件不存在");
        }
        AbstractGridFormat format = GridFormatFinder.findFormat(geoTifFile);

        if (!(format instanceof GeoTiffFormat)) {
            throw new RuntimeException(String.format("传入文件\"%s\"不是geoTif格式文件", geoTifPath));
        }
        Hints hints = new Hints(Hints.FORCE_LONGITUDE_FIRST_AXIS_ORDER, Boolean.TRUE);

        AbstractGridCoverage2DReader geoTifReader = format.getReader(geoTifFile, hints);
        try {
            return geoTifReader.read(null);
        } catch (Exception e) {
            throw new RuntimeException(e.getMessage(), e);
        }
    }

    public enum ImageFormat {
        FORMAT_WEBP("webp", "webp"),
        FORMAT_PNG("png", "png");
        private final String formatName;
        private final String suffixName;
        ImageFormat(String formatName, String suffixName) {
            this.formatName = formatName;
            this.suffixName = suffixName;
        }
        public String getFormatName() {
            return formatName;
        }
        public String getSuffixName() {
            return suffixName;
        }
    }
}

这里需要说明的一下,private static GridCoverage2D cropCoverage(GridCoverage2D gridCoverage, Envelope envelope)方法也可以有如下的实现:

private static GridCoverage2D cropCoverage(GridCoverage2D gridCoverage, Envelope envelope) {
  Operations ops = new Operations(null);
  return (GridCoverage2D) ops.crop(gridCoverage, envelope);
}

4. 方法调用

ImageTiles.handlerImage2Tiles("C:\\Users\\lianghj\\Desktop\\TB.tif","C:\\Users\\Admin\\Desktop\\tile", ImageTiles.ImageFormat.FORMAT_WEBP);
ImageTiles.handlerImage2Tiles("C:\\Users\\lianghj\\Desktop\\TB.tif","C:\\Users\\Admin\\Desktop\\tile", ImageTiles.ImageFormat.FORMAT_PNG);

5. 效果展示

这里主要展示两种不同的格式数据量的大小,这里差距还是挺大的。所以,现在还是很流行的图片格式。
在这里插入图片描述

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

surpassLiang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值