git.cworth.org Git - apitrace/blob - gui/saverthread.cpp

   1 #include "saverthread.h"
   2
   3 #include "trace_writer.hpp"
   4 #include "trace_model.hpp"
   5 #include "trace_parser.hpp"
   6
   7 #include <QFile>
   8 #include <QHash>
   9 #include <QUrl>
  10
  11 #include <QDebug>
  12
  13 #if 0
  14 static Trace::FunctionSig *
  15 createFunctionSig(ApiTraceCall *call, unsigned id)
  16 {
  17     Trace::FunctionSig *sig = new Trace::FunctionSig();
  18
  19     sig->id = id;
  20     sig->name = qstrdup(call->name().toLocal8Bit());
  21
  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());
  27     }
  28
  29     return sig;
  30 }
  31
  32 static void
  33 deleteFunctionSig(Trace::FunctionSig *sig)
  34 {
  35     for (int i = 0; i < sig->num_args; ++i) {
  36         delete [] sig->arg_names[i];
  37     }
  38     delete [] sig->arg_names;
  39     delete [] sig->name;
  40     delete sig;
  41 }
  42
  43 static Trace::StructSig *
  44 createStructSig(const ApiStruct &str, unsigned id)
  45 {
  46     ApiStruct::Signature aSig = str.signature();
  47
  48     Trace::StructSig *sig = new Trace::StructSig();
  49     sig->id = id;
  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());
  56     }
  57     return sig;
  58 }
  59
  60 static void
  61 deleteStructSig(Trace::StructSig *sig)
  62 {
  63     for (int i = 0; i < sig->num_members; ++i) {
  64         delete [] sig->member_names[i];
  65     }
  66     delete [] sig->member_names;
  67     delete [] sig->name;
  68     delete sig;
  69 }
  70
  71 static Trace::EnumSig *
  72 createEnumSig(const ApiEnum &en, unsigned id)
  73 {
  74     Trace::EnumSig *sig = new Trace::EnumSig();
  75
  76     sig->id = id;
  77     sig->name = qstrdup(en.name().toLocal8Bit());
  78     sig->value = en.value().toLongLong();
  79
  80     return sig;
  81 }
  82
  83 static void
  84 deleteEnumSig(Trace::EnumSig *sig)
  85 {
  86     delete [] sig->name;
  87     delete sig;
  88 }
  89
  90 static Trace::BitmaskSig *
  91 createBitmaskSig(const ApiBitmask &bt, unsigned id)
  92 {
  93     ApiBitmask::Signature bsig = bt.signature();
  94     ApiBitmask::Signature::const_iterator itr;
  95
  96     Trace::BitmaskSig *sig = new Trace::BitmaskSig();
  97     Trace::BitmaskFlag *flags = new Trace::BitmaskFlag[bsig.count()];
  98
  99     sig->id = id;
 100     sig->num_flags = bsig.count();
 101     sig->flags = flags;
 102
 103     int i = 0;
 104     for (itr = bsig.constBegin(); itr != bsig.constEnd(); ++itr, ++i) {
 105         flags[i].name = qstrdup(itr->first.toLocal8Bit());
 106         flags[i].value = itr->second;
 107     }
 108
 109     return sig;
 110 }
 111
 112 static void
 113 deleteBitmaskSig(Trace::BitmaskSig *sig)
 114 {
 115     for (int i = 0; i < sig->num_flags; ++i) {
 116         delete [] sig->flags[i].name;
 117     }
 118     delete [] sig->flags;
 119     delete sig;
 120 }
 121
 122 static void
 123 writeValue(Trace::Writer &writer, const QVariant &var, unsigned &id)
 124 {
 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();
 131
 132     switch(type) {
 133     case QVariant::Bool:
 134         writer.writeBool(var.toBool());
 135         break;
 136     case QVariant::ByteArray: {
 137         QByteArray ba = var.toByteArray();
 138         writer.writeBlob((const void*)ba.constData(), ba.size());
 139     }
 140         break;
 141     case QVariant::Double:
 142         writer.writeDouble(var.toDouble());
 143         break;
 144     case QMetaType::Float:
 145         writer.writeFloat(var.toFloat());
 146         break;
 147     case QVariant::Int:
 148         writer.writeSInt(var.toInt());
 149         break;
 150     case QVariant::LongLong:
 151         writer.writeSInt(var.toLongLong());
 152         break;
 153     case QVariant::String: {
 154         QString str = var.toString();
 155         writer.writeString(str.toLocal8Bit().constData(), str.length());
 156     }
 157         break;
 158     case QVariant::UInt:
 159         writer.writeUInt(var.toInt());
 160         break;
 161     case QVariant::ULongLong:
 162         writer.writeUInt(var.toLongLong());
 163         break;
 164     default:
 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);
 172                 writer.endElement();
 173             }
 174             writer.endArray();
 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);
 187             }
 188             writer.endStruct();
 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();
 196             //writer.endArray();
 197         } else if (type == enumType) {
 198             ApiEnum apiEnum = var.value<ApiEnum>();
 199             Trace::EnumSig *sig = createEnumSig(apiEnum, ++id);
 200             writer.writeEnum(sig);
 201             deleteEnumSig(sig);
 202         } else {
 203             qWarning()<<"Unsupported write variant : "
 204                       << QMetaType::typeName(type);
 205         }
 206     }
 207 }
 208 #endif
 209
 210 class EditVisitor : public Trace::Visitor
 211 {
 212 public:
 213     EditVisitor(const QVariant &variant)
 214         : m_variant(variant),
 215           m_editedValue(0)
 216     {}
 217     virtual void visit(Trace::Null *val)
 218     {
 219         m_editedValue = val;
 220     }
 221
 222     virtual void visit(Trace::Bool *node)
 223     {
 224 //        Q_ASSERT(m_variant.userType() == QVariant::Bool);
 225         bool var = m_variant.toBool();
 226         m_editedValue = new Trace::Bool(var);
 227     }
 228
 229     virtual void visit(Trace::SInt *node)
 230     {
 231 //        Q_ASSERT(m_variant.userType() == QVariant::Int);
 232         m_editedValue = new Trace::SInt(m_variant.toInt());
 233     }
 234
 235     virtual void visit(Trace::UInt *node)
 236     {
 237 //        Q_ASSERT(m_variant.userType() == QVariant::UInt);
 238         m_editedValue = new Trace::SInt(m_variant.toUInt());
 239     }
 240
 241     virtual void visit(Trace::Float *node)
 242     {
 243         m_editedValue = new Trace::Float(m_variant.toFloat());
 244     }
 245
 246     virtual void visit(Trace::String *node)
 247     {
 248         QString str = m_variant.toString();
 249         m_editedValue = new Trace::String(str.toLocal8Bit().constData());
 250     }
 251
 252     virtual void visit(Trace::Enum *e)
 253     {
 254         m_editedValue = e;
 255     }
 256
 257     virtual void visit(Trace::Bitmask *bitmask)
 258     {
 259         m_editedValue = bitmask;
 260     }
 261
 262     virtual void visit(Trace::Struct *str)
 263     {
 264         m_editedValue = str;
 265     }
 266
 267     virtual void visit(Trace::Array *array)
 268     {
 269         ApiArray apiArray = m_variant.value<ApiArray>();
 270         QVector<QVariant> vals = apiArray.values();
 271
 272         Trace::Array *newArray = new Trace::Array(vals.count());
 273         for (int i = 0; i < vals.count(); ++i) {
 274             EditVisitor visitor(vals[i]);
 275
 276             array->values[i]->visit(visitor);
 277             if (array->values[i] == visitor.value()) {
 278                 //non-editabled
 279                 delete newArray;
 280                 m_editedValue = array;
 281                 return;
 282             }
 283
 284             newArray->values.push_back(visitor.value());
 285         }
 286         m_editedValue = newArray;
 287     }
 288
 289     virtual void visit(Trace::Blob *blob)
 290     {
 291         m_editedValue = blob;
 292     }
 293
 294     virtual void visit(Trace::Pointer *ptr)
 295     {
 296         m_editedValue = ptr;
 297     }
 298
 299     Trace::Value *value() const
 300     {
 301         return m_editedValue;
 302     }
 303 private:
 304     QVariant m_variant;
 305     Trace::Value *m_editedValue;
 306 };
 307
 308 static void
 309 overwriteValue(Trace::Call *call, const QVariant &val, int index)
 310 {
 311     EditVisitor visitor(val);
 312     Trace::Value *origValue = call->args[index];
 313     origValue->visit(visitor);
 314
 315     if (visitor.value() && origValue != visitor.value()) {
 316         delete origValue;
 317         call->args[index] = visitor.value();
 318     }
 319 }
 320
 321 SaverThread::SaverThread(QObject *parent)
 322     : QThread(parent)
 323 {
 324 }
 325
 326 void SaverThread::saveFile(const QString &writeFileName,
 327                            const QString &readFileName,
 328                            const QSet<ApiTraceCall*> &editedCalls)
 329 {
 330     m_writeFileName = writeFileName;
 331     m_readFileName = readFileName;
 332     m_editedCalls = editedCalls;
 333     start();
 334 }
 335
 336 void SaverThread::run()
 337 {
 338     qDebug() << "Saving  " << m_readFileName
 339              << ", to " << m_writeFileName;
 340     QMap<int, ApiTraceCall*> callIndexMap;
 341
 342     foreach(ApiTraceCall *call, m_editedCalls) {
 343         callIndexMap.insert(call->index(), call);
 344     }
 345
 346     Trace::Writer writer;
 347     writer.open(m_writeFileName.toLocal8Bit());
 348
 349     Trace::Parser parser;
 350     parser.open(m_readFileName.toLocal8Bit());
 351
 352     Trace::Call *call;
 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);
 359             }
 360             writer.writeCall(call);
 361         } else {
 362             writer.writeCall(call);
 363         }
 364     }
 365
 366     writer.close();
 367
 368     emit traceSaved();
 369 }
 370
 371 #include "saverthread.moc"