X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=pv%2Fdata%2Fmathsignal.hpp;h=76427c56e73079a4757d9b468e32b947478bd0a7;hb=36e62b1700d580adde53c8f4489037f0071fa6a0;hp=04380cdc813f5d734f9dcaf7c356c0de5cb41747;hpb=b0773a8aa01735d7220284ab7a3e8b5d02b48e9e;p=pulseview.git

diff --git a/pv/data/mathsignal.hpp b/pv/data/mathsignal.hpp
index 04380cdc..76427c56 100644
--- a/pv/data/mathsignal.hpp
+++ b/pv/data/mathsignal.hpp
@@ -20,13 +20,19 @@
 #ifndef PULSEVIEW_PV_DATA_MATHSIGNAL_HPP
 #define PULSEVIEW_PV_DATA_MATHSIGNAL_HPP
 
-#include <QSettings>
+#include <limits>
+
 #include <QString>
 
+#include <pv/exprtk.hpp>
+#include <pv/util.hpp>
 #include <pv/data/analog.hpp>
 #include <pv/data/signalbase.hpp>
-#include <pv/util.hpp>
 
+using std::atomic;
+using std::condition_variable;
+using std::mutex;
+using std::numeric_limits;
 using std::shared_ptr;
 
 namespace pv {
@@ -35,11 +41,25 @@ class Session;
 namespace data {
 
 class SignalBase;
-class SignalData;
+
+template<typename T>
+struct fnc_sample;
+
+struct signal_data {
+	signal_data(const shared_ptr<SignalBase> _sb) :
+		sb(_sb), sample_num(numeric_limits<uint64_t>::max()), sample_value(0), ref(nullptr)
+	{}
+
+	const shared_ptr<SignalBase> sb;
+	uint64_t sample_num;
+	double sample_value;
+	double* ref;
+};
 
 class MathSignal : public SignalBase
 {
 	Q_OBJECT
+	Q_PROPERTY(QString expression READ get_expression WRITE set_expression NOTIFY expression_changed)
 
 private:
 	static const int64_t ChunkLength;
@@ -51,6 +71,35 @@ public:
 	virtual void save_settings(QSettings &settings) const;
 	virtual void restore_settings(QSettings &settings);
 
+	QString get_expression() const;
+	void set_expression(QString expression);
+
+private:
+	void set_error(uint8_t type, QString msg);
+
+	/**
+	 * Returns the number of samples that can be worked on,
+	 * i.e. the number of samples where samples are available
+	 * for all connected channels.
+	 * If the math signal uses no input channels, this is the
+	 * number of samples in the session.
+	 */
+	uint64_t get_working_sample_count(uint32_t segment_id) const;
+
+	void update_completeness(uint32_t segment_id, uint64_t output_sample_count);
+
+	void reset_generation();
+	virtual void begin_generation();
+
+	void generate_samples(uint32_t segment_id, const uint64_t start_sample,
+		const int64_t sample_count);
+	void generation_proc();
+
+	signal_data* signal_from_name(const std::string& name);
+	void update_signal_sample(signal_data* sig_data, uint32_t segment_id, uint64_t sample_num);
+
+	bool all_input_signals_enabled(QString &disabled_signals) const;
+
 Q_SIGNALS:
 	void samples_cleared();
 
@@ -59,19 +108,41 @@ Q_SIGNALS:
 
 	void min_max_changed(float min, float max);
 
-//private Q_SLOTS:
-//	void on_data_cleared();
-//	void on_data_received();
+	void expression_changed(QString expression);
 
-//	void on_samples_added(SharedPtrToSegment segment, uint64_t start_sample,
-//		uint64_t end_sample);
+private Q_SLOTS:
+	void on_capture_state_changed(int state);
+	void on_data_received();
 
-//	void on_min_max_changed(float min, float max);
+	void on_enabled_changed();
 
 private:
 	pv::Session &session_;
 
-	QString error_message_;
+	uint64_t custom_sample_rate_;
+	uint64_t custom_sample_count_;
+	bool use_custom_sample_rate_, use_custom_sample_count_;
+	map<std::string, signal_data> input_signals_;
+
+	QString expression_;
+
+	uint8_t error_type_;
+
+	mutable mutex input_mutex_;
+	mutable condition_variable gen_input_cond_;
+
+	std::thread gen_thread_;
+	atomic<bool> gen_interrupt_;
+
+	exprtk::symbol_table<double> *exprtk_unknown_symbol_table_, *exprtk_symbol_table_;
+	exprtk::expression<double> *exprtk_expression_;
+	exprtk::parser<double> *exprtk_parser_;
+	double exprtk_current_time_, exprtk_current_sample_;
+
+	fnc_sample<double>* fnc_sample_;
+
+	// Give sig_sample access to the private helper functions
+	friend struct fnc_sample<double>;
 };
 
 } // namespace data