52 lines
1.1 KiB
C++
52 lines
1.1 KiB
C++
#ifndef OPENMANUS_TOOL_COLLECTION_H
|
||
#define OPENMANUS_TOOL_COLLECTION_H
|
||
|
||
#include <string>
|
||
#include <vector>
|
||
#include <memory>
|
||
#include <unordered_map>
|
||
#include "tool_base.h"
|
||
|
||
namespace openmanus {
|
||
|
||
/**
|
||
* @class ToolCollection
|
||
* @brief 工具集合类，管理所有可用的工具
|
||
*/
|
||
class ToolCollection {
|
||
public:
|
||
ToolCollection();
|
||
~ToolCollection() = default;
|
||
|
||
/**
|
||
* @brief 添加工具到集合中
|
||
* @param tool 要添加的工具
|
||
*/
|
||
void addTool(std::unique_ptr<ToolBase> tool);
|
||
|
||
/**
|
||
* @brief 根据名称获取工具
|
||
* @param name 工具名称
|
||
* @return 工具指针，如果不存在则返回nullptr
|
||
*/
|
||
ToolBase* getTool(const std::string& name);
|
||
|
||
/**
|
||
* @brief 获取所有工具的名称
|
||
* @return 工具名称列表
|
||
*/
|
||
std::vector<std::string> getToolNames() const;
|
||
|
||
/**
|
||
* @brief 获取工具数量
|
||
* @return 工具数量
|
||
*/
|
||
size_t size() const { return tools_.size(); }
|
||
|
||
private:
|
||
std::unordered_map<std::string, std::unique_ptr<ToolBase>> tools_;
|
||
};
|
||
|
||
} // namespace openmanus
|
||
|
||
#endif // OPENMANUS_TOOL_COLLECTION_H
|