void PortGroup::processStreamConfigList(int portIndex,
PbRpcController *controller)
{
OstProto::StreamConfigList *streamConfigList
= static_cast<OstProto::StreamConfigList*>(controller->response());
qDebug("In %s", __PRETTY_FUNCTION__);
Q_ASSERT(portIndex < numPorts());
if (controller->Failed())
{
qDebug("%s: rpc failed(%s)", __FUNCTION__,
qPrintable(controller->ErrorString()));
goto _exit;
}
Q_ASSERT(portIndex < numPorts());
if (streamConfigList->port_id().id() != mPorts[portIndex]->id())
{
qDebug("Invalid portId %d (expected %d) received for portIndex %d",
streamConfigList->port_id().id(), mPorts[portIndex]->id(), portIndex);
goto _exit;
}
for(int i = 0; i < streamConfigList->stream_size(); i++)
{
uint streamId;
streamId = streamConfigList->stream(i).stream_id().id();
mPorts[portIndex]->updateStream(streamId,
streamConfigList->mutable_stream(i));
}
// Are we done for all ports?
if (portIndex >= numPorts())
{
// FIXME(HI): some way to reset streammodel
}
_exit:
delete controller;
}
bool PythonFileFormat::save(const OstProto::StreamConfigList streams,
const QString fileName, QString &error)
{
QFile file(fileName);
QTextStream out(&file);
QSet<QString> imports;
if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate))
goto _open_fail;
// import standard modules
emit status("Writing imports ...");
emit target(0);
writeStandardImports(out);
emit target(streams.stream_size());
// import protocols from respective modules
// build the import list using a QSet to eliminate duplicates
for (int i = 0; i < streams.stream_size(); i++) {
const OstProto::Stream &stream = streams.stream(i);
for (int j = 0 ; j < stream.protocol_size(); j++) {
const OstProto::Protocol &protocol = stream.protocol(j);
const Reflection *refl = protocol.GetReflection();
std::vector<const FieldDescriptor*> fields;
refl->ListFields(protocol, &fields);
for (uint k = 0; k < fields.size(); k++) {
// skip non extension fields
if (!fields.at(k)->is_extension())
continue;
if (fields.at(k)->file()->name() !=
fields.at(k)->message_type()->file()->name()) {
imports.insert(
QString("%1 import %2").arg(
QString(fields.at(k)->message_type()
->file()->name().c_str())
.replace(".proto", "_pb2"),
fields.at(k)->message_type()->name().c_str()));
imports.insert(
QString("%1 import %2").arg(
QString(fields.at(k)
->file()->name().c_str())
.replace(".proto", "_pb2"),
fields.at(k)->name().c_str()));
}
else {
imports.insert(
QString("%1 import %2, %3").arg(
QString(fields.at(k)->file()->name().c_str())
.replace(".proto", "_pb2"),
fields.at(k)->message_type()->name().c_str(),
fields.at(k)->name().c_str()));
}
}
}
emit progress(i);
}
// write the import statements
out << "# import ostinato modules\n";
out << "from ostinato.core import DroneProxy, ost_pb\n";
foreach (QString str, imports)
out << "from ostinato.protocols." << str << "\n";
out << "\n";
// start of script - init, connect to drone etc.
emit status("Writing prologue ...");
emit target(0);
writePrologue(out);
// Add streams
emit status("Writing stream adds ...");
emit target(streams.stream_size());
out << " # ------------#\n";
out << " # add streams #\n";
out << " # ------------#\n";
out << " stream_id = ost_pb.StreamIdList()\n";
out << " stream_id.port_id.id = tx_port_number\n";
for (int i = 0; i < streams.stream_size(); i++) {
out << " stream_id.stream_id.add().id = "
<< streams.stream(i).stream_id().id() << "\n";
emit progress(i);
}
out << " drone.addStream(stream_id)\n";
out << "\n";
// Configure streams with actual values
emit status("Writing stream configuration ...");
emit target(streams.stream_size());
out << " # ------------------#\n";
out << " # configure streams #\n";
out << " # ------------------#\n";
out << " stream_cfg = ost_pb.StreamConfigList()\n";
out << " stream_cfg.port_id.id = tx_port_number\n";
for (int i = 0; i < streams.stream_size(); i++) {
const OstProto::Stream &stream = streams.stream(i);
const Reflection *refl;
std::vector<const FieldDescriptor*> fields;
out << "\n";
out << " # stream " << stream.stream_id().id() << " "
<< stream.core().name().c_str() << "\n";
out << " s = stream_cfg.stream.add()\n";
out << " s.stream_id.id = "
<< stream.stream_id().id() << "\n";
// Stream Core values
refl = stream.core().GetReflection();
refl->ListFields(stream.core(), &fields);
for (uint j = 0; j < fields.size(); j++) {
writeFieldAssignment(out, QString(" s.core.")
.append(fields.at(j)->name().c_str()),
stream.core(), refl, fields.at(j));
}
// Stream Control values
refl = stream.control().GetReflection();
refl->ListFields(stream.control(), &fields);
for (uint j = 0; j < fields.size(); j++) {
writeFieldAssignment(out, QString(" s.control.")
.append(fields.at(j)->name().c_str()),
stream.control(), refl, fields.at(j));
}
// Protocols
for (int j = 0 ; j < stream.protocol_size(); j++) {
const OstProto::Protocol &protocol = stream.protocol(j);
out << "\n"
<< " p = s.protocol.add()\n"
<< " p.protocol_id.id = "
<< QString(OstProto::Protocol_k_descriptor()
->FindValueByNumber(protocol.protocol_id().id())
->full_name().c_str())
.replace("OstProto", "ost_pb");
out << "\n";
refl = protocol.GetReflection();
refl->ListFields(protocol, &fields);
for (uint k = 0; k < fields.size(); k++) {
// skip protocol_id field
if (fields.at(k)->number() ==
OstProto::Protocol::kProtocolIdFieldNumber)
continue;
QString pfx(" p.Extensions[X]");
pfx.replace(fields.at(k)->is_extension()? "X": "Extensions[X]",
fields.at(k)->name().c_str());
writeFieldAssignment(out, pfx, protocol,
refl, fields.at(k));
}
}
emit progress(i);
}
out << "\n";
out << " drone.modifyStream(stream_cfg)\n";
// end of script - transmit streams, disconnect from drone etc.
emit status("Writing epilogue ...");
emit target(0);
writeEpilogue(out);
out.flush();
file.close();
return true;
_open_fail:
error = QString(tr("Error opening %1 (Error Code = %2)"))
.arg(fileName)
.arg(file.error());
return false;
}
bool Port::openStreams(QString fileName, bool append, QString &error)
{
bool ret = false;
QDialog *optDialog;
QProgressDialog progress("Opening Streams", "Cancel", 0, 0, mainWindow);
OstProto::StreamConfigList streams;
AbstractFileFormat *fmt = AbstractFileFormat::fileFormatFromFile(fileName);
if (fmt == NULL)
goto _fail;
if ((optDialog = fmt->openOptionsDialog()))
{
int ret;
optDialog->setParent(mainWindow, Qt::Dialog);
ret = optDialog->exec();
optDialog->setParent(0, Qt::Dialog);
if (ret == QDialog::Rejected)
goto _user_opt_cancel;
}
progress.setAutoReset(false);
progress.setAutoClose(false);
progress.setMinimumDuration(0);
progress.show();
mainWindow->setDisabled(true);
progress.setEnabled(true); // to override the mainWindow disable
connect(fmt, SIGNAL(status(QString)),&progress,SLOT(setLabelText(QString)));
connect(fmt, SIGNAL(target(int)), &progress, SLOT(setMaximum(int)));
connect(fmt, SIGNAL(progress(int)), &progress, SLOT(setValue(int)));
connect(&progress, SIGNAL(canceled()), fmt, SLOT(cancel()));
fmt->openStreamsOffline(fileName, streams, error);
qDebug("after open offline");
while (!fmt->isFinished())
qApp->processEvents();
qDebug("wait over for offline operation");
if (!fmt->result())
goto _fail;
// process any remaining events posted from the thread
for (int i = 0; i < 10; i++)
qApp->processEvents();
if (!append)
{
int n = numStreams();
progress.setLabelText("Deleting existing streams...");
progress.setRange(0, n);
for (int i = 0; i < n; i++)
{
if (progress.wasCanceled())
goto _user_cancel;
deleteStreamAt(0);
progress.setValue(i);
if (i % 32 == 0)
qApp->processEvents();
}
}
progress.setLabelText("Constructing new streams...");
progress.setRange(0, streams.stream_size());
for (int i = 0; i < streams.stream_size(); i++)
{
if (progress.wasCanceled())
goto _user_cancel;
newStreamAt(mStreams.size(), &streams.stream(i));
progress.setValue(i);
if (i % 32 == 0)
qApp->processEvents();
}
_user_cancel:
emit streamListChanged(mPortGroupId, mPortId);
_user_opt_cancel:
ret = true;
_fail:
progress.close();
mainWindow->setEnabled(true);
recalculateAverageRates();
return ret;
}
