org.esa.beam.occci.S2EoProduct.java Source code

Introduction

Here is the source code for org.esa.beam.occci.S2EoProduct.java

Source

/*
 * Copyright (C) 2015 Brockmann Consult GmbH (info@brockmann-consult.de)
 *
 * 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 3 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, see http://www.gnu.org/licenses/
 */

package org.esa.beam.occci;

import com.google.common.geometry.S2CellId;
import com.google.common.geometry.S2CellUnion;
import com.google.common.geometry.S2LatLng;
import com.google.common.geometry.S2Loop;
import com.google.common.geometry.S2Point;
import com.google.common.geometry.S2Polygon;
import com.google.common.geometry.S2RegionCoverer;

import java.io.DataOutputStream;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.text.ParseException;
import java.util.ArrayList;

/**
 * Created by marcoz on 17.08.15.
 */
public class S2EoProduct extends AbstractEoProduct {

    private static final S2WKTReader wktReader = new S2WKTReader();
    private static final S2Polygon THE_PRODUCT;
    //    private static final S2CellUnion THE_PRODUCT_UNION;

    private static Field S2X;
    private static Field S2Y;
    private static Field S2Z;

    static {
        THE_PRODUCT = createPolygon(EoProduct.OVERLAP_WKT);
        //        S2RegionCoverer coverer = new S2RegionCoverer();
        //        coverer.setMinLevel(0);
        //        coverer.setMaxLevel(3);
        //        coverer.setMaxCells(500);
        //        THE_PRODUCT_UNION = coverer.getCovering(THE_PRODUCT);
        try {
            S2X = S2Point.class.getDeclaredField("x");
            S2Y = S2Point.class.getDeclaredField("y");
            S2Z = S2Point.class.getDeclaredField("z");
            S2X.setAccessible(true);
            S2Y.setAccessible(true);
            S2Z.setAccessible(true);
        } catch (NoSuchFieldException e) {
            e.printStackTrace();
        }
    }

    private final String wkt;
    private S2Polygon polygon;
    S2CellUnion cellUnion;

    public S2EoProduct(String name, long startTime, long endTime, String wkt) {
        super(name, startTime, endTime);
        this.wkt = wkt;
    }

    @Override
    public boolean contains(double lon, double lat) {
        S2Polygon poly = getPolygon();
        if (poly != null) {
            S2Point s2Point = S2LatLng.fromDegrees(lat, lon).toPoint();
            return poly.contains(s2Point);
        }
        return false;
    }

    @Override
    public boolean overlaps() {
        S2Polygon poly = getPolygon();
        if (poly != null) {
            return poly.intersects(THE_PRODUCT);
        }
        return false;
    }

    @Override
    public void reset() {
        polygon = null;
        cellUnion = null;
    }

    @Override
    public void createGeo() {
        getPolygon();
        S2RegionCoverer coverer = new S2RegionCoverer();
        coverer.setMinLevel(0);
        coverer.setMaxLevel(3);
        coverer.setMaxCells(500);
        cellUnion = coverer.getCovering(polygon);
    }

    private S2Polygon getPolygon() {
        if (polygon == null) {
            polygon = createPolygon(wkt);
        }
        return polygon;
    }

    public static S2Polygon createPolygon(String wkt) {
        return (S2Polygon) wktReader.read(wkt);
    }

    public static EoProduct parse(String line) throws ParseException {
        String[] splits = line.split("\t");
        String name = splits[0];
        long startTime = AbstractEoProduct.DATE_FORMAT.parse(DateUtils.getNoFractionString(splits[1])).getTime();
        long endTime = AbstractEoProduct.DATE_FORMAT.parse(DateUtils.getNoFractionString(splits[2])).getTime();
        String wkt = splits[3];
        return new S2EoProduct(name, startTime, endTime, wkt);
    }

    public String getLoopAsString() {
        return singleLoopToString(polygon);
    }

    public String getCellUnionAsString() {
        return cellUnionToString(cellUnion);
    }

    public String getIndexString() {
        String points = singleLoopToString(polygon);
        String cells = cellUnionToString(cellUnion);
        return getName() + "\t" + getStartTime() + "\t" + getEndTime() + "\t" + points + "\t" + cells + "\n";

    }

    private static String cellUnionToString(S2CellUnion cellUnion) {
        ArrayList<S2CellId> s2CellIds = cellUnion.cellIds();
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < s2CellIds.size(); i++) {
            S2CellId s2CellId = s2CellIds.get(i);
            sb.append(s2CellId.id());
            if (i < s2CellIds.size() - 1) {
                sb.append(";");
            }
        }
        return sb.toString();
    }

    private static String singleLoopToString(S2Polygon s2polygon) {
        S2Loop loop = s2polygon.loop(0);
        int numVertices = loop.numVertices();
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < numVertices; i++) {
            sb.append(loop.vertex(i).toString());
            if (i < numVertices - 1) {
                sb.append(";");
            }
        }
        return sb.toString();
    }

    public void writePolygone(DataOutputStream dos) throws IOException {
        S2Loop loop = polygon.loop(0);
        int numVertices = loop.numVertices();
        dos.writeInt(numVertices);

        for (int i = 0; i < numVertices; i++) {
            S2Point vertex = loop.vertex(i);
            try {
                double x = (double) S2X.get(vertex);
                double y = (double) S2Y.get(vertex);
                double z = (double) S2Z.get(vertex);
                dos.writeDouble(x);
                dos.writeDouble(y);
                dos.writeDouble(z);
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
    }

    public void writeCellUnion(DataOutputStream dos) throws IOException {
        ArrayList<S2CellId> s2CellIds = cellUnion.cellIds();
        dos.writeInt(s2CellIds.size());

        for (S2CellId s2CellId : s2CellIds) {
            dos.writeLong(s2CellId.id());
        }
    }
}