mirror of
				https://github.com/f4exb/sdrangel.git
				synced 2025-10-31 04:50:29 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			334 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			334 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include "dsp/scopevis.h"
 | |
| #include "gui/glscope.h"
 | |
| #include "dsp/dspcommands.h"
 | |
| #include "util/messagequeue.h"
 | |
| #include <algorithm>
 | |
| 
 | |
| #include <QDebug>
 | |
| 
 | |
| MESSAGE_CLASS_DEFINITION(ScopeVis::MsgConfigureScopeVis, Message)
 | |
| 
 | |
| const uint ScopeVis::m_traceChunkSize = 4800;
 | |
| 
 | |
| ScopeVis::ScopeVis(GLScope* glScope) :
 | |
| 	m_glScope(glScope),
 | |
|     m_tracebackCount(0),
 | |
| 	m_fill(0),
 | |
| 	m_triggerState(Untriggered),
 | |
| 	m_triggerChannel(TriggerFreeRun),
 | |
| 	m_triggerLevel(0.0),
 | |
| 	m_triggerPositiveEdge(true),
 | |
| 	m_triggerBothEdges(false),
 | |
| 	m_triggerPre(0),
 | |
|     m_triggerDelay(0),
 | |
|     m_triggerDelayCount(0),
 | |
| 	m_triggerOneShot(false),
 | |
| 	m_armed(false),
 | |
| 	m_sampleRate(0)
 | |
| {
 | |
| 	setObjectName("ScopeVis");
 | |
| 	m_trace.reserve(100*m_traceChunkSize);
 | |
| 	m_trace.resize(20*m_traceChunkSize);
 | |
| 	m_traceback.resize(20*m_traceChunkSize);
 | |
| }
 | |
| 
 | |
| ScopeVis::~ScopeVis()
 | |
| {
 | |
| }
 | |
| 
 | |
| void ScopeVis::configure(MessageQueue* msgQueue, 
 | |
|     TriggerChannel triggerChannel, 
 | |
|     Real triggerLevel, 
 | |
|     bool triggerPositiveEdge, 
 | |
|     bool triggerBothEdges,
 | |
|     uint triggerPre,
 | |
|     uint triggerDelay,
 | |
|     uint traceSize)
 | |
| {
 | |
| 	Message* cmd = MsgConfigureScopeVis::create(triggerChannel,
 | |
| 			triggerLevel,
 | |
| 			triggerPositiveEdge,
 | |
| 			triggerBothEdges,
 | |
| 			triggerPre,
 | |
| 			triggerDelay,
 | |
| 			traceSize);
 | |
| 	msgQueue->push(cmd);
 | |
| }
 | |
| 
 | |
| void ScopeVis::feed(const SampleVector::const_iterator& cbegin, const SampleVector::const_iterator& end, bool positiveOnly)
 | |
| {
 | |
| 	SampleVector::const_iterator begin(cbegin);
 | |
| 
 | |
| 	if (m_triggerChannel == TriggerFreeRun) {
 | |
| 		m_triggerPoint = begin;
 | |
| 	}
 | |
| 	else if (m_triggerState == Triggered) {
 | |
| 		m_triggerPoint = begin;
 | |
| 	}
 | |
| 	else if (m_triggerState == Untriggered) {
 | |
| 		m_triggerPoint = end;
 | |
| 	}
 | |
| 	else if (m_triggerState == WaitForReset) {
 | |
| 		m_triggerPoint = end;
 | |
| 	}
 | |
| 	else {
 | |
| 		m_triggerPoint = begin;
 | |
| 	}
 | |
| 
 | |
| 	while(begin < end)
 | |
| 	{
 | |
| 		if (m_triggerChannel == TriggerFreeRun)
 | |
| 		{
 | |
| 			int count = end - begin;
 | |
| 
 | |
| 			if(count > (int)(m_trace.size() - m_fill))
 | |
| 			{
 | |
| 				count = m_trace.size() - m_fill;
 | |
| 			}
 | |
| 
 | |
| 			std::vector<Complex>::iterator it = m_trace.begin() + m_fill;
 | |
| 
 | |
| 			for(int i = 0; i < count; ++i)
 | |
| 			{
 | |
| 				*it++ = Complex(begin->real() / 32768.0f, begin->imag() / 32768.0f);
 | |
| 				++begin;
 | |
| 			}
 | |
| 
 | |
| 			m_fill += count;
 | |
| 
 | |
| 			if(m_fill >= m_trace.size())
 | |
| 			{
 | |
| 				m_glScope->newTrace(m_trace, m_sampleRate);
 | |
| 				m_fill = 0;
 | |
| 			}
 | |
| 		}
 | |
| 		else
 | |
| 		{
 | |
| 			if(m_triggerState == WaitForReset)
 | |
| 			{
 | |
| 				break;
 | |
| 			}
 | |
| 			if(m_triggerState == Config)
 | |
| 			{
 | |
|                 m_glScope->newTrace(m_trace, m_sampleRate); // send a dummy trace
 | |
| 				m_triggerState = Untriggered;
 | |
| 			}
 | |
|             if(m_triggerState == Delay)
 | |
|             {
 | |
|                 int count = end - begin;
 | |
| 				if (count > (int)(m_trace.size() - m_fill))
 | |
|                 {
 | |
| 					count = m_trace.size() - m_fill;
 | |
|                 }
 | |
|                 begin += count;
 | |
|                 m_fill += count;
 | |
| 				if(m_fill >= m_trace.size()) 
 | |
|                 {
 | |
|                     m_fill = 0;
 | |
|                     m_triggerDelayCount--;
 | |
|                     if (m_triggerDelayCount == 0)
 | |
|                     {
 | |
|                         m_triggerState = Triggered;
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
| 
 | |
| 			if(m_triggerState == Untriggered)
 | |
| 			{
 | |
| 				while(begin < end)
 | |
| 				{
 | |
|                     bool triggerCdt = triggerCondition(begin);
 | |
| 
 | |
|                     if (m_tracebackCount > m_triggerPre)
 | |
|                     {
 | |
|                         bool trigger;
 | |
| 
 | |
|                         if (m_triggerBothEdges) {
 | |
|                         	trigger = m_prevTrigger ^ triggerCdt;
 | |
|                         } else {
 | |
|                         	trigger = triggerCdt ^ !m_triggerPositiveEdge;
 | |
|                         }
 | |
| 
 | |
|                         if (trigger)
 | |
| 						{
 | |
| 							if (m_armed)
 | |
| 							{
 | |
| 								m_armed = false;
 | |
|                                 if (m_triggerDelay > 0)
 | |
|                                 {
 | |
|                                     m_triggerDelayCount = m_triggerDelay;
 | |
|                                     m_fill = 0;
 | |
|                                     m_triggerState = Delay;
 | |
|                                 }
 | |
|                                 else
 | |
|                                 {
 | |
|                                     m_triggerState = Triggered;
 | |
|                                     m_triggerPoint = begin;
 | |
|                                     // fill beginning of m_trace with delayed samples from the trace memory FIFO. Increment m_fill accordingly.
 | |
|                                     if (m_triggerPre) { // do this process only if there is a pre-trigger delay
 | |
|                                         std::copy(m_traceback.end() - m_triggerPre - 1, m_traceback.end() - 1, m_trace.begin());
 | |
|                                         m_fill = m_triggerPre; // Increment m_fill accordingly (from 0).
 | |
|                                     }
 | |
|                                 }
 | |
| 								break;
 | |
| 							}
 | |
| 						}
 | |
| 						else
 | |
| 						{
 | |
| 							m_armed = true;
 | |
| 						}
 | |
|                     }
 | |
|                     m_prevTrigger = triggerCdt;
 | |
| 					++begin;
 | |
| 				}
 | |
| 			}
 | |
| 
 | |
| 			if(m_triggerState == Triggered)
 | |
| 			{
 | |
| 				int count = end - begin;
 | |
| 
 | |
| 				if(count > (int)(m_trace.size() - m_fill))
 | |
| 				{
 | |
| 					count = m_trace.size() - m_fill;
 | |
| 				}
 | |
| 
 | |
| 				std::vector<Complex>::iterator it = m_trace.begin() + m_fill;
 | |
| 
 | |
| 				for(int i = 0; i < count; ++i)
 | |
| 				{
 | |
| 					*it++ = Complex(begin->real() / 32768.0f, begin->imag() / 32768.0f);
 | |
| 					++begin;
 | |
| 				}
 | |
| 
 | |
| 				m_fill += count;
 | |
| 
 | |
| 				if(m_fill >= m_trace.size())
 | |
| 				{
 | |
| 					m_glScope->newTrace(m_trace, m_sampleRate);
 | |
| 					m_fill = 0;
 | |
| 
 | |
| 					if (m_triggerOneShot) {
 | |
| 						m_triggerState = WaitForReset;
 | |
| 					} else {
 | |
|                         m_tracebackCount = 0;
 | |
| 						m_triggerState = Untriggered;
 | |
| 					}
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void ScopeVis::start()
 | |
| {
 | |
| }
 | |
| 
 | |
| void ScopeVis::stop()
 | |
| {
 | |
| }
 | |
| 
 | |
| bool ScopeVis::handleMessage(const Message& message)
 | |
| {
 | |
| 	qDebug() << "ScopeVis::handleMessage";
 | |
| 
 | |
| 	if (DSPSignalNotification::match(message))
 | |
| 	{
 | |
| 		DSPSignalNotification& notif = (DSPSignalNotification&) message;
 | |
| 		m_sampleRate = notif.getSampleRate();
 | |
| 		qDebug() << "  - DSPSignalNotification: m_sampleRate: " << m_sampleRate;
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 	else if (MsgConfigureScopeVis::match(message))
 | |
| 	{
 | |
| 		MsgConfigureScopeVis& conf = (MsgConfigureScopeVis&) message;
 | |
| 
 | |
| 		m_tracebackCount = 0;
 | |
| 		m_triggerState = Config;
 | |
| 		m_triggerChannel = (TriggerChannel) conf.getTriggerChannel();
 | |
| 		m_triggerLevel = conf.getTriggerLevel();
 | |
| 		m_triggerPositiveEdge = conf.getTriggerPositiveEdge();
 | |
| 		m_triggerBothEdges = conf.getTriggerBothEdges();
 | |
| 		m_triggerPre = conf.getTriggerPre();
 | |
| 
 | |
|         if (m_triggerPre >= m_traceback.size())
 | |
|         {
 | |
|         	m_triggerPre = m_traceback.size() - 1; // top sample in FIFO is always the triggering one (pre-trigger delay = 0)
 | |
|         }
 | |
| 
 | |
|         m_triggerDelay = conf.getTriggerDelay();
 | |
|         uint newSize = conf.getTraceSize();
 | |
| 
 | |
|         if (newSize != m_trace.size())
 | |
|         {
 | |
|             m_trace.resize(newSize);
 | |
|         }
 | |
| 
 | |
|         if (newSize > m_traceback.size()) // fitting the exact required space is not a requirement for the back trace
 | |
|         {
 | |
|             m_traceback.resize(newSize);
 | |
|         }
 | |
| 
 | |
| 		qDebug() << "  - MsgConfigureScopeVis:"
 | |
| 				<< " m_triggerChannel: " << m_triggerChannel
 | |
| 				<< " m_triggerLevel: " << m_triggerLevel
 | |
| 				<< " m_triggerPositiveEdge: " << (m_triggerPositiveEdge ? "edge+" : "edge-")
 | |
| 				<< " m_triggerBothEdges: " << (m_triggerBothEdges ? "yes" : "no")
 | |
| 				<< " m_preTrigger: " << m_triggerPre
 | |
| 				<< " m_triggerDelay: " << m_triggerDelay
 | |
| 				<< " m_traceSize: " << m_trace.size();
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 	else
 | |
| 	{
 | |
| 		return false;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void ScopeVis::setSampleRate(int sampleRate)
 | |
| {
 | |
| 	m_sampleRate = sampleRate;
 | |
| }
 | |
| 
 | |
| bool ScopeVis::triggerCondition(SampleVector::const_iterator& it)
 | |
| {
 | |
| 	Complex c(it->real()/32768.0f, it->imag()/32768.0f);
 | |
|     m_traceback.push_back(c); // store into trace memory FIFO
 | |
|     
 | |
|     if (m_tracebackCount < m_traceback.size()) { // increment count up to trace memory size
 | |
|         m_tracebackCount++;
 | |
|     }
 | |
|     
 | |
| 	if (m_triggerChannel == TriggerChannelI) {
 | |
| 		return c.real() > m_triggerLevel;
 | |
| 	}
 | |
| 	else if (m_triggerChannel == TriggerChannelQ) {
 | |
| 		return c.imag() > m_triggerLevel;
 | |
| 	}
 | |
| 	else if (m_triggerChannel == TriggerMagLin) {
 | |
| 		return abs(c) > m_triggerLevel;
 | |
| 	}
 | |
| 	else if (m_triggerChannel == TriggerMagDb) {
 | |
| 		Real mult = (10.0f / log2f(10.0f));
 | |
| 		Real v = c.real() * c.real() + c.imag() * c.imag();
 | |
| 		return mult * log2f(v) > m_triggerLevel;
 | |
| 	}
 | |
| 	else if (m_triggerChannel == TriggerPhase) {
 | |
| 		return arg(c) / M_PI > m_triggerLevel;
 | |
| 	}
 | |
| 	else {
 | |
| 		return false;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void ScopeVis::setOneShot(bool oneShot)
 | |
| {
 | |
| 	m_triggerOneShot = oneShot;
 | |
| 
 | |
| 	if ((m_triggerState == WaitForReset) && !oneShot) {
 | |
|         m_tracebackCount = 0;
 | |
| 		m_triggerState = Untriggered;
 | |
| 	}
 | |
| }
 |