source: flair-src/branches/sanscv/tools/FlairGCS/src/file_ui.cpp@ 451

Last change on this file since 451 was 324, checked in by Sanahuja Guillaume, 5 years ago

removing opencv dependency
File size: 14.5 KB
Line 
1// %flair:license{
2// This file is part of the Flair framework distributed under the
3// CECILL-C License, Version 1.0.
4// %flair:license}
5#include "file_ui.h"
6#include "communication.h"
7
8#include <stdio.h>
9#include <cstring>
10#include <cstdlib>
11#include <fstream>
12#include <iostream>
13#include <QDir>
14#include <QDate>
15#include <QTextStream>
16#include <QGridLayout>
17#include <io_hdfile.h>
18#include <QtEndian>
19#include <QComboBox>
20#include <QPushButton>
21#include <QTextEdit>
22#include <QDialog>
23#include <QStringList>
24#include <QFormLayout>
25#include <QThread>
26
27#include <unistd.h>
28using namespace std;
29
30file_ui::file_ui(UDTSOCKET socket,QString name): QObject() {
31 this->socket=socket;
32 this->name=name;
33
34 bool blocking = true;
35 if (UDT::setsockopt(socket, 0, UDT_SNDSYN, &blocking, sizeof(bool))!= 0) {
36 fprintf(stderr,"UDT::setsockopt error (UDT_SNDSYN) %s\n",UDT::getlasterror().getErrorMessage());
37 }
38
39 if (UDT::setsockopt(socket, 0, UDT_RCVSYN, &blocking, sizeof(bool)) != 0) {
40 fprintf(stderr,"UDT::setsockopt error (UDT_RCVSYN) %s\n",UDT::getlasterror().getErrorMessage());
41 }
42
43 linger _linger;
44 _linger.l_onoff=1;
45 _linger.l_linger=180;
46
47 if (UDT::setsockopt(socket, 0, UDT_LINGER, &_linger, sizeof(struct linger)) != 0)
48 fprintf(stderr,"UDT::setsockopt error (UDT_LINGER) %s\n",UDT::getlasterror().getErrorMessage());
49
50 dialog = new QDialog();
51
52 dialog->setWindowTitle("log files");
53 QGridLayout *main_layout = new QGridLayout(dialog);
54 ok_button = new QPushButton("Ok", dialog);
55 log_text = new QTextEdit(dialog);
56 log_text->setReadOnly(true);
57 input_text = new QTextEdit(dialog);
58 input_text->setText("add your log comment here");
59 input_cleared = false;
60 ok_button->setEnabled(false);
61
62 main_layout->addWidget(log_text, 0, 0);
63 main_layout->addWidget(input_text, 1, 0);
64
65 QWidget *widget = new QWidget(dialog);
66 QFormLayout *formLayout = new QFormLayout(widget);
67 csv_combo = new QComboBox(widget);
68 formLayout->addRow(tr("save all log with following base time"), csv_combo);
69 csv_combo->addItem("(no base time)");
70 main_layout->addWidget(widget, 2, 0);
71 main_layout->addWidget(ok_button, 3, 0);
72
73 //connect for multithreaded stuffs
74 connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),Qt::QueuedConnection);
75 connect(input_text, SIGNAL(cursorPositionChanged()), this,SLOT(clearInputText()),Qt::DirectConnection);
76 connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)));
77
78 file_names = new QStringList();
79
80 dialog->show();
81}
82
83file_ui::~file_ui() {
84 delete dialog;
85}
86
87void file_ui::receive(void) {
88 char *recv_buf;
89 int bytesRead;
90 bool flag_new_seq = true;
91 QString folder_name;
92//ffprintf(stderr,stderr,"file_ui thread %x\n",thread());
93 while(1) {
94 // receive file info
95 recv_buf = (char *)malloc(1024);
96 bytesRead = UDT::recvmsg(socket, recv_buf, 1024);
97 if (bytesRead <= 0) {
98 free(recv_buf);
99 break;
100 }
101
102 int size;
103 memcpy(&size, &recv_buf[1], sizeof(int));
104 if (recv_buf[0] == FILE_INFO_BIG_ENDIAN)
105 size = qFromBigEndian(size);
106
107 // fprintf(stderr,"file_ui recu %i %x\n",bytesRead,recv_buf[0]);
108 if ((recv_buf[0]==FILE_INFO_LITTLE_ENDIAN || recv_buf[0]==FILE_INFO_BIG_ENDIAN) && size>0) {
109 if (flag_new_seq == true) {
110 // create directory for storage
111 QDateTime dateTime = QDateTime::currentDateTime();
112 folder_name = dateTime.toString("yyyyMMdd_hhmm") + "_" + name;
113 if (QDir().exists(folder_name) == true) {
114 folder_name = dateTime.toString("yyyyMMdd_hhmm_ss") + "_" + name;
115 }
116 QDir().mkdir(folder_name);
117
118 flag_new_seq = false;
119 appendToLog("Creating directory " + folder_name);
120 }
121
122 QString file_name=QString::fromAscii((const char *)&recv_buf[5], bytesRead - 5);
123 QString file_path=folder_name+"/"+file_name;
124 appendToLog(QString("receiving %1 (%2 bytes)").arg(file_name).arg(size));
125 QFile fichier(file_path);
126
127 if (!fichier.open(QIODevice::WriteOnly)) {
128 appendToLog(" could not write to file!");
129 } else {
130 // receive file
131 recv_buf = (char *)realloc((void *)recv_buf, size);
132 bytesRead = UDT::recvmsg(socket, recv_buf, size);
133 if (bytesRead != size) {
134 appendToLog(QString(" error receiving file! (%1/%2): %3").arg(bytesRead).arg(size).arg(UDT::getlasterror().getErrorMessage()));
135 free(recv_buf);
136 break;
137 } else {
138 appendToLog(" ok");
139 }
140
141 QDataStream stream(&fichier);
142 stream.writeRawData(recv_buf, size);
143 fichier.close();
144
145 addFile(file_path);
146 }
147
148 free(recv_buf);
149 } else if (recv_buf[0] == END_SENDING_FILES) {
150 //end ack
151 UDT::sendmsg(socket,&recv_buf[0],1);
152 endOfFiles();
153 UDT::close(socket);
154 fprintf(stderr,"disconnected from log files\n");
155 break;
156 }
157 }
158}
159
160void file_ui::addFile(QString file_path) {
161 // framework sends dbt file then txt file
162 // when we receive txt, we have both files
163 // and we can convert it to .csv
164 if (file_path.endsWith(".dbt") == true) {
165 QString name =
166 file_path.section('/', -1); // remove path for displaying on combobox
167 csv_combo->addItem(name.replace(QString(".dbt"), QString(".csv")));
168 file_names->append(file_path.replace(QString(".dbt"), QString(".csv")));
169 }
170
171 if (file_path.endsWith(".txt") == true) {
172 dbt2csv(file_path.replace(QString(".txt"), QString(".dbt")));
173 }
174}
175
176void file_ui::endOfFiles(void) {
177 ok_button->setEnabled(true);
178
179 qint64 max_file_size = 0;
180 for (int i = 0; i < file_names->count(); i++) {
181 QFileInfo info(file_names->at(i));
182 if (info.size() > max_file_size) {
183 max_file_size = info.size();
184 csv_combo->setCurrentIndex(i+1); // first item of combobox is already taken
185 }
186 }
187}
188
189void file_ui::dbt2csv(QString file_path) {
190 hdfile_t *dbtFile = NULL;
191 char *data;
192 QStringList data_type;
193
194 QString filename =file_path.section('/', -1); // remove path for displaying on logs
195 appendToLog(QString("converting %1 to csv").arg(filename));
196
197 // open csv file
198 QString csv_filename = file_path;
199 csv_filename.replace(QString(".dbt"), QString(".csv"));
200 QFile csv_file(csv_filename);
201
202 if (!csv_file.open(QIODevice::WriteOnly | QIODevice::Text)) {
203 appendToLog(" error opening csv file!");
204 return;
205 }
206 QTextStream out(&csv_file);
207
208 // open txt file
209 QString txt_filename = file_path;
210 txt_filename.replace(QString(".dbt"), QString(".txt"));
211 QFile txt_file(txt_filename);
212
213 if (!txt_file.open(QIODevice::ReadOnly | QIODevice::Text)) {
214 appendToLog(" error opening txt file!");
215 return;
216 }
217
218 // read txt file
219 QTextStream txt_in(&txt_file);
220 txt_in.readLine(); // time us
221 txt_in.readLine(); // time ns
222 while (1) {
223 if (txt_in.atEnd() == true)
224 break;
225 QString txt_line = txt_in.readLine();
226 data_type.append(txt_line.section("(",-1)); // on part de la fin pour trouver la premiere parenthese ouvrante
227 // fprintf(stderr,"type %s\n",txt_line.section("(",-1).toLocal8Bit().constData());
228 }
229 txt_file.close();
230
231 dbtFile = open_hdfile(file_path.toLocal8Bit().data(), READ_MODE);
232
233 if (!dbtFile) {
234 appendToLog(" error opening dbt file!");
235 return;
236 }
237 data = (char *)malloc(dbtFile->h.DataSize);
238 if (data == NULL) {
239 appendToLog(" error malloc!");
240 return;
241 }
242
243 bool dataWritten = false;
244 while (1) {
245 road_time_t time;
246 road_timerange_t tr = 0;
247 int offset = 0;
248 QTextStream csv_line;
249
250 if (read_hdfile(dbtFile, (void *)data, &time, &tr) == 0) {
251 break;
252 }
253 dataWritten = true;
254
255 out << time << "," << tr;
256 for (int i = 0; i < data_type.size(); i++) {
257 if (data_type.at(i) == "double)") {
258 double *value = (double *)(data + offset);
259 offset += sizeof(double);
260 out << "," << *value;
261 } else if (data_type.at(i) == "float)") {
262 float *value = (float *)(data + offset);
263 offset += sizeof(float);
264 out << "," << *value;
265 } else if (data_type.at(i) == "int8_t)") {
266 int8_t *value = (int8_t *)(data + offset);
267 offset += sizeof(int8_t);
268 out << "," << *value;
269 } else if (data_type.at(i) == "uint8_t)") {
270 uint8_t *value = (uint8_t *)(data + offset);
271 offset += sizeof(uint8_t);
272 out << "," << *value;
273 } else if (data_type.at(i) == "int16_t)") {
274 int16_t *value = (int16_t *)(data + offset);
275 offset += sizeof(int16_t);
276 out << "," << *value;
277 } else if (data_type.at(i) == "uint16_t)") {
278 uint16_t *value = (uint16_t *)(data + offset);
279 offset += sizeof(uint16_t);
280 out << "," << *value;
281 } else {
282 appendToLog(QString(" unhandled type: %1").arg(data_type.at(i)));
283 }
284 }
285
286 out << "\n";
287 }
288
289 if (!dataWritten) {
290 // empty file!
291 out << "0,0"; // timr
292 for (int i = 0; i < data_type.size(); i++) {
293 out << ",0";
294 }
295 out << "\n";
296 }
297
298 csv_file.close();
299 close_hdfile(dbtFile);
300 if (data != NULL)
301 free(data);
302
303 appendToLog(" ok");
304}
305
306void file_ui::clearInputText(void) {
307 if (input_cleared == false) {
308 input_cleared = true;
309 input_text->clear();
310 }
311}
312
313void file_ui::save(void) {
314 save_comment();
315 if (csv_combo->currentIndex() != 0) {
316 save_csv();
317 save_txt();
318 }
319
320 emit finished();
321}
322
323void file_ui::save_comment(void) {
324 QString folder_name = file_names->at(0).section('/', 0, -2);
325
326 QString filename = folder_name + "/commentaire.txt";
327 QFile file(filename);
328 if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
329 fprintf(stderr,"file_ui::save_comment: erreur ouverture fichier %s\n",
330 filename.toLocal8Bit().constData());
331 QTextStream out(&file);
332
333 out << input_text->toPlainText();
334 file.close();
335}
336
337void file_ui::save_csv(void) {
338 // global csv file
339 QString folder_name = file_names->at(0).section('/', 0, -2);
340 QString filename = folder_name + "/all_logs.csv";
341 QFile global_file(filename);
342 if (!global_file.open(QIODevice::WriteOnly | QIODevice::Text))
343 fprintf(stderr,"file_ui::save_csv: erreur ouverture fichier %s\n",
344 filename.toLocal8Bit().constData());
345 QTextStream out(&global_file);
346
347 // reference csv file
348 filename = file_names->at(csv_combo->currentIndex() - 1);
349 QFile ref_file(filename);
350 // fprintf(stderr,"file_ui::save_csv: ref %s\n",filename.toLocal8Bit().constData());
351 if (!ref_file.open(QIODevice::ReadOnly | QIODevice::Text))
352 fprintf(stderr,"file_ui::save_csv: erreur ouverture ficher %s\n",
353 filename.toLocal8Bit().constData());
354
355 // other csv files
356 int j = 0;
357 QFile m_file[file_names->count() - 1];
358 QTextStream m_in[file_names->count() - 1];
359 for (int i = 0; i < file_names->count(); i++) {
360 if (i == csv_combo->currentIndex() - 1)
361 continue;
362 filename = file_names->at(i);
363 m_file[j].setFileName(filename);
364 if (!m_file[j].open(QIODevice::ReadOnly | QIODevice::Text))
365 fprintf(stderr,"file_ui::save_csv: erreur ouverture ficher %s\n",
366 filename.toLocal8Bit().constData());
367 m_in[j].setDevice(&m_file[j]);
368 j++;
369 }
370
371 // init
372 QTextStream ref_in(&ref_file);
373 QString m_line[file_names->count() - 1];
374 QString m_line_prev[file_names->count() - 1];
375 for (int i = 0; i < file_names->count() - 1; i++) {
376 m_line[i] = m_in[i].readLine();
377 m_line_prev[i] = m_line[i];
378 }
379
380 // organize csv files in one file
381 while (1) {
382 if (ref_in.atEnd() == true)
383 break;
384 QString ref_line = ref_in.readLine();
385
386 qint64 ref_us = ref_line.section(',', 0, 0).toLongLong();
387 int ref_ns = ref_line.section(',', 1, 1).toInt();
388 // fprintf(stderr,"ref %lld %i\n",ref_us,ref_ns);
389
390 for (int i = 0; i < file_names->count() - 1; i++) {
391 qint64 csv_us = m_line[i].section(',', 0, 0).toLongLong();
392 int csv_ns = m_line[i].section(',', 1, 1).toInt();
393 // fprintf(stderr,"m %lld %i\n",csv_us,csv_ns);
394
395 while (is_greater(ref_us, csv_us, ref_ns, csv_ns) == true) {
396 m_line_prev[i] = m_line[i];
397 if (m_in[i].atEnd() == true)
398 break;
399 m_line[i] = m_in[i].readLine();
400 csv_us = m_line[i].section(',', 0, 0).toLongLong();
401 csv_ns = m_line[i].section(',', 1, 1).toInt();
402 // fprintf(stderr,"m %lld %i\n",csv_us,csv_ns);
403 }
404 csv_us = m_line_prev[i].section(',', 0, 0).toLongLong();
405 csv_ns = m_line_prev[i].section(',', 1, 1).toInt();
406 // fprintf(stderr,"m ok %lld %i\n",csv_us,csv_ns);
407
408 ref_line += "," + m_line_prev[i].section(',', 2);
409 }
410
411 out << ref_line << "\n";
412 }
413
414 global_file.close();
415 ref_file.close();
416 for (int i = 0; i < file_names->count() - 1; i++)
417 m_file[i].close();
418}
419
420void file_ui::save_txt(void) {
421 // global txt file
422 QString folder_name = file_names->at(0).section('/', 0, -2);
423 QString filename = folder_name + "/all_logs.txt";
424 QFile global_file(filename);
425 if (!global_file.open(QIODevice::WriteOnly | QIODevice::Text))
426 fprintf(stderr,"file_ui::save_txt: erreur ouverture ficher %s\n",
427 filename.toLocal8Bit().constData());
428 QTextStream out(&global_file);
429
430 // reference txt file
431 filename = file_names->at(csv_combo->currentIndex() - 1);
432 filename.replace(QString(".csv"), QString(".txt"));
433 QFile ref_file(filename);
434 if (!ref_file.open(QIODevice::ReadOnly | QIODevice::Text))
435 fprintf(stderr,"file_ui::save_txt: erreur ouverture ficher %s\n",
436 filename.toLocal8Bit().constData());
437
438 QTextStream ref_in(&ref_file);
439 QString current_line = ref_in.readLine();
440 int nb_lines = 1;
441 while (current_line != NULL) {
442 out << current_line << "\n";
443 ;
444 current_line = ref_in.readLine();
445 nb_lines++;
446 }
447
448 // other txt files
449 for (int i = 0; i < file_names->count(); i++) {
450 if (i == csv_combo->currentIndex() - 1)
451 continue;
452 filename = file_names->at(i);
453 filename.replace(QString(".csv"), QString(".txt"));
454 QFile txt_file(filename);
455 if (!txt_file.open(QIODevice::ReadOnly | QIODevice::Text))
456 fprintf(stderr,"file_ui::save_txt: erreur ouverture ficher %s\n",
457 filename.toLocal8Bit().constData());
458 QTextStream txt_in(&txt_file);
459 txt_in.readLine(); // time us
460 txt_in.readLine(); // time ns
461 current_line = txt_in.readLine();
462 while (current_line != NULL) {
463 out << nb_lines << ":" << current_line.section(':', 1) << "\n";
464 ;
465 current_line = txt_in.readLine();
466 nb_lines++;
467 }
468 txt_file.close();
469 }
470 global_file.close();
471 ref_file.close();
472}
473
474bool file_ui::is_greater(qint64 ref_us, qint64 csv_us, int ref_ns, int csv_ns) {
475 if (ref_us == csv_us) {
476 if (ref_ns > csv_ns) {
477 return true;
478 } else {
479 return false;
480 }
481 }
482 if (ref_us > csv_us) {
483 return true;
484 } else {
485 return false;
486 }
487}
Note: See TracBrowser for help on using the repository browser.