mainguiwindow.cpp 31.2 KB
Newer Older
roangel's avatar
roangel committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
//    Teacher's GUI main window header.
//    Copyright (C) 2017  Angel Romero
//
//    This program is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    This program is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with this program.  If not, see <http://www.gnu.org/licenses/>.

17
#include "mainguiwindow.h"
18
#include "ui_mainguiwindow.h"
19
#include "crazyFlyZoneTab.h"
20
21
#include "myGraphicsScene.h"
#include "myGraphicsView.h"
22
#include "channelLUT.h"
23
24
25
26
27

#include <QObject>
#include <QDoubleSpinBox>
#include <QTextEdit>
#include <QString>
28
#include <QMetaType>
29
#include <QDir>
30
#include <regex>
31

32
33
#ifdef CATKIN_MAKE
#include "d_fall_pps/UnlabeledMarker.h"
34
#include "d_fall_pps/CMRead.h"
35
#include "d_fall_pps/CrazyflieEntry.h"
36
37
38
#include "d_fall_pps/CMUpdate.h"
#include "d_fall_pps/CMCommand.h"
#include "CentralManagerService.h"
39
40
41

#include <ros/ros.h>
#include <ros/network.h>
42
#include <ros/package.h>
43

44
45
#endif

46
#include <string>
47

48
#define N_MAX_CRAZYFLIES           20 // protection number
49

tiagos's avatar
tiagos committed
50
51
52
53
#define UWB_UPDATE_DISABLE      0
#define UWB_UPDATE_ENABLE       1
#define UWB_UPDATE_ANCHORS      5

roangel's avatar
roangel committed
54
#ifdef CATKIN_MAKE
55
using namespace d_fall_pps;
roangel's avatar
roangel committed
56
#endif
57

58
59
60
61
MainGUIWindow::MainGUIWindow(int argc, char **argv, QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainGUIWindow)
{
62
    #ifdef CATKIN_MAKE
63
    _rosNodeThread = new rosNodeThread(argc, argv, "my_GUI");
64
    #endif
65
66
67
    ui->setupUi(this);
    _init();
}
68

69

70
71
72
73
74
MainGUIWindow::~MainGUIWindow()
{
    delete ui;
}

75
76
77
78
79
80
81
82
83
84
85
86
87
88
int MainGUIWindow::getTabIndexFromName(QString name)
{
    int found_name = -1;
    for(int i = 0; i < ui->tabWidget->count(); i++)
    {
        qDebug("name: %s", name.toStdString().c_str());
        qDebug("tabText: %s", ui->tabWidget->tabText(i).toStdString().c_str());
        if(name == ui->tabWidget->tabText(i))
        {
            found_name = i;
        }
    }
    return found_name;
}
89

roangel's avatar
roangel committed
90
void MainGUIWindow::doNumCrazyFlyZonesChanged(int n)
91
{
roangel's avatar
roangel committed
92
    // tabs number management, maybe do it in a different way so we dont have to remove and add everything?
93
    // first check if size of tabs is greater than size of vector or viceversa. Have we removed or added a zone?
94

95
    qDebug("tabWidgetCount : %d", ui->tabWidget->count());
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
    if(ui->tabWidget->count() > scene->crazyfly_zones.size())
    {
        // we removed one crazyfly_zone, n means index of the one we removed. Look for that index tab and remove it
        QString qstr = "CrazyFly ";
        qstr.append(QString::number(n+1));
        if(scene->crazyfly_zones.size() == 0)
        {
            ui->tabWidget->clear();
        }
        int found_index = getTabIndexFromName(qstr);
        if(found_index != -1)
        {
            ui->tabWidget->widget(found_index)->deleteLater();
            // ui->tabWidget->removeTab(found_index);
        }

        //  now unlink it from table also:
        #ifdef CATKIN_MAKE
        if(cf_linker->isCFZoneLinked(n))
        {
            cf_linker->unlink_cf_zone(n);
        }
        #endif
    }
    else if(ui->tabWidget->count() < scene->crazyfly_zones.size())
    {
        // we added one crazyfly_zone, n means index of the new one. New tab will be labeld index + 1
        QString qstr = "CrazyFly ";
        qstr.append(QString::number(n+1));
        crazyFlyZoneTab* widget = new crazyFlyZoneTab(n);
        ui->tabWidget->insertTab(n, widget, qstr);
        connect(widget, SIGNAL(centerButtonClickedSignal(int)), this, SLOT(centerViewIndex(int)));
    }
129

130
    updateComboBoxesCFZones();
131
132
133
134
}

void MainGUIWindow::_init()
{
135
136
137
138
139
140
141
142
    // initialize checkboxes, spinboxes,....
    ui->scaleSpinBox->setRange(0.1, 100);
    ui->scaleSpinBox->setSingleStep(0.1);
    ui->scaleSpinBox->setValue(1);

    ui->checkBox_vicon_crazyflies->setChecked(false);
    ui->scaleSpinBox->setEnabled(false);

143

144
    ui->graphicsView->setViewportUpdateMode(QGraphicsView::FullViewportUpdate);
145
146
    ui->graphicsView->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOn);
    ui->graphicsView->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOn);
147

148
149
150
151
152
    // error messages
    // ui->err_message_cf->hide();
    // ui->err_message_cf_zone->hide();
    // ui->err_message_student_id->hide();

153
154
155
156
157
158
159
    ui->radioAddress_text->setReadOnly(true);

    QPalette *palette = new QPalette();
    palette->setColor(QPalette::Base,Qt::lightGray);
    palette->setColor(QPalette::Text,Qt::darkGray);
    ui->radioAddress_text->setPalette(*palette);

160
161
162
163
    ui->err_message_cf->setStyleSheet("QLabel { color : red; }");
    ui->err_message_cf_zone->setStyleSheet("QLabel { color : red; }");
    ui->err_message_student_id->setStyleSheet("QLabel { color : red; }");
    ui->err_message_radio_address->setStyleSheet("QLabel { color : red; }");
164

165
166
167
168
    ui->err_message_cf->clear();
    ui->err_message_cf_zone->clear();
    ui->err_message_student_id->clear();
    ui->err_message_radio_address->clear();
169

170
    // initialize table_links
171
    ui->table_links->setColumnCount(4);
172

173
174
175
176
177
    QFont fnt;
    fnt.setPointSize(7);
    ui->table_links->horizontalHeader()->setFont(fnt);

    ui->table_links->horizontalHeader()->setDefaultSectionSize(90);
178
    ui->table_links->verticalHeader()->setDefaultSectionSize(20);
179
180
181
182
183
184
185
186
187
188
189

    const int rowCount = ui->table_links->rowCount();
    const int columnCount = ui->table_links->columnCount();
    for(int i = 0; i < rowCount; ++i)
    {
    	for(int j = 0; j < columnCount; ++j)
        {
    		QTableWidgetItem* selectedItem = ui->table_links->item(i, j);
    		selectedItem->setFont(fnt);
    	}
    }
roangel's avatar
roangel committed
190
    ui->table_links->setSelectionBehavior(QAbstractItemView::SelectRows);
191
    QStringList horizontal_header;
192
    horizontal_header << "Student ID" << "CrazyFly" << "CrazyFly Zone" << "Radio Address";
193
    ui->table_links->setHorizontalHeaderLabels(horizontal_header);
194

195
    // scene
196
    scene = new myGraphicsScene(ui->frame_drawing);
197
    scene->setSceneRect(-100 * FROM_METERS_TO_UNITS, -100 * FROM_METERS_TO_UNITS, 200 * FROM_METERS_TO_UNITS, 200 * FROM_METERS_TO_UNITS);
198

199
    ui->graphicsView->setScene(scene);
200

201
    // after scene is created, create CFlinker
202
    #ifdef CATKIN_MAKE
203
    cf_linker = new CFLinker(ui, &crazyflies_vector, &scene->crazyfly_zones);
204
    #endif
205
    // connections
206
    QObject::connect(ui->tabWidget, SIGNAL(tabCloseRequested(int)), this, SLOT(doTabClosed(int)));
roangel's avatar
roangel committed
207
    QObject::connect(scene, SIGNAL(numCrazyFlyZonesChanged(int)), this, SLOT(doNumCrazyFlyZonesChanged(int)));
208
    QObject::connect(scene, SIGNAL(crazyFlyZoneSelected(int)), this, SLOT(setTabIndex(int)));
209
210
    QObject::connect(scene, SIGNAL(modeChanged(int)), this, SLOT(transitionToMode(int)));
    QObject::connect(scene, SIGNAL(numTablePiecesChanged(int)), this, SLOT(handleTablePiecesNumChanged(int)));
roangel's avatar
roangel committed
211
212

    ui->checkBox_vicon_highlight_markers->setEnabled(false);
213

roangel's avatar
roangel committed
214
    #ifdef CATKIN_MAKE
215
    _rosNodeThread->init();
216
    qRegisterMetaType<ptrToMessage>("ptrToMessage");
217
218
    qRegisterMetaType<std_msgs::Int32MultiArray>("std_msgs::Int32MultiArray");
    qRegisterMetaType<CrazyfliePositionData>("CrazyfliePositionData");
219
    QObject::connect(_rosNodeThread, SIGNAL(newViconData(const ptrToMessage&)), this, SLOT(updateNewViconData(const ptrToMessage&)));
220
221
    QObject::connect(_rosNodeThread, SIGNAL(newStudentIDs(const std_msgs::Int32MultiArray&)), this, SLOT(updateStudentIDs(const std_msgs::Int32MultiArray&)));
    QObject::connect(_rosNodeThread, SIGNAL(newPositionData(const CrazyfliePositionData&)), this, SLOT(updateCFPositions(const CrazyfliePositionData&)));
222
    QObject::connect(cf_linker, SIGNAL(updateComboBoxes()), this, SLOT(updateComboBoxes()));
223
224
225

    ros::NodeHandle nodeHandle("~");
    DBChangedPublisher = nodeHandle.advertise<std_msgs::Int32>("DBChanged", 1);
226
    emergencyStopPublisher = nodeHandle.advertise<std_msgs::Int32>("emergencyStop", 1);
227
228

    refreshStudents_publisher = nodeHandle.advertise<std_msgs::Int32>("refreshStudents", 1);
tiagos's avatar
tiagos committed
229
230
231
    UWBServiceClientUpdate_publisher = nodeHandle.advertise<std_msgs::Int32>("UWBUpdate", 1);
    // TODO RENAME
    UWBSettings_publisher = nodeHandle.advertise<std_msgs::Int32>("enableUWB", 1);
232
233

    QObject::connect(ui->checkBox_enable_UWB, SIGNAL(stateChanged(int)), this, SLOT(on_checkBox_enable_UWB_toggled(bool)));
tiagos's avatar
tiagos committed
234
    QObject::connect(ui->reloadAnchors_button, SIGNAL(pressed()), this, SLOT(on_reloadAnchors_button_pressed()));
235

roangel's avatar
roangel committed
236
    #endif
237
}
238

239
240
241
void MainGUIWindow::doTabClosed(int tab_index)
{
    QString name = ui->tabWidget->tabText(tab_index);
roangel's avatar
roangel committed
242
    #ifdef CATKIN_MAKE
roangel's avatar
roangel committed
243
244
    int cf_zone_index = cf_linker->getCFZoneIndexFromName(name);
    scene->removeCrazyFlyZone(cf_zone_index);
roangel's avatar
roangel committed
245
    #endif
246
247
248
249
250
251
252
253
254
}

void MainGUIWindow::setTabIndex(int index)
{
    QString qstr = "CrazyFly ";
    qstr.append(QString::number(index + 1));
    ui->tabWidget->setCurrentIndex(getTabIndexFromName(qstr));
}

255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
void MainGUIWindow::updateComboBoxes()
{
    updateComboBoxesCFs();
    updateComboBoxesCFZones();
}
void MainGUIWindow::updateComboBoxesCFs()
{
    #ifdef CATKIN_MAKE
    ui->comboBoxCFs->clear();
    for(int i = 0; i < crazyflies_vector.size(); i++)
    {
        if(!cf_linker->isCFLinked(crazyflies_vector[i]->getName()))
        {
            QString qstr = QString::fromStdString(crazyflies_vector[i]->getName());
            ui->comboBoxCFs->addItem(qstr);
        }
    }
    #endif
}

void MainGUIWindow::updateComboBoxesCFZones()
{
    ui->comboBoxCFZones->clear();
    #ifdef CATKIN_MAKE
    for(int i = 0; i < scene->crazyfly_zones.size(); i++)
    {
        if(!cf_linker->isCFZoneLinked(scene->crazyfly_zones[i]->getIndex()))
        {
283
            int cf_zone_index = scene->crazyfly_zones[i]->getIndex();
284
            QString qstr = "CrazyFlyZone ";
285
            qstr.append(QString::number(cf_zone_index + 1));
286
287
288
289
290
291
292
            ui->comboBoxCFZones->addItem(qstr);
        }
    }
    #endif
}


roangel's avatar
roangel committed
293
#ifdef CATKIN_MAKE
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
void MainGUIWindow::updateCFPositions(const CrazyfliePositionData& data)
{
    bool inScene = false;
    int index;

    // Reconvert CFPD to CFD
    CrazyflieData cfdata;
    cfdata.crazyflieName = std::to_string(data.id);
    cfdata.x = data.x;
    cfdata.y = data.y;
    cfdata.z = data.z;
    cfdata.roll = data.roll;
    cfdata.pitch = data.pitch;
    cfdata.yaw = data.yaw;
    cfdata.acquiringTime = data.acquiringTime;
    cfdata.occluded = false;

    for(int i = 0; i < cf_positionsVector.size(); ++i)
    {
        if(cf_positionsVector[i]->getName() == cfdata.crazyflieName)
        {
            cf_positionsVector[i]->updateCF(&cfdata);
            inScene = true;
            index = i;
            break;
        }
    }

    if(!inScene)
    {
        QString filename(":/images/center_rect.svg");
        crazyFly* tmp_p_crazyfly = new crazyFly(&cfdata, filename);
        cf_positionsVector.push_back(tmp_p_crazyfly);
        index = cf_positionsVector.size() - 1;
    }

    if(ui->checkBox_vicon_crazyflies->checkState() == Qt::Checked)
    {
        if(!cf_positionsVector[index]->isAddedToScene())
        {
            scene->addItem(cf_positionsVector[index]);
            cf_positionsVector[index]->setAddedToScene(true);
        }
    }
}

void MainGUIWindow::updateStudentIDs(const std_msgs::Int32MultiArray& ids)
341
{
342
343
344
345
346
    ui->list_discovered_student_ids->clear();

    for(int i = 0; i < ids.data.size(); ++i)
        ui->list_discovered_student_ids->addItem(std::to_string(ids.data[i]).c_str());
}
347

348
349
void MainGUIWindow::updateNewViconData(const ptrToMessage& p_msg) //connected to newViconData, from node
{
350
351
    // update Markers

352
    if(p_msg->markers.size() < markers_vector.size()) // some markers have dissapeared, received stuff is smaller than what we have
353
354
355
    {
        for(int i = p_msg->markers.size(); i < markers_vector.size(); i++)
        {
356
            scene->removeItem(markers_vector[i]); // remove objects from scene
357
            // ROS_INFO_STREAM("element index: " << i << " removed");
358
        }
359
        markers_vector.erase(markers_vector.begin() + p_msg->markers.size(), markers_vector.end()); //delete them
360
361
    }

362
    // ROS_INFO_STREAM("markers.size: " << p_msg->markers.size());
363

364
365
366
367
    for(int i = 0; i < p_msg->markers.size(); i++) // here, or new markers message is equal to current messages, or greater (some new markers)
    {
        if(i >= markers_vector.size()) //some new markers coming
        {
368
            // ROS_INFO_STREAM("element index: " << i << " added");
369
            Marker* tmp_p_marker = new Marker(&(p_msg->markers[i]));
370
371
            markers_vector.push_back(tmp_p_marker); // what happens with the new indexes? check if this is correct

372
            if(ui->checkBox_vicon_markers->checkState() == Qt::Checked) //only if markers checkbox info is checked..
373
374
375
376
377
378
379
            {
                scene->addItem(markers_vector[i]);
                if(ui->checkBox_vicon_highlight_markers->checkState() == Qt::Checked)
                {
                    markers_vector[i]->setHighlighted();
                }
            }
380
381
382
        }
        else
        {
383
            // ROS_INFO_STREAM("element index: " << i << " moved, already existed");
384
            markers_vector[i]->updateMarker(&(p_msg->markers[i]));
385
386
        }
    }
387
388

    // update Crazyflies
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
    // also: what happens if we dont go through one of the names? we need to remove that crazyfly
    int crazyfly_vector_size_before = crazyflies_vector.size(); //initial size of vector
    // in this loop, add new ones and update old ones
    for(int i = 0; i < p_msg->crazyflies.size(); i++)
    {
        bool name_found = false; // for each iteration, name_found starts in false
        int index_name_found;
        for(int j = 0; j < crazyfly_vector_size_before; j++)
        {
            if(crazyflies_vector[j]->getName() == p_msg->crazyflies[i].crazyflieName)
            {
                name_found = true; // name found. This can only happen once per i-iteration, names are unique
                index_name_found = j; // index in already existing vector, to update it later (really needed?)
            }
        }

        if(name_found)
        {
            crazyflies_vector[index_name_found]->updateCF(&(p_msg->crazyflies[i]));
        }
409
        else                    //name not found, newly arrived, add it to the vector
410
        {
roangel's avatar
roangel committed
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
            // now, if name follows our format, put the corresponding number. If not, put the unknown image
            std::string s = p_msg->crazyflies[i].crazyflieName;
            std::smatch m;
            std::regex e ("PPS_CF([0-9]{2})");

            QString filename(":/images/drone_fixed_");

            if(std::regex_search(s, m, e))
            {
                std::string found_string = m[1].str();
                filename.append(QString::fromStdString(found_string));
                filename.append(".svg");
            }
            else
            {
                filename.append("unk.svg");
            }

            crazyFly* tmp_p_crazyfly = new crazyFly(&(p_msg->crazyflies[i]), filename);
430
431
432
433
434
435
            crazyflies_vector.push_back(tmp_p_crazyfly);
        }

        if(ui->checkBox_vicon_crazyflies->checkState() == Qt::Checked)
        {
            for(int i = 0; i < crazyflies_vector.size(); i++) //check for occlussion
436
            {
437
438
                if(crazyflies_vector[i]->isOccluded())
                {
roangel's avatar
roangel committed
439
                    // ROS_INFO("===================OCCLUDED");
440
441
442
443
444
445
446
447
448
449
450
451
452
453
                    if(crazyflies_vector[i]->isAddedToScene())
                    {
                        scene->removeItem(crazyflies_vector[i]);
                        crazyflies_vector[i]->setAddedToScene(false);
                    }
                }
                else
                {
                    if(!crazyflies_vector[i]->isAddedToScene())
                    {
                        scene->addItem(crazyflies_vector[i]);
                        crazyflies_vector[i]->setAddedToScene(true);
                    }
                }
454
            }
455
456
        }
    }
457

458
    // in this loop, clean the ones that are not present anymore. UPDATE: this will apparently only happen when we tick and untick in Vicon
459
    int crazyfly_vector_size_after = crazyflies_vector.size();
460

461
    for(int j = crazyfly_vector_size_after - 1; j >= 0; j--)
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
    {
        bool name_found = false;
        for(int i = 0; i < p_msg->crazyflies.size(); i++)
        {
            if(crazyflies_vector[j]->getName() == p_msg->crazyflies[i].crazyflieName)
            {
                name_found = true;
            }
        }
        if(!name_found)
        {
            scene->removeItem(crazyflies_vector[j]);
            crazyflies_vector.erase(crazyflies_vector.begin() + j);
        }
    }
477
}
roangel's avatar
roangel committed
478
#endif
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553


void MainGUIWindow::on_removeTable_clicked()
{
    if(scene->getMode() == myGraphicsScene::mode_table)
    {
        scene->removeTable();
    }
}

void MainGUIWindow::transitionToMode(int mode)
{
    switch(mode)
    {
        case myGraphicsScene::mode_table:
        {
            ui->removeTable->setDisabled(false);
            break;
        }
        case myGraphicsScene::mode_crazyfly_zones:
        {
            ui->removeTable->setDisabled(true);
            break;
        }
    }
}

void MainGUIWindow::on_radioButton_table_mode_toggled(bool checked)
{
    switch(scene->getMode())
    {
        case myGraphicsScene::mode_table:
        {
            // already in the mode we want, do nothing
            break;
        }
        case myGraphicsScene::mode_crazyfly_zones:
        {
            scene->setMode(myGraphicsScene::mode_table);
            break;
        }
        case myGraphicsScene::mode_locked:
        {
            scene->setMode(myGraphicsScene::mode_table);
            break;
        }
    }

}


void MainGUIWindow::on_radioButton_crazyfly_zones_mode_toggled(bool checked)
{
    switch(scene->getMode())
    {
        case myGraphicsScene::mode_table:
        {
            scene->setMode(myGraphicsScene::mode_crazyfly_zones);
            break;
        }
        case myGraphicsScene::mode_crazyfly_zones:
        {
            // already in the mode we want, do nothing
            break;
        }
        case myGraphicsScene::mode_locked:
        {
            scene->setMode(myGraphicsScene::mode_crazyfly_zones);
            break;
        }
    }
}

void MainGUIWindow::handleTablePiecesNumChanged(int newNum)
{
554

555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
}

void MainGUIWindow::on_radioButton_lock_mode_toggled(bool checked)
{
    switch(scene->getMode())
    {
        case myGraphicsScene::mode_table:
        {
            scene->setMode(myGraphicsScene::mode_locked);
            break;
        }
        case myGraphicsScene::mode_crazyfly_zones:
        {
            scene->setMode(myGraphicsScene::mode_locked);
            break;
        }
        case myGraphicsScene::mode_locked:
        {
            break;
        }
    }
}

void MainGUIWindow::on_checkBox_grid_toggled(bool checked)
{
    scene->setGrid(checked);
}

void MainGUIWindow::on_checkBox_table_toggled(bool checked)
{
    if(checked)
    {
        scene->showTable();
    }
    else
    {
        scene->hideTable();
    }
}

void MainGUIWindow::on_checkBox_crazyfly_zones_toggled(bool checked)
{
    if(checked)
    {
        scene->showCrazyFlyZones();
    }
    else
    {
        scene->hideCrazyFlyZones();
    }
}

void MainGUIWindow::on_tabWidget_currentChanged(int index)
{
609
610
611
612
613
614
    // // this index is tab index. Need to go to cf index
    // QString name = ui->tabWidget->tabText(index);
    // #ifdef CATKIN_MAKE
    // int cf_index = cf_linker->getCFZoneIndexFromName(name);
    // scene->setSelectedCrazyFlyZone(cf_index);
    // #endif
615
616
617
618
619
620
621
622
623
624
625
626
627
628
}

void MainGUIWindow::centerViewIndex(int index)
{
    ui->graphicsView->fitInView(scene->getRectFCrazyFlyZone(index), Qt::KeepAspectRatio);
    ui->graphicsView->scale(0.95, 0.95); // A bit back zoom, so we can see everything better
}


void MainGUIWindow::on_pushButton_fitAll_clicked()
{
    ui->graphicsView->fitInView(scene->itemsBoundingRect(), Qt::KeepAspectRatio);
    ui->graphicsView->scale(0.95, 0.95); // A bit back zoom, so we can see everything better
}
629
630
631
632
633

void MainGUIWindow::on_checkBox_vicon_markers_toggled(bool checked)
{
    if(checked)
    {
634
        #ifdef CATKIN_MAKE
635
636
637
638
        for(int i = 0; i < markers_vector.size(); i++)
        {
            scene->addItem(markers_vector[i]);
        }
639
        #endif
640
        ui->checkBox_vicon_highlight_markers->setCheckable(true);
roangel's avatar
roangel committed
641
        ui->checkBox_vicon_highlight_markers->setEnabled(true);
642
643
644
    }
    else
    {
645
        #ifdef CATKIN_MAKE
646
647
648
649
        for(int i = 0; i < markers_vector.size(); i++)
        {
            scene->removeItem(markers_vector[i]);
        }
650
        #endif
651
652
        ui->checkBox_vicon_highlight_markers->setChecked(false);
        ui->checkBox_vicon_highlight_markers->setCheckable(false);
roangel's avatar
roangel committed
653
        ui->checkBox_vicon_highlight_markers->setEnabled(false);
654
655
656
657
658
659
660
    }
}

void MainGUIWindow::on_checkBox_vicon_highlight_markers_toggled(bool checked)
{
    if(checked)
    {
661
        #ifdef CATKIN_MAKE
662
663
664
665
        for(int i = 0; i < markers_vector.size(); i++)
        {
            markers_vector[i]->setHighlighted();
        }
666
        #endif
667
668
669
    }
    else
    {
670
        #ifdef CATKIN_MAKE
671
672
673
674
        for(int i = 0; i < markers_vector.size(); i++)
        {
            markers_vector[i]->clearHighlighted();
        }
675
676
677
678
679
680
681
682
683
684
685
        #endif
    }
}

void MainGUIWindow::on_checkBox_vicon_crazyflies_toggled(bool checked)
{
    if(checked)
    {
        #ifdef CATKIN_MAKE
        for(int i = 0; i < crazyflies_vector.size(); i++)
        {
686
687
688
689
690
            if(!crazyflies_vector[i]->isAddedToScene())
            {
                scene->addItem(crazyflies_vector[i]);
                crazyflies_vector[i]->setAddedToScene(true);
            }
691
692
693
694
695
696
697
698
699
        }
        #endif
        ui->scaleSpinBox->setEnabled(true);
    }
    else
    {
        #ifdef CATKIN_MAKE
        for(int i = 0; i < crazyflies_vector.size(); i++)
        {
700
701
702
703
704
            if(crazyflies_vector[i]->isAddedToScene())
            {
                scene->removeItem(crazyflies_vector[i]);
                crazyflies_vector[i]->setAddedToScene(false);
            }
705
706
707
708
709
710
711
712
713
714
715
716
        }
        #endif
        ui->scaleSpinBox->setEnabled(false);
    }
}

void MainGUIWindow::on_scaleSpinBox_valueChanged(double arg1)
{
    #ifdef CATKIN_MAKE
    for(int i = 0; i < crazyflies_vector.size(); i++)
    {
        crazyflies_vector[i]->setScaleCFs(arg1);
717
    }
718
    #endif
719
}
720
721
722

void MainGUIWindow::on_refresh_cfs_button_clicked()
{
723
    updateComboBoxesCFs();
724
725
726
727
728
729
}

void MainGUIWindow::on_refresh_student_ids_button_clicked()
{
    #ifdef CATKIN_MAKE

730
731
732
733
    // Publish the button being pressed
    std_msgs::Int32 msg;
    msg.data = 1;
    this->refreshStudents_publisher.publish(msg);
734
735
736

    #endif
}
737
738
739
740

void MainGUIWindow::on_link_button_clicked()
{
    #ifdef CATKIN_MAKE
741
742
743
744
745

    bool error = false;
    if(ui->comboBoxCFs->count() == 0)
    {
        // plot error message
746
        ui->err_message_cf->setText("CF box is empty");
747
748
        error = true;
    }
749
750
751
752
    else
    {
        ui->err_message_cf->clear();
    }
753

754
755
756
    if(ui->comboBoxCFZones->count() == 0)
    {
        // plot error message
757
        ui->err_message_cf_zone->setText("CFZone box is empty");
758
759
        error = true;
    }
760
761
762
763
    else
    {
        ui->err_message_cf_zone->clear();
    }
764

765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
    if(cf_linker->isRadioAddressLinked(ui->radioAddress_text->text().toStdString()))
    {
        ui->err_message_radio_address->setText("Already in use");
        error = true;
    }
    else if(ui->radioAddress_text->text().toStdString() == "")
    {
        ui->err_message_radio_address->setText("Field is empty");
        error = true;
    }
    else
    {
        ui->err_message_radio_address->clear();
    }

780
    if(cf_linker->isStudentIDLinked(ui->spinBox_student_ids->value()))
781
    {
782
783
784
785
786
787
788
        // plot error message
        ui->err_message_student_id->setText("This StudentID has already been linked");
        error = true;
    }
    else
    {
        ui->err_message_student_id->clear();
789
790
791
792
    }

    if(!error)
    {
793
        cf_linker->link(ui->spinBox_student_ids->value(), cf_linker->getCFZoneIndexFromName(ui->comboBoxCFZones->currentText()), ui->comboBoxCFs->currentText().toStdString(), ui->radioAddress_text->text().toStdString());
794
    }
795
796
    #endif
}
797
798
799
800

void MainGUIWindow::on_unlink_button_clicked()
{
    #ifdef CATKIN_MAKE
801
    cf_linker->unlink_selection();
802
803
    #endif
}
804
805
806
807

void MainGUIWindow::on_save_in_DB_button_clicked()
{
    // we need to update and then save?
808
    CrazyflieDB tmp_db;
809
    for(int i = 0; i < cf_linker->links.size(); i++)
810
    {
811
812
        CrazyflieEntry tmp_entry;
        tmp_entry.crazyflieContext.crazyflieName = cf_linker->links[i].cf_name;
813
        tmp_entry.crazyflieContext.crazyflieAddress = cf_linker->links[i].radio_address;
814
815
        tmp_entry.crazyflieContext.localArea.crazyfly_zone_index = cf_linker->links[i].cf_zone_index;
        tmp_entry.studentID = cf_linker->links[i].student_id;
816

817
818
819
820
        for(int j = 0; j < scene->crazyfly_zones.size(); j++)
        {
            if(cf_linker->links[i].cf_zone_index == scene->crazyfly_zones[j]->getIndex())
            {
821
822
823
824
825
826
827
828
829
830
                double x_min = scene->crazyfly_zones[j]->sceneBoundingRect().bottomLeft().x();
                double y_min = - scene->crazyfly_zones[j]->sceneBoundingRect().bottomLeft().y();

                double x_max = scene->crazyfly_zones[j]->sceneBoundingRect().topRight().x();
                double y_max = -scene->crazyfly_zones[j]->sceneBoundingRect().topRight().y();

                tmp_entry.crazyflieContext.localArea.xmin = x_min * FROM_UNITS_TO_METERS;
                tmp_entry.crazyflieContext.localArea.xmax = x_max * FROM_UNITS_TO_METERS;
                tmp_entry.crazyflieContext.localArea.ymin = y_min * FROM_UNITS_TO_METERS;
                tmp_entry.crazyflieContext.localArea.ymax = y_max * FROM_UNITS_TO_METERS;
831
832
833

                tmp_entry.crazyflieContext.localArea.zmin = -0.2;
                tmp_entry.crazyflieContext.localArea.zmax = 2.0;
834
835
836
            }
        }
        tmp_db.crazyflieEntries.push_back(tmp_entry);
837
    }
838
839
840
841

    m_data_base = tmp_db;

    ROS_INFO_STREAM("database:\n" << m_data_base);
842
843
844
845

    // save the database in the file

    fill_database_file();
846
847
848
849
850

    // Now also publish a ROS message stating that we changed the DB, so the nodes can update it
    std_msgs::Int32 msg;
    msg.data = 1;
    this->DBChangedPublisher.publish(msg);
851
852
}

853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
void MainGUIWindow::clear_database_file()
{
    CrazyflieDB tmp_db;
    if(read_database_from_file(tmp_db) == 0)
    {
        for(int i = 0; i < tmp_db.crazyflieEntries.size(); i++)
        {
            CMUpdate updateCall;
            updateCall.request.mode = ENTRY_REMOVE;
            updateCall.request.crazyflieEntry.crazyflieContext.crazyflieName = tmp_db.crazyflieEntries[i].crazyflieContext.crazyflieName;
            if(_rosNodeThread->m_update_db_client.call(updateCall))
            {
                ROS_INFO("database changed in central manager service");
            }
            else
            {
                ROS_ERROR("Failed to remove entry in DB");
            }
        }
        save_database_file();
    }
    else
    {
        ROS_INFO("Failed to read DB");
    }
}

void MainGUIWindow::fill_database_file()
{
    clear_database_file();
    ROS_INFO("cleared data base file");
    ROS_INFO_STREAM("database:\n" << m_data_base);
    for(int i = 0; i < m_data_base.crazyflieEntries.size(); i++)
    {
        ROS_INFO("inserted 1 item in DB");
        insert_or_update_entry_database(m_data_base.crazyflieEntries[i]);
    }
    save_database_file();
}

void MainGUIWindow::save_database_file()
{
    CMCommand commandCall;
    commandCall.request.command = CMD_SAVE;
    if(_rosNodeThread->m_command_db_client.call(commandCall))
    {
        ROS_INFO("successfully saved db");
    }
    else
    {
        ROS_ERROR("failed to save db");
    }
}

void MainGUIWindow::insert_or_update_entry_database(CrazyflieEntry entry)
{
    CMUpdate updateCall;
    updateCall.request.mode = ENTRY_INSERT_OR_UPDATE;
    updateCall.request.crazyflieEntry = entry;
    _rosNodeThread->m_update_db_client.call(updateCall);
}

int MainGUIWindow::read_database_from_file(CrazyflieDB &read_db)
916
917
918
919
920
{
    CMRead getDBCall;
    _rosNodeThread->m_read_db_client.waitForExistence(ros::Duration(-1));
    if(_rosNodeThread->m_read_db_client.call(getDBCall))
    {
921
922
923
924
925
926
927
928
929
930
931
932
        read_db = getDBCall.response.crazyflieDB;
        return 0;
    }
    else
    {
        return -1;
    }
}

void MainGUIWindow::on_load_from_DB_button_clicked()
{
    CrazyflieDB tmp_db;
933

934
935
936
937
    if(read_database_from_file(tmp_db) == 0)
    {
		ROS_INFO_STREAM("database:\n" << tmp_db);
        m_data_base = tmp_db;
roangel's avatar
roangel committed
938

939
        cf_linker->clear_all_links();
roangel's avatar
roangel committed
940
        // remove all cf_zones existing
941
        scene->removeAllCFZones();
roangel's avatar
roangel committed
942
943
944
945

        int size = scene->crazyfly_zones.size();
        ROS_INFO("vector_cf_zones_size %d", size);

946
947
948
        for(int i = 0; i < m_data_base.crazyflieEntries.size(); i++)
        {
            std::string cf_name = m_data_base.crazyflieEntries[i].crazyflieContext.crazyflieName;
949
            std::string radio_address = m_data_base.crazyflieEntries[i].crazyflieContext.crazyflieAddress;
950
            int cf_zone_index = m_data_base.crazyflieEntries[i].crazyflieContext.localArea.crazyfly_zone_index;
951

952
            // we should first create the cf zones that are in the database?
953
954
955
            bool cf_zone_exists;
            qreal width = m_data_base.crazyflieEntries[i].crazyflieContext.localArea.xmax - m_data_base.crazyflieEntries[i].crazyflieContext.localArea.xmin;
            qreal height = m_data_base.crazyflieEntries[i].crazyflieContext.localArea.ymax - m_data_base.crazyflieEntries[i].crazyflieContext.localArea.ymin;
956
            QRectF tmp_rect(m_data_base.crazyflieEntries[i].crazyflieContext.localArea.xmin * FROM_METERS_TO_UNITS,
957
                            - m_data_base.crazyflieEntries[i].crazyflieContext.localArea.ymax * FROM_METERS_TO_UNITS, // minus sign because qt has y-axis inverted
958
959
                            width * FROM_METERS_TO_UNITS,
                            height * FROM_METERS_TO_UNITS);
960
            int student_id = m_data_base.crazyflieEntries[i].studentID;
roangel's avatar
roangel committed
961
962
963
964
965


            scene->addCFZone(tmp_rect, cf_zone_index);


966
            cf_linker->link(student_id, cf_zone_index, cf_name, radio_address);
roangel's avatar
roangel committed
967

968
969
970
971
972
973
974
        }
    }
    else
    {
        ROS_ERROR("Failed to read DB");
    }
}
975

976
977
void MainGUIWindow::on_checkBox_enable_UWB_toggled(bool checked)
{
tiagos's avatar
tiagos committed
978
979
980
981
982
983
    // Send Command to enable UWB to UWBServiceClient
    std_msgs::Int32 server_msg;
    server_msg.data = UWB_UPDATE_ENABLE;
    UWBServiceClientUpdate_publisher.publish(server_msg);


984

tiagos's avatar
tiagos committed
985
986
987
988
989
990
991
992
993
994
995
996
997
998
    // Publish message that clients have to reload data from UWBServiceClient
    //std_msgs::Int32 msg;
    //msg.data = 1;
   // UWBSettings_publisher.publish(msg);
}

void MainGUIWindow::on_reloadAnchors_button_pressed()
{
    // Load Data from UWBServiceClient

    // Publish message that clients have to reload data from UWBServiceClient
    std_msgs::Int32 msg;
    msg.data = 1;
    UWBSettings_publisher.publish(msg);
999
1000
}

For faster browsing, not all history is shown. View entire blame