source: flair-src/trunk/tools/FlairGCS/src/ConnectionLayout.cpp@ 222

Last change on this file since 222 was 222, checked in by Sanahuja Guillaume, 4 years ago

allows realloc in graphs

File size: 5.0 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 "ConnectionLayout.h"
6#include "UdtSocket.h"
7#include "DataRemote.h"
8#include <qendian.h>
9#include "communication.h"
10#include <zlib.h>
11#include <assert.h>
12
13#define COMPRESS_CHUNK 1024
14#define RX_TIME 5000
15
16ConnectionLayout::ConnectionLayout(UdtSocket *socket, QString name)
17 : Layout(NULL, name, "root") {
18 isRemoteNameDefined = false;
19 this->socket = socket;
20
21 total_received=0;
22 receive_timer = new QTimer(this);
23 connect(receive_timer, SIGNAL(timeout()), this, SLOT(computeRxRate()));
24 receive_timer->start(RX_TIME);
25}
26
27ConnectionLayout::~ConnectionLayout() {
28 receive_timer->stop();
29}
30
31void ConnectionLayout::computeRxRate(void) {
32 float receive_rate=((float)total_received/(RX_TIME/1000))/1000;//in Ko/s
33 total_received=0;
34 computedRxRate(receive_rate);
35}
36
37void ConnectionLayout::receive(char *buf, int size) {
38 total_received+=size-1;
39 // printf("trame %x\n",buf[0]);
40 // for(int i=0; i<size;i++) printf("%x ",buf[i]);
41 // printf("\n");
42 switch (buf[0]) {
43 case ZLIB_HEADER: {
44 ssize_t out_size;
45 char *uncompressbuf;
46 uncompressBuffer(buf, size, &uncompressbuf, &out_size);
47 handleUncompressedFrame(uncompressbuf, out_size);
48 free(uncompressbuf);
49 break;
50 }
51 default:
52 handleUncompressedFrame(buf, size);
53 }
54 free(buf);
55}
56
57void ConnectionLayout::XmlToSend(QDomDocument doc) {
58 // printf("xml to send\n%s\n",doc.toString().toLocal8Bit().constData());
59 // xml to send a mettre dans le manager
60 QMetaObject::invokeMethod(
61 socket, "write", Qt::BlockingQueuedConnection,
62 Q_ARG(const char *, doc.toString().toLocal8Bit().constData()),
63 Q_ARG(qint64, doc.toString().toLocal8Bit().length()));
64}
65
66void ConnectionLayout::LoadXml(QDomDocument to_parse) {
67 if (!isRemoteNameDefined) {
68 printf("load xml: name not defined!\n");
69 return;
70 }
71
72 QDomElement tmp = to_parse.firstChildElement("root");
73 while (tmp.attribute("name") != remoteName && !tmp.isNull())
74 tmp = to_parse.nextSiblingElement("root");
75
76 if (!tmp.isNull()) {
77 XmlWidget::LoadXml(tmp);
78 } else {
79 printf("%s not found in xml file \n", remoteName.toLocal8Bit().constData());
80 }
81}
82
83void ConnectionLayout::handleUncompressedFrame(char *buf, ssize_t size) {
84 switch ((unsigned char)buf[0]) {
85 case XML_HEADER: {
86 QString xml;
87 QDomDocument doc;
88 xml = QString((char *)buf);
89 xml.resize(size);
90
91 // printf("recu %i\n%s\n",size,xml.toLocal8Bit().constData());
92 if (!doc.setContent(xml)) {
93 printf("prob setContent fichier\n");
94 }
95
96 if (!isRemoteNameDefined) {
97 isRemoteNameDefined = true;
98 remoteName = doc.firstChildElement("root").attribute("name");
99 setRemoteName(remoteName);
100 SetAttribute("name", remoteName);
101 }
102
103 ParseXml(doc.firstChildElement("root").firstChildElement());
104 break;
105 }
106 case DATAS_BIG_ENDIAN: {
107 // for(int i=0;i<size;i++) printf("%x ",buf[i]);
108 // printf("\n");
109 uint16_t period;
110 memcpy(&period, &buf[1], sizeof(uint16_t));
111 period = qFromBigEndian(period);
112 drawDatas(&buf[3], size - 3, period, true);
113 break;
114 }
115 case DATAS_LITTLE_ENDIAN: {
116 // for(int i=0;i<size;i++) printf("%x ",buf[i]);
117 // printf("\n");
118 uint16_t period;
119 memcpy(&period, &buf[1], sizeof(uint16_t));
120 // printf("recu %i period %i\n",size,period);
121 drawDatas(&buf[3], size - 3, period);
122 break;
123 }
124 default:
125 printf("trame non supportée %x\n", buf[0]);
126 }
127}
128
129void ConnectionLayout::removeDataRemote(DataRemote *data) {
130 dataremotes.removeOne(data);
131}
132
133void ConnectionLayout::addDataRemote(DataRemote *data) {
134 dataremotes.append(data);
135}
136
137QString ConnectionLayout::getRemoteName() { return remoteName; }
138
139void ConnectionLayout::drawDatas(char *buf, int buf_size, uint16_t period,
140 bool big_endian) {
141 for (int i = 0; i < dataremotes.count(); i++) {
142 dataremotes.at(i)->BufEvent(&buf, &buf_size, period, big_endian);
143 }
144}
145
146int ConnectionLayout::uncompressBuffer(char *in, ssize_t in_size, char **out,
147 ssize_t *out_size) {
148 int ret;
149 unsigned have;
150 z_stream strm;
151
152 // allocate inflate state
153 strm.zalloc = Z_NULL;
154 strm.zfree = Z_NULL;
155 strm.opaque = Z_NULL;
156 strm.avail_in = 0;
157 strm.next_in = Z_NULL;
158 ret = inflateInit(&strm);
159 if (ret != Z_OK)
160 return ret;
161
162 *out = (char *)malloc(COMPRESS_CHUNK);
163 if (!(*out))
164 return Z_BUF_ERROR;
165
166 strm.avail_in = in_size;
167 strm.next_in = (unsigned char *)in;
168 strm.avail_out = COMPRESS_CHUNK;
169 strm.next_out = (unsigned char *)*out;
170
171 ret = inflate(&strm, Z_NO_FLUSH);
172 assert(ret != Z_STREAM_ERROR); // state not clobbered
173 switch (ret) {
174 case Z_NEED_DICT:
175 ret = Z_DATA_ERROR; // and fall through
176 case Z_DATA_ERROR:
177 case Z_MEM_ERROR:
178 (void)inflateEnd(&strm);
179 return ret;
180 }
181 have = COMPRESS_CHUNK - strm.avail_out;
182 *out_size = have;
183
184 // printf("%i -> %i\n",in_size,have);
185 // printf("%s\n",*out);
186 // clean up and return
187 (void)inflateEnd(&strm);
188 return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
189}
Note: See TracBrowser for help on using the repository browser.