Branch data Line data Source code
1 : : // Copyright (c) 2018-2021 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 <wallet/coincontrol.h>
6 : :
7 : : #include <common/args.h>
8 : :
9 : : namespace wallet {
10 [ + - ]: 2522 : CCoinControl::CCoinControl()
11 : : {
12 [ + - + - ]: 2522 : m_avoid_partial_spends = gArgs.GetBoolArg("-avoidpartialspends", DEFAULT_AVOIDPARTIALSPENDS);
13 : 2522 : }
14 : :
15 : 991136 : bool CCoinControl::HasSelected() const
16 : : {
17 : 991136 : return !m_selected.empty();
18 : : }
19 : :
20 : 43202 : bool CCoinControl::IsSelected(const COutPoint& outpoint) const
21 : : {
22 : 43202 : return m_selected.count(outpoint) > 0;
23 : : }
24 : :
25 : 12332 : bool CCoinControl::IsExternalSelected(const COutPoint& outpoint) const
26 : : {
27 : 12332 : const auto it = m_selected.find(outpoint);
28 [ + + + + ]: 12332 : return it != m_selected.end() && it->second.HasTxOut();
29 : : }
30 : :
31 : 76 : std::optional<CTxOut> CCoinControl::GetExternalOutput(const COutPoint& outpoint) const
32 : : {
33 : 76 : const auto it = m_selected.find(outpoint);
34 [ + - - + ]: 76 : if (it == m_selected.end() || !it->second.HasTxOut()) {
35 : 0 : return std::nullopt;
36 : : }
37 : 76 : return it->second.GetTxOut();
38 : : }
39 : :
40 : 3211 : PreselectedInput& CCoinControl::Select(const COutPoint& outpoint)
41 : : {
42 : 3211 : auto& input = m_selected[outpoint];
43 : 3211 : input.SetPosition(m_selection_pos);
44 : 3211 : ++m_selection_pos;
45 : 3211 : return input;
46 : : }
47 : 0 : void CCoinControl::UnSelect(const COutPoint& outpoint)
48 : : {
49 : 0 : m_selected.erase(outpoint);
50 : 0 : }
51 : :
52 : 0 : void CCoinControl::UnSelectAll()
53 : : {
54 : 0 : m_selected.clear();
55 : 0 : }
56 : :
57 : 894 : std::vector<COutPoint> CCoinControl::ListSelected() const
58 : : {
59 : 894 : std::vector<COutPoint> outpoints;
60 [ + - ]: 894 : std::transform(m_selected.begin(), m_selected.end(), std::back_inserter(outpoints),
61 : 11250 : [](const std::map<COutPoint, PreselectedInput>::value_type& pair) {
62 : 11250 : return pair.first;
63 : : });
64 : 894 : return outpoints;
65 : 0 : }
66 : :
67 : 2489 : void CCoinControl::SetInputWeight(const COutPoint& outpoint, int64_t weight)
68 : : {
69 : 2489 : m_selected[outpoint].SetInputWeight(weight);
70 : 2489 : }
71 : :
72 : 19408 : std::optional<int64_t> CCoinControl::GetInputWeight(const COutPoint& outpoint) const
73 : : {
74 : 19408 : const auto it = m_selected.find(outpoint);
75 [ + + ]: 19408 : return it != m_selected.end() ? it->second.GetInputWeight() : std::nullopt;
76 : : }
77 : :
78 : 13744 : std::optional<uint32_t> CCoinControl::GetSequence(const COutPoint& outpoint) const
79 : : {
80 : 13744 : const auto it = m_selected.find(outpoint);
81 [ + + ]: 13744 : return it != m_selected.end() ? it->second.GetSequence() : std::nullopt;
82 : : }
83 : :
84 : 13744 : std::pair<std::optional<CScript>, std::optional<CScriptWitness>> CCoinControl::GetScripts(const COutPoint& outpoint) const
85 : : {
86 : 13744 : const auto it = m_selected.find(outpoint);
87 [ + + ]: 13744 : return it != m_selected.end() ? m_selected.at(outpoint).GetScripts() : std::make_pair(std::nullopt, std::nullopt);
88 : : }
89 : :
90 : 27 : void PreselectedInput::SetTxOut(const CTxOut& txout)
91 : : {
92 : 27 : m_txout = txout;
93 : 27 : }
94 : :
95 : 76 : CTxOut PreselectedInput::GetTxOut() const
96 : : {
97 [ - + ]: 76 : assert(m_txout.has_value());
98 : 76 : return m_txout.value();
99 : : }
100 : :
101 : 2324 : bool PreselectedInput::HasTxOut() const
102 : : {
103 : 2324 : return m_txout.has_value();
104 : : }
105 : :
106 : 2489 : void PreselectedInput::SetInputWeight(int64_t weight)
107 : : {
108 : 2489 : m_weight = weight;
109 : 2489 : }
110 : :
111 : 9113 : std::optional<int64_t> PreselectedInput::GetInputWeight() const
112 : : {
113 : 9113 : return m_weight;
114 : : }
115 : :
116 : 2662 : void PreselectedInput::SetSequence(uint32_t sequence)
117 : : {
118 : 2662 : m_sequence = sequence;
119 : 2662 : }
120 : :
121 : 3449 : std::optional<uint32_t> PreselectedInput::GetSequence() const
122 : : {
123 : 3449 : return m_sequence;
124 : : }
125 : :
126 : 2662 : void PreselectedInput::SetScriptSig(const CScript& script)
127 : : {
128 : 2662 : m_script_sig = script;
129 : 2662 : }
130 : :
131 : 2662 : void PreselectedInput::SetScriptWitness(const CScriptWitness& script_wit)
132 : : {
133 : 2662 : m_script_witness = script_wit;
134 : 2662 : }
135 : :
136 : 0 : bool PreselectedInput::HasScripts() const
137 : : {
138 [ # # # # ]: 0 : return m_script_sig.has_value() || m_script_witness.has_value();
139 : : }
140 : :
141 : 3449 : std::pair<std::optional<CScript>, std::optional<CScriptWitness>> PreselectedInput::GetScripts() const
142 : : {
143 : 3449 : return {m_script_sig, m_script_witness};
144 : : }
145 : :
146 : 3211 : void PreselectedInput::SetPosition(unsigned int pos)
147 : : {
148 : 3211 : m_pos = pos;
149 : 3211 : }
150 : :
151 : 29783 : std::optional<unsigned int> PreselectedInput::GetPosition() const
152 : : {
153 : 29783 : return m_pos;
154 : : }
155 : : } // namespace wallet
|