void RasterImageLayer::loadFile()
{
GDALDataset *ds = static_cast<GDALDataset*>(GDALOpen(m_filename.toLocal8Bit(), GA_ReadOnly));
if (ds == nullptr)
{
qWarning() << "Error opening file:" << m_filename;
}
else
{
projection().setGeogCS(new OGRSpatialReference(ds->GetProjectionRef()));
projection().setProjCS(new OGRSpatialReference(ds->GetProjectionRef()));
projection().setDomain({-180., -90., 360., 180.});
std::vector<double> geoTransform(6);
int xsize = ds->GetRasterXSize();
int ysize = ds->GetRasterYSize();
ds->GetGeoTransform(geoTransform.data());
vertData.resize(4);
vertData[0] = QVector2D(geoTransform[0], geoTransform[3]);
vertData[1] = QVector2D(geoTransform[0] + geoTransform[2] * ysize,
geoTransform[3] + geoTransform[5] * ysize);
vertData[2] = QVector2D(geoTransform[0] + geoTransform[1] * xsize + geoTransform[2] * ysize,
geoTransform[3] + geoTransform[4] * xsize + geoTransform[5] * ysize);
vertData[3] = QVector2D(geoTransform[0] + geoTransform[1] * xsize,
geoTransform[3] + geoTransform[4] * xsize);
texData = {{0., 0.}, {0., 1.}, {1., 1.}, {1., 0.}};
int numBands = ds->GetRasterCount();
qDebug() << "Bands:" << numBands;
imData = QImage(xsize, ysize, QImage::QImage::Format_RGBA8888);
imData.fill(QColor(255, 255, 255, 255));
// Bands start at 1
for (int i = 1; i <= numBands; ++i)
{
GDALRasterBand *band = ds->GetRasterBand(i);
switch(band->GetColorInterpretation())
{
case GCI_RedBand:
band->RasterIO(GF_Read, 0, 0, xsize, ysize, imData.bits(),
xsize, ysize, GDT_Byte, 4, 0);
break;
case GCI_GreenBand:
band->RasterIO(GF_Read, 0, 0, xsize, ysize, imData.bits() + 1,
xsize, ysize, GDT_Byte, 4, 0);
break;
case GCI_BlueBand:
band->RasterIO(GF_Read, 0, 0, xsize, ysize, imData.bits() + 2,
xsize, ysize, GDT_Byte, 4, 0);
break;
default:
qWarning() << "Unhandled color interpretation:" << band->GetColorInterpretation();
}
}
GDALClose(ds);
newFile = true;
}
}
void FBXConverter::processAnimations( FbxNode* node , std::vector<JointData> &skeleton , std::vector<VertexData> &verts , std::vector<IndexData> &indices )
{
FbxMesh* theMesh = node->GetMesh();
FbxAnimStack* animationStack = node->GetScene()->GetSrcObject<FbxAnimStack>();
FbxAnimLayer* animationLayer = animationStack->GetMember<FbxAnimLayer>();
std::vector<FbxTime> frames;
for ( int curveNodeIndex = 0; curveNodeIndex < animationLayer->GetMemberCount(); ++curveNodeIndex )
{
FbxAnimCurveNode* currentCurve = animationLayer->GetMember<FbxAnimCurveNode>( curveNodeIndex );
for ( unsigned int channelIndex = 0; channelIndex < currentCurve->GetChannelsCount(); ++channelIndex )
{
for ( int curve = 0; curve < currentCurve->GetCurveCount( channelIndex ); ++curve )
{
FbxAnimCurve* theCurveVictim = currentCurve->GetCurve( channelIndex , curve );
for ( int keyIndex = 0; keyIndex < theCurveVictim->KeyGetCount(); ++keyIndex )
{
bool alreadyIn = false;
for ( unsigned int frameIndex = 0; frameIndex < frames.size(); ++frameIndex )
{
if ( ( unsigned int ) frames[frameIndex].GetFrameCount() == ( unsigned int ) theCurveVictim->KeyGet( keyIndex ).GetTime().GetFrameCount() )
{
alreadyIn = true;
//std::cout << frames[frameIndex].GetFrameCount() << " " << theCurveVictim->KeyGet( keyIndex ).GetTime().GetFrameCount() << " " << alreadyIn << std::endl;
break;
}
}
std::cout << theCurveVictim->KeyGet( keyIndex ).GetTime().GetFrameCount() << " " << alreadyIn << std::endl;
if(!alreadyIn) frames.push_back(theCurveVictim->KeyGet( keyIndex ).GetTime());
}
}
}
}
std::sort( frames.begin() , frames.end() , frameCompare );
FbxAMatrix geoTransform( node->GetGeometricTranslation( FbxNode::eSourcePivot ) , node->GetGeometricRotation( FbxNode::eSourcePivot ) , node->GetGeometricScaling( FbxNode::eSourcePivot ) );
for ( int i = 0; i < theMesh->GetDeformerCount(); ++i )
{
FbxSkin* theSkin = reinterpret_cast< FbxSkin* >( theMesh->GetDeformer( i , FbxDeformer::eSkin ) );
if ( !theSkin ) continue;
for ( int j = 0; j < theSkin->GetClusterCount(); ++j )
{
FbxCluster* cluster = theSkin->GetCluster( j );
std::string jointName = cluster->GetLink()->GetName();
int currentJointIndex = -1;
for ( unsigned int k = 0; k < skeleton.size(); ++k )
{
if ( !skeleton[k].name.compare( jointName ) )
{
currentJointIndex = k;
break;
}
}
if ( currentJointIndex < 0 )
{
std::cout << "wrong bone" << std::endl;
continue;
}
FbxAMatrix transformMatrix, transformLinkMatrix, offsetMatrix;
cluster->GetTransformMatrix( transformMatrix );
cluster->GetTransformLinkMatrix( transformLinkMatrix );
//offsetMatrix = transformLinkMatrix.Inverse() * transformMatrix * geoTransform;
FbxMatrix realMatrix( transformLinkMatrix.Inverse() * transformMatrix );
for ( unsigned int row = 0; row < 4; ++row )
{
for ( unsigned int column = 0; column < 4; ++column )
{
skeleton[currentJointIndex].offsetMatrix[row][column] = (float)realMatrix.Get( row , column );
}
}
for ( int k = 0; k < cluster->GetControlPointIndicesCount(); ++k )
{
BlendingIndexWeightPair weightPair;
weightPair.blendingIndex = currentJointIndex;
weightPair.blendingWeight = (float)cluster->GetControlPointWeights()[k];
unsigned int controlPoint = cluster->GetControlPointIndices()[k];
for ( unsigned int z = 0; z < indices.size(); ++z )
{
if ( indices[z].oldControlPoint == controlPoint )
{
if ( verts[indices[z].index].blendingInfo.size() > 4 )
{
std::cout << "Warning: vert has more than 4 bones connected, ignoring additional bone." << std::endl;
}
//else verts[indices[z].index].blendingInfo.push_back( weightPair );
bool found = false;
for ( unsigned int omg = 0; omg < verts[indices[z].index].blendingInfo.size(); ++omg )
{
if ( verts[indices[z].index].blendingInfo[omg].blendingIndex == weightPair.blendingIndex )
{
found = true;
break;
}
}
if ( !found ) verts[indices[z].index].blendingInfo.push_back( weightPair );
}
}
}
for ( unsigned int frameIndex = 0; frameIndex < frames.size(); ++frameIndex )
{
FbxVector4 translation = cluster->GetLink()->EvaluateLocalTranslation( frames[frameIndex] );
FbxVector4 rotation = cluster->GetLink()->EvaluateLocalRotation( frames[frameIndex] );
FbxVector4 scale = cluster->GetLink()->EvaluateLocalScaling( frames[frameIndex] );
AnimationData animData;
animData.frame = (int)frames[frameIndex].GetFrameCount();
animData.translation = glm::vec3(translation[0],translation[1],translation[2]);
animData.rotation = glm::angleAxis( glm::radians( (float)rotation[2] ), glm::vec3(0,0,1)) * glm::angleAxis(glm::radians((float) rotation[1]), glm::vec3(0,1,0)) * glm::angleAxis(glm::radians((float)rotation[0]), glm::vec3(1,0,0));
animData.scale = glm::vec3(scale[0],scale[1],scale[2]);
skeleton[currentJointIndex].animation.push_back( animData );
}
}
}
}
