source: flair-src/branches/mavlink/tools/FlairGCS/src/file_ui.cpp @ 77

Last change on this file since 77 was 15, checked in by Bayard Gildas, 6 years ago

sources reformatted with flair-format-dir script

File size: 11.3 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 <stdio.h>
7
8#include <cstring>
9#include <cstdlib>
10#include <fstream>
11#include <iostream>
12#include <QDir>
13#include <QDate>
14#include <QTextStream>
15#include <QGridLayout>
16#include <io_hdfile.h>
17#include <QtEndian>
18#include <QComboBox>
19#include <QPushButton>
20#include <QTextEdit>
21#include <QDialog>
22#include <QStringList>
23#include <QFormLayout>
24
25#ifndef WIN32
26#include <arpa/inet.h>
27#else
28#include <winsock2.h>
29#include <ws2tcpip.h>
30#endif
31
32using namespace std;
33
34file_ui::file_ui() {
35  dialog = new QDialog();
36  dialog->setWindowTitle("log files");
37  QGridLayout *main_layout = new QGridLayout(dialog);
38  ok_button = new QPushButton("Ok", dialog);
39  log_text = new QTextEdit(dialog);
40  log_text->setReadOnly(true);
41  input_text = new QTextEdit(dialog);
42  input_text->setText("add your log comment here");
43  input_cleared = false;
44
45  ok_button->setEnabled(false);
46
47  main_layout->addWidget(log_text, 0, 0);
48  main_layout->addWidget(input_text, 1, 0);
49
50  QWidget *widget = new QWidget(dialog);
51  QFormLayout *formLayout = new QFormLayout(widget);
52  csv_combo = new QComboBox(widget);
53  formLayout->addRow(tr("save all log with following base time"), csv_combo);
54  csv_combo->addItem("(no base time)");
55  main_layout->addWidget(widget, 2, 0);
56  main_layout->addWidget(ok_button, 3, 0);
57
58  connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),
59          Qt::QueuedConnection);
60  connect(this, SIGNAL(showDialog()), dialog, SLOT(show()),
61          Qt::QueuedConnection);
62  connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)),
63          Qt::QueuedConnection);
64  connect(input_text, SIGNAL(cursorPositionChanged()), this,
65          SLOT(clearInputText()), Qt::QueuedConnection);
66
67  file_names = new QStringList();
68}
69
70file_ui::~file_ui() { delete dialog; }
71
72void file_ui::log(QString text) { appendToLog(text); }
73
74void file_ui::addFile(QString file_path) {
75  // framework sends dbt file then txt file
76  // when we receive txt, we have both files
77  // and we can convert it to .csv
78  if (file_path.endsWith(".dbt") == true) {
79    QString name =
80        file_path.section('/', -1); // remove path for displaying on combobox
81    csv_combo->addItem(name.replace(QString(".dbt"), QString(".csv")));
82    file_names->append(file_path.replace(QString(".dbt"), QString(".csv")));
83  }
84
85  if (file_path.endsWith(".txt") == true) {
86    dbt2csv(file_path.replace(QString(".txt"), QString(".dbt")));
87  }
88
89  if (file_names->size() == 1) {
90    input_cleared = false;
91    showDialog();
92  }
93}
94
95void file_ui::endOfFiles(void) {
96  ok_button->setEnabled(true);
97
98  qint64 max_file_size = 0;
99  for (int i = 0; i < file_names->count(); i++) {
100    QFileInfo info(file_names->at(i));
101    if (info.size() > max_file_size) {
102      max_file_size = info.size();
103      csv_combo->setCurrentIndex(i +
104                                 1); // first item of combobox is already taken
105    }
106  }
107}
108
109void file_ui::dbt2csv(QString file_path) {
110  hdfile_t *dbtFile = NULL;
111  char *data;
112  QStringList data_type;
113
114  QString filename =
115      file_path.section('/', -1); // remove path for displaying on logs
116  appendToLog(QString("converting %1 to csv").arg(filename));
117
118  // open csv file
119  QString csv_filename = file_path;
120  csv_filename.replace(QString(".dbt"), QString(".csv"));
121  QFile csv_file(csv_filename);
122
123  if (!csv_file.open(QIODevice::WriteOnly | QIODevice::Text)) {
124    appendToLog("      error opening csv file!");
125    return;
126  }
127  QTextStream out(&csv_file);
128
129  // open txt file
130  QString txt_filename = file_path;
131  txt_filename.replace(QString(".dbt"), QString(".txt"));
132  QFile txt_file(txt_filename);
133
134  if (!txt_file.open(QIODevice::ReadOnly | QIODevice::Text)) {
135    appendToLog("      error opening txt file!");
136    return;
137  }
138
139  // read txt file
140  QTextStream txt_in(&txt_file);
141  txt_in.readLine(); // time us
142  txt_in.readLine(); // time ns
143  while (1) {
144    if (txt_in.atEnd() == true)
145      break;
146    QString txt_line = txt_in.readLine();
147    data_type.append(txt_line.section(
148        "(",
149        -1)); // on part de la fin pour trouver la premiere parenthese ouvrante
150    // printf("type %s\n",txt_line.section("(",-1).toLocal8Bit().constData());
151  }
152  txt_file.close();
153
154  dbtFile = open_hdfile(file_path.toLocal8Bit().data(), READ_MODE);
155
156  if (!dbtFile) {
157    appendToLog("      error opening dbt file!");
158    return;
159  }
160  data = (char *)malloc(dbtFile->h.DataSize);
161  if (data == NULL) {
162    appendToLog("      error malloc!");
163    return;
164  }
165
166  bool dataWritten = false;
167  while (1) {
168    road_time_t time;
169    road_timerange_t tr = 0;
170    int offset = 0;
171    QTextStream csv_line;
172
173    if (read_hdfile(dbtFile, (void *)data, &time, &tr) == 0) {
174      break;
175    }
176    dataWritten = true;
177
178    out << time << "," << tr;
179    for (int i = 0; i < data_type.size(); i++) {
180      if (data_type.at(i) == "float)") {
181        float *value = (float *)(data + offset);
182        offset += sizeof(float);
183        out << "," << *value;
184      } else if (data_type.at(i) == "int8_t)") {
185        int8_t *value = (int8_t *)(data + offset);
186        offset += sizeof(int8_t);
187        float fl = (float)*value;
188        out << "," << *value;
189      } else {
190        appendToLog(QString("      unhandled type: %1").arg(data_type.at(i)));
191      }
192    }
193
194    out << "\n";
195  }
196
197  if (!dataWritten) {
198    // empty file!
199    out << "0,0"; // timr
200    for (int i = 0; i < data_type.size(); i++) {
201      out << ",0";
202    }
203    out << "\n";
204  }
205
206  csv_file.close();
207  close_hdfile(dbtFile);
208  if (data != NULL)
209    free(data);
210
211  appendToLog("      ok");
212}
213
214void file_ui::clearInputText(void) {
215  if (input_cleared == false) {
216    input_cleared = true;
217    input_text->clear();
218  }
219}
220
221void file_ui::save(void) {
222  save_comment();
223  if (csv_combo->currentIndex() != 0) {
224    save_csv();
225    save_txt();
226  }
227
228  log_text->clear();
229  input_cleared = true; // avoid clearing it with setText
230  input_text->setText("add your log comment here");
231  file_names->clear();
232  csv_combo->clear();
233  csv_combo->addItem(QString("(no base time)"));
234
235  dialog->setVisible(false);
236  ok_button->setEnabled(false);
237  emit finished();
238}
239
240void file_ui::save_comment(void) {
241  QString folder_name = file_names->at(0).section('/', 0, -2);
242
243  QString filename = folder_name + "/commentaire.txt";
244  QFile file(filename);
245  if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
246    printf("file_ui::save_comment: erreur ouverture fichier %s\n",
247           filename.toLocal8Bit().constData());
248  QTextStream out(&file);
249
250  out << input_text->toPlainText();
251  file.close();
252}
253
254void file_ui::save_csv(void) {
255  // global csv file
256  QString folder_name = file_names->at(0).section('/', 0, -2);
257  QString filename = folder_name + "/all_logs.csv";
258  QFile global_file(filename);
259  if (!global_file.open(QIODevice::WriteOnly | QIODevice::Text))
260    printf("file_ui::save_csv: erreur ouverture fichier %s\n",
261           filename.toLocal8Bit().constData());
262  QTextStream out(&global_file);
263
264  // reference csv file
265  filename = file_names->at(csv_combo->currentIndex() - 1);
266  QFile ref_file(filename);
267  // printf("file_ui::save_csv: ref %s\n",filename.toLocal8Bit().constData());
268  if (!ref_file.open(QIODevice::ReadOnly | QIODevice::Text))
269    printf("file_ui::save_csv: erreur ouverture ficher %s\n",
270           filename.toLocal8Bit().constData());
271
272  // other csv files
273  int j = 0;
274  QFile m_file[file_names->count() - 1];
275  QTextStream m_in[file_names->count() - 1];
276  for (int i = 0; i < file_names->count(); i++) {
277    if (i == csv_combo->currentIndex() - 1)
278      continue;
279    filename = file_names->at(i);
280    m_file[j].setFileName(filename);
281    if (!m_file[j].open(QIODevice::ReadOnly | QIODevice::Text))
282      printf("file_ui::save_csv: erreur ouverture ficher %s\n",
283             filename.toLocal8Bit().constData());
284    m_in[j].setDevice(&m_file[j]);
285    j++;
286  }
287
288  // init
289  QTextStream ref_in(&ref_file);
290  QString m_line[file_names->count() - 1];
291  QString m_line_prev[file_names->count() - 1];
292  for (int i = 0; i < file_names->count() - 1; i++) {
293    m_line[i] = m_in[i].readLine();
294    m_line_prev[i] = m_line[i];
295  }
296
297  // organize csv files in one file
298  while (1) {
299    if (ref_in.atEnd() == true)
300      break;
301    QString ref_line = ref_in.readLine();
302
303    qint64 ref_us = ref_line.section(',', 0, 0).toLongLong();
304    int ref_ns = ref_line.section(',', 1, 1).toInt();
305    // printf("ref %lld %i\n",ref_us,ref_ns);
306
307    for (int i = 0; i < file_names->count() - 1; i++) {
308      qint64 csv_us = m_line[i].section(',', 0, 0).toLongLong();
309      int csv_ns = m_line[i].section(',', 1, 1).toInt();
310      // printf("m %lld %i\n",csv_us,csv_ns);
311
312      while (is_greater(ref_us, csv_us, ref_ns, csv_ns) == true) {
313        m_line_prev[i] = m_line[i];
314        if (m_in[i].atEnd() == true)
315          break;
316        m_line[i] = m_in[i].readLine();
317        csv_us = m_line[i].section(',', 0, 0).toLongLong();
318        csv_ns = m_line[i].section(',', 1, 1).toInt();
319        // printf("m %lld %i\n",csv_us,csv_ns);
320      }
321      csv_us = m_line_prev[i].section(',', 0, 0).toLongLong();
322      csv_ns = m_line_prev[i].section(',', 1, 1).toInt();
323      // printf("m ok %lld %i\n",csv_us,csv_ns);
324
325      ref_line += "," + m_line_prev[i].section(',', 2);
326    }
327
328    out << ref_line << "\n";
329  }
330
331  global_file.close();
332  ref_file.close();
333  for (int i = 0; i < file_names->count() - 1; i++)
334    m_file[i].close();
335}
336
337void file_ui::save_txt(void) {
338  // global txt file
339  QString folder_name = file_names->at(0).section('/', 0, -2);
340  QString filename = folder_name + "/all_logs.txt";
341  QFile global_file(filename);
342  if (!global_file.open(QIODevice::WriteOnly | QIODevice::Text))
343    printf("file_ui::save_txt: erreur ouverture ficher %s\n",
344           filename.toLocal8Bit().constData());
345  QTextStream out(&global_file);
346
347  // reference txt file
348  filename = file_names->at(csv_combo->currentIndex() - 1);
349  filename.replace(QString(".csv"), QString(".txt"));
350  QFile ref_file(filename);
351  if (!ref_file.open(QIODevice::ReadOnly | QIODevice::Text))
352    printf("file_ui::save_txt: erreur ouverture ficher %s\n",
353           filename.toLocal8Bit().constData());
354
355  QTextStream ref_in(&ref_file);
356  QString current_line = ref_in.readLine();
357  int nb_lines = 1;
358  while (current_line != NULL) {
359    out << current_line << "\n";
360    ;
361    current_line = ref_in.readLine();
362    nb_lines++;
363  }
364
365  // other txt files
366  for (int i = 0; i < file_names->count(); i++) {
367    if (i == csv_combo->currentIndex() - 1)
368      continue;
369    filename = file_names->at(i);
370    filename.replace(QString(".csv"), QString(".txt"));
371    QFile txt_file(filename);
372    if (!txt_file.open(QIODevice::ReadOnly | QIODevice::Text))
373      printf("file_ui::save_txt: erreur ouverture ficher %s\n",
374             filename.toLocal8Bit().constData());
375    QTextStream txt_in(&txt_file);
376    txt_in.readLine(); // time us
377    txt_in.readLine(); // time ns
378    current_line = txt_in.readLine();
379    while (current_line != NULL) {
380      out << nb_lines << ":" << current_line.section(':', 1) << "\n";
381      ;
382      current_line = txt_in.readLine();
383      nb_lines++;
384    }
385    txt_file.close();
386  }
387  global_file.close();
388  ref_file.close();
389}
390
391bool file_ui::is_greater(qint64 ref_us, qint64 csv_us, int ref_ns, int csv_ns) {
392  if (ref_us == csv_us) {
393    if (ref_ns > csv_ns) {
394      return true;
395    } else {
396      return false;
397    }
398  }
399  if (ref_us > csv_us) {
400    return true;
401  } else {
402    return false;
403  }
404}
Note: See TracBrowser for help on using the repository browser.