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.writePointer(apiPtr.value());
 193         } else if (type == enumType) {
 194             ApiEnum apiEnum = var.value<ApiEnum>();
 195             trace::EnumSig *sig = createEnumSig(apiEnum, ++id);
 196             writer.writeEnum(sig);
 197             deleteEnumSig(sig);
 198         } else {
 199             qWarning()<<"Unsupported write variant : "
 200                       << QMetaType::typeName(type);
 201         }
 202     }
 203 }
 204 #endif
 205
 206 class EditVisitor : public trace::Visitor
 207 {
 208 public:
 209     EditVisitor(const QVariant &variant)
 210         : m_variant(variant),
 211           m_editedValue(0)
 212     {}
 213     virtual void visit(trace::Null *val)
 214     {
 215         m_editedValue = val;
 216     }
 217
 218     virtual void visit(trace::Bool *node)
 219     {
 220 //        Q_ASSERT(m_variant.userType() == QVariant::Bool);
 221         bool var = m_variant.toBool();
 222         m_editedValue = new trace::Bool(var);
 223     }
 224
 225     virtual void visit(trace::SInt *node)
 226     {
 227 //        Q_ASSERT(m_variant.userType() == QVariant::Int);
 228         m_editedValue = new trace::SInt(m_variant.toInt());
 229     }
 230
 231     virtual void visit(trace::UInt *node)
 232     {
 233 //        Q_ASSERT(m_variant.userType() == QVariant::UInt);
 234         m_editedValue = new trace::SInt(m_variant.toUInt());
 235     }
 236
 237     virtual void visit(trace::Float *node)
 238     {
 239         m_editedValue = new trace::Float(m_variant.toFloat());
 240     }
 241
 242     virtual void visit(trace::Double *node)
 243     {
 244         m_editedValue = new trace::Double(m_variant.toDouble());
 245     }
 246
 247     virtual void visit(trace::String *node)
 248     {
 249         QString str = m_variant.toString();
 250         m_editedValue = new trace::String(str.toLocal8Bit().constData());
 251     }
 252
 253     virtual void visit(trace::Enum *e)
 254     {
 255         m_editedValue = e;
 256     }
 257
 258     virtual void visit(trace::Bitmask *bitmask)
 259     {
 260         m_editedValue = bitmask;
 261     }
 262
 263     virtual void visit(trace::Struct *str)
 264     {
 265         m_editedValue = str;
 266     }
 267
 268     virtual void visit(trace::Array *array)
 269     {
 270         ApiArray apiArray = m_variant.value<ApiArray>();
 271         QVector<QVariant> vals = apiArray.values();
 272
 273         trace::Array *newArray = new trace::Array(vals.count());
 274         for (int i = 0; i < vals.count(); ++i) {
 275             EditVisitor visitor(vals[i]);
 276
 277             array->values[i]->visit(visitor);
 278             if (array->values[i] == visitor.value()) {
 279                 //non-editabled
 280                 delete newArray;
 281                 m_editedValue = array;
 282                 return;
 283             }
 284
 285             newArray->values.push_back(visitor.value());
 286         }
 287         m_editedValue = newArray;
 288     }
 289
 290     virtual void visit(trace::Blob *blob)
 291     {
 292         m_editedValue = blob;
 293     }
 294
 295     virtual void visit(trace::Pointer *ptr)
 296     {
 297         m_editedValue = ptr;
 298     }
 299
 300     trace::Value *value() const
 301     {
 302         return m_editedValue;
 303     }
 304 private:
 305     QVariant m_variant;
 306     trace::Value *m_editedValue;
 307 };
 308
 309 static void
 310 overwriteValue(trace::Call *call, const QVariant &val, int index)
 311 {
 312     EditVisitor visitor(val);
 313     trace::Value *origValue = call->args[index].value;
 314     origValue->visit(visitor);
 315
 316     if (visitor.value() && origValue != visitor.value()) {
 317         delete origValue;
 318         call->args[index].value = visitor.value();
 319     }
 320 }
 321
 322 SaverThread::SaverThread(QObject *parent)
 323     : QThread(parent)
 324 {
 325 }
 326
 327 void SaverThread::saveFile(const QString &writeFileName,
 328                            const QString &readFileName,
 329                            const QSet<ApiTraceCall*> &editedCalls)
 330 {
 331     m_writeFileName = writeFileName;
 332     m_readFileName = readFileName;
 333     m_editedCalls = editedCalls;
 334     start();
 335 }
 336
 337 void SaverThread::run()
 338 {
 339     qDebug() << "Saving  " << m_readFileName
 340              << ", to " << m_writeFileName;
 341     QMap<int, ApiTraceCall*> callIndexMap;
 342
 343     foreach(ApiTraceCall *call, m_editedCalls) {
 344         callIndexMap.insert(call->index(), call);
 345     }
 346
 347     trace::Writer writer;
 348     writer.open(m_writeFileName.toLocal8Bit());
 349
 350     trace::Parser parser;
 351     parser.open(m_readFileName.toLocal8Bit());
 352
 353     trace::Call *call;
 354     while ((call = parser.parse_call())) {
 355         if (callIndexMap.contains(call->no)) {
 356             QVector<QVariant> values = callIndexMap[call->no]->editedValues();
 357             for (int i = 0; i < values.count(); ++i) {
 358                 const QVariant &val = values[i];
 359                 overwriteValue(call, val, i);
 360             }
 361             writer.writeCall(call);
 362         } else {
 363             writer.writeCall(call);
 364         }
 365     }
 366
 367     writer.close();
 368
 369     emit traceSaved();
 370 }
 371
 372 #include "saverthread.moc"