src/TortoiseMerge/Undo.cpp

   1 // TortoiseMerge - a Diff/Patch program
   2
   3 // Copyright (C) 2006-2007 - TortoiseSVN
   4 // Copyright (C) 2011 Sven Strickroth <email@cs-ware.de>
   5
   6 // This program is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU General Public License
   8 // as published by the Free Software Foundation; either version 2
   9 // of the License, or (at your option) any later version.
  10
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software Foundation,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20
  21 #include "StdAfx.h"
  22 #include "Undo.h"
  23
  24 #include "BaseView.h"
  25
  26 void viewstate::AddLineFormView(CBaseView *pView, int nLine, bool bAddEmptyLine)
  27 {
  28         if (!pView || !pView->m_pViewData)
  29                 return;
  30         difflines[nLine] = pView->m_pViewData->GetLine(nLine);
  31         linestates[nLine] = pView->m_pViewData->GetState(nLine);
  32         linesEOL[nLine] = pView->m_pViewData->GetLineEnding(nLine);
  33         if (bAddEmptyLine)
  34         {
  35                 addedlines.push_back(nLine + 1);
  36                 pView->AddEmptyLine(nLine);
  37         }
  38 }
  39
  40 CUndo& CUndo::GetInstance()
  41 {
  42         static CUndo instance;
  43         return instance;
  44 }
  45
  46 CUndo::CUndo()
  47 {
  48         m_originalstate = 0;
  49 }
  50
  51 CUndo::~CUndo()
  52 {
  53 }
  54
  55 void CUndo::AddState(const viewstate& leftstate, const viewstate& rightstate, const viewstate& bottomstate, POINT pt)
  56 {
  57         m_viewstates.push_back(bottomstate);
  58         m_viewstates.push_back(rightstate);
  59         m_viewstates.push_back(leftstate);
  60         m_caretpoints.push_back(pt);
  61 }
  62
  63 bool CUndo::Undo(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
  64 {
  65         if (!CanUndo())
  66                 return false;
  67
  68         if (!m_groups.empty() && m_groups.back() == m_caretpoints.size())
  69         {
  70                 m_groups.pop_back();
  71                 std::list<int>::size_type b = m_groups.back();
  72                 m_groups.pop_back();
  73                 while (b < m_caretpoints.size())
  74                         UndoOne(pLeft, pRight, pBottom);
  75         }
  76         else
  77                 UndoOne(pLeft, pRight, pBottom);
  78
  79         CBaseView * pActiveView = pLeft;
  80
  81         if (pRight && pRight->HasCaret())
  82                 pActiveView = pRight;
  83
  84         if (pBottom && pBottom->HasCaret())
  85                 pActiveView = pBottom;
  86
  87         if (pActiveView) {
  88                 pActiveView->ClearSelection();
  89                 pActiveView->EnsureCaretVisible();
  90                 pActiveView->UpdateCaret();
  91                 pActiveView->SetModified(m_viewstates.size() != m_originalstate);
  92                 pActiveView->RefreshViews();
  93         }
  94
  95         if (m_viewstates.size() < m_originalstate)
  96                 // Can never get back to original state now
  97                 m_originalstate = 1; // size() is always a multiple of 3
  98
  99         return true;
 100 }
 101
 102 void CUndo::UndoOne(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
 103 {
 104         viewstate state = m_viewstates.back();
 105         Undo(state, pLeft);
 106         m_viewstates.pop_back();
 107         state = m_viewstates.back();
 108         Undo(state, pRight);
 109         m_viewstates.pop_back();
 110         state = m_viewstates.back();
 111         Undo(state, pBottom);
 112         m_viewstates.pop_back();
 113         if ((pLeft)&&(pLeft->HasCaret()))
 114         {
 115                 pLeft->SetCaretPosition(m_caretpoints.back());
 116                 pLeft->EnsureCaretVisible();
 117         }
 118         if ((pRight)&&(pRight->HasCaret()))
 119         {
 120                 pRight->SetCaretPosition(m_caretpoints.back());
 121                 pRight->EnsureCaretVisible();
 122         }
 123         if ((pBottom)&&(pBottom->HasCaret()))
 124         {
 125                 pBottom->SetCaretPosition(m_caretpoints.back());
 126                 pBottom->EnsureCaretVisible();
 127         }
 128         m_caretpoints.pop_back();
 129 }
 130
 131 void CUndo::Undo(const viewstate& state, CBaseView * pView)
 132 {
 133         if (!pView)
 134                 return;
 135
 136         for (std::list<int>::const_iterator it = state.addedlines.begin(); it != state.addedlines.end(); ++it)
 137         {
 138                 if (pView->m_pViewData)
 139                         pView->m_pViewData->RemoveData(*it);
 140         }
 141         for (std::map<int, DWORD>::const_iterator it = state.linelines.begin(); it != state.linelines.end(); ++it)
 142         {
 143                 if (pView->m_pViewData)
 144                 {
 145                         pView->m_pViewData->SetLineNumber(it->first, it->second);
 146                 }
 147         }
 148         for (std::map<int, DWORD>::const_iterator it = state.linestates.begin(); it != state.linestates.end(); ++it)
 149         {
 150                 if (pView->m_pViewData)
 151                 {
 152                         pView->m_pViewData->SetState(it->first, (DiffStates)it->second);
 153                 }
 154         }
 155         for (std::map<int, EOL>::const_iterator it = state.linesEOL.begin(); it != state.linesEOL.end(); ++it)
 156         {
 157                 if (pView->m_pViewData)
 158                 {
 159                         pView->m_pViewData->SetLineEnding(it->first, (EOL)it->second);
 160                 }
 161         }
 162         for (std::map<int, CString>::const_iterator it = state.difflines.begin(); it != state.difflines.end(); ++it)
 163         {
 164                 if (pView->m_pViewData)
 165                 {
 166                         pView->m_pViewData->SetLine(it->first, it->second);
 167                 }
 168         }
 169         for (std::map<int, viewdata>::const_iterator it = state.removedlines.begin(); it != state.removedlines.end(); ++it)
 170         {
 171                 if (pView->m_pViewData)
 172                 {
 173                         pView->m_pViewData->InsertData(it->first, it->second.sLine, it->second.state, it->second.linenumber, it->second.ending);
 174                 }
 175         }
 176 }
 177
 178
 179 void CUndo::Clear()
 180 {
 181         m_viewstates.clear();
 182         m_caretpoints.clear();
 183         m_groups.clear();
 184         m_originalstate = 0;
 185 }