Option to apply calibration and de-embedding while/after importing

Software/PC_Application/Traces/sparamtraceselector.cpp

@@ -0,0 +1,172 @@
+#include "sparamtraceselector.h"
+
+#include <QFormLayout>
+#include <QLabel>
+
+using namespace std;
+
+SparamTraceSelector::SparamTraceSelector(const TraceModel &model, unsigned int num_ports, bool empty_allowed, std::set<unsigned int> skip)
+    : model(model),
+      num_ports(num_ports),
+      empty_allowed(empty_allowed)
+{
+    // Create comboboxes
+    auto layout = new QFormLayout;
+    setLayout(layout);
+    for(unsigned int i=0;i<num_ports;i++) {
+        for(unsigned int j=0;j<num_ports;j++) {
+            auto label = new QLabel("S"+QString::number(i+1)+QString::number(j+1)+":");
+            auto box = new QComboBox();
+            connect(box, qOverload<int>(&QComboBox::currentIndexChanged), [=](int) {
+               traceSelectionChanged(box);
+            });
+            boxes.push_back(box);
+            layout->addRow(label, box);
+            if(skip.count(i*num_ports + j)) {
+                label->setVisible(false);
+                box->setVisible(false);
+            }
+        }
+    }
+
+    setInitialChoices();
+}
+
+std::vector<Trace*> SparamTraceSelector::getTraces()
+{
+    vector<Trace*> ret;
+    for(auto b : boxes) {
+        if(b->currentIndex() == 0) {
+            ret.push_back(nullptr);
+        } else {
+            auto trace = qvariant_cast<Trace*>(b->itemData(b->currentIndex()));
+            ret.push_back(trace);
+        }
+    }
+    return ret;
+}
+
+void SparamTraceSelector::setInitialChoices()
+{
+    for(unsigned int i=0;i<num_ports*num_ports;i++) {
+        boxes[i]->blockSignals(true);
+        boxes[i]->clear();
+        boxes[i]->addItem("None");
+        for(auto t : model.getTraces()) {
+            if(t->outputType() != Trace::DataType::Frequency) {
+                // only frequency domain traces allowed
+                continue;
+            }
+            if(t->size() == 0) {
+                // can't select empty traces
+                continue;
+            }
+            bool reflectionRequired = i%(num_ports+1) == 0 ? true : false;
+            if(reflectionRequired != t->isReflection()) {
+                // invalid S parameter
+                continue;
+            }
+            boxes[i]->addItem(t->name(), QVariant::fromValue<Trace*>(t));
+        }
+        boxes[i]->blockSignals(false);
+    }
+    points = 0;
+    valid = empty_allowed;
+}
+
+void SparamTraceSelector::traceSelectionChanged(QComboBox *cb)
+{
+    // update available traces in combo boxes
+    if(cb->currentIndex() != 0 && points == 0) {
+        // the first trace has been selected, extract frequency info
+        Trace *t = qvariant_cast<Trace*>(cb->itemData(cb->currentIndex()));
+        points = t->size();
+        if(points > 0) {
+            minFreq = t->minX();
+            maxFreq = t->maxX();
+        }
+        // remove all trace options with incompatible frequencies
+        for(auto c : boxes) {
+            if(!c->isVisible()) {
+                // skip invisible selections
+                continue;
+            }
+            for(int i=1;i<c->count();i++) {
+                Trace *t = qvariant_cast<Trace*>(c->itemData(i));
+                if(t->size() != points || (points > 0 && (t->minX() != minFreq || t->maxX() != maxFreq))) {
+                    // this trace is not available anymore
+                    c->removeItem(i);
+                    // decrement to check the next index in the next loop iteration
+                    i--;
+                }
+            }
+        }
+        // try to find matching traces based on their names (convenient autoselection of matching traces)
+        auto text = cb->currentText();
+        text.chop(2);
+        if(text.endsWith("S")) {
+            // tracename ended in Sxx, probably other traces with matching prefix available
+            for(unsigned int i=0;i<num_ports*num_ports;i++) {
+                auto b = boxes[i];
+                if(b == cb) {
+                    // skip this box
+                    continue;
+                }
+                for(int j=0;j<b->count();j++) {
+                    auto candidate = b->itemText(j);
+                    // check if correct parameter
+                    QString expectedSparam = QString::number(i/num_ports+1)+QString::number(i%num_ports+1);
+                    if(!candidate.endsWith(expectedSparam)) {
+                        // wrong S parameter, skip
+                        continue;
+                    }
+                    candidate.chop(2);
+                    if(candidate != text) {
+                        // trace name does not match
+                        continue;
+                    }
+                    // possible match, select for this box
+                    b->setCurrentIndex(j);
+                    break;
+                }
+            }
+        }
+
+    } else if(cb->currentIndex() == 0 && points > 0) {
+        emit selectionValid(false);
+        // Check if all trace selections are set for none
+        for(auto c : boxes) {
+            if(!c->isVisible()) {
+                // skip invisible selections
+                continue;
+            }
+            if(c->currentIndex() != 0) {
+                // some trace is still selected, abort
+                return;
+            }
+        }
+        // all traces set for none
+        points = 0;
+        minFreq = 0;
+        maxFreq = 0;
+        setInitialChoices();
+    }
+    if(empty_allowed) {
+        // always valid
+        emit selectionValid(true);
+    } else {
+        // actually need to check
+        valid = true;
+        for(auto c : boxes) {
+            if(!c->isVisible()) {
+                // skip invisible selections
+                continue;
+            }
+            if (c->currentIndex() == 0) {
+                valid = false;
+                break;
+            }
+        }
+        emit selectionValid(valid);
+    }
+}