forked from microsoft/onnxruntime-extensions
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
21 changed files
with
1,070 additions
and
328 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,97 @@ | ||
| // Copyright (c) Microsoft Corporation. All rights reserved. | ||
| // Licensed under the MIT License. | ||
|
|
||
| #pragma once | ||
|
|
||
| #include "ortx_utils.h" | ||
|
|
||
| namespace ort_extensions { | ||
|
|
||
| template <typename T> | ||
| class OrtxDeleter { | ||
| public: | ||
| void operator()(T* p) const { | ||
| if (p) { | ||
| OrtxDisposeOnly(p); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| /** | ||
| * @brief A smart pointer class that manages the lifetime of an OrtxObject. | ||
| * | ||
| * This class is derived from std::unique_ptr and provides additional functionality | ||
| * specific to OrtxObject. It automatically calls the OrtxDeleter to release the | ||
| * owned object when it goes out of scope. | ||
| * | ||
| * @tparam T The type of the object being managed. | ||
| */ | ||
| template <typename T> | ||
| class OrtxObjectPtr : public std::unique_ptr<T, OrtxDeleter<T>> { | ||
| public: | ||
| /** | ||
| * @brief Default constructor. | ||
| * | ||
| * Constructs an OrtxObjectPtr with a null pointer. | ||
| */ | ||
| OrtxObjectPtr() : std::unique_ptr<T, OrtxDeleter<T>>(nullptr) {} | ||
|
|
||
| /** | ||
| * @brief Constructor that creates an OrtxObjectPtr from a function call. | ||
| * | ||
| * This constructor calls the specified function with the given arguments to | ||
| * create an OrtxObject. If the function call succeeds, the created object is | ||
| * owned by the OrtxObjectPtr. | ||
| * | ||
| * @tparam TFn The type of the function pointer or function object. | ||
| * @tparam Args The types of the arguments to be passed to the function. | ||
| * @param fn The function pointer or function object used to create the OrtxObject. | ||
| * @param args The arguments to be passed to the function. | ||
| */ | ||
| template <typename TFn, typename... Args> | ||
| OrtxObjectPtr(TFn fn, Args&&... args) { | ||
| OrtxObject* proc = nullptr; | ||
| err_ = fn(&proc, std::forward<Args>(args)...); | ||
| if (err_ == kOrtxOK) { | ||
| this->reset(static_cast<T*>(proc)); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * @brief Get the error code associated with the creation of the OrtxObject. | ||
| * | ||
| * @return The error code. | ||
| */ | ||
| extError_t Code() const { return err_; } | ||
|
|
||
| private: | ||
| extError_t err_ = kOrtxOK; /**< The error code associated with the creation of the OrtxObject. */ | ||
| }; | ||
|
|
||
| template <typename T> | ||
| struct PointerAssigner { | ||
| OrtxObject* obj_{}; | ||
| OrtxObjectPtr<T>& ptr_; | ||
| PointerAssigner(OrtxObjectPtr<T>& ptr) : ptr_(ptr){}; | ||
|
|
||
| ~PointerAssigner() { ptr_.reset(static_cast<T*>(obj_)); }; | ||
|
|
||
| operator T**() { return reinterpret_cast<T**>(&obj_); }; | ||
| }; | ||
|
|
||
| /** | ||
| * @brief A wrapper function for OrtxObjectPtr that can be used as a function parameter on creation. | ||
| * | ||
| * This function creates a PointerAssigner object for the given OrtxObjectPtr. The PointerAssigner | ||
| * object can be used to assign a pointer value to the OrtxObjectPtr. | ||
| * | ||
| * @tparam T The type of the object pointed to by the OrtxObjectPtr. | ||
| * @param ptr The OrtxObjectPtr to create the PointerAssigner for. | ||
| * @return A PointerAssigner object for the given OrtxObjectPtr. | ||
| */ | ||
| template <typename T> | ||
| PointerAssigner<T> ptr(OrtxObjectPtr<T>& ptr) { | ||
| return PointerAssigner<T>{ptr}; | ||
| }; | ||
|
|
||
| } // namespace ort_extensions |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.