Export functionality for transaction list

[novacoin.git] / src / qt / transactiontablemodel.h

diff --git a/src/qt/transactiontablemodel.h b/src/qt/transactiontablemodel.h
index 5974c0e..85bfeeb 100644 (file)
--- a/src/qt/transactiontablemodel.h
+++ b/src/qt/transactiontablemodel.h
@@ -4,34 +4,49 @@
 #include <QAbstractTableModel>
 #include <QStringList>
 
+class CWallet;
 class TransactionTablePriv;
 class TransactionRecord;
+class WalletModel;
 
 class TransactionTableModel : public QAbstractTableModel
 {
     Q_OBJECT
 public:
-    explicit TransactionTableModel(QObject *parent = 0);
+    explicit TransactionTableModel(CWallet* wallet, WalletModel *parent = 0);
     ~TransactionTableModel();
 
     enum {
         Status = 0,
         Date = 1,
-        Description = 2,
-        Debit = 3,
-        Credit = 4
+        Type = 2,
+        ToAddress = 3,
+        Amount = 4
     } ColumnIndex;
 
+    // Roles to get specific information from a transaction row
+    // These are independent of column
     enum {
+        // Type of transaction
         TypeRole = Qt::UserRole,
-        LongDescriptionRole = Qt::UserRole+1
+        // Date and time this transaction was created
+        DateRole,
+        // Long description (HTML format)
+        LongDescriptionRole,
+        // Address of transaction
+        AddressRole,
+        // Label of address related to transaction
+        LabelRole,
+        // Absolute net amount of transaction, for filtering
+        AbsoluteAmountRole,
+        // Unique identifier
+        TxIDRole,
+        // Is transaction confirmed?
+        ConfirmedRole,
+        // Formatted amount, without brackets when unconfirmed
+        FormattedAmountRole
     } RoleIndex;
 
-    /* TypeRole values */
-    static const QString Sent;
-    static const QString Received;
-    static const QString Other;
-
     int rowCount(const QModelIndex &parent) const;
     int columnCount(const QModelIndex &parent) const;
     QVariant data(const QModelIndex &index, int role) const;
@@ -39,14 +54,18 @@ public:
     Qt::ItemFlags flags(const QModelIndex &index) const;
     QModelIndex index(int row, int column, const QModelIndex & parent = QModelIndex()) const;
 private:
+    CWallet* wallet;
+    WalletModel *walletModel;
     QStringList columns;
     TransactionTablePriv *priv;
 
+    QString lookupAddress(const std::string &address) const;
     QVariant formatTxStatus(const TransactionRecord *wtx) const;
     QVariant formatTxDate(const TransactionRecord *wtx) const;
-    QVariant formatTxDescription(const TransactionRecord *wtx) const;
-    QVariant formatTxDebit(const TransactionRecord *wtx) const;
-    QVariant formatTxCredit(const TransactionRecord *wtx) const;
+    QVariant formatTxType(const TransactionRecord *wtx) const;
+    QVariant formatTxToAddress(const TransactionRecord *wtx) const;
+    QVariant formatTxAmount(const TransactionRecord *wtx, bool showUnconfirmed=true) const;
+    QVariant formatTxDecoration(const TransactionRecord *wtx) const;
 
 private slots:
     void update();