src/qt/walletmodeltransaction.cpp

   1 // Copyright (c) 2011-2016 The Bitcoin Core developers
   2 // Distributed under the MIT software license, see the accompanying
   3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #include "walletmodeltransaction.h"
   6
   7 #include "policy/policy.h"
   8 #include "wallet/wallet.h"
   9
  10 WalletModelTransaction::WalletModelTransaction(const QList<SendCoinsRecipient> &_recipients) :
  11     recipients(_recipients),
  12     walletTransaction(0),
  13     keyChange(0),
  14     fee(0)
  15 {
  16     walletTransaction = new CWalletTx();
  17 }
  18
  19 WalletModelTransaction::~WalletModelTransaction()
  20 {
  21     delete keyChange;
  22     delete walletTransaction;
  23 }
  24
  25 QList<SendCoinsRecipient> WalletModelTransaction::getRecipients()
  26 {
  27     return recipients;
  28 }
  29
  30 CWalletTx *WalletModelTransaction::getTransaction()
  31 {
  32     return walletTransaction;
  33 }
  34
  35 unsigned int WalletModelTransaction::getTransactionSize()
  36 {
  37     return (!walletTransaction ? 0 : ::GetVirtualTransactionSize(*walletTransaction));
  38 }
  39
  40 CAmount WalletModelTransaction::getTransactionFee()
  41 {
  42     return fee;
  43 }
  44
  45 void WalletModelTransaction::setTransactionFee(const CAmount& newFee)
  46 {
  47     fee = newFee;
  48 }
  49
  50 void WalletModelTransaction::reassignAmounts(int nChangePosRet)
  51 {
  52     int i = 0;
  53     for (QList<SendCoinsRecipient>::iterator it = recipients.begin(); it != recipients.end(); ++it)
  54     {
  55         SendCoinsRecipient& rcp = (*it);
  56
  57         if (rcp.paymentRequest.IsInitialized())
  58         {
  59             CAmount subtotal = 0;
  60             const payments::PaymentDetails& details = rcp.paymentRequest.getDetails();
  61             for (int j = 0; j < details.outputs_size(); j++)
  62             {
  63                 const payments::Output& out = details.outputs(j);
  64                 if (out.amount() <= 0) continue;
  65                 if (i == nChangePosRet)
  66                     i++;
  67                 subtotal += walletTransaction->tx->vout[i].nValue;
  68                 i++;
  69             }
  70             rcp.amount = subtotal;
  71         }
  72         else // normal recipient (no payment request)
  73         {
  74             if (i == nChangePosRet)
  75                 i++;
  76             rcp.amount = walletTransaction->tx->vout[i].nValue;
  77             i++;
  78         }
  79     }
  80 }
  81
  82 CAmount WalletModelTransaction::getTotalTransactionAmount()
  83 {
  84     CAmount totalTransactionAmount = 0;
  85     for (const SendCoinsRecipient &rcp : recipients)
  86     {
  87         totalTransactionAmount += rcp.amount;
  88     }
  89     return totalTransactionAmount;
  90 }
  91
  92 void WalletModelTransaction::newPossibleKeyChange(CWallet *wallet)
  93 {
  94     keyChange = new CReserveKey(wallet);
  95 }
  96
  97 CReserveKey *WalletModelTransaction::getPossibleKeyChange()
  98 {
  99     return keyChange;
 100 }