Moved include/osgTerrain/DataSet/.cpp, gdal plugin and osgdem out into the VirtualPlaneBuilder project.
Updated examples to remove DataSet/GDAL dependency. Updated wrappers
This commit is contained in:
@@ -1,738 +0,0 @@
|
||||
#include <osg/Image>
|
||||
#include <osg/Notify>
|
||||
#include <osg/Geode>
|
||||
#include <osg/GL>
|
||||
|
||||
#include <osgDB/Registry>
|
||||
#include <osgDB/FileNameUtils>
|
||||
#include <osgDB/FileUtils>
|
||||
#include <osgDB/ImageOptions>
|
||||
|
||||
#include <OpenThreads/ScopedLock>
|
||||
#include <OpenThreads/ReentrantMutex>
|
||||
|
||||
#include <gdal_priv.h>
|
||||
|
||||
#define SERIALIZER() OpenThreads::ScopedLock<OpenThreads::ReentrantMutex> lock(_serializerMutex)
|
||||
|
||||
// From easyrgb.com
|
||||
float Hue_2_RGB( float v1, float v2, float vH )
|
||||
{
|
||||
if ( vH < 0 ) vH += 1;
|
||||
if ( vH > 1 ) vH -= 1;
|
||||
if ( ( 6 * vH ) < 1 ) return ( v1 + ( v2 - v1 ) * 6 * vH );
|
||||
if ( ( 2 * vH ) < 1 ) return ( v2 );
|
||||
if ( ( 3 * vH ) < 2 ) return ( v1 + ( v2 - v1 ) * ( ( 2 / 3 ) - vH ) * 6 );
|
||||
return ( v1 );
|
||||
}
|
||||
|
||||
class ReaderWriterGDAL : public osgDB::ReaderWriter
|
||||
{
|
||||
public:
|
||||
virtual const char* className() const { return "GDAL Image Reader"; }
|
||||
virtual bool acceptsExtension(const std::string& extension) const
|
||||
{
|
||||
return osgDB::equalCaseInsensitive(extension,"gdal") || osgDB::equalCaseInsensitive(extension,"gdal");
|
||||
}
|
||||
|
||||
virtual ReadResult readImage(const std::string& fileName, const osgDB::ReaderWriter::Options* options) const
|
||||
{
|
||||
OpenThreads::ScopedLock<OpenThreads::ReentrantMutex> lock(_serializerMutex);
|
||||
return const_cast<ReaderWriterGDAL*>(this)->local_readImage(fileName, options);
|
||||
}
|
||||
|
||||
virtual ReadResult readHeightField(const std::string& fileName, const osgDB::ReaderWriter::Options* options) const
|
||||
{
|
||||
OpenThreads::ScopedLock<OpenThreads::ReentrantMutex> lock(_serializerMutex);
|
||||
return const_cast<ReaderWriterGDAL*>(this)->local_readHeightField(fileName, options);
|
||||
}
|
||||
|
||||
virtual ReadResult local_readImage(const std::string& file, const osgDB::ReaderWriter::Options* options)
|
||||
{
|
||||
// Looks like gdal's GDALRasterBand::GetColorInterpretation()
|
||||
// is not giving proper values for ecw images. There is small
|
||||
// hack to get around
|
||||
bool ecwLoad = osgDB::equalCaseInsensitive(osgDB::getFileExtension(file),"ecw");
|
||||
|
||||
//if (!acceptsExtension(ext)) return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
osg::notify(osg::INFO) << "GDAL : " << file << std::endl;
|
||||
|
||||
std::string fileName = osgDB::findDataFile( file, options );
|
||||
if (fileName.empty()) return ReadResult::FILE_NOT_FOUND;
|
||||
|
||||
initGDAL();
|
||||
|
||||
std::auto_ptr<GDALDataset> dataset((GDALDataset*)GDALOpen(fileName.c_str(),GA_ReadOnly));
|
||||
if (!dataset.get()) return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
int dataWidth = dataset->GetRasterXSize();
|
||||
int dataHeight = dataset->GetRasterYSize();
|
||||
|
||||
int windowX = 0;
|
||||
int windowY = 0;
|
||||
int windowWidth = dataWidth;
|
||||
int windowHeight = dataHeight;
|
||||
|
||||
int destX = 0;
|
||||
int destY = 0;
|
||||
int destWidth = osg::minimum(dataWidth,1024);
|
||||
int destHeight = osg::minimum(dataHeight,1024);
|
||||
// int destWidth = osg::minimum(dataWidth,4096);
|
||||
// int destHeight = osg::minimum(dataHeight,4096);
|
||||
|
||||
|
||||
osgDB::ImageOptions::TexCoordRange* texCoordRange = 0;
|
||||
|
||||
osgDB::ImageOptions* imageOptions = dynamic_cast<osgDB::ImageOptions*>(const_cast<osgDB::ReaderWriter::Options*>(options));
|
||||
if (imageOptions)
|
||||
{
|
||||
osg::notify(osg::INFO)<<"Got ImageOptions"<<std::endl;
|
||||
|
||||
int margin = 0;
|
||||
switch(imageOptions->_sourceImageWindowMode)
|
||||
{
|
||||
case(osgDB::ImageOptions::RATIO_WINDOW):
|
||||
{
|
||||
double desiredX = (double)dataWidth * imageOptions->_sourceRatioWindow.windowX;
|
||||
double desiredY = (double)dataHeight * imageOptions->_sourceRatioWindow.windowY;
|
||||
double desiredWidth = (double)dataWidth * imageOptions->_sourceRatioWindow.windowWidth;
|
||||
double desiredHeight = (double)dataHeight * imageOptions->_sourceRatioWindow.windowHeight;
|
||||
|
||||
windowX = osg::maximum((int)(floor(desiredX))-margin,0);
|
||||
windowY = osg::maximum((int)(floor(desiredY))-margin,0);
|
||||
windowWidth = osg::minimum((int)(ceil(desiredX + desiredWidth))+margin,dataWidth)-windowX;
|
||||
windowHeight = osg::minimum((int)(ceil(desiredY + desiredHeight))+margin,dataHeight)-windowY;
|
||||
|
||||
texCoordRange = new osgDB::ImageOptions::TexCoordRange;
|
||||
texCoordRange->set((desiredX-(double)windowX)/(double)windowWidth,
|
||||
((double)(windowY+windowHeight) -(desiredY+desiredHeight))/(double)windowHeight,
|
||||
(desiredWidth)/(double)windowWidth,
|
||||
(desiredHeight)/(double)windowHeight);
|
||||
osg::notify(osg::INFO)<<"tex coord range "<<texCoordRange->_x<<" "<<texCoordRange->_y<<" "<<texCoordRange->_w<<" "<<texCoordRange->_h<<std::endl;
|
||||
}
|
||||
break;
|
||||
case(osgDB::ImageOptions::PIXEL_WINDOW):
|
||||
windowX = imageOptions->_sourcePixelWindow.windowX;
|
||||
windowY = imageOptions->_sourcePixelWindow.windowY;
|
||||
windowWidth = imageOptions->_sourcePixelWindow.windowWidth;
|
||||
windowHeight = imageOptions->_sourcePixelWindow.windowHeight;
|
||||
break;
|
||||
default:
|
||||
// leave source window dimensions as whole image.
|
||||
break;
|
||||
}
|
||||
|
||||
// reapply the window coords to the pixel window so that calling code
|
||||
// knows the original pixel size
|
||||
imageOptions->_sourcePixelWindow.windowX = windowX;
|
||||
imageOptions->_sourcePixelWindow.windowY = windowY;
|
||||
imageOptions->_sourcePixelWindow.windowWidth = windowWidth;
|
||||
imageOptions->_sourcePixelWindow.windowHeight = windowHeight;
|
||||
|
||||
switch(imageOptions->_destinationImageWindowMode)
|
||||
{
|
||||
case(osgDB::ImageOptions::RATIO_WINDOW):
|
||||
destX = (unsigned int)(floor((double)dataWidth * imageOptions->_destinationRatioWindow.windowX));
|
||||
destY = (unsigned int)(floor((double)dataHeight * imageOptions->_destinationRatioWindow.windowY));
|
||||
destWidth = (unsigned int)(ceil((double)dataWidth * (imageOptions->_destinationRatioWindow.windowX + imageOptions->_destinationRatioWindow.windowWidth)))-windowX;
|
||||
destHeight = (unsigned int)(ceil((double)dataHeight * (imageOptions->_destinationRatioWindow.windowY + imageOptions->_destinationRatioWindow.windowHeight)))-windowY;
|
||||
break;
|
||||
case(osgDB::ImageOptions::PIXEL_WINDOW):
|
||||
destX = imageOptions->_destinationPixelWindow.windowX;
|
||||
destY = imageOptions->_destinationPixelWindow.windowY;
|
||||
destWidth = imageOptions->_destinationPixelWindow.windowWidth;
|
||||
destHeight = imageOptions->_destinationPixelWindow.windowHeight;
|
||||
break;
|
||||
default:
|
||||
// leave source window dimensions as whole image.
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// windowX = 0;
|
||||
// windowY = 0;
|
||||
// windowWidth = destWidth;
|
||||
// windowHeight = destHeight;
|
||||
|
||||
osg::notify(osg::INFO) << " windowX = "<<windowX<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowY = "<<windowY<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowWidth = "<<windowWidth<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowHeight = "<<windowHeight<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " destX = "<<destX<<std::endl;
|
||||
osg::notify(osg::INFO) << " destY = "<<destY<<std::endl;
|
||||
osg::notify(osg::INFO) << " destWidth = "<<destWidth<<std::endl;
|
||||
osg::notify(osg::INFO) << " destHeight = "<<destHeight<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " GetRaterCount() "<< dataset->GetRasterCount()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetProjectionRef() "<< dataset->GetProjectionRef()<<std::endl;
|
||||
|
||||
|
||||
double geoTransform[6];
|
||||
if (dataset->GetGeoTransform(geoTransform)==CE_None)
|
||||
{
|
||||
osg::notify(osg::INFO) << " GetGeoTransform "<< std::endl;
|
||||
osg::notify(osg::INFO) << " Origin = "<<geoTransform[0]<<" "<<geoTransform[3]<<std::endl;
|
||||
osg::notify(osg::INFO) << " Pixel X = "<<geoTransform[1]<<" "<<geoTransform[4]<<std::endl;
|
||||
osg::notify(osg::INFO) << " Pixel Y = "<<geoTransform[2]<<" "<<geoTransform[5]<<std::endl;
|
||||
}
|
||||
|
||||
int numBands = dataset->GetRasterCount();
|
||||
|
||||
|
||||
GDALRasterBand* bandGray = 0;
|
||||
GDALRasterBand* bandRed = 0;
|
||||
GDALRasterBand* bandGreen = 0;
|
||||
GDALRasterBand* bandBlue = 0;
|
||||
GDALRasterBand* bandAlpha = 0;
|
||||
GDALRasterBand* bandPalette = 0;
|
||||
|
||||
int internalFormat = GL_LUMINANCE;
|
||||
unsigned int pixelFormat = GL_LUMINANCE;
|
||||
unsigned int dataType = 0;
|
||||
unsigned int numBytesPerPixel = 0;
|
||||
|
||||
GDALDataType targetGDALType = GDT_Byte;
|
||||
|
||||
for(int b=1;b<=numBands;++b)
|
||||
{
|
||||
|
||||
GDALRasterBand* band = dataset->GetRasterBand(b);
|
||||
|
||||
osg::notify(osg::INFO) << " Band "<<b<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " GetOverviewCount() = "<< band->GetOverviewCount()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetColorTable() = "<< band->GetColorTable()<<std::endl;
|
||||
osg::notify(osg::INFO) << " DataTypeName() = "<< GDALGetDataTypeName(band->GetRasterDataType())<<std::endl;
|
||||
osg::notify(osg::INFO) << " ColorIntepretationName() = "<< GDALGetColorInterpretationName(band->GetColorInterpretation())<<std::endl;
|
||||
|
||||
bool bandNotHandled = true;
|
||||
if (ecwLoad)
|
||||
{
|
||||
bandNotHandled = false;
|
||||
switch (b)
|
||||
{
|
||||
case 1:
|
||||
bandRed = band;
|
||||
break;
|
||||
case 2:
|
||||
bandGreen = band;
|
||||
break;
|
||||
case 3:
|
||||
bandBlue = band;
|
||||
break;
|
||||
default:
|
||||
bandNotHandled = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (bandNotHandled)
|
||||
{
|
||||
if (band->GetColorInterpretation()==GCI_GrayIndex) bandGray = band;
|
||||
else if (band->GetColorInterpretation()==GCI_RedBand) bandRed = band;
|
||||
else if (band->GetColorInterpretation()==GCI_GreenBand) bandGreen = band;
|
||||
else if (band->GetColorInterpretation()==GCI_BlueBand) bandBlue = band;
|
||||
else if (band->GetColorInterpretation()==GCI_AlphaBand) bandAlpha = band;
|
||||
else if (band->GetColorInterpretation()==GCI_PaletteIndex) bandPalette = band;
|
||||
else bandGray = band;
|
||||
}
|
||||
|
||||
if (bandPalette)
|
||||
{
|
||||
osg::notify(osg::INFO) << " Palette Interpretation: " << GDALGetPaletteInterpretationName(band->GetColorTable()->GetPaletteInterpretation()) << std::endl;
|
||||
}
|
||||
|
||||
// int gotMin,gotMax;
|
||||
// double minmax[2];
|
||||
//
|
||||
// minmax[0] = band->GetMinimum(&gotMin);
|
||||
// minmax[1] = band->GetMaximum(&gotMax);
|
||||
// if (!(gotMin && gotMax))
|
||||
// {
|
||||
// osg::notify(osg::INFO)<<" computing min max"<<std::endl;
|
||||
// GDALComputeRasterMinMax(band,TRUE,minmax);
|
||||
// }
|
||||
//
|
||||
// osg::notify(osg::INFO) << " min "<<minmax[0]<<std::endl;
|
||||
// osg::notify(osg::INFO) << " max "<<minmax[1]<<std::endl;
|
||||
|
||||
if (dataType==0)
|
||||
{
|
||||
targetGDALType = band->GetRasterDataType();
|
||||
switch(band->GetRasterDataType())
|
||||
{
|
||||
case(GDT_Byte): dataType = GL_UNSIGNED_BYTE; numBytesPerPixel = 1; break;
|
||||
case(GDT_UInt16): dataType = GL_UNSIGNED_SHORT; numBytesPerPixel = 2; break;
|
||||
case(GDT_Int16): dataType = GL_SHORT; numBytesPerPixel = 2; break;
|
||||
case(GDT_UInt32): dataType = GL_UNSIGNED_INT; numBytesPerPixel = 4; break;
|
||||
case(GDT_Int32): dataType = GL_INT; numBytesPerPixel = 4; break;
|
||||
case(GDT_Float32): dataType = GL_FLOAT; numBytesPerPixel = 4; break;
|
||||
case(GDT_Float64): dataType = GL_DOUBLE; numBytesPerPixel = 8; break; // not handled
|
||||
default: dataType = 0; numBytesPerPixel = 0; break; // not handled
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int s = destWidth;
|
||||
int t = destHeight;
|
||||
int r = 1;
|
||||
|
||||
|
||||
if (dataType==0)
|
||||
{
|
||||
dataType = GL_UNSIGNED_BYTE;
|
||||
numBytesPerPixel = 1;
|
||||
targetGDALType = GDT_Byte;
|
||||
}
|
||||
|
||||
unsigned char* imageData = 0;
|
||||
|
||||
if (bandRed && bandGreen && bandBlue)
|
||||
{
|
||||
if (bandAlpha)
|
||||
{
|
||||
// RGBA
|
||||
|
||||
int pixelSpace=4*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
imageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
pixelFormat = GL_RGBA;
|
||||
internalFormat = GL_RGBA;
|
||||
|
||||
osg::notify(osg::INFO) << "reading RGBA"<<std::endl;
|
||||
|
||||
bandRed->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+0),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandGreen->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+1),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandBlue->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+2),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandAlpha->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+3),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
// RGB
|
||||
|
||||
int pixelSpace=3*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
imageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
pixelFormat = GL_RGB;
|
||||
internalFormat = GL_RGB;
|
||||
|
||||
osg::notify(osg::INFO) << "reading RGB"<<std::endl;
|
||||
|
||||
bandRed->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+0),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandGreen->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+1),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandBlue->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+2),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
|
||||
}
|
||||
}
|
||||
else if (bandGray)
|
||||
{
|
||||
if (bandAlpha)
|
||||
{
|
||||
// Luminance alpha
|
||||
int pixelSpace=2*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
imageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
pixelFormat = GL_LUMINANCE_ALPHA;
|
||||
internalFormat = GL_LUMINANCE_ALPHA;
|
||||
|
||||
osg::notify(osg::INFO) << "reading grey + alpha"<<std::endl;
|
||||
|
||||
bandGray->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+0),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
bandAlpha->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+1),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Luminance map
|
||||
int pixelSpace=1*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
imageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
pixelFormat = GL_LUMINANCE;
|
||||
internalFormat = GL_LUMINANCE;
|
||||
|
||||
osg::notify(osg::INFO) << "reading grey"<<std::endl;
|
||||
|
||||
bandGray->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+0),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
}
|
||||
}
|
||||
else if (bandAlpha)
|
||||
{
|
||||
// alpha map
|
||||
int pixelSpace=1*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
imageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
pixelFormat = GL_ALPHA;
|
||||
internalFormat = GL_ALPHA;
|
||||
|
||||
osg::notify(osg::INFO) << "reading alpha"<<std::endl;
|
||||
|
||||
bandAlpha->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(imageData+0),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
|
||||
}
|
||||
else if (bandPalette)
|
||||
{
|
||||
// Paletted map
|
||||
int pixelSpace=1*numBytesPerPixel;
|
||||
int lineSpace=destWidth * pixelSpace;
|
||||
|
||||
unsigned char *rawImageData;
|
||||
rawImageData = new unsigned char[destWidth * destHeight * pixelSpace];
|
||||
imageData = new unsigned char[destWidth * destHeight * 4/*RGBA*/];
|
||||
pixelFormat = GL_RGBA;
|
||||
internalFormat = GL_RGBA;
|
||||
|
||||
osg::notify(osg::INFO) << "reading palette"<<std::endl;
|
||||
osg::notify(osg::INFO) << "numBytesPerPixel: " << numBytesPerPixel << std::endl;
|
||||
|
||||
bandPalette->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(rawImageData),destWidth,destHeight,targetGDALType,pixelSpace,lineSpace);
|
||||
|
||||
// Map the indexes to an actual RGBA Value.
|
||||
for (int i = 0; i < destWidth * destHeight; i++)
|
||||
{
|
||||
const GDALColorEntry *colorEntry = bandPalette->GetColorTable()->GetColorEntry(rawImageData[i]);
|
||||
GDALPaletteInterp interp = bandPalette->GetColorTable()->GetPaletteInterpretation();
|
||||
if (!colorEntry)
|
||||
{
|
||||
//FIXME: What to do here?
|
||||
|
||||
//osg::notify(osg::INFO) << "NO COLOR ENTRY FOR COLOR " << rawImageData[i] << std::endl;
|
||||
imageData[4*i+0] = 255;
|
||||
imageData[4*i+1] = 0;
|
||||
imageData[4*i+2] = 0;
|
||||
imageData[4*i+3] = 1;
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
if (interp == GPI_RGB)
|
||||
{
|
||||
imageData[4*i+0] = colorEntry->c1;
|
||||
imageData[4*i+1] = colorEntry->c2;
|
||||
imageData[4*i+2] = colorEntry->c3;
|
||||
imageData[4*i+3] = colorEntry->c4;
|
||||
}
|
||||
else if (interp == GPI_CMYK)
|
||||
{
|
||||
// from wikipedia.org
|
||||
short C = colorEntry->c1;
|
||||
short M = colorEntry->c2;
|
||||
short Y = colorEntry->c3;
|
||||
short K = colorEntry->c4;
|
||||
imageData[4*i+0] = 255 - C*(255 - K) - K;
|
||||
imageData[4*i+1] = 255 - M*(255 - K) - K;
|
||||
imageData[4*i+2] = 255 - Y*(255 - K) - K;
|
||||
imageData[4*i+3] = 255;
|
||||
}
|
||||
else if (interp == GPI_HLS)
|
||||
{
|
||||
// from easyrgb.com
|
||||
float H = colorEntry->c1;
|
||||
float S = colorEntry->c3;
|
||||
float L = colorEntry->c2;
|
||||
float R, G, B;
|
||||
if ( S == 0 ) //HSL values = 0 - 1
|
||||
{
|
||||
R = L; //RGB results = 0 - 1
|
||||
G = L;
|
||||
B = L;
|
||||
}
|
||||
else
|
||||
{
|
||||
float var_2, var_1;
|
||||
if ( L < 0.5 )
|
||||
var_2 = L * ( 1 + S );
|
||||
else
|
||||
var_2 = ( L + S ) - ( S * L );
|
||||
|
||||
var_1 = 2 * L - var_2;
|
||||
|
||||
R = Hue_2_RGB( var_1, var_2, H + ( 1 / 3 ) );
|
||||
G = Hue_2_RGB( var_1, var_2, H );
|
||||
B = Hue_2_RGB( var_1, var_2, H - ( 1 / 3 ) );
|
||||
}
|
||||
imageData[4*i+0] = static_cast<unsigned char>(R*255.0f);
|
||||
imageData[4*i+1] = static_cast<unsigned char>(G*255.0f);
|
||||
imageData[4*i+2] = static_cast<unsigned char>(B*255.0f);
|
||||
imageData[4*i+3] = static_cast<unsigned char>(255.0f);
|
||||
}
|
||||
else if (interp == GPI_Gray)
|
||||
{
|
||||
imageData[4*i+0] = static_cast<unsigned char>(colorEntry->c1*255.0f);
|
||||
imageData[4*i+1] = static_cast<unsigned char>(colorEntry->c1*255.0f);
|
||||
imageData[4*i+2] = static_cast<unsigned char>(colorEntry->c1*255.0f);
|
||||
imageData[4*i+3] = static_cast<unsigned char>(255.0f);
|
||||
}
|
||||
}
|
||||
}
|
||||
delete [] rawImageData;
|
||||
}
|
||||
else
|
||||
{
|
||||
osg::notify(osg::INFO) << "not found any usable bands in file."<<std::endl;
|
||||
}
|
||||
|
||||
|
||||
//GDALOpen(dataset);
|
||||
|
||||
if (imageData)
|
||||
{
|
||||
osg::Image* image = new osg::Image;
|
||||
image->setFileName(fileName.c_str());
|
||||
image->setImage(s,t,r,
|
||||
internalFormat,
|
||||
pixelFormat,
|
||||
dataType,
|
||||
(unsigned char *)imageData,
|
||||
osg::Image::USE_NEW_DELETE);
|
||||
|
||||
if (texCoordRange) image->setUserData(texCoordRange);
|
||||
|
||||
image->flipVertical();
|
||||
|
||||
return image;
|
||||
|
||||
}
|
||||
|
||||
return 0;
|
||||
|
||||
}
|
||||
|
||||
|
||||
ReadResult local_readHeightField(const std::string& fileName, const osgDB::ReaderWriter::Options* options)
|
||||
{
|
||||
//std::string ext = osgDB::getFileExtension(fileName);
|
||||
//if (!acceptsExtension(ext)) return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
initGDAL();
|
||||
|
||||
std::auto_ptr<GDALDataset> dataset((GDALDataset*)GDALOpen(fileName.c_str(),GA_ReadOnly));
|
||||
if (!dataset.get()) return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
int dataWidth = dataset->GetRasterXSize();
|
||||
int dataHeight = dataset->GetRasterYSize();
|
||||
|
||||
int windowX = 0;
|
||||
int windowY = 0;
|
||||
int windowWidth = dataWidth;
|
||||
int windowHeight = dataHeight;
|
||||
|
||||
int destX = 0;
|
||||
int destY = 0;
|
||||
int destWidth = osg::minimum(dataWidth,1024);
|
||||
int destHeight = osg::minimum(dataHeight,1024);
|
||||
|
||||
osgDB::ImageOptions::TexCoordRange* texCoordRange = 0;
|
||||
|
||||
const osgDB::ImageOptions* imageOptions = dynamic_cast<const osgDB::ImageOptions*>(options);
|
||||
if (imageOptions)
|
||||
{
|
||||
osg::notify(osg::INFO)<<"Got ImageOptions"<<std::endl;
|
||||
|
||||
int margin = 0;
|
||||
switch(imageOptions->_sourceImageWindowMode)
|
||||
{
|
||||
case(osgDB::ImageOptions::RATIO_WINDOW):
|
||||
{
|
||||
double desiredX = (double)dataWidth * imageOptions->_sourceRatioWindow.windowX;
|
||||
double desiredY = (double)dataHeight * imageOptions->_sourceRatioWindow.windowY;
|
||||
double desiredWidth = (double)dataWidth * imageOptions->_sourceRatioWindow.windowWidth;
|
||||
double desiredHeight = (double)dataHeight * imageOptions->_sourceRatioWindow.windowHeight;
|
||||
|
||||
windowX = osg::maximum((int)(floor(desiredX))-margin,0);
|
||||
windowY = osg::maximum((int)(floor(desiredY))-margin,0);
|
||||
windowWidth = osg::minimum((int)(ceil(desiredX + desiredWidth))+margin,dataWidth)-windowX;
|
||||
windowHeight = osg::minimum((int)(ceil(desiredY + desiredHeight))+margin,dataHeight)-windowY;
|
||||
|
||||
texCoordRange = new osgDB::ImageOptions::TexCoordRange;
|
||||
texCoordRange->set((desiredX-(double)windowX)/(double)windowWidth,
|
||||
((double)(windowY+windowHeight) -(desiredY+desiredHeight))/(double)windowHeight,
|
||||
(desiredWidth)/(double)windowWidth,
|
||||
(desiredHeight)/(double)windowHeight);
|
||||
osg::notify(osg::INFO)<<"tex coord range "<<texCoordRange->_x<<" "<<texCoordRange->_y<<" "<<texCoordRange->_w<<" "<<texCoordRange->_h<<std::endl;
|
||||
}
|
||||
break;
|
||||
case(osgDB::ImageOptions::PIXEL_WINDOW):
|
||||
windowX = imageOptions->_sourcePixelWindow.windowX;
|
||||
windowY = imageOptions->_sourcePixelWindow.windowY;
|
||||
windowWidth = imageOptions->_sourcePixelWindow.windowWidth;
|
||||
windowHeight = imageOptions->_sourcePixelWindow.windowHeight;
|
||||
break;
|
||||
default:
|
||||
// leave source window dimensions as whole image.
|
||||
break;
|
||||
}
|
||||
|
||||
switch(imageOptions->_destinationImageWindowMode)
|
||||
{
|
||||
case(osgDB::ImageOptions::RATIO_WINDOW):
|
||||
destX = (unsigned int)(floor((double)dataWidth * imageOptions->_destinationRatioWindow.windowX));
|
||||
destY = (unsigned int)(floor((double)dataHeight * imageOptions->_destinationRatioWindow.windowY));
|
||||
destWidth = (unsigned int)(ceil((double)dataWidth * (imageOptions->_destinationRatioWindow.windowX + imageOptions->_destinationRatioWindow.windowWidth)))-windowX;
|
||||
destHeight = (unsigned int)(ceil((double)dataHeight * (imageOptions->_destinationRatioWindow.windowY + imageOptions->_destinationRatioWindow.windowHeight)))-windowY;
|
||||
break;
|
||||
case(osgDB::ImageOptions::PIXEL_WINDOW):
|
||||
destX = imageOptions->_destinationPixelWindow.windowX;
|
||||
destY = imageOptions->_destinationPixelWindow.windowY;
|
||||
destWidth = imageOptions->_destinationPixelWindow.windowWidth;
|
||||
destHeight = imageOptions->_destinationPixelWindow.windowHeight;
|
||||
break;
|
||||
default:
|
||||
// leave source window dimensions as whole image.
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// windowX = 0;
|
||||
// windowY = 0;
|
||||
// windowWidth = destWidth;
|
||||
// windowHeight = destHeight;
|
||||
|
||||
osg::notify(osg::INFO) << " windowX = "<<windowX<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowY = "<<windowY<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowWidth = "<<windowWidth<<std::endl;
|
||||
osg::notify(osg::INFO) << " windowHeight = "<<windowHeight<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " destX = "<<destX<<std::endl;
|
||||
osg::notify(osg::INFO) << " destY = "<<destY<<std::endl;
|
||||
osg::notify(osg::INFO) << " destWidth = "<<destWidth<<std::endl;
|
||||
osg::notify(osg::INFO) << " destHeight = "<<destHeight<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " GetRaterCount() "<< dataset->GetRasterCount()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetProjectionRef() "<< dataset->GetProjectionRef()<<std::endl;
|
||||
|
||||
|
||||
double geoTransform[6];
|
||||
osg::notify(osg::INFO) << " GetGeoTransform == "<< dataset->GetGeoTransform(geoTransform)<<" == CE_None"<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetGeoTransform "<< std::endl;
|
||||
osg::notify(osg::INFO) << " Origin = "<<geoTransform[0]<<" "<<geoTransform[3]<<std::endl;
|
||||
osg::notify(osg::INFO) << " Pixel X = "<<geoTransform[1]<<" "<<geoTransform[4]<<std::endl;
|
||||
osg::notify(osg::INFO) << " Pixel Y = "<<geoTransform[2]<<" "<<geoTransform[5]<<std::endl;
|
||||
|
||||
|
||||
double TopLeft[2],BottomLeft[2],BottomRight[2],TopRight[2];
|
||||
TopLeft[0] = geoTransform[0];
|
||||
TopLeft[1] = geoTransform[1];
|
||||
BottomLeft[0] = TopLeft[0]+geoTransform[2]*(dataHeight-1);
|
||||
BottomLeft[1] = TopLeft[1]+geoTransform[5]*(dataHeight-1);
|
||||
BottomRight[0] = BottomLeft[0]+geoTransform[1]*(dataWidth-1);
|
||||
BottomRight[1] = BottomLeft[1]+geoTransform[4]*(dataWidth-1);
|
||||
TopRight[0] = TopLeft[0]+geoTransform[1]*(dataWidth-1);
|
||||
TopRight[1] = TopLeft[1]+geoTransform[4]*(dataWidth-1);
|
||||
|
||||
|
||||
osg::notify(osg::INFO) << "TopLeft "<<TopLeft[0]<<"\t"<<TopLeft[1]<<std::endl;
|
||||
osg::notify(osg::INFO) << "BottomLeft "<<BottomLeft[0]<<"\t"<<BottomLeft[1]<<std::endl;
|
||||
osg::notify(osg::INFO) << "BottomRight "<<BottomRight[0]<<"\t"<<BottomRight[1]<<std::endl;
|
||||
osg::notify(osg::INFO) << "TopRight "<<TopRight[0]<<"\t"<<TopRight[1]<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO)<<" GDALGetGCPCount "<<dataset->GetGCPCount()<<std::endl;
|
||||
|
||||
int numBands = dataset->GetRasterCount();
|
||||
|
||||
|
||||
GDALRasterBand* bandGray = 0;
|
||||
GDALRasterBand* bandRed = 0;
|
||||
GDALRasterBand* bandGreen = 0;
|
||||
GDALRasterBand* bandBlue = 0;
|
||||
GDALRasterBand* bandAlpha = 0;
|
||||
|
||||
for(int b=1;b<=numBands;++b)
|
||||
{
|
||||
|
||||
GDALRasterBand* band = dataset->GetRasterBand(b);
|
||||
|
||||
osg::notify(osg::INFO) << " Band "<<b<<std::endl;
|
||||
|
||||
osg::notify(osg::INFO) << " GetOverviewCount() = "<< band->GetOverviewCount()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetColorTable() = "<< band->GetColorTable()<<std::endl;
|
||||
osg::notify(osg::INFO) << " DataTypeName() = "<< GDALGetDataTypeName(band->GetRasterDataType())<<std::endl;
|
||||
osg::notify(osg::INFO) << " ColorIntepretationName() = "<< GDALGetColorInterpretationName(band->GetColorInterpretation())<<std::endl;
|
||||
|
||||
|
||||
osg::notify(osg::INFO) << std::endl;
|
||||
osg::notify(osg::INFO) << " GetNoDataValue() = "<< band->GetNoDataValue()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetMinimum() = "<< band->GetMinimum()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetMaximum() = "<< band->GetMaximum()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetOffset() = "<< band->GetOffset()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetScale() = "<< band->GetScale()<<std::endl;
|
||||
osg::notify(osg::INFO) << " GetUnitType() = '"<< band->GetUnitType()<<"'"<<std::endl;
|
||||
|
||||
|
||||
if (band->GetColorInterpretation()==GCI_GrayIndex) bandGray = band;
|
||||
else if (band->GetColorInterpretation()==GCI_RedBand) bandRed = band;
|
||||
else if (band->GetColorInterpretation()==GCI_GreenBand) bandGreen = band;
|
||||
else if (band->GetColorInterpretation()==GCI_BlueBand) bandBlue = band;
|
||||
else if (band->GetColorInterpretation()==GCI_AlphaBand) bandAlpha = band;
|
||||
else bandGray = band;
|
||||
|
||||
}
|
||||
|
||||
|
||||
GDALRasterBand* bandSelected = 0;
|
||||
if (!bandSelected && bandGray) bandSelected = bandGray;
|
||||
else if (!bandSelected && bandAlpha) bandSelected = bandAlpha;
|
||||
else if (!bandSelected && bandRed) bandSelected = bandRed;
|
||||
else if (!bandSelected && bandGreen) bandSelected = bandGreen;
|
||||
else if (!bandSelected && bandBlue) bandSelected = bandBlue;
|
||||
|
||||
if (bandSelected)
|
||||
{
|
||||
osg::HeightField* hf = new osg::HeightField;
|
||||
hf->allocate(destWidth,destHeight);
|
||||
|
||||
bandSelected->RasterIO(GF_Read,windowX,windowY,windowWidth,windowHeight,(void*)(&(hf->getHeightList().front())),destWidth,destHeight,GDT_Float32,0,0);
|
||||
|
||||
// now need to flip since the OSG's origin is in lower left corner.
|
||||
osg::notify(osg::INFO)<<"flipping"<<std::endl;
|
||||
unsigned int copy_r = hf->getNumRows()-1;
|
||||
for(unsigned int r=0;r<copy_r;++r,--copy_r)
|
||||
{
|
||||
for(unsigned int c=0;c<hf->getNumColumns();++c)
|
||||
{
|
||||
float temp = hf->getHeight(c,r);
|
||||
hf->setHeight(c,r,hf->getHeight(c,copy_r));
|
||||
hf->setHeight(c,copy_r,temp);
|
||||
}
|
||||
}
|
||||
|
||||
return hf;
|
||||
}
|
||||
|
||||
return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
}
|
||||
|
||||
void initGDAL()
|
||||
{
|
||||
static bool s_initialized = false;
|
||||
if (!s_initialized)
|
||||
{
|
||||
s_initialized = true;
|
||||
GDALAllRegister();
|
||||
}
|
||||
}
|
||||
|
||||
mutable OpenThreads::ReentrantMutex _serializerMutex;
|
||||
|
||||
};
|
||||
|
||||
// now register with Registry to instantiate the above
|
||||
// reader/writer.
|
||||
osgDB::RegisterReaderWriterProxy<ReaderWriterGDAL> g_readerWriter_GDAL_Proxy;
|
||||
Reference in New Issue
Block a user