src/TortoiseProc/RevisionGraph/CollapseTreeUpward.cpp

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