2014-11-18 03:02:13 -08:00
|
|
|
package ptrie
|
|
|
|
|
2014-12-23 09:35:36 -08:00
|
|
|
import "fmt"
|
|
|
|
|
2014-11-18 03:02:13 -08:00
|
|
|
type FullNode struct {
|
|
|
|
trie *Trie
|
|
|
|
nodes [17]Node
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewFullNode(t *Trie) *FullNode {
|
|
|
|
return &FullNode{trie: t}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *FullNode) Dirty() bool { return true }
|
|
|
|
func (self *FullNode) Value() Node {
|
|
|
|
self.nodes[16] = self.trie.trans(self.nodes[16])
|
|
|
|
return self.nodes[16]
|
|
|
|
}
|
2014-11-19 06:05:08 -08:00
|
|
|
func (self *FullNode) Branches() []Node {
|
|
|
|
return self.nodes[:16]
|
|
|
|
}
|
2014-11-18 03:02:13 -08:00
|
|
|
|
2014-11-19 07:21:28 -08:00
|
|
|
func (self *FullNode) Copy() Node {
|
|
|
|
nnode := NewFullNode(self.trie)
|
|
|
|
for i, node := range self.nodes {
|
|
|
|
nnode.nodes[i] = node
|
|
|
|
}
|
|
|
|
|
|
|
|
return nnode
|
|
|
|
}
|
2014-11-18 03:02:13 -08:00
|
|
|
|
|
|
|
// Returns the length of non-nil nodes
|
|
|
|
func (self *FullNode) Len() (amount int) {
|
|
|
|
for _, node := range self.nodes {
|
|
|
|
if node != nil {
|
|
|
|
amount++
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *FullNode) Hash() interface{} {
|
|
|
|
return self.trie.store(self)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *FullNode) RlpData() interface{} {
|
|
|
|
t := make([]interface{}, 17)
|
|
|
|
for i, node := range self.nodes {
|
|
|
|
if node != nil {
|
|
|
|
t[i] = node.Hash()
|
|
|
|
} else {
|
|
|
|
t[i] = ""
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return t
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *FullNode) set(k byte, value Node) {
|
2014-12-23 09:35:36 -08:00
|
|
|
if _, ok := value.(*ValueNode); ok && k != 16 {
|
|
|
|
fmt.Println(value, k)
|
|
|
|
panic(":(")
|
|
|
|
}
|
|
|
|
|
2014-11-18 03:02:13 -08:00
|
|
|
self.nodes[int(k)] = value
|
|
|
|
}
|
|
|
|
|
2014-11-19 06:05:08 -08:00
|
|
|
func (self *FullNode) branch(i byte) Node {
|
2014-11-18 03:02:13 -08:00
|
|
|
if self.nodes[int(i)] != nil {
|
|
|
|
self.nodes[int(i)] = self.trie.trans(self.nodes[int(i)])
|
|
|
|
|
|
|
|
return self.nodes[int(i)]
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|