FileGeoTIFF.h

/*
 
MIT License
 
Copyright (c) 2017 FMI Open Development / Markus Peura, first.last@fmi.fi
 
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
 
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
 
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
 
*/
/*
Part of Rack development has been done in the BALTRAD projects part-financed
by the European Union (European Regional Development Fund and European
Neighbourhood Partnership Instrument, Baltic Sea Region Programme 2007-2013)
*/
#ifndef DRAIN_FILE_GEO_TIFF
#define DRAIN_FILE_GEO_TIFF
 
#include "FileTIFF.h"
 
#ifndef USE_GEOTIFF_NO
 
#include <geotiff.h>
#include <geotiffio.h>
#include <geo_normalize.h>
 
#include "drain/util/EnumFlags.h"
#include "TreeXML-GDAL.h"
 
 
 
namespace drain
{
 
namespace image
{
 
 
 
class FileGeoTIFF : public FileTIFF {
public:
 
    // typedef std::map<short, std::list<std::pair<geokey_t, drain::Variable> > > epsg_map_t;
    // static epsg_map_t epsgConf;
 
    FileGeoTIFF() : FileTIFF(), gtif(nullptr){
        //gdalInfo.data.setType(GDAL::tag_t::ROOT);
        gdalMetadata(GDAL::tag_t::ROOT);
    }
 
    FileGeoTIFF(const std::string & path, const std::string & mode = "w") : FileTIFF(), gtif(nullptr){
        //gdalInfo.data.setType(GDAL::tag_t::ROOT);
        gdalMetadata(GDAL::tag_t::ROOT);
        open(path, mode);
    }
 
    virtual inline
    ~FileGeoTIFF(){
        //gt_close();
        close();
    }
 
    virtual
    void open(const std::string & path, const std::string & mode = "w");
 
 
    template <typename T>
    void setGeoTiffField(geokey_t tag, T value){
        if ((std::is_enum<T>::value) || (std::is_integral<T>::value)){
            GTIFKeySet(gtif, tag, TYPE_SHORT, 1, static_cast<short int>(value));
        }
        else if (std::is_floating_point<T>::value){
            GTIFKeySet(gtif, tag, TYPE_FLOAT, 1, value);
        }
        else {
            drain::Logger mout(__FILE__, __FUNCTION__);
            mout.warn("TAG: ", tag, ", value:", value, ", type=", typeid(T).name()); // , drain::Type::call<drain::nameGetter>(typeid(T)));
            mout.error("Not implemented");
        }
    }
 
 
    void writeMetadata();
 
 
    virtual
    void close();
 
    inline virtual
    bool isOpen() const {
        return FileTIFF::isOpen() && (gtif != nullptr);
    }
 
 
    // static
    // bool strictCompliance;
 
    typedef enum {UNDEFINED=0, TIFF=1, GEOTIFF=2, EPSG=6, STRICT=10} TiffCompliance;
 
    //typedef drain::EnumFlagger<drain::SingleFlagger<TiffCompliance> > tiffComplianceFlagger;
    typedef drain::EnumFlagger<drain::MultiFlagger<TiffCompliance> > complianceFlagger;
 
    static
    complianceFlagger compliancyFlagger;
 
    static
    std::string compliancy;
    //TiffCompliance compliance;
    // static   bool plainEPSG;
 
    TreeGDAL gdalMetadata;
    void setGdalScale(double scale=1.0, double offset=0.0);
 
 
    void setGdalNoData(const std::string & nodata);
 
 
    template <class T>
    void setGdal(const std::string & key, const T & value, int sample=-1, const std::string & role = ""){
 
        TreeGDAL & elem =  gdalMetadata[key](GDAL::ITEM);
        //elem->setType(GDAL::UNDEFINED);
        //elem->setType(GDAL::ITEM);
 
        //NodeGDAL gdalItem =  gdalMetadata[key](NodeGDAL::ITEM).data;
 
        NodeGDAL & item =  elem.data;
        item.name = key;
        item.setText(value);
        /*
        this->name   = name;
        this->ctext  = ctext.toStr();
        this->sample = sample;
        */
 
        if (sample >= 0){
            //item.sample = sample;
            item.set("sample", sample); // create Variable
            if (role.empty()){
                item.role = key;
                drain::StringTools::lowerCase(item.role);
            }
            else {
                item.role = role;
            }
        }
 
        //item.set("name", "gdalItem.name"); // KLUDGE
        //iItem.set("role", "gdalItem.role"); // KLUDGE
    }
 
 
    void setGeoMetaData(const drain::image::GeoFrame & frame);
 
 
    void setProjectionEPSG(short epsg);
 
 
    void setProjection(const std::string & proj);
 
    void setProjection(const drain::Proj6 & proj);
 
    void setProjectionLongLat();
 
 
 
    /*
    template <typename T>
    inline
    tagtype_t getTagType(){
        const typename tagtype_map_t::const_iterator it = tagtype_map.find(&typeid(T));
        if (it != tagtype_map.end()){
            return it->second;
        }
        else {
            drain::Logger mout(__FILE__, __FUNCTION__);
            mout.warn("unknown/unsupported GTIFF tag basetype: ", typeid(T).name());
            return TYPE_UNKNOWN;
        }
    }
    */
 
protected:
 
 
    GTIF *gtif;
 
    /*
    typedef std::map<const std::type_info *,tagtype_t> tagtype_map_t;
 
    static
    const tagtype_map_t tagtype_map;
    */
 
};
 
template <>
inline
void FileGeoTIFF::setGeoTiffField(geokey_t tag, const char *value){
    GTIFKeySet(gtif, tag, TYPE_ASCII, strlen(value)+1, value);
}
 
template <>
inline
void FileGeoTIFF::setGeoTiffField(geokey_t tag, const std::string & value){
    GTIFKeySet(gtif, tag, TYPE_ASCII, value.size()+1, value.c_str());
}
 
 
} // image::
 
//template <>
// const drain::FlagResolver::dict_t drain::EnumDict<image::FileGeoTIFF::TiffCompliance>::dict;
 
/*
template <>
template <>
image::TreeGDAL & image::TreeGDAL::operator()(const image::NodeGDAL::tag_t & type);
*/
 
} // drain::
#endif
 
#endif
 
 
/*
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<unsigned char>(){ return TYPE_BYTE;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<unsigned short>(){ return TYPE_SHORT;}
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<unsigned long>(){ return TYPE_LONG;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<std::string>(){ return TYPE_ASCII;};
 
template  <>
inline
tagtype_t FileGeoTIFF::getTagType<float>(){ return TYPE_FLOAT;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<double>(){ return TYPE_DOUBLE;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<signed char>(){ return TYPE_SBYTE;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<signed short>(){ return TYPE_SSHORT;};
 
template <>
inline
tagtype_t  FileGeoTIFF::getTagType<signed long>(){ return TYPE_SLONG;};
//inline getTagType<>(){ return TYPE_UNKNOWN);
//inline getTagType<>(){ return TYPE_RATIONAL);
*/