1 #include "saverthread.h"
3 #include "trace_writer.hpp"
4 #include "trace_model.hpp"
5 #include "trace_parser.hpp"
14 static Trace::FunctionSig *
15 createFunctionSig(ApiTraceCall *call, unsigned id)
17 Trace::FunctionSig *sig = new Trace::FunctionSig();
20 sig->name = qstrdup(call->name().toLocal8Bit());
22 QStringList args = call->argNames();
23 sig->num_args = args.count();
24 sig->arg_names = new const char*[args.count()];
25 for (int i = 0; i < args.count(); ++i) {
26 sig->arg_names[i] = qstrdup(args[i].toLocal8Bit());
33 deleteFunctionSig(Trace::FunctionSig *sig)
35 for (int i = 0; i < sig->num_args; ++i) {
36 delete [] sig->arg_names[i];
38 delete [] sig->arg_names;
43 static Trace::StructSig *
44 createStructSig(const ApiStruct &str, unsigned id)
46 ApiStruct::Signature aSig = str.signature();
48 Trace::StructSig *sig = new Trace::StructSig();
50 sig->name = qstrdup(aSig.name.toLocal8Bit());
51 sig->num_members = aSig.memberNames.count();
52 char **member_names = new char*[aSig.memberNames.count()];
53 sig->member_names = (const char **)member_names;
54 for (int i = 0; i < aSig.memberNames.count(); ++i) {
55 member_names[i] = qstrdup(aSig.memberNames[i].toLocal8Bit());
61 deleteStructSig(Trace::StructSig *sig)
63 for (int i = 0; i < sig->num_members; ++i) {
64 delete [] sig->member_names[i];
66 delete [] sig->member_names;
71 static Trace::EnumSig *
72 createEnumSig(const ApiEnum &en, unsigned id)
74 Trace::EnumSig *sig = new Trace::EnumSig();
77 sig->name = qstrdup(en.name().toLocal8Bit());
78 sig->value = en.value().toLongLong();
84 deleteEnumSig(Trace::EnumSig *sig)
90 static Trace::BitmaskSig *
91 createBitmaskSig(const ApiBitmask &bt, unsigned id)
93 ApiBitmask::Signature bsig = bt.signature();
94 ApiBitmask::Signature::const_iterator itr;
96 Trace::BitmaskSig *sig = new Trace::BitmaskSig();
97 Trace::BitmaskFlag *flags = new Trace::BitmaskFlag[bsig.count()];
100 sig->num_flags = bsig.count();
104 for (itr = bsig.constBegin(); itr != bsig.constEnd(); ++itr, ++i) {
105 flags[i].name = qstrdup(itr->first.toLocal8Bit());
106 flags[i].value = itr->second;
113 deleteBitmaskSig(Trace::BitmaskSig *sig)
115 for (int i = 0; i < sig->num_flags; ++i) {
116 delete [] sig->flags[i].name;
118 delete [] sig->flags;
123 writeValue(Trace::Writer &writer, const QVariant &var, unsigned &id)
125 int arrayType = QMetaType::type("ApiArray");
126 int bitmaskType = QMetaType::type("ApiBitmask");
127 int structType = QMetaType::type("ApiStruct");
128 int pointerType = QMetaType::type("ApiPointer");
129 int enumType = QMetaType::type("ApiEnum");
130 int type = var.userType();
134 writer.writeBool(var.toBool());
136 case QVariant::ByteArray: {
137 QByteArray ba = var.toByteArray();
138 writer.writeBlob((const void*)ba.constData(), ba.size());
141 case QVariant::Double:
142 writer.writeDouble(var.toDouble());
144 case QMetaType::Float:
145 writer.writeFloat(var.toFloat());
148 writer.writeSInt(var.toInt());
150 case QVariant::LongLong:
151 writer.writeSInt(var.toLongLong());
153 case QVariant::String: {
154 QString str = var.toString();
155 writer.writeString(str.toLocal8Bit().constData(), str.length());
159 writer.writeUInt(var.toInt());
161 case QVariant::ULongLong:
162 writer.writeUInt(var.toLongLong());
165 if (type == arrayType) {
166 ApiArray array = var.value<ApiArray>();
167 QVector<QVariant> vals = array.values();
168 writer.beginArray(vals.count());
169 foreach(QVariant el, vals) {
170 writer.beginElement();
171 writeValue(writer, el, ++id);
175 } else if (type == bitmaskType) {
176 ApiBitmask bm = var.value<ApiBitmask>();
177 Trace::BitmaskSig *sig = createBitmaskSig(bm, ++id);
178 writer.writeBitmask(sig, bm.value());
179 deleteBitmaskSig(sig);
180 } else if (type == structType) {
181 ApiStruct apiStr = var.value<ApiStruct>();
182 QList<QVariant> vals = apiStr.values();
183 Trace::StructSig *str = createStructSig(apiStr, ++id);
184 writer.beginStruct(str);
185 foreach(QVariant val, vals) {
186 writeValue(writer, val, ++id);
189 deleteStructSig(str);
190 } else if (type == pointerType) {
191 ApiPointer apiPtr = var.value<ApiPointer>();
192 //writer.beginArray(1);
193 //writer.beginElement();
194 writer.writeOpaque((const void*)apiPtr.value());
195 //writer.endElement();
197 } else if (type == enumType) {
198 ApiEnum apiEnum = var.value<ApiEnum>();
199 Trace::EnumSig *sig = createEnumSig(apiEnum, ++id);
200 writer.writeEnum(sig);
203 qWarning()<<"Unsupported write variant : "
204 << QMetaType::typeName(type);
210 class EditVisitor : public Trace::Visitor
213 EditVisitor(const QVariant &variant)
214 : m_variant(variant),
217 virtual void visit(Trace::Null *val)
222 virtual void visit(Trace::Bool *node)
224 // Q_ASSERT(m_variant.userType() == QVariant::Bool);
225 bool var = m_variant.toBool();
226 m_editedValue = new Trace::Bool(var);
229 virtual void visit(Trace::SInt *node)
231 // Q_ASSERT(m_variant.userType() == QVariant::Int);
232 m_editedValue = new Trace::SInt(m_variant.toInt());
235 virtual void visit(Trace::UInt *node)
237 // Q_ASSERT(m_variant.userType() == QVariant::UInt);
238 m_editedValue = new Trace::SInt(m_variant.toUInt());
241 virtual void visit(Trace::Float *node)
243 m_editedValue = new Trace::Float(m_variant.toFloat());
246 virtual void visit(Trace::String *node)
248 QString str = m_variant.toString();
249 m_editedValue = new Trace::String(str.toLocal8Bit().constData());
252 virtual void visit(Trace::Enum *e)
257 virtual void visit(Trace::Bitmask *bitmask)
259 m_editedValue = bitmask;
262 virtual void visit(Trace::Struct *str)
267 virtual void visit(Trace::Array *array)
269 ApiArray apiArray = m_variant.value<ApiArray>();
270 QVector<QVariant> vals = apiArray.values();
272 Trace::Array *newArray = new Trace::Array(vals.count());
273 for (int i = 0; i < vals.count(); ++i) {
274 EditVisitor visitor(vals[i]);
276 array->values[i]->visit(visitor);
277 if (array->values[i] == visitor.value()) {
280 m_editedValue = array;
284 newArray->values.push_back(visitor.value());
286 m_editedValue = newArray;
289 virtual void visit(Trace::Blob *blob)
291 m_editedValue = blob;
294 virtual void visit(Trace::Pointer *ptr)
299 Trace::Value *value() const
301 return m_editedValue;
305 Trace::Value *m_editedValue;
309 overwriteValue(Trace::Call *call, const QVariant &val, int index)
311 EditVisitor visitor(val);
312 Trace::Value *origValue = call->args[index];
313 origValue->visit(visitor);
315 if (visitor.value() && origValue != visitor.value()) {
317 call->args[index] = visitor.value();
321 SaverThread::SaverThread(QObject *parent)
326 void SaverThread::saveFile(const QString &writeFileName,
327 const QString &readFileName,
328 const QSet<ApiTraceCall*> &editedCalls)
330 m_writeFileName = writeFileName;
331 m_readFileName = readFileName;
332 m_editedCalls = editedCalls;
336 void SaverThread::run()
338 qDebug() << "Saving " << m_readFileName
339 << ", to " << m_writeFileName;
340 QMap<int, ApiTraceCall*> callIndexMap;
342 foreach(ApiTraceCall *call, m_editedCalls) {
343 callIndexMap.insert(call->index(), call);
346 Trace::Writer writer;
347 writer.open(m_writeFileName.toLocal8Bit());
349 Trace::Parser parser;
350 parser.open(m_readFileName.toLocal8Bit());
353 while ((call = parser.parse_call())) {
354 if (callIndexMap.contains(call->no)) {
355 QVector<QVariant> values = callIndexMap[call->no]->editedValues();
356 for (int i = 0; i < values.count(); ++i) {
357 const QVariant &val = values[i];
358 overwriteValue(call, val, i);
360 writer.writeCall(call);
362 writer.writeCall(call);
371 #include "saverthread.moc"