You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
			
				
					102 lines
				
				2.7 KiB
			
		
		
			
		
	
	
					102 lines
				
				2.7 KiB
			| 
								 
											6 years ago
										 
									 | 
							
								//==============================================================================
							 | 
						||
| 
								 | 
							
								//
							 | 
						||
| 
								 | 
							
								//  Copyright (c) 2019 Qualcomm Technologies, Inc.
							 | 
						||
| 
								 | 
							
								//  All Rights Reserved.
							 | 
						||
| 
								 | 
							
								//  Confidential and Proprietary - Qualcomm Technologies, Inc.
							 | 
						||
| 
								 | 
							
								//
							 | 
						||
| 
								 | 
							
								//==============================================================================
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#ifndef PSNPE_APPLICATIONBUFFERMAP_HPP
							 | 
						||
| 
								 | 
							
								#define PSNPE_APPLICATIONBUFFERMAP_HPP
							 | 
						||
| 
								 | 
							
								#include <vector>
							 | 
						||
| 
								 | 
							
								#include <string>
							 | 
						||
| 
								 | 
							
								#include <unordered_map>
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#include "DlSystem/UserBufferMap.hpp"
							 | 
						||
| 
								 | 
							
								#include "DlSystem/ZdlExportDefine.hpp"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								namespace zdl
							 | 
						||
| 
								 | 
							
								{
							 | 
						||
| 
								 | 
							
								namespace PSNPE
							 | 
						||
| 
								 | 
							
								{
							 | 
						||
| 
								 | 
							
								/** @addtogroup c_plus_plus_apis C++
							 | 
						||
| 
								 | 
							
								@{ */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/**
							 | 
						||
| 
								 | 
							
								 * @brief .
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * A class representing the UserBufferMap of Input and Output asynchronous mode.
							 | 
						||
| 
								 | 
							
								 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								class ZDL_EXPORT ApplicationBufferMap final
							 | 
						||
| 
								 | 
							
								{
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								 public:
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief Adds a name and the corresponding buffer
							 | 
						||
| 
								 | 
							
								    *        to the map
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @param[in] name The name of the UserBuffer
							 | 
						||
| 
								 | 
							
								    * @param[in] buffer The vector of the uint8_t data
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @note If a UserBuffer with the same name already exists, the new
							 | 
						||
| 
								 | 
							
								    *       UserBuffer pointer would be updated.
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   void add(const char* name, std::vector<uint8_t>& buff) noexcept;
							 | 
						||
| 
								 | 
							
								   void add(const char* name, std::vector<float>& buff) noexcept;
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief Removes a mapping of one UserBuffer and its name by its name
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @param[in] name The name of UserBuffer to be removed
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @note If no UserBuffer with the specified name is found, nothing
							 | 
						||
| 
								 | 
							
								    *       is done.
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   void remove(const char* name) noexcept;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief Returns the number of UserBuffers in the map
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   size_t size() const noexcept;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief .
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * Removes all UserBuffers from the map
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   void clear() noexcept;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief Returns the UserBuffer given its name.
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @param[in] name The name of the UserBuffer to get.
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @return nullptr if no UserBuffer with the specified name is
							 | 
						||
| 
								 | 
							
								    *         found; otherwise, a valid pointer to the UserBuffer.
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   const std::vector<uint8_t>& getUserBuffer(const char* name) const;
							 | 
						||
| 
								 | 
							
								   const std::vector<uint8_t>& operator[](const char* name) const;
							 | 
						||
| 
								 | 
							
								   /**
							 | 
						||
| 
								 | 
							
								    * @brief .
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * Returns the names of all UserAsyncBufferMap
							 | 
						||
| 
								 | 
							
								    *
							 | 
						||
| 
								 | 
							
								    * @return A list of UserBuffer names.
							 | 
						||
| 
								 | 
							
								    */
							 | 
						||
| 
								 | 
							
								   zdl::DlSystem::StringList getUserBufferNames() const;
							 | 
						||
| 
								 | 
							
								   const std::unordered_map<std::string, std::vector<uint8_t>>& getUserBuffer() const;
							 | 
						||
| 
								 | 
							
								   explicit ApplicationBufferMap();
							 | 
						||
| 
								 | 
							
								   ~ApplicationBufferMap();
							 | 
						||
| 
								 | 
							
								   explicit ApplicationBufferMap(
							 | 
						||
| 
								 | 
							
								     const std::unordered_map<std::string, std::vector<uint8_t>> buffer);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								 private:
							 | 
						||
| 
								 | 
							
								   std::unordered_map<std::string, std::vector<uint8_t>> m_UserMap;
							 | 
						||
| 
								 | 
							
								};
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/** @} */ /* end_addtogroup c_plus_plus_apis C++ */
							 | 
						||
| 
								 | 
							
								} // namespace PSNPE
							 | 
						||
| 
								 | 
							
								} // namespace zdl
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#endif // PSNPE_APPLICATIONBUFFERMAP_HPP
							 |