Export streams as a python script for use with python-ostinato
This commit is contained in:
Srivats P.
parent
b617c44e61
commit
10d921adfc
@ -636,7 +636,8 @@ _retry:
|
||||
goto _exit;
|
||||
|
||||
fileType = fileType.remove(QRegExp("\\(.*\\)")).trimmed();
|
||||
if (!fileType.startsWith("Ostinato"))
|
||||
if (!fileType.startsWith("Ostinato")
|
||||
&& !fileType.startsWith("Python"))
|
||||
{
|
||||
if (QMessageBox::warning(this, tr("Ostinato"),
|
||||
QString("You have chosen to save in %1 format. All stream "
|
||||
|
||||
@ -22,6 +22,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
#include "fileformat.h"
|
||||
#include "pcapfileformat.h"
|
||||
#include "pdmlfileformat.h"
|
||||
#include "pythonfileformat.h"
|
||||
|
||||
#include <QStringList>
|
||||
|
||||
@ -49,7 +50,8 @@ QStringList AbstractFileFormat::supportedFileTypes()
|
||||
return QStringList()
|
||||
<< "Ostinato (*)"
|
||||
<< "PCAP (*)"
|
||||
<< "PDML (*.pdml)";
|
||||
<< "PDML (*.pdml)"
|
||||
<< "PythonScript (*.py)";
|
||||
}
|
||||
|
||||
void AbstractFileFormat::openStreamsOffline(const QString fileName,
|
||||
@ -110,6 +112,9 @@ AbstractFileFormat* AbstractFileFormat::fileFormatFromType(
|
||||
if (pcapFileFormat.isMyFileType(fileType))
|
||||
return &pcapFileFormat;
|
||||
|
||||
if (pythonFileFormat.isMyFileType(fileType))
|
||||
return &pythonFileFormat;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
@ -40,6 +40,7 @@ HEADERS = \
|
||||
ipv6addressdelegate.h \
|
||||
pcapfileformat.h \
|
||||
pdmlfileformat.h \
|
||||
pythonfileformat.h \
|
||||
pdmlprotocol.h \
|
||||
pdmlprotocols.h \
|
||||
pdmlreader.h
|
||||
@ -80,6 +81,7 @@ SOURCES += \
|
||||
fileformat.cpp \
|
||||
pcapfileformat.cpp \
|
||||
pdmlfileformat.cpp \
|
||||
pythonfileformat.cpp \
|
||||
pdmlprotocol.cpp \
|
||||
pdmlprotocols.cpp \
|
||||
pdmlreader.cpp \
|
||||
|
||||
553
common/pythonfileformat.cpp
Normal file
553
common/pythonfileformat.cpp
Normal file
@ -0,0 +1,553 @@
|
||||
/*
|
||||
Copyright (C) 2015 Srivats P.
|
||||
|
||||
This file is part of "Ostinato"
|
||||
|
||||
This is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
|
||||
#include "pythonfileformat.h"
|
||||
|
||||
#include <google/protobuf/descriptor.h>
|
||||
|
||||
#include <QFile>
|
||||
#include <QSet>
|
||||
|
||||
#include <cctype>
|
||||
#include <vector>
|
||||
|
||||
using google::protobuf::Message;
|
||||
using google::protobuf::Reflection;
|
||||
using google::protobuf::FieldDescriptor;
|
||||
|
||||
PythonFileFormat pythonFileFormat;
|
||||
|
||||
extern char *version;
|
||||
extern char *revision;
|
||||
|
||||
PythonFileFormat::PythonFileFormat()
|
||||
{
|
||||
// Nothing to do
|
||||
}
|
||||
|
||||
PythonFileFormat::~PythonFileFormat()
|
||||
{
|
||||
// Nothing to do
|
||||
}
|
||||
|
||||
bool PythonFileFormat::openStreams(const QString fileName,
|
||||
OstProto::StreamConfigList &streams, QString &error)
|
||||
{
|
||||
// NOT SUPPORTED!
|
||||
return false;
|
||||
}
|
||||
|
||||
bool PythonFileFormat::saveStreams(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 protocol_id field
|
||||
if (fields.at(k)->number() ==
|
||||
OstProto::Protocol::kProtocolIdFieldNumber)
|
||||
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("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:
|
||||
return false;
|
||||
}
|
||||
|
||||
bool PythonFileFormat::isMyFileFormat(const QString fileName)
|
||||
{
|
||||
// isMyFileFormat() is used for file open case to detect
|
||||
// file format - Open not supported for Python Scripts
|
||||
return false;
|
||||
}
|
||||
|
||||
bool PythonFileFormat::isMyFileType(const QString fileType)
|
||||
{
|
||||
if (fileType.startsWith("PythonScript"))
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
|
||||
//
|
||||
// Private Member Functions
|
||||
//
|
||||
void PythonFileFormat::writeStandardImports(QTextStream &out)
|
||||
{
|
||||
out << "#! /usr/bin/env python\n";
|
||||
out << "\n";
|
||||
out << "# This script was programmatically generated\n"
|
||||
<< "# by Ostinato version " << version
|
||||
<< " revision " << revision << "\n"
|
||||
<< "# The script should work out of the box mostly,\n"
|
||||
<< "# but occassionally might need minor tweaking\n"
|
||||
<< "# Please report any bugs at http://ostinato.org\n";
|
||||
out << "\n";
|
||||
out << "# standard modules\n";
|
||||
out << "import logging\n";
|
||||
out << "import os\n";
|
||||
out << "import sys\n";
|
||||
out << "import time\n";
|
||||
out << "\n";
|
||||
}
|
||||
|
||||
void PythonFileFormat::writePrologue(QTextStream &out)
|
||||
{
|
||||
out << "# initialize the below variables appropriately "
|
||||
<< "to avoid manual input\n";
|
||||
out << "host_name = ''\n";
|
||||
out << "tx_port_number = -1\n";
|
||||
out << "\n";
|
||||
out << "# setup logging\n";
|
||||
out << "log = logging.getLogger(__name__)\n";
|
||||
out << "logging.basicConfig(level=logging.INFO)\n";
|
||||
out << "\n";
|
||||
out << "# get inputs, if required\n";
|
||||
out << "while len(host_name) == 0:\n";
|
||||
out << " host_name = raw_input('Drone\\'s Hostname/IP: ')\n";
|
||||
out << "while tx_port_number < 0:\n";
|
||||
out << " tx_port_number = int(raw_input('Tx Port Number: '))\n";
|
||||
out << "\n";
|
||||
out << "drone = DroneProxy(host_name)\n";
|
||||
out << "\n";
|
||||
out << "try:\n";
|
||||
out << " # connect to drone\n";
|
||||
out << " log.info('connecting to drone(%s:%d)' \n";
|
||||
out << " % (drone.hostName(), drone.portNumber()))\n";
|
||||
out << " drone.connect()\n";
|
||||
out << "\n";
|
||||
out << " # setup tx port list\n";
|
||||
out << " tx_port = ost_pb.PortIdList()\n";
|
||||
out << " tx_port.port_id.add().id = tx_port_number;\n";
|
||||
out << "\n";
|
||||
}
|
||||
|
||||
void PythonFileFormat::writeEpilogue(QTextStream &out)
|
||||
{
|
||||
out << " # clear tx/rx stats\n";
|
||||
out << " log.info('clearing tx stats')\n";
|
||||
out << " drone.clearStats(tx_port)\n";
|
||||
out << "\n";
|
||||
out << " log.info('starting transmit')\n";
|
||||
out << " drone.startTransmit(tx_port)\n";
|
||||
out << "\n";
|
||||
out << " # wait for transmit to finish\n";
|
||||
out << " log.info('waiting for transmit to finish ...')\n";
|
||||
out << " while True:\n";
|
||||
out << " time.sleep(5)\n";
|
||||
out << " tx_stats = drone.getStats(tx_port)\n";
|
||||
out << " if tx_stats.port_stats[0].state.is_transmit_on == False:\n";
|
||||
out << " break\n";
|
||||
out << "\n";
|
||||
out << " # stop transmit and capture\n";
|
||||
out << " log.info('stopping transmit')\n";
|
||||
out << " drone.stopTransmit(tx_port)\n";
|
||||
out << "\n";
|
||||
out << " # get tx stats\n";
|
||||
out << " log.info('retreiving stats')\n";
|
||||
out << " tx_stats = drone.getStats(tx_port)\n";
|
||||
out << "\n";
|
||||
out << " log.info('tx pkts = %d' % (tx_stats.port_stats[0].tx_pkts))\n";
|
||||
out << "\n";
|
||||
out << " # delete streams\n";
|
||||
out << " log.info('deleting tx_streams')\n";
|
||||
out << " drone.deleteStream(stream_id)\n";
|
||||
out << "\n";
|
||||
out << " # bye for now\n";
|
||||
out << " drone.disconnect()\n";
|
||||
out << "\n";
|
||||
out << "except Exception as ex:\n";
|
||||
out << " log.exception(ex)\n";
|
||||
out << " sys.exit(1)\n";
|
||||
}
|
||||
|
||||
void PythonFileFormat::writeFieldAssignment(
|
||||
QTextStream &out,
|
||||
QString fieldName,
|
||||
const Message &msg,
|
||||
const Reflection *refl,
|
||||
const FieldDescriptor *fieldDesc,
|
||||
int index)
|
||||
{
|
||||
// for a repeated field,
|
||||
// if index < 0 => we are writing a repeated aggregate
|
||||
// if index >= 0 => we are writing a repeated element
|
||||
if (fieldDesc->is_repeated() && (index < 0)) {
|
||||
int n = refl->FieldSize(msg, fieldDesc);
|
||||
QString var = singularize(fieldDesc->name().c_str());
|
||||
for (int i = 0; i < n; i++) {
|
||||
out << " " << var << " = " << fieldName.trimmed() << ".add()\n";
|
||||
writeFieldAssignment(out, QString(" ").append(var),
|
||||
msg, refl, fieldDesc, i);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// Ideally fields should not be set if they have the same
|
||||
// value as the default value - but currently protocols don't
|
||||
// check this when setting values in the protobuf data object
|
||||
// so here we check that explicitly for each field and if true
|
||||
// we don't output anything
|
||||
switch(fieldDesc->cpp_type()) {
|
||||
case FieldDescriptor::CPPTYPE_INT32:
|
||||
{
|
||||
qint32 val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedInt32(msg, fieldDesc, index) :
|
||||
refl->GetInt32(msg, fieldDesc);
|
||||
if (val != fieldDesc->default_value_int32())
|
||||
out << fieldName << " = " << val << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_INT64:
|
||||
{
|
||||
qint64 val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedInt64(msg, fieldDesc, index) :
|
||||
refl->GetInt64(msg, fieldDesc);
|
||||
if (val != fieldDesc->default_value_int64())
|
||||
out << fieldName << " = " << val << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_UINT32:
|
||||
{
|
||||
quint32 val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedUInt32(msg, fieldDesc, index) :
|
||||
refl->GetUInt32(msg, fieldDesc);
|
||||
QString valStr;
|
||||
|
||||
if (useDecimalBase(fieldName))
|
||||
valStr.setNum(val);
|
||||
else
|
||||
valStr.setNum(val, 16).prepend("0x");
|
||||
|
||||
if (val != fieldDesc->default_value_uint32())
|
||||
out << fieldName << " = " << valStr << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_UINT64:
|
||||
{
|
||||
quint64 val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedUInt64(msg, fieldDesc, index) :
|
||||
refl->GetUInt64(msg, fieldDesc);
|
||||
QString valStr;
|
||||
|
||||
if (useDecimalBase(fieldName))
|
||||
valStr.setNum(val);
|
||||
else
|
||||
valStr.setNum(val, 16).prepend("0x");
|
||||
|
||||
if (val != fieldDesc->default_value_uint64())
|
||||
out << fieldName << " = " << valStr << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_DOUBLE:
|
||||
{
|
||||
double val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedDouble(msg, fieldDesc, index) :
|
||||
refl->GetDouble(msg, fieldDesc);
|
||||
if (val != fieldDesc->default_value_double())
|
||||
out << fieldName << " = " << val << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_FLOAT:
|
||||
{
|
||||
float val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedFloat(msg, fieldDesc, index) :
|
||||
refl->GetFloat(msg, fieldDesc);
|
||||
if (val != fieldDesc->default_value_float())
|
||||
out << fieldName << " = " << val << "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_BOOL:
|
||||
{
|
||||
bool val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedBool(msg, fieldDesc, index) :
|
||||
refl->GetBool(msg, fieldDesc);
|
||||
if (val != fieldDesc->default_value_bool())
|
||||
out << fieldName
|
||||
<< " = "
|
||||
<< (refl->GetBool(msg, fieldDesc) ? "True" : "False")
|
||||
<< "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_STRING:
|
||||
{
|
||||
std::string val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedStringReference(msg, fieldDesc, index, &val) :
|
||||
refl->GetStringReference(msg, fieldDesc, &val);
|
||||
QString escVal = escapeString(val.c_str());
|
||||
if (val != fieldDesc->default_value_string())
|
||||
out << fieldName << " = '" << escVal << "'\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_ENUM:
|
||||
{
|
||||
// Fields defined in protocol.proto are within ost_pb scope
|
||||
QString module = fieldDesc->file()->name() == "protocol.proto" ?
|
||||
"ost_pb." : "";
|
||||
std::string val = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedEnum(msg, fieldDesc, index)->full_name() :
|
||||
refl->GetEnum(msg, fieldDesc)->full_name();
|
||||
if (val != fieldDesc->default_value_enum()->full_name())
|
||||
out << fieldName << " = " << QString::fromStdString(val)
|
||||
.replace("OstProto.", module)
|
||||
<< "\n";
|
||||
break;
|
||||
}
|
||||
case FieldDescriptor::CPPTYPE_MESSAGE:
|
||||
{
|
||||
QString pfxStr(fieldName);
|
||||
const Message &msg2 = fieldDesc->is_repeated() ?
|
||||
refl->GetRepeatedMessage(msg, fieldDesc, index) :
|
||||
refl->GetMessage(msg, fieldDesc);
|
||||
const Reflection *refl2 = msg2.GetReflection();
|
||||
std::vector<const FieldDescriptor*> fields2;
|
||||
QList<std::string> autoFields;
|
||||
|
||||
refl2->ListFields(msg2, &fields2);
|
||||
|
||||
// Unfortunately, auto-calculated fields such as cksum, length
|
||||
// and protocol-type etc. may be set in the protobuf even if
|
||||
// they are not being overridden;
|
||||
// Intelligence regarding them is inside the respective protocol
|
||||
// implementation, not inside the protobuf objects - the latter
|
||||
// is all we have available here to work with;
|
||||
// We attempt a crude hack here to detect such fields and avoid
|
||||
// writing assignment statements for them
|
||||
for (uint i = 0; i < fields2.size(); i++) {
|
||||
std::string name = fields2.at(i)->name();
|
||||
if ((fields2.at(i)->cpp_type()
|
||||
== FieldDescriptor::CPPTYPE_BOOL)
|
||||
&& (name.find("is_override_") == 0)
|
||||
&& (refl2->GetBool(msg2, fields2.at(i)) == false)) {
|
||||
name.erase(0, sizeof("is_override_") - 1);
|
||||
autoFields.append(name);
|
||||
}
|
||||
}
|
||||
|
||||
for (uint i = 0 ; i < fields2.size(); i++) {
|
||||
// skip auto fields that are not overridden
|
||||
if (autoFields.contains(fields2.at(i)->name()))
|
||||
continue;
|
||||
|
||||
writeFieldAssignment(out,
|
||||
QString("%1.%2").arg(pfxStr,
|
||||
fields2.at(i)->name().c_str()),
|
||||
msg2, refl2, fields2.at(i));
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
qWarning("unable to write field of unsupported type %d",
|
||||
fieldDesc->cpp_type());
|
||||
}
|
||||
}
|
||||
|
||||
QString PythonFileFormat::singularize(QString plural)
|
||||
{
|
||||
QString singular = plural;
|
||||
|
||||
// Apply some heuristics
|
||||
if (plural.endsWith("ies"))
|
||||
singular.replace(singular.length()-3, 3, "y");
|
||||
else if (plural.endsWith("ses"))
|
||||
singular.chop(2);
|
||||
else if (plural.endsWith("s"))
|
||||
singular.chop(1);
|
||||
|
||||
return singular;
|
||||
}
|
||||
|
||||
QString PythonFileFormat::escapeString(QString str)
|
||||
{
|
||||
QString escStr = "";
|
||||
for (int i=0; i < str.length(); i++) {
|
||||
uchar c = str[i].cell();
|
||||
if ((c < 128) && isprint(c)) {
|
||||
if (c == '\'')
|
||||
escStr.append("\\'");
|
||||
else
|
||||
escStr.append(str[i]);
|
||||
}
|
||||
else
|
||||
escStr.append(QString("\\x%1").arg(int(c), 2, 16, QChar('0')));
|
||||
}
|
||||
return escStr;
|
||||
}
|
||||
|
||||
bool PythonFileFormat::useDecimalBase(QString fieldName)
|
||||
{
|
||||
// Heuristic - use Hex base for all except for the following
|
||||
return fieldName.endsWith("count")
|
||||
|| fieldName.endsWith("length")
|
||||
|| fieldName.endsWith("len")
|
||||
|| fieldName.endsWith("time");
|
||||
}
|
||||
|
||||
59
common/pythonfileformat.h
Normal file
59
common/pythonfileformat.h
Normal file
@ -0,0 +1,59 @@
|
||||
/*
|
||||
Copyright (C) 2015 Srivats P.
|
||||
|
||||
This file is part of "Ostinato"
|
||||
|
||||
This is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
|
||||
#ifndef _PYTHON_FILE_FORMAT_H
|
||||
#define _PYTHON_FILE_FORMAT_H
|
||||
|
||||
#include "abstractfileformat.h"
|
||||
|
||||
#include <QTextStream>
|
||||
|
||||
class PythonFileFormat : public AbstractFileFormat
|
||||
{
|
||||
public:
|
||||
PythonFileFormat();
|
||||
~PythonFileFormat();
|
||||
|
||||
virtual bool openStreams(const QString fileName,
|
||||
OstProto::StreamConfigList &streams, QString &error);
|
||||
virtual bool saveStreams(const OstProto::StreamConfigList streams,
|
||||
const QString fileName, QString &error);
|
||||
|
||||
bool isMyFileFormat(const QString fileName);
|
||||
bool isMyFileType(const QString fileType);
|
||||
|
||||
private:
|
||||
void writeStandardImports(QTextStream &out);
|
||||
void writePrologue(QTextStream &out);
|
||||
void writeEpilogue(QTextStream &out);
|
||||
void writeFieldAssignment(QTextStream &out,
|
||||
QString fieldName,
|
||||
const google::protobuf::Message &msg,
|
||||
const google::protobuf::Reflection *refl,
|
||||
const google::protobuf::FieldDescriptor *fieldDesc,
|
||||
int index = -1);
|
||||
QString singularize(QString plural);
|
||||
QString escapeString(QString str);
|
||||
bool useDecimalBase(QString fieldName);
|
||||
};
|
||||
|
||||
extern PythonFileFormat pythonFileFormat;
|
||||
|
||||
#endif
|
||||
|
||||
Loading…
Reference in New Issue
Block a user