GeoTIFFIO.cpp

Go to the documentation of this file.

//
// Created by ndettmer on 07.02.19.
//

#include <iostream>

#include "lvr2/io/GeoTIFFIO.hpp"
#include "lvr2/io/Timestamp.hpp"

namespace lvr2
{
    

GeoTIFFIO::GeoTIFFIO(std::string filename, int cols, int rows, int bands) : m_cols(cols), m_rows(rows), m_bands(bands)
{
    GDALAllRegister();
    m_gtif_driver = GetGDALDriverManager()->GetDriverByName("GTiff");
    m_gtif_dataset = m_gtif_driver->Create(filename.c_str(), m_cols, m_rows, m_bands, GDT_UInt16, NULL);
}

GeoTIFFIO::GeoTIFFIO(std::string filename)
{
    GDALAllRegister();
    m_gtif_dataset = (GDALDataset *) GDALOpen(filename.c_str(), GA_ReadOnly);
}

int GeoTIFFIO::writeBand(cv::Mat *mat, int band)
{
    if (!m_gtif_dataset)
    {
        std::cout << timestamp << "GeoTIFF dataset not initialized!" << std::endl;
        return -1;
    }

    uint16_t *rowBuff = (uint16_t *) CPLMalloc(sizeof(uint16_t) * m_cols);
    for (int row = 0; row < m_rows; row++)
    {
        for (int col = 0; col < m_cols; col++)
        {
            rowBuff[col] = mat->at<uint16_t>(row, col);
        }
        if (m_gtif_dataset->GetRasterBand(band)->RasterIO(
                GF_Write, 0, row, m_cols, 1, rowBuff, m_cols, 1, GDT_UInt16, 0, 0) != CPLE_None)
        {
            std::cout << timestamp << "An error occurred in GDAL while writing band "
                << band << " in row " << row << "." << std::endl;
            return -1;
        }
    }
    return 0;
}

int GeoTIFFIO::getRasterWidth()
{
    if(m_gtif_dataset)
    {
        return m_gtif_dataset->GetRasterXSize();
    }
    else
    {
        return 0;
    }
    
}

int GeoTIFFIO::getRasterHeight()
{
    if(m_gtif_dataset)
    {
        return m_gtif_dataset->GetRasterYSize();
    }
    else
    {
        return 0;
    }
    
}

int GeoTIFFIO::getNumBands()
{
    if(m_gtif_dataset)
    {
        return m_gtif_dataset->GetRasterCount();
    }
    return 0;
}

cv::Mat *GeoTIFFIO::readBand(int index)
{
    GDALRasterBand *band = m_gtif_dataset->GetRasterBand(index);
    if(band)
    {
        int nXSize = band->GetXSize();
        int nYSize = band->GetYSize();
        uint16_t *buf = (uint16_t *) CPLMalloc(sizeof(uint16_t) * nXSize * nYSize);

        CPLErr error = band->RasterIO(GF_Read, 0, 0, nXSize, nYSize, buf, nXSize, nYSize, GDT_UInt16, 0, 0);

        cv::Mat *mat = new cv::Mat(nXSize, nYSize, CV_16UC1, buf);
        
        return mat;
    }
    else
    {
        std::cout << timestamp << "Error getting raster band" << std::endl;
        return new cv::Mat;
    }
    
}

GeoTIFFIO::~GeoTIFFIO()
{
    GDALClose(m_gtif_dataset);
    GDALDestroyDriverManager();
}

} // namespace lvr2