SDRPlusPlus/core/src/dsp/operator.h

66 wiersze
1.8 KiB
C++

#pragma once
#include "block.h"
namespace dsp {
template <class A, class B, class O>
class Operator : public block {
using base_type = block;
public:
Operator() {}
Operator(stream<A>* a, stream<B>* b) { init(a, b); }
virtual ~Operator() {}
virtual void init(stream<A>* a, stream<B>* b) {
_a = a;
_b = b;
base_type::registerInput(_a);
base_type::registerInput(_b);
base_type::registerOutput(&out);
base_type::_block_init = true;
}
virtual void setInputs(stream<A>* a, stream<B>* b) {
assert(_block_init);
std::lock_guard<std::recursive_mutex> lck(ctrlMtx);
base_type::tempStop();
base_type::unregisterInput(_a);
base_type::unregisterInput(_b);
_a = a;
_b = b;
base_type::registerInput(_a);
base_type::registerInput(_b);
base_type::tempStart();
}
virtual void setInputA(stream<A>* a) {
assert(_block_init);
std::lock_guard<std::recursive_mutex> lck(ctrlMtx);
base_type::tempStop();
base_type::unregisterInput(_a);
_a = a;
base_type::registerInput(_a);
base_type::tempStart();
}
virtual void setInputB(stream<B>* b) {
assert(_block_init);
std::lock_guard<std::recursive_mutex> lck(ctrlMtx);
base_type::tempStop();
base_type::unregisterInput(_b);
_b = b;
base_type::registerInput(_b);
base_type::tempStart();
}
virtual int run() = 0;
stream<O> out;
protected:
stream<A>* _a;
stream<B>* _b;
};
}