/****************************************************************************** * $Id: gdalpansharpen.h 33715 2016-03-13 08:52:06Z goatbar $ * * Project: GDAL Pansharpening module * Purpose: Prototypes, and definitions for pansharpening related work. * Author: Even Rouault * ****************************************************************************** * Copyright (c) 2015, Even Rouault * * 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. ****************************************************************************/ #ifndef GDALPANSHARPEN_H_INCLUDED #define GDALPANSHARPEN_H_INCLUDED #include "gdal.h" CPL_C_START /** * \file gdalpansharpen.h * * GDAL pansharpening related entry points and definitions. * * @since GDAL 2.1 */ /** Pansharpening algorithms. */ typedef enum { /*! Weighted Brovery. */ GDAL_PSH_WEIGHTED_BROVEY } GDALPansharpenAlg; /** Pansharpening options. */ typedef struct { /*! Pan sharpening algorithm/method. Only weighed Brovey for now. */ GDALPansharpenAlg ePansharpenAlg; /*! Resampling algorithm to upsample spectral bands to pan band resolution. */ GDALRIOResampleAlg eResampleAlg; /*! Bit depth of the spectral bands. Can be let to 0 for default behaviour. */ int nBitDepth; /*! Number of weight coefficients in padfWeights. */ int nWeightCount; /*! Array of nWeightCount weights used by weighted Brovey. */ double *padfWeights; /*! Panchromatic band. */ GDALRasterBandH hPanchroBand; /*! Number of input spectral bands. */ int nInputSpectralBands; /** Array of nInputSpectralBands input spectral bands. The spectral band have * generally a coarser resolution than the panchromatic band, but they * are assumed to have the same spatial extent (and projection) at that point. * Necessary spatial adjustments must be done beforehand, for example by wrapping * inside a VRT dataset. */ GDALRasterBandH *pahInputSpectralBands; /*! Number of output pansharpened spectral bands. */ int nOutPansharpenedBands; /*! Array of nOutPansharpendBands values such as panOutPansharpenedBands[k] is a value in the range [0,nInputSpectralBands-1] . */ int *panOutPansharpenedBands; /*! Whether the panchromatic and spectral bands have a noData value. */ int bHasNoData; /** NoData value of the panchromatic and spectral bands (only taken into account if bHasNoData = TRUE). This will also be use has the output nodata value. */ double dfNoData; /** Number of threads or -1 to mean ALL_CPUS. By default (0), single threaded mode is enabled * unless the GDAL_NUM_THREADS configuration option is set to an integer or ALL_CPUS. */ int nThreads; double dfMSShiftX; double dfMSShiftY; } GDALPansharpenOptions; GDALPansharpenOptions CPL_DLL * GDALCreatePansharpenOptions(void); void CPL_DLL GDALDestroyPansharpenOptions( GDALPansharpenOptions * ); GDALPansharpenOptions CPL_DLL * GDALClonePansharpenOptions( const GDALPansharpenOptions* psOptions); /*! Pansharpening operation handle. */ typedef void* GDALPansharpenOperationH; GDALPansharpenOperationH CPL_DLL GDALCreatePansharpenOperation(const GDALPansharpenOptions* ); void CPL_DLL GDALDestroyPansharpenOperation( GDALPansharpenOperationH ); CPLErr CPL_DLL GDALPansharpenProcessRegion( GDALPansharpenOperationH hOperation, int nXOff, int nYOff, int nXSize, int nYSize, void *pDataBuf, GDALDataType eBufDataType); CPL_C_END #ifdef __cplusplus #include #include "gdal_priv.h" #include "cpl_worker_thread_pool.h" #ifdef DEBUG_TIMING #include #endif class GDALPansharpenOperation; typedef struct { GDALPansharpenOperation* poPansharpenOperation; GDALDataType eWorkDataType; GDALDataType eBufDataType; const void* pPanBuffer; const void* pUpsampledSpectralBuffer; void* pDataBuf; int nValues; int nBandValues; GUInt32 nMaxValue; #ifdef DEBUG_TIMING struct timeval* ptv; #endif CPLErr eErr; } GDALPansharpenJob; typedef struct { GDALDataset* poMEMDS; int nXOff; int nYOff; int nXSize; int nYSize; double dfXOff; double dfYOff; double dfXSize; double dfYSize; void *pBuffer; GDALDataType eDT; int nBufXSize; int nBufYSize; int nBandCount; GDALRIOResampleAlg eResampleAlg; GSpacing nBandSpace; #ifdef DEBUG_TIMING struct timeval* ptv; #endif } GDALPansharpenResampleJob; /** Pansharpening operation class. */ class GDALPansharpenOperation { GDALPansharpenOptions* psOptions; std::vector anInputBands; std::vector aVDS; // to destroy std::vector aMSBands; // original multispectral bands potentially warped into a VRT int bPositiveWeights; CPLWorkerThreadPool* poThreadPool; int nKernelRadius; static void PansharpenJobThreadFunc(void* pUserData); static void PansharpenResampleJobThreadFunc(void* pUserData); template void WeightedBroveyWithNoData( const WorkDataType* pPanBuffer, const WorkDataType* pUpsampledSpectralBuffer, OutDataType* pDataBuf, int nValues, int nBandValues, WorkDataType nMaxValue) const; template void WeightedBrovey3( const WorkDataType* pPanBuffer, const WorkDataType* pUpsampledSpectralBuffer, OutDataType* pDataBuf, int nValues, int nBandValues, WorkDataType nMaxValue) const; template void WeightedBrovey( const WorkDataType* pPanBuffer, const WorkDataType* pUpsampledSpectralBuffer, OutDataType* pDataBuf, int nValues, int nBandValues, WorkDataType nMaxValue) const; template CPLErr WeightedBrovey( const WorkDataType* pPanBuffer, const WorkDataType* pUpsampledSpectralBuffer, void *pDataBuf, GDALDataType eBufDataType, int nValues, int nBandValues, WorkDataType nMaxValue) const; template CPLErr WeightedBrovey( const WorkDataType* pPanBuffer, const WorkDataType* pUpsampledSpectralBuffer, void *pDataBuf, GDALDataType eBufDataType, int nValues, int nBandValues) const; void WeightedBroveyPositiveWeights( const GUInt16* pPanBuffer, const GUInt16* pUpsampledSpectralBuffer, GUInt16* pDataBuf, int nValues, int nBandValues, GUInt16 nMaxValue) const; template int WeightedBroveyPositiveWeightsInternal( const GUInt16* pPanBuffer, const GUInt16* pUpsampledSpectralBuffer, GUInt16* pDataBuf, int nValues, int nBandValues, GUInt16 nMaxValue) const; CPLErr PansharpenChunk( GDALDataType eWorkDataType, GDALDataType eBufDataType, const void* pPanBuffer, const void* pUpsampledSpectralBuffer, void* pDataBuf, int nValues, int nBandValues, GUInt32 nMaxValue) const; public: GDALPansharpenOperation(); ~GDALPansharpenOperation(); CPLErr Initialize(const GDALPansharpenOptions* psOptions); CPLErr ProcessRegion(int nXOff, int nYOff, int nXSize, int nYSize, void *pDataBuf, GDALDataType eBufDataType); GDALPansharpenOptions* GetOptions(); }; #endif /* __cplusplus */ #endif /* GDALPANSHARPEN_H_INCLUDED */