onnxruntime/winml/lib/Api/ImageFeatureDescriptor.cpp
Sheil Kumar 2717c178cc
Fork the WinML APIs into the Microsoft namespace (#3503)
* Migrate winml to Microsoft Namespace (packaging changes are pending)

* add ns_prefix toggle

* fix packaging

* Users/sheilk/add missing raw header (#3484)

* add dualapipartition

* wrong variable for repo root

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>

* remove existence check to force failures

* extra paren

* dualapipartition needs to be referenced from the source

* add microsoft.ai.machinelearning.dll to the output dir

* rename the idl file so that assembly info is correctly added into the winmd

* fix namespaces

* update namespaces

* default to microsoft, and add namespace override as build argument

* update cmakesetings.json as well

* remove from cmakelists.txt

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
Co-authored-by: Changming Sun <chasun@microsoft.com>
2020-04-17 06:18:54 -07:00

123 lines
3.1 KiB
C++

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#include "pch.h"
#include "ImageFeatureDescriptor.h"
#include <windows.graphics.imaging.h>
namespace WINMLP {
ImageFeatureDescriptor::ImageFeatureDescriptor(
const char* name,
const char* description,
winml::TensorKind tensor_kind,
const std::vector<int64_t>& shape,
bool is_required,
wgi::BitmapPixelFormat pixel_format,
wgi::BitmapAlphaMode alpha_mode,
uint32_t width,
uint32_t height,
ImageNominalPixelRange nominal_pixel_range,
ImageColorSpaceGamma color_space_gamma) : name_(_winml::Strings::HStringFromUTF8(name)),
description_(_winml::Strings::HStringFromUTF8(description)),
tensor_kind_(tensor_kind),
shape_(shape),
is_required_(is_required),
pixel_format_(pixel_format),
alpha_mode_(alpha_mode),
width_(width),
height_(height),
nominal_pixel_range_(nominal_pixel_range),
color_space_gamma_(color_space_gamma) {
}
wgi::BitmapPixelFormat
ImageFeatureDescriptor::BitmapPixelFormat() try {
return pixel_format_;
}
WINML_CATCH_ALL
wgi::BitmapAlphaMode
ImageFeatureDescriptor::BitmapAlphaMode() try {
return alpha_mode_;
}
WINML_CATCH_ALL
uint32_t
ImageFeatureDescriptor::Width() try {
return width_;
}
WINML_CATCH_ALL
uint32_t
ImageFeatureDescriptor::Height() try {
return height_;
}
WINML_CATCH_ALL
hstring
ImageFeatureDescriptor::Name() try {
return name_;
}
WINML_CATCH_ALL
hstring
ImageFeatureDescriptor::Description() try {
return description_;
}
WINML_CATCH_ALL
winml::LearningModelFeatureKind
ImageFeatureDescriptor::Kind() try {
return LearningModelFeatureKind::Image;
}
WINML_CATCH_ALL
bool ImageFeatureDescriptor::IsRequired() try {
return is_required_;
}
WINML_CATCH_ALL
winml::TensorKind
ImageFeatureDescriptor::TensorKind() {
return tensor_kind_;
}
wfc::IVectorView<int64_t>
ImageFeatureDescriptor::Shape() {
return winrt::single_threaded_vector<int64_t>(
std::vector<int64_t>(
std::begin(shape_),
std::end(shape_)))
.GetView();
}
HRESULT
ImageFeatureDescriptor::GetName(
const wchar_t** name,
uint32_t* cchName) {
*name = name_.data();
*cchName = static_cast<uint32_t>(name_.size());
return S_OK;
}
HRESULT
ImageFeatureDescriptor::GetDescription(
const wchar_t** description,
uint32_t* cchDescription) {
*description = description_.data();
*cchDescription = static_cast<uint32_t>(description_.size());
return S_OK;
}
ImageNominalPixelRange
ImageFeatureDescriptor::GetNominalPixelRange() {
return nominal_pixel_range_;
}
ImageColorSpaceGamma
ImageFeatureDescriptor::GetColorSpaceGamma() {
return color_space_gamma_;
}
} // namespace WINMLP