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

   1 #include "saverthread.h"
   2
   3 #include "trace_writer.hpp"
   4
   5 #include <QFile>
   6 #include <QHash>
   7 #include <QUrl>
   8
   9 #include <QDebug>
  10
  11
  12 static Trace::FunctionSig *
  13 createFunctionSig(ApiTraceCall *call, unsigned id)
  14 {
  15     Trace::FunctionSig *sig = new Trace::FunctionSig();
  16
  17     sig->id = id;
  18     sig->name = qstrdup(call->name().toLocal8Bit());
  19
  20     QStringList args = call->argNames();
  21     sig->num_args = args.count();
  22     sig->arg_names = new const char*[args.count()];
  23     for (int i = 0; i < args.count(); ++i) {
  24         sig->arg_names[i] = qstrdup(args[i].toLocal8Bit());
  25     }
  26
  27     return sig;
  28 }
  29
  30 static void
  31 deleteFunctionSig(Trace::FunctionSig *sig)
  32 {
  33     for (int i = 0; i < sig->num_args; ++i) {
  34         delete [] sig->arg_names[i];
  35     }
  36     delete [] sig->arg_names;
  37     delete [] sig->name;
  38     delete sig;
  39 }
  40
  41 static Trace::StructSig *
  42 createStructSig(const ApiStruct &str, unsigned id)
  43 {
  44     ApiStruct::Signature aSig = str.signature();
  45
  46     Trace::StructSig *sig = new Trace::StructSig();
  47     sig->id = id;
  48     sig->name = qstrdup(aSig.name.toLocal8Bit());
  49     sig->num_members = aSig.memberNames.count();
  50     char **member_names = new char*[aSig.memberNames.count()];
  51     sig->member_names = (const char **)member_names;
  52     for (int i = 0; i < aSig.memberNames.count(); ++i) {
  53         member_names[i] = qstrdup(aSig.memberNames[i].toLocal8Bit());
  54     }
  55     return sig;
  56 }
  57
  58 static void
  59 deleteStructSig(Trace::StructSig *sig)
  60 {
  61     for (int i = 0; i < sig->num_members; ++i) {
  62         delete [] sig->member_names[i];
  63     }
  64     delete [] sig->member_names;
  65     delete [] sig->name;
  66     delete sig;
  67 }
  68
  69 static Trace::EnumSig *
  70 createEnumSig(const ApiEnum &en, unsigned id)
  71 {
  72     Trace::EnumSig *sig = new Trace::EnumSig();
  73
  74     sig->id = id;
  75     sig->name = qstrdup(en.name().toLocal8Bit());
  76     sig->value = en.value().toLongLong();
  77
  78     return sig;
  79 }
  80
  81 static void
  82 deleteEnumSig(Trace::EnumSig *sig)
  83 {
  84     delete [] sig->name;
  85     delete sig;
  86 }
  87
  88 static Trace::BitmaskSig *
  89 createBitmaskSig(const ApiBitmask &bt, unsigned id)
  90 {
  91     ApiBitmask::Signature bsig = bt.signature();
  92     ApiBitmask::Signature::const_iterator itr;
  93
  94     Trace::BitmaskSig *sig = new Trace::BitmaskSig();
  95     Trace::BitmaskFlag *flags = new Trace::BitmaskFlag[bsig.count()];
  96
  97     sig->id = id;
  98     sig->num_flags = bsig.count();
  99     sig->flags = flags;
 100
 101     int i = 0;
 102     for (itr = bsig.constBegin(); itr != bsig.constEnd(); ++itr, ++i) {
 103         flags[i].name = qstrdup(itr->first.toLocal8Bit());
 104         flags[i].value = itr->second;
 105     }
 106
 107     return sig;
 108 }
 109
 110 static void
 111 deleteBitmaskSig(Trace::BitmaskSig *sig)
 112 {
 113     for (int i = 0; i < sig->num_flags; ++i) {
 114         delete [] sig->flags[i].name;
 115     }
 116     delete [] sig->flags;
 117     delete sig;
 118 }
 119
 120 static void
 121 writeValue(Trace::Writer &writer, const QVariant &var, unsigned &id)
 122 {
 123     int arrayType   = QMetaType::type("ApiArray");
 124     int bitmaskType = QMetaType::type("ApiBitmask");
 125     int structType  = QMetaType::type("ApiStruct");
 126     int pointerType = QMetaType::type("ApiPointer");
 127     int enumType    = QMetaType::type("ApiEnum");
 128     int type = var.userType();
 129
 130     switch(type) {
 131     case QVariant::Bool:
 132         writer.writeBool(var.toBool());
 133         break;
 134     case QVariant::ByteArray: {
 135         QByteArray ba = var.toByteArray();
 136         writer.writeBlob((const void*)ba.constData(), ba.size());
 137     }
 138         break;
 139     case QVariant::Double:
 140         writer.writeDouble(var.toDouble());
 141         break;
 142     case QMetaType::Float:
 143         writer.writeFloat(var.toFloat());
 144         break;
 145     case QVariant::Int:
 146         writer.writeSInt(var.toInt());
 147         break;
 148     case QVariant::LongLong:
 149         writer.writeSInt(var.toLongLong());
 150         break;
 151     case QVariant::String: {
 152         QString str = var.toString();
 153         writer.writeString(str.toLocal8Bit().constData(), str.length());
 154     }
 155         break;
 156     case QVariant::UInt:
 157         writer.writeUInt(var.toInt());
 158         break;
 159     case QVariant::ULongLong:
 160         writer.writeUInt(var.toLongLong());
 161         break;
 162     default:
 163         if (type == arrayType) {
 164             ApiArray array = var.value<ApiArray>();
 165             QVector<QVariant> vals = array.values();
 166             writer.beginArray(vals.count());
 167             foreach(QVariant el, vals) {
 168                 writer.beginElement();
 169                 writeValue(writer, el, ++id);
 170                 writer.endElement();
 171             }
 172             writer.endArray();
 173         } else if (type == bitmaskType) {
 174             ApiBitmask bm = var.value<ApiBitmask>();
 175             Trace::BitmaskSig *sig = createBitmaskSig(bm, ++id);
 176             writer.writeBitmask(sig, bm.value());
 177             deleteBitmaskSig(sig);
 178         } else if (type == structType) {
 179             ApiStruct apiStr = var.value<ApiStruct>();
 180             QList<QVariant> vals = apiStr.values();
 181             Trace::StructSig *str = createStructSig(apiStr, ++id);
 182             writer.beginStruct(str);
 183             foreach(QVariant val, vals) {
 184                 writeValue(writer, val, ++id);
 185             }
 186             writer.endStruct();
 187             deleteStructSig(str);
 188         } else if (type == pointerType) {
 189             ApiPointer apiPtr = var.value<ApiPointer>();
 190             //writer.beginArray(1);
 191             //writer.beginElement();
 192             writer.writeOpaque((const void*)apiPtr.value());
 193             //writer.endElement();
 194             //writer.endArray();
 195         } else if (type == enumType) {
 196             ApiEnum apiEnum = var.value<ApiEnum>();
 197             Trace::EnumSig *sig = createEnumSig(apiEnum, ++id);
 198             writer.writeEnum(sig);
 199             deleteEnumSig(sig);
 200         } else {
 201             qWarning()<<"Unsupported write variant : "
 202                       << QMetaType::typeName(type);
 203         }
 204     }
 205 }
 206
 207 SaverThread::SaverThread(QObject *parent)
 208     : QThread(parent)
 209 {
 210 }
 211
 212 void SaverThread::saveFile(const QString &fileName,
 213                            const QVector<ApiTraceCall*> &calls)
 214 {
 215     m_fileName = fileName;
 216     m_calls = calls;
 217     start();
 218 }
 219
 220 void SaverThread::run()
 221 {
 222     unsigned id = 0;
 223     qDebug() << "Saving  : " << m_fileName;
 224     Trace::Writer writer;
 225     writer.open(m_fileName.toLocal8Bit());
 226     for (int i = 0; i < m_calls.count(); ++i) {
 227         ApiTraceCall *call = m_calls[i];
 228         Trace::FunctionSig *funcSig = createFunctionSig(call, ++id);
 229         unsigned callNo = writer.beginEnter(funcSig);
 230         {
 231             //args
 232             QVector<QVariant> vars = call->arguments();
 233             int index = 0;
 234             foreach(QVariant var, vars) {
 235                 writer.beginArg(index++);
 236                 writeValue(writer, var, ++id);
 237                 writer.endArg();
 238             }
 239         }
 240         writer.endEnter();
 241         writer.beginLeave(callNo);
 242         {
 243             QVariant ret = call->returnValue();
 244             if (!ret.isNull()) {
 245                 writer.beginReturn();
 246                 writeValue(writer, ret, ++id);
 247                 writer.endReturn();
 248             }
 249         }
 250         writer.endLeave();
 251
 252         deleteFunctionSig(funcSig);
 253     }
 254     writer.close();
 255
 256     emit traceSaved();
 257 }
 258
 259 #include "saverthread.moc"