123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317 |
- #ifndef CNSTREAM_MODULE_HPP_
- #define CNSTREAM_MODULE_HPP_
- #include <cxxabi.h>
- #include <unistd.h>
- #include <atomic>
- #include <functional>
- #include <list>
- #include <memory>
- #include <string>
- #include <thread>
- #include <typeinfo>
- #include <unordered_map>
- #include <utility>
- #include <vector>
- #include "cnstream_common.hpp"
- #include "cnstream_config.hpp"
- #include "cnstream_eventbus.hpp"
- #include "cnstream_frame.hpp"
- #include "cnstream_logging.hpp"
- #include "private/cnstream_module_pri.hpp"
- #include "util/cnstream_queue.hpp"
- #include "util/cnstream_rwlock.hpp"
- namespace cnstream {
- class Pipeline;
- class ModuleProfiler;
- struct NodeContext;
- class IModuleObserver {
- public:
-
- virtual void notify(std::shared_ptr<CNFrameInfo> data) = 0;
-
- virtual ~IModuleObserver() = default;
- };
- class Module : private NonCopyable {
- public:
-
- explicit Module(const std::string &name) : name_(name) {}
-
- virtual ~Module();
-
- void SetObserver(IModuleObserver *observer) {
- RwLockWriteGuard guard(observer_lock_);
- observer_ = observer;
- }
-
- virtual bool Open(ModuleParamSet param_set) = 0;
-
- virtual void Close() = 0;
-
- virtual int Process(std::shared_ptr<CNFrameInfo> data) = 0;
-
- virtual void OnEos(const std::string &stream_id) {}
-
- inline std::string GetName() const { return name_; }
-
- bool PostEvent(EventType type, const std::string &msg);
-
- bool PostEvent(Event e);
-
- bool TransmitData(std::shared_ptr<CNFrameInfo> data);
-
- virtual bool CheckParamSet(const ModuleParamSet ¶mSet) const { return true; }
-
- Pipeline* GetContainer() const { return container_; }
-
- ModuleProfiler* GetProfiler();
-
- bool HasTransmit() const { return hasTransmit_.load(); }
-
- ParamRegister param_register_;
- #ifdef UNIT_TEST
- public:
- #else
- protected:
- #endif
- friend class Pipeline;
- friend class CNFrameInfo;
-
- void SetContainer(Pipeline *container);
-
- int DoProcess(std::shared_ptr<CNFrameInfo> data);
- Pipeline *container_ = nullptr;
- RwLock container_lock_;
- std::string name_;
- std::atomic<bool> hasTransmit_{false};
- #ifdef UNIT_TEST
- public:
- #else
- private:
- #endif
- IModuleObserver *observer_ = nullptr;
- RwLock observer_lock_;
- void NotifyObserver(std::shared_ptr<CNFrameInfo> data) {
- RwLockReadGuard guard(observer_lock_);
- if (observer_) {
- observer_->notify(data);
- }
- }
- int DoTransmitData(std::shared_ptr<CNFrameInfo> data);
- size_t GetId();
- size_t id_ = INVALID_MODULE_ID;
- NodeContext* context_ = nullptr;
- };
- class ModuleEx : public Module {
- public:
-
- explicit ModuleEx(const std::string &name) : Module(name) { hasTransmit_.store(true); }
- };
- }
- #endif
|