Changeset 234 in flair-src for trunk/tools/FlairGCS/src/file_ui.cpp

Timestamp:

04/10/18 17:05:27 (6 years ago)

Author:

Sanahuja Guillaume

Message:

create file oscket only when necessary

File:

• trunk/tools/FlairGCS/src/file_ui.cpp (modified) (9 diffs)

trunk/tools/FlairGCS/src/file_ui.cpp

@@ -4 +4 @@
 // %flair:license}
 #include "file_ui.h"
+#include "communication.h"
+
 #include <stdio.h>
-
 #include <cstring>
 #include <cstdlib>
@@ -22 +23 @@
 #include <QStringList>
 #include <QFormLayout>
-
-#ifndef WIN32
-#include <arpa/inet.h>
-#else
-#include <winsock2.h>
-#include <ws2tcpip.h>
-#endif
-
+#include <QThread>
+
+#include <unistd.h>
 using namespace std;
 
-file_ui::file_ui() {
+file_ui::file_ui(UDTSOCKET socket,QString name): QObject() {
+  this->socket=socket;
+  this->name=name;
+  
+  bool blocking = true;
+  if (UDT::setsockopt(socket, 0, UDT_SNDSYN, &blocking, sizeof(bool))!= 0) {
+    printf("UDT::setsockopt error (UDT_SNDSYN) %s\n",UDT::getlasterror().getErrorMessage());
+  }
+    
+  linger _linger;
+  _linger.l_onoff=1;
+  _linger.l_linger=180;
+  
+  if (UDT::setsockopt(socket, 0, UDT_LINGER, &_linger, sizeof(struct linger)) != 0)
+    printf("UDT::setsockopt error (UDT_LINGER) %s\n",UDT::getlasterror().getErrorMessage());
+    
   dialog = new QDialog();
+//fprintf(stderr,"creator file ui %x\n",thread());
   dialog->setWindowTitle("log files");
   QGridLayout *main_layout = new QGridLayout(dialog);
@@ -42 +54 @@
   input_text->setText("add your log comment here");
   input_cleared = false;
-
   ok_button->setEnabled(false);
 
@@ -56 +67 @@
   main_layout->addWidget(ok_button, 3, 0);
 
-  connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),
-          Qt::QueuedConnection);
-  connect(this, SIGNAL(showDialog()), dialog, SLOT(show()),
-          Qt::QueuedConnection);
-  connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)),
-          Qt::QueuedConnection);
-  connect(input_text, SIGNAL(cursorPositionChanged()), this,
-          SLOT(clearInputText()), Qt::QueuedConnection);
+  //connect for multithreaded stuffs
+  connect(ok_button, SIGNAL(clicked()), this, SLOT(save()),Qt::QueuedConnection);
+  connect(input_text, SIGNAL(cursorPositionChanged()), this,SLOT(clearInputText()),Qt::DirectConnection);
+  connect(this, SIGNAL(appendToLog(QString)), log_text, SLOT(append(QString)));
 
   file_names = new QStringList();
-}
-
-file_ui::~file_ui() { delete dialog; }
-
-void file_ui::log(QString text) { appendToLog(text); }
+  
+  dialog->show();
+}
+
+file_ui::~file_ui() { 
+  delete dialog;
+}
+
+void file_ui::receive(void) {
+  char *recv_buf;
+  int bytesRead;
+  bool flag_new_seq = true;
+  QString folder_name;
+//fprintf(stderr,"file_ui thread %x\n",thread());
+  while(1) {
+    // receive file info
+    recv_buf = (char *)malloc(1024);
+    bytesRead = UDT::recvmsg(socket, recv_buf, 1024);
+    if (bytesRead <= 0) {
+      free(recv_buf);
+      break;
+    }
+
+    int size;
+    memcpy(&size, &recv_buf[1], sizeof(int));
+    if (recv_buf[0] == FILE_INFO_BIG_ENDIAN)
+      size = qFromBigEndian(size);
+
+    // printf("file_ui recu %i %x\n",bytesRead,recv_buf[0]);
+    if ((recv_buf[0]==FILE_INFO_LITTLE_ENDIAN || recv_buf[0]==FILE_INFO_BIG_ENDIAN) && size>0) {
+      if (flag_new_seq == true) {
+        // create directory for storage
+        QDateTime dateTime = QDateTime::currentDateTime();
+        folder_name = dateTime.toString("yyyyMMdd_hhmm") + "_" + name;
+        if (QDir().exists(folder_name) == true) {
+          folder_name = dateTime.toString("yyyyMMdd_hhmm_ss") + "_" + name;
+        }
+        QDir().mkdir(folder_name);
+
+        flag_new_seq = false;
+        appendToLog("Creating directory " + folder_name);
+      }
+
+      QString file_name=QString::fromAscii((const char *)&recv_buf[5], bytesRead - 5);
+      QString file_path=folder_name+"/"+file_name;
+      appendToLog(QString("receiving %1 (%2 bytes)").arg(file_name).arg(size));
+      QFile fichier(file_path);
+
+      if (!fichier.open(QIODevice::WriteOnly)) {
+        appendToLog("      could not write to file!");
+      } else {
+        // receive file
+        recv_buf = (char *)realloc((void *)recv_buf, size);
+        bytesRead = UDT::recvmsg(socket, recv_buf, size);
+        if (bytesRead != size) {
+          appendToLog(QString("      error receiving file! (%1/%2)").arg(bytesRead).arg(size));
+          free(recv_buf);
+          break;
+        } else {
+          appendToLog("      ok");
+        }
+
+        QDataStream stream(&fichier);
+        stream.writeRawData(recv_buf, size);
+        fichier.close();
+
+        addFile(file_path);
+      }
+
+      free(recv_buf);
+    } else if (recv_buf[0] == END_SENDING_FILES) {
+      //end ack
+      UDT::sendmsg(socket,&recv_buf[0],1);
+      endOfFiles();
+      UDT::close(socket);
+      printf("disconnected from log files\n");
+      break;
+    }
+  }
+}
 
 void file_ui::addFile(QString file_path) {
@@ -86 +168 @@
     dbt2csv(file_path.replace(QString(".txt"), QString(".dbt")));
   }
-
-  if (file_names->size() == 1) {
-    input_cleared = false;
-    showDialog();
-  }
 }
 
@@ -101 +178 @@
     if (info.size() > max_file_size) {
       max_file_size = info.size();
-      csv_combo->setCurrentIndex(i +
-                                 1); // first item of combobox is already taken
+      csv_combo->setCurrentIndex(i+1); // first item of combobox is already taken
     }
   }
@@ -112 +188 @@
   QStringList data_type;
 
-  QString filename =
-      file_path.section('/', -1); // remove path for displaying on logs
+  QString filename =file_path.section('/', -1); // remove path for displaying on logs
   appendToLog(QString("converting %1 to csv").arg(filename));
 
@@ -145 +220 @@
       break;
     QString txt_line = txt_in.readLine();
-    data_type.append(txt_line.section(
-        "(",
-        -1)); // on part de la fin pour trouver la premiere parenthese ouvrante
+    data_type.append(txt_line.section("(",-1)); // on part de la fin pour trouver la premiere parenthese ouvrante
     // printf("type %s\n",txt_line.section("(",-1).toLocal8Bit().constData());
   }
@@ -241 +314 @@
   }
 
-  log_text->clear();
-  input_cleared = true; // avoid clearing it with setText
-  input_text->setText("add your log comment here");
-  file_names->clear();
-  csv_combo->clear();
-  csv_combo->addItem(QString("(no base time)"));
-
-  dialog->setVisible(false);
-  ok_button->setEnabled(false);
   emit finished();
 }