From f4ff104c0dcbc2937f9a7110a30680461d6da89c Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@irstea.fr>
Date: Sun, 2 Sep 2018 13:09:51 +0000
Subject: [PATCH] ENH: expose tiling layout (tile hint) in output image
metadata
---
include/otbTensorflowMultisourceModelFilter.h | 4 ++++
include/otbTensorflowMultisourceModelFilter.hxx | 7 +++++++
2 files changed, 11 insertions(+)
diff --git a/include/otbTensorflowMultisourceModelFilter.h b/include/otbTensorflowMultisourceModelFilter.h
index 37be3aa..04a952b 100644
--- a/include/otbTensorflowMultisourceModelFilter.h
+++ b/include/otbTensorflowMultisourceModelFilter.h
@@ -21,6 +21,10 @@
#include "otbTensorflowDataTypeBridge.h"
#include "otbTensorflowCopyUtils.h"
+// Tile hint
+#include "itkMetaDataObject.h"
+#include "otbMetaDataKey.h"
+
namespace otb
{
diff --git a/include/otbTensorflowMultisourceModelFilter.hxx b/include/otbTensorflowMultisourceModelFilter.hxx
index e00be69..9aaeeda 100644
--- a/include/otbTensorflowMultisourceModelFilter.hxx
+++ b/include/otbTensorflowMultisourceModelFilter.hxx
@@ -288,6 +288,13 @@ TensorflowMultisourceModelFilter<TInputImage, TOutputImage>
m_NullPixel.SetSize(outputPtr->GetNumberOfComponentsPerPixel());
m_NullPixel.Fill(0);
+ //////////////////////////////////////////////////////////////////////////////////////////
+ // Set the tiling layout hint in metadata
+ //////////////////////////////////////////////////////////////////////////////////////////
+
+ itk::EncapsulateMetaData(outputPtr->GetMetaDataDictionary(), MetaDataKey::TileHintX, m_OutputGridSize[0]);
+ itk::EncapsulateMetaData(outputPtr->GetMetaDataDictionary(), MetaDataKey::TileHintY, m_OutputGridSize[1]);
+
}
template <class TInputImage, class TOutputImage>
--
GitLab