Lab12/TreeNode.h at master · scaneer/Lab12 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#if !defined (TREENODE_H)
#define TREENODE_H

   #include "Color.h"
   #include "Circle.h"
   #include "String.h"
   using CSC2110::String;
   #include "wx/wx.h"

   template < class T >
   class TreeNode
   {
      private:
         TreeNode* left;
         TreeNode* right;
         T* item;
      public:
         TreeNode(T* item);
         ~TreeNode();

         TreeNode<T>* getLeft();
         TreeNode<T>* getRight();
         T* getItem();

         void setLeft(TreeNode<T>* tNode);
         void setRight(TreeNode<T>* tNode);
         void setItem(T* item);

         void draw(wxDC& dc, int x, int y);
   };

   template < class T >
   TreeNode<T>::TreeNode(T* itm)
   {
      left = NULL;
      right = NULL;
      item = itm;
   }

   template < class T >
   TreeNode<T>::~TreeNode()
   {}

   template < class T >
   TreeNode<T>* TreeNode<T>::getLeft()
   {
      return left;
   }

   template < class T >
   TreeNode<T>* TreeNode<T>::getRight()
   {
      return right;
   }

   template < class T >
   T* TreeNode<T>::getItem()
   {
      return item;
   }

   template < class T >
   void TreeNode<T>::setLeft(TreeNode<T>* l)
   {
      left = l;
   }

   template < class T >
   void TreeNode<T>::setRight(TreeNode<T>* r)
   {
      right = r;
   }

   template < class T >
   void TreeNode<T>::setItem(T* itm)
   {
      item = itm;
   }

   template < class T >
   void TreeNode<T>::draw(wxDC& dc, int x, int y)
   {
      T* item = getItem();
      String* key = item->getKey();
      Color* color = new Color(0.886, 0.49, 0.067);
      Circle* circle = new Circle(color, 20, key);
      circle->draw(dc, x, y);

      delete circle;
   }

#endif