IH5Dataset.h

//-*- C++ -*-
//-*- coding: utf-8 -*-
//
// Author: Piyush Agram
// Copyright 2019

#pragma once

#include "forward.h"

#include <gdal_pam.h>
#include <gdal_priv.h>
#include <gdal_rat.h>
#include "IH5.h"


namespace isce3 {
namespace io {

/************************************************************************/
/*                            IH5Dataset                                */
/************************************************************************/

class IH5Dataset final: public GDALDataset
{
    CPL_DISALLOW_COPY_ASSIGN(IH5Dataset)

    friend class IH5RasterBand;

    //Standard stuff expected from all formats
    int bGeoTransformSet; 
    double adfGeoTransform[6];

    char *pszProjection;
    CPLString pszGCPProjection;
    GDAL_GCP *pasGCPList;
    int  nGCPCount; 
    OGRSpatialReference oSRS;

    //IDataSet from Francois's IH5
    isce3::io::IDataSet* _dataset;
    H5::DataType nativeType;
    H5::DataType actualType;
    int ndims;
    int dimensions[3];
    int chunks[3];

    protected:
        CPLErr populateFromDataset();

    public:
        IH5Dataset(const hid_t &inputds, GDALAccess eAccess);

        virtual ~IH5Dataset();

#if GDAL_VERSION_MAJOR == 2
        virtual const char *GetProjectionRef() override;
        virtual CPLErr SetProjection( const char * ) override;
        virtual const char *GetGCPProjection() override;
        virtual CPLErr SetGCPs( int nGCPCount, const GDAL_GCP *pasGCPList,
                            const char *pszGCPProjection ) override;
#elif GDAL_VERSION_MAJOR == 3
        //https://github.com/OSGeo/gdal/blob/master/gdal/MIGRATION_GUIDE.TXT
        const char *_GetProjectionRef() override;
        const OGRSpatialReference* GetSpatialRef() const override {
            return GetSpatialRefFromOldGetProjectionRef();
        }

        CPLErr _SetProjection( const char * ) override;
        CPLErr SetSpatialRef(const OGRSpatialReference* poSRS) override {
            return OldSetProjectionFromSetSpatialRef(poSRS);
        }

        const char* _GetGCPProjection() override;
        const OGRSpatialReference* GetGCPSpatialRef() const override {
            return GetGCPSpatialRefFromOldGetGCPProjection();
        }

        CPLErr _SetGCPs(int nGCPCount, const GDAL_GCP *pasGCPList, const char *pszGCPProjection) override;
        using GDALDataset::SetGCPs;
        CPLErr SetGCPs( int nGCPCount, const GDAL_GCP *pasGCPList,
                    const OGRSpatialReference* poSRS ) override {
            return OldSetGCPsFromNew(nGCPCount, pasGCPList, poSRS);
        }
#endif

        virtual int GetGCPCount() override;
        virtual const GDAL_GCP *GetGCPs() override;

        void *GetInternalHandle (const char *) override;

        virtual CPLErr GetGeoTransform( double *padfTransform ) override;
        virtual CPLErr SetGeoTransform( double * ) override;
        static GDALDataset *Open(GDALOpenInfo *info);
        static int Identify(GDALOpenInfo *info);
};


/************************************************************************/
/*                            IH5RasterBand                             */
/************************************************************************/

class IH5RasterBand : public GDALPamRasterBand
{
    protected:
        friend class IH5Dataset;
        bool bNoDataSet;
        double dfNoData;

    public:
        IH5RasterBand(IH5Dataset *ds, int band, 
                      GDALDataType eTypeIn);

        virtual ~IH5RasterBand();

        virtual CPLErr IReadBlock( int, int, void * ) override;
        virtual CPLErr IWriteBlock( int, int, void * ) override;
        virtual double GetNoDataValue( int *pbSuccess = nullptr ) override;
        virtual CPLErr SetNoDataValue( double ) override;
};


void GDALRegister_IH5();

}} // namespace isce3::io