GUI::GUI() {
IGTLinkStreamer::pointer streamer = IGTLinkStreamer::New();
setWidth(getScreenWidth());
setHeight(getScreenHeight());
enableMaximized();
setTitle("FAST Neural Network Despeckling");
auto resizer = ImageResizer::New();
resizer->setInputConnection(streamer->getOutputPort());
resizer->setWidth(512);
resizer->setHeight(512);
UltrasoundImageEnhancement::pointer enhancer1 = UltrasoundImageEnhancement::New();
enhancer1->setInputConnection(resizer->getOutputPort());
ImageRenderer::pointer renderer1 = ImageRenderer::New();
renderer1->addInputConnection(enhancer1->getOutputPort());
View* viewOrig = createView();
viewOrig->setBackgroundColor(Color::Black());
viewOrig->set2DMode();
viewOrig->addRenderer(renderer1);
ImageToImageNetwork::pointer network = ImageToImageNetwork::New();
network->load("/home/smistad/Downloads/filname_test.pb");
network->addOutputNode(0, "activation_8/Tanh:0", NodeType::TENSOR);
network->setSignedInputNormalization(true);
network->setScaleFactor(1.0f/255.0f);
network->setInputConnection(streamer->getOutputPort());
UltrasoundImageEnhancement::pointer enhancer2 = UltrasoundImageEnhancement::New();
enhancer2->setInputConnection(network->getOutputPort());
ImageRenderer::pointer renderer2 = ImageRenderer::New();
renderer2->addInputConnection(enhancer2->getOutputPort());
View* view = createView();
view->setBackgroundColor(Color::Black());
view->set2DMode();
view->addRenderer(renderer2);
QHBoxLayout* layout = new QHBoxLayout;
layout->addWidget(viewOrig);
layout->addWidget(view);
mWidget->setLayout(layout);
}
TabBarViewController::TabBarViewController() : ViewController()
{
_tabCount = 0;
View * masterView = getMasterView();
CGColor temp = CGColor();
temp.setColorWithHSB(0.0, 0.0, 0.0);
masterView->setBackgroundColor(temp);
}
GUI::GUI() {
mStreamingMode = STREAMING_MODE_STORE_ALL_FRAMES;
QScreen* screen = QGuiApplication::primaryScreen();
menuWidth = screen->geometry().width()*(1.0f/6.0f);
mPipelineWidget = nullptr;
mStreamer = ImageFileStreamer::New();
QVBoxLayout* viewLayout = new QVBoxLayout;
View* view = createView();
view->set2DMode();
view->setBackgroundColor(Color::Black());
setWidth(screen->geometry().width());
setHeight(screen->geometry().height());
enableMaximized();
setTitle("FAST - Viewer");
viewLayout->addWidget(view);
menuLayout = new QVBoxLayout;
menuLayout->setAlignment(Qt::AlignTop);
// Logo
QImage* image = new QImage;
image->load((Config::getDocumentationPath() + "images/FAST_logo_square.png").c_str());
QLabel* logo = new QLabel;
logo->setPixmap(QPixmap::fromImage(image->scaled(menuWidth, ((float)menuWidth/image->width())*image->height(), Qt::IgnoreAspectRatio, Qt::SmoothTransformation)));
logo->adjustSize();
menuLayout->addWidget(logo);
// Title label
QLabel* title = new QLabel;
title->setText("Viewer");
QFont font;
font.setPixelSize(24 * getScalingFactor());
font.setWeight(QFont::Bold);
title->setFont(font);
title->setAlignment(Qt::AlignCenter);
menuLayout->addWidget(title);
// Quit button
QPushButton* quitButton = new QPushButton;
quitButton->setText("Quit (q)");
quitButton->setStyleSheet("QPushButton { background-color: red; color: white; }");
quitButton->setFixedWidth(menuWidth);
menuLayout->addWidget(quitButton);
// Connect the clicked signal of the quit button to the stop method for the window
QObject::connect(quitButton, &QPushButton::clicked, std::bind(&Window::stop, this));
QLabel* inputListLabel = new QLabel;
//inputListLabel->setFixedHeight(24);
inputListLabel->setText("Input data");
inputListLabel->setStyleSheet("QLabel { font-weight: bold; }");
menuLayout->addWidget(inputListLabel);
mList = new QListWidget;
mList->setFixedWidth(menuWidth);
mList->setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Minimum);
mList->setFixedHeight(200);
mList->setSelectionMode(QAbstractItemView::ExtendedSelection); // Allow multiple items to be selected
QObject::connect(mList, &QListWidget::itemSelectionChanged, std::bind(&GUI::selectInputData, this));
menuLayout->addWidget(mList);
QPushButton* addButton = new QPushButton;
addButton->setText("Add input data");
addButton->setFixedWidth(menuWidth);
QObject::connect(addButton, &QPushButton::clicked, std::bind(&GUI::addInputData, this));
menuLayout->addWidget(addButton);
QLabel* selectPipelineLabel = new QLabel;
selectPipelineLabel->setText("Active pipeline");
selectPipelineLabel->setStyleSheet("QLabel { font-weight: bold; }");
//selectPipelineLabel->setFixedHeight(24);
menuLayout->addWidget(selectPipelineLabel);
mSelectPipeline = new QComboBox;
mSelectPipeline->setFixedWidth(menuWidth);
mPipelines = getAvailablePipelines();
int index = 0;
int counter = 0;
for(auto pipeline : mPipelines) {
mSelectPipeline->addItem((pipeline.getName() + " (" + pipeline.getDescription() + ")").c_str());
if(pipeline.getName() == "Image renderer") {
index = counter;
}
++counter;
}
mSelectPipeline->setCurrentIndex(index);
menuLayout->addWidget(mSelectPipeline);
QObject::connect(mSelectPipeline, static_cast<void (QComboBox::*)(int)>(&QComboBox::activated), std::bind(&GUI::selectPipeline, this));
QPushButton* refreshPipeline = new QPushButton;
refreshPipeline->setText("Refresh pipeline");
refreshPipeline->setStyleSheet("QPushButton { background-color: blue; color: white; }");
refreshPipeline->setFixedWidth(menuWidth);
QObject::connect(refreshPipeline, &QPushButton::clicked, std::bind(&GUI::selectPipeline, this));
menuLayout->addWidget(refreshPipeline);
QPushButton* editPipeline = new QPushButton;
editPipeline->setText("Edit pipeline");
editPipeline->setStyleSheet("QPushButton { background-color: blue; color: white; }");
editPipeline->setFixedWidth(menuWidth);
QObject::connect(editPipeline, &QPushButton::clicked, std::bind(&GUI::editPipeline, this));
menuLayout->addWidget(editPipeline);
QPushButton* newPipeline = new QPushButton;
newPipeline->setText("New pipeline");
newPipeline->setStyleSheet("QPushButton { background-color: blue; color: white; }");
newPipeline->setFixedWidth(menuWidth);
QObject::connect(newPipeline, &QPushButton::clicked, std::bind(&GUI::newPipeline, this));
menuLayout->addWidget(newPipeline);
// Playback
QHBoxLayout* playbackLayout = new QHBoxLayout;
mPlayPauseButton = new QPushButton;
mPlayPauseButton->setText("Play");
mPlayPauseButton->setStyleSheet("QPushButton { background-color: green; color: white; }");
//mPlayPauseButton->setFixedHeight(100);
QObject::connect(mPlayPauseButton, &QPushButton::clicked, std::bind(&GUI::playPause, this));
playbackLayout->addWidget(mPlayPauseButton);
mTimestepSlider = new QSlider(Qt::Horizontal);
// Improve style of slider
mTimestepSlider->setStyleSheet("QSlider:horizontal { min-height: 50px; } QSlider::groove:horizontal {\n"
" border: 1px solid #999999;\n"
" height: 10px; /* the groove expands to the size of the slider by default. by giving it a height, it has a fixed size */\n"
" background: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 #B1B1B1, stop:1 #c4c4c4);\n"
" margin: 2px 0;\n"
"}\n"
"\n"
"QSlider::handle:horizontal {\n"
" background: qlineargradient(x1:0, y1:0, x2:1, y2:1, stop:0 #81BEF7, stop:1 #2E9AFE);\n"
" border: 1px solid #5c5c5c;\n"
" width: 18px;\n"
" height: 25px;\n"
" margin: -10px 0; /* handle is placed by default on the contents rect of the groove. Expand outside the groove */\n"
" border-radius: 3px;\n"
"}");
playbackLayout->addWidget(mTimestepSlider);
mTimestepSlider->setTickPosition(QSlider::NoTicks);
mTimestepSlider->setRange(0, 1234);
mTimestepSlider->setPageStep(1);
mTimestepSlider->setSingleStep(1);
mTimestepSlider->setStyle(new MyStyle(mTimestepSlider->style())); // Fixes issues with direct jump with slider
QObject::connect(mTimestepSlider, &QSlider::sliderMoved, std::bind(&GUI::setTimestep, this));
QObject::connect(mTimestepSlider, &QSlider::sliderPressed, std::bind(&GUI::setTimestep, this));
viewLayout->addLayout(playbackLayout);
// Add menu and view to main layout
QHBoxLayout* layout = new QHBoxLayout;
layout->addLayout(menuLayout);
layout->addLayout(viewLayout);
mWidget->setLayout(layout);
std::cout << "Finished viewer setup" << std::endl;
}
