185 lines
4.7 KiB
C++
185 lines
4.7 KiB
C++
#ifndef _OTB_TRISKELE_I_IMAGE_TPP
|
|
#define _OTB_TRISKELE_I_IMAGE_TPP
|
|
|
|
// ========================================
|
|
template<typename PixelT>
|
|
inline void
|
|
Raster<PixelT>::setSize (const Size &size) {
|
|
DEF_LOG ("Raster::setSize", "size: " << size);
|
|
if (this->size == size)
|
|
return;
|
|
pixels.resize (DimImg (size.width)*DimImg (size.height));
|
|
this->size = size;
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline const Size &
|
|
Raster<PixelT>::getSize () const {
|
|
return size;
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline const PixelT *
|
|
Raster<PixelT>::getPixels () const {
|
|
return &pixels[0];
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline PixelT *
|
|
Raster<PixelT>::getPixels () {
|
|
return &pixels[0];
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline vector<PixelT> &
|
|
Raster<PixelT>::getPixelsVector () {
|
|
return pixels;
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline DimNodeId
|
|
Raster<PixelT>::pointIdx (const Point &p) const {
|
|
return point2idx (size, p);
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline PixelT
|
|
Raster<PixelT>::getValue (const DimImg &idx) const {
|
|
return pixels[idx];
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline PixelT
|
|
Raster<PixelT>::getValue (const Point &point) const {
|
|
return pixels [pointIdx (point)];
|
|
}
|
|
|
|
|
|
template<typename PixelT>
|
|
inline
|
|
Raster<PixelT>::Raster (const Size &size)
|
|
: size (NullSize),
|
|
pixels () {
|
|
setSize (size);
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline
|
|
Raster<PixelT>::Raster (const Raster &raster, const Rect &tile)
|
|
: size (tile.width, tile.height),
|
|
pixels () {
|
|
tile.cpToTile (raster.size, &raster.pixels[0], pixels);
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline
|
|
Raster<PixelT>::~Raster () {
|
|
}
|
|
|
|
// ========================================
|
|
inline void
|
|
IImage::setFileName (string fileName) {
|
|
if (this->fileName == fileName)
|
|
return;
|
|
this->fileName = fileName;
|
|
if (read)
|
|
close ();
|
|
}
|
|
|
|
inline void
|
|
IImage::getGeo (string& projectionRef, vector<double> &geoTransform) const {
|
|
projectionRef = this->projectionRef;
|
|
geoTransform = this->geoTransform;
|
|
}
|
|
|
|
inline void
|
|
IImage::setGeo (const string& projectionRef, vector<double> geoTransform, const Point &topLeft) {
|
|
DEF_LOG ("IImage::setGeo", "projectionRef:" << projectionRef);
|
|
// move TopLeft cf WordFile
|
|
double
|
|
x (geoTransform[0]),
|
|
xScale (geoTransform[1]),
|
|
xSkew (geoTransform[2]),
|
|
y (geoTransform[3]),
|
|
ySkew (geoTransform[4]),
|
|
yScale (geoTransform[5]);
|
|
double
|
|
xTrans = x + topLeft.x*xScale + topLeft.y*xSkew,
|
|
yTrans = y + topLeft.x*ySkew + topLeft.y*yScale;
|
|
geoTransform[0] = xTrans;
|
|
geoTransform[3] = yTrans;
|
|
CPLErr err = gdalOutputDataset->SetGeoTransform (&geoTransform[0]);
|
|
if (err != CE_None)
|
|
cerr << "IImage::setGeo: can't set geoTransform for " << fileName << endl;
|
|
err = gdalOutputDataset->SetProjection (projectionRef.c_str ());
|
|
if (err != CE_None)
|
|
cerr << "IImage::setGeo: can't set projection for " << fileName << endl;
|
|
}
|
|
|
|
|
|
inline const string &
|
|
IImage::getFileName () const {
|
|
return fileName;
|
|
}
|
|
|
|
inline const Size &
|
|
IImage::getSize () const {
|
|
return size;
|
|
}
|
|
|
|
inline const DimChanel &
|
|
IImage::getBandCount () const {
|
|
return bandCount;
|
|
}
|
|
|
|
inline GDALDataType
|
|
IImage::getDataType () const {
|
|
return dataType;
|
|
}
|
|
|
|
inline const bool
|
|
IImage::isRead () const {
|
|
return read;
|
|
}
|
|
|
|
inline const bool
|
|
IImage::isEmpty () const {
|
|
return size == NullSize;
|
|
}
|
|
|
|
// ========================================
|
|
template<typename PixelT>
|
|
inline void
|
|
IImage::readBand (Raster<PixelT> &raster, const DimChanel &band) const {
|
|
readBand (raster, band, NullPoint, size);
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline void
|
|
IImage::readBand (Raster<PixelT> &raster, DimChanel band, const Point &cropOrig, const Size &cropSize) const {
|
|
DEF_LOG ("IImage::readBand", "band: " << band << " crop: " << cropOrig << " - " << cropSize);
|
|
BOOST_ASSERT (gdalInputDataset);
|
|
raster.setSize (cropSize);
|
|
band++; // !!! GDAL bands start at 1 (not 0 :-( )
|
|
GDALRasterBand &poBand = *gdalInputDataset->GetRasterBand (band);
|
|
GDALDataType bandType = poBand.GetRasterDataType ();
|
|
CPLErr err = poBand.RasterIO (GF_Read, cropOrig.x, cropOrig.y, cropSize.width, cropSize.height,
|
|
raster.getPixels (), cropSize.width, cropSize.height, bandType, 0, 0);
|
|
if (err != CE_None)
|
|
cerr << "IImage::readBand: can't acces " << fileName << endl;
|
|
}
|
|
|
|
template<typename PixelT>
|
|
inline void
|
|
IImage::writeBand (PixelT *pixels, DimChanel band) const {
|
|
DEF_LOG ("IImage::writeBand", "band: " << band);
|
|
BOOST_ASSERT (gdalOutputDataset);
|
|
band++; // !!! GDAL layers starts at 1 (not 0 :-( )
|
|
GDALRasterBand &poBand = *gdalOutputDataset->GetRasterBand (band);
|
|
CPLErr err = poBand.RasterIO (GF_Write, 0, 0, size.width, size.height, pixels, size.width, size.height, dataType, 0, 0);
|
|
if (err != CE_None)
|
|
cerr << "IImage::writeBand: can't acces " << fileName << endl;
|
|
}
|
|
|
|
#endif // _OTB_TRISKELE_I_IMAGE_TPP
|