Click here to Skip to main content
Click here to Skip to main content

Three State Treeview - Part 2

, 22 Mar 2014
Rate this:
Please Sign up or sign in to vote.
Treeview with Checkboxes supporting 3-state-logic

Introduction

My search for a Treeview with 3-state-checkboxes led me to this article. It explains the logic properly and promises a part 2, in which the problems about ownerdrawing the checkboxes should be solved.

But that part 2 has, for whatever reason, never been written. But the author allowed the audience to feel free to write part 2. So I felt free. Wink | ;)

Repeat the Logic

The user can only check or uncheck Treenodes - not set to indeterminate. Checking/unchecking a Node sets all childnodes to that new state. If a ParentNode contains nodes of different states, then it will display the Indeterminate - state.

The Code

The principle is to use the StateImageList-property with 3 Images: Unchecked, Checked, Indeterminate. The dual logic is done well by the Treeview as it is. Treeview uses the first two images properly to display Checked/Unchecked. Nevertheless I carefully set the proper StateImageIndices, although that's not needed (for dual logic).
But I need it to persist 3 states. When it comes to draw, I only have to draw the Indeterminated Checkbox.

A problem was that I need to use TreeViewDrawMode.OwnerDrawAll to figure out which node to draw. But I don't want to draw the nodes completely, because that's quite difficult (Checkbox, optional Icon, SelectedIcon, Text, SelectedText, Focus). I just want to add my Indeterminated-Checkbox, if necessary.
Unfortunately DrawMode.OwnerDrawAll disables the _Paint-Event, and there is no "AfterDrawNode"-Event. So I had to subclass the windowmessages, observing, when the WM_PAINT-windowmessage has passed. At that moment, I can draw my indeterminated-Checkboxes, and they will not be overdrawn by the Treeview.

So here you can look at the most important parts of the ThreeStateTreeview, and I hope, it is commented well enough to make more explanations redundant.

protected override void OnAfterCheck(TreeViewEventArgs e) {
   /* Logic: All children of an (un)checked Node inherit its Checkstate
    * Parents recompute their state: if all children of a parent have same state, 
    * that one will be taken over as parents state - otherwise take Indeterminate 
    */
   if(_skipCheckEvents) return;/* changing any Treenodes .Checked-Property will raise 
                                     another Before- and After-Check. Skip'em */
   _skipCheckEvents = true;
   try {
      TreeNode nd = e.Node;
      /* uninitialized Nodes have StateImageIndex -1, 
       * so I associate StateImageIndex as follows:
       * -1: Unchecked
       *  0: Checked
       *  1: Indeterminate
       *  That corresponds to the System.Windows.Forms.Checkstate - enumeration, 
       *  but 1 less.
       *  Furthermore I ordered the images in that manner
       */
      int state = nd.StateImageIndex == 0 ? -1 : 0;      /* this state is already toggled.
          Note: -1 (Unchecked) and 1 (Indeterminate) both toggle to 0, 
                        that means: Checked */
      if((state == 0) != nd.Checked) return;       //suppress redundant AfterCheck-event
      InheritCheckstate(nd, state);         // inherit Checkstate to children
      // Parents recompute their state
      nd = nd.Parent;
      while(nd != null) {
         // At Indeterminate (==1) skip the children-query - 
         // every parent becomes Indeterminate
         if(state != 1) {
            foreach(TreeNode ndChild in nd.Nodes) {
               if(ndChild.StateImageIndex != state) {
                  state = 1;
                  break;
               }
            }
         }
         AssignState(nd, state);
         nd = nd.Parent;
      }
      base.OnAfterCheck(e);
   } finally { _skipCheckEvents = false; }
}

private void AssignState(TreeNode nd, int state) {
   bool ck = state == 0;
   bool stateInvalid = nd.StateImageIndex != state;
   if(stateInvalid) nd.StateImageIndex = state;
   if(nd.Checked != ck) {
      nd.Checked = ck;                // changing .Checked-Property raises 
                // Invalidating internally
   } else if(stateInvalid) {
      // in general: the less and small the invalidated area, the less flickering
      // so avoid calling Invalidate() if possible, and only call, if really needed.
      this.Invalidate(GetCheckRect(nd));
   }
}

private void InheritCheckstate(TreeNode nd, int state) {
   AssignState(nd, state);
   foreach(TreeNode ndChild in nd.Nodes) {
      InheritCheckstate(ndChild, state);
   }
}

public System.Windows.Forms.CheckState GetState(TreeNode nd) {
   // compute the System.Windows.Forms.CheckState from a StateImageIndex 
   // is not that complicated
   return (CheckState)nd.StateImageIndex + 1;
}

protected override void OnDrawNode(DrawTreeNodeEventArgs e) {
   // here nothing is drawn. Only collect Indeterminated Nodes, 
   // to draw them later (in WndProc())
   // because drawing Treenodes properly (Text, Icon(s) Focus, Selection...) 
   // is very complicated
   if(e.Node.StateImageIndex == 1) _indeterminateds.Add(e.Node);
   e.DrawDefault = true;
   base.OnDrawNode(e);
}

protected override void WndProc(ref Message m) {
   const int WM_Paint = 15;
   base.WndProc(ref m);
   if(m.Msg == WM_Paint) {
      // at that point built-in drawing is completed - 
      // and I paint over the Indeterminate-Checkboxes
      foreach(TreeNode nd in _indeterminateds) {
         _graphics.DrawImage(_imgIndeterminate, GetCheckRect(nd).Location);
      }
      _indeterminateds.Clear();
   }
}

Credits

  • Three State Treeview - Part 1 - Although I didn't use a line of that code, it gave me the idea of how to synchronize the Checked-Property with the 3 options of StateImageIndex, and how to avoid multiple Before-/After-Check-Events while updating the Treenode states.

History

  • 1st April, 2009: Initial post
  • 18th May, 2010: Bugfix: Christo667 reported a well hidden bug, when programmatical set a nodes Checked-property to the same value, it had before (see on Message-board). The bug-reason was, in that case the common TreeView raises redundant Before-/After-Checked-Events, and ThreeStateTreeview toggled the nodes appearance, although it shouldn't.
    Now ThreeStateTreeview suppresses those redundant Events. That may be a bug-workaround for the common TreeView as well.
    Thank you, Christo!
  • BugFix of Standard-Treeview, when doubleclicking the Checkbox of a node.

License

This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)

Share

About the Author

Mr.PoorEnglish

Germany Germany
No Biography provided

Comments and Discussions

 
GeneralLot of flicker on re-size Pinmembersurya chundru14-Jan-11 10:56 
QuestionCan this work in a x64 OS? PinmemberCody Tang30-Sep-10 17:13 
Answersorry - I don't know PinmemberMr.PoorEnglish30-Sep-10 19:48 
GeneralRe: sorry - I don't know PinmemberCody Tang3-Oct-10 20:42 
Questionwhich is better control ?? Pinmemberalhambra-eidos15-Jun-10 21:07 
AnswerRe: which is better control ?? [modified] PinmemberCody Tang30-Sep-10 17:19 
GeneralThree State Treeview - Part 1 not found !!! Pinmemberalhambra-eidos15-Jun-10 20:23 
GeneralHas a problem when leaf nodes not checked consistently. PinmemberChristo66717-May-10 12:10 
GeneralRe: Has a problem when leaf nodes not checked consistently (on initialization) [modified] PinmemberChristo66717-May-10 13:20 
GeneralFixed PinmemberMr.PoorEnglish18-May-10 10:59 
GeneralRe: Has a problem when leaf nodes not checked consistently (on initialization) PinmemberChristo66718-May-10 12:28 
General"HerrGrottigDeutsche" PinmemberMr.PoorEnglish18-May-10 12:59 
GeneralArgh, seems to have bugs if you rapidly double click nodes Pinmembergrahamoneale29-Apr-10 17:37 
GeneralRe: Argh, seems to have bugs if you rapidly double click nodes PinmemberMr.PoorEnglish30-Apr-10 3:10 
GeneralRe: Argh, seems to have bugs if you rapidly double click nodes PinmemberMr.PoorEnglish30-Apr-10 7:37 
GeneralAjaj Pinmemberkonikula6-Jan-10 21:38 
GeneralPS Pinmemberkonikula6-Jan-10 22:01 
GeneralDraw standard indeterminated checkboxes PinmemberMatthias Frensel26-Nov-09 2:44 
GeneralRe: Draw standard indeterminated checkboxes PinmemberChristo66719-May-10 5:46 
SuggestionRe: Draw standard indeterminated checkboxes Pinmemberjeffyoung123430-Aug-11 23:05 
General(hopefully) fixed PinmemberMr.PoorEnglish31-Aug-11 0:02 
GeneralOne bug Pinmembera010405022-Nov-09 2:39 
GeneralRe: One bug PinmemberMr.PoorEnglish22-Nov-09 5:40 
GeneralRe: One bug Pinmembera010405022-Nov-09 6:59 
GeneralRe: One bug PinmemberMr.PoorEnglish22-Nov-09 7:38 
QuestionHow to make the checkboxes look like real Windows checkboxes Pinmemberjdmwood21-May-09 6:18 
AnswerRe: How to make the checkboxes look like real Windows checkboxes Pinmembergrahamoneale29-Apr-10 14:15 
GeneralRe: How to make the checkboxes look like real Windows checkboxes Pinmemberjdmwood224-Jun-10 4:24 

General General    News News    Suggestion Suggestion    Question Question    Bug Bug    Answer Answer    Joke Joke    Rant Rant    Admin Admin   

Use Ctrl+Left/Right to switch messages, Ctrl+Up/Down to switch threads, Ctrl+Shift+Left/Right to switch pages.

| Advertise | Privacy | Mobile
Web03 | 2.8.140827.1 | Last Updated 22 Mar 2014
Article Copyright 2009 by Mr.PoorEnglish
Everything else Copyright © CodeProject, 1999-2014
Terms of Service
Layout: fixed | fluid