src/TortoiseMerge/Undo.cpp

   1 // TortoiseMerge - a Diff/Patch program
   2
   3 // Copyright (C) 2006-2007 - 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
  20 #include "StdAfx.h"
  21 #include "Undo.h"
  22
  23 #include "BaseView.h"
  24
  25 void viewstate::AddLineFormView(CBaseView *pView, int nLine, bool bAddEmptyLine)
  26 {
  27         if (!pView || !pView->m_pViewData)
  28                 return;
  29         difflines[nLine] = pView->m_pViewData->GetLine(nLine);
  30         linestates[nLine] = pView->m_pViewData->GetState(nLine);
  31         if (bAddEmptyLine)
  32         {
  33                 addedlines.push_back(nLine + 1);
  34                 pView->AddEmptyLine(nLine);
  35         }
  36 }
  37
  38 CUndo& CUndo::GetInstance()
  39 {
  40         static CUndo instance;
  41         return instance;
  42 }
  43
  44 CUndo::CUndo()
  45 {
  46         m_originalstate = 0;
  47 }
  48
  49 CUndo::~CUndo()
  50 {
  51 }
  52
  53 void CUndo::AddState(const viewstate& leftstate, const viewstate& rightstate, const viewstate& bottomstate, POINT pt)
  54 {
  55         m_viewstates.push_back(bottomstate);
  56         m_viewstates.push_back(rightstate);
  57         m_viewstates.push_back(leftstate);
  58         m_caretpoints.push_back(pt);
  59 }
  60
  61 bool CUndo::Undo(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
  62 {
  63         if (!CanUndo())
  64                 return false;
  65
  66         if (m_groups.size() && m_groups.back() == m_caretpoints.size())
  67         {
  68                 m_groups.pop_back();
  69                 std::list<int>::size_type b = m_groups.back();
  70                 m_groups.pop_back();
  71                 while (b < m_caretpoints.size())
  72                         UndoOne(pLeft, pRight, pBottom);
  73         }
  74         else
  75                 UndoOne(pLeft, pRight, pBottom);
  76
  77         CBaseView * pActiveView = pLeft;
  78
  79         if (pRight && pRight->HasCaret())
  80                 pActiveView = pRight;
  81
  82         if (pBottom && pBottom->HasCaret())
  83                 pActiveView = pBottom;
  84
  85         if (pActiveView) {
  86                 pActiveView->ClearSelection();
  87                 pActiveView->EnsureCaretVisible();
  88                 pActiveView->UpdateCaret();
  89                 pActiveView->SetModified(m_viewstates.size() != m_originalstate);
  90                 pActiveView->RefreshViews();
  91         }
  92
  93         if (m_viewstates.size() < m_originalstate)
  94                 // Can never get back to original state now
  95                 m_originalstate = 1; // size() is always a multiple of 3
  96
  97         return true;
  98 }
  99
 100 void CUndo::UndoOne(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
 101 {
 102         viewstate state = m_viewstates.back();
 103         Undo(state, pLeft);
 104         m_viewstates.pop_back();
 105         state = m_viewstates.back();
 106         Undo(state, pRight);
 107         m_viewstates.pop_back();
 108         state = m_viewstates.back();
 109         Undo(state, pBottom);
 110         m_viewstates.pop_back();
 111         if ((pLeft)&&(pLeft->HasCaret()))
 112         {
 113                 pLeft->SetCaretPosition(m_caretpoints.back());
 114                 pLeft->EnsureCaretVisible();
 115         }
 116         if ((pRight)&&(pRight->HasCaret()))
 117         {
 118                 pRight->SetCaretPosition(m_caretpoints.back());
 119                 pRight->EnsureCaretVisible();
 120         }
 121         if ((pBottom)&&(pBottom->HasCaret()))
 122         {
 123                 pBottom->SetCaretPosition(m_caretpoints.back());
 124                 pBottom->EnsureCaretVisible();
 125         }
 126         m_caretpoints.pop_back();
 127 }
 128
 129 void CUndo::Undo(const viewstate& state, CBaseView * pView)
 130 {
 131         if (!pView)
 132                 return;
 133
 134         for (std::list<int>::const_iterator it = state.addedlines.begin(); it != state.addedlines.end(); ++it)
 135         {
 136                 if (pView->m_pViewData)
 137                         pView->m_pViewData->RemoveData(*it);
 138         }
 139         for (std::map<int, DWORD>::const_iterator it = state.linelines.begin(); it != state.linelines.end(); ++it)
 140         {
 141                 if (pView->m_pViewData)
 142                 {
 143                         pView->m_pViewData->SetLineNumber(it->first, it->second);
 144                 }
 145         }
 146         for (std::map<int, DWORD>::const_iterator it = state.linestates.begin(); it != state.linestates.end(); ++it)
 147         {
 148                 if (pView->m_pViewData)
 149                 {
 150                         pView->m_pViewData->SetState(it->first, (DiffStates)it->second);
 151                 }
 152         }
 153         for (std::map<int, CString>::const_iterator it = state.difflines.begin(); it != state.difflines.end(); ++it)
 154         {
 155                 if (pView->m_pViewData)
 156                 {
 157                         pView->m_pViewData->SetLine(it->first, it->second);
 158                 }
 159         }
 160         for (std::map<int, viewdata>::const_iterator it = state.removedlines.begin(); it != state.removedlines.end(); ++it)
 161         {
 162                 if (pView->m_pViewData)
 163                 {
 164                         pView->m_pViewData->InsertData(it->first, it->second.sLine, it->second.state, it->second.linenumber, it->second.ending);
 165                 }
 166         }
 167 }
 168
 169
 170 void CUndo::Clear()
 171 {
 172         m_viewstates.clear();
 173         m_caretpoints.clear();
 174         m_groups.clear();
 175         m_originalstate = 0;
 176 }