src/TortoiseProc/RevisionGraph/CutTrees.cpp

   1 // TortoiseSVN - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2008 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #include "StdAfx.h"
  20 #include "CutTrees.h"
  21 #include "VisibleGraphNode.h"
  22 #include "GraphNodeState.h"
  23
  24 // construction
  25
  26 CCutTrees::CCutTrees
  27     ( CRevisionGraphOptionList& list
  28     , const CGraphNodeStates* nodeStates)
  29     : inherited (list)
  30     , nodeStates (nodeStates)
  31 {
  32 }
  33
  34 // implement IModificationOption
  35
  36 void CCutTrees::Apply (CVisibleGraph* graph, CVisibleGraphNode* node)
  37 {
  38     // short-cut: most nodes won't have a state info
  39
  40     DWORD state = nodeStates->GetFlags (node, true);
  41     if (state == 0)
  42         return;
  43
  44     // cut at this tree node, if requested by the node state
  45
  46     if (state & CGraphNodeStates::SPLIT_ABOVE)
  47     {
  48         node->MakeRoot (graph, false);
  49     }
  50
  51     if (state & CGraphNodeStates::SPLIT_BELOW)
  52     {
  53         CVisibleGraphNode* next = node->GetNext();
  54         if (next)
  55             next->MakeRoot (graph, false);
  56     }
  57 }