src/TortoiseMerge/Undo.cpp

   1 // TortoiseGitMerge - a Diff/Patch program
   2
   3 // Copyright (C) 2006-2007, 2010-2011 - 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::AddViewLineFromView(CBaseView *pView, int nViewLine, bool bAddEmptyLine)
  26 {
  27         // is undo good place for this ?
  28         if (!pView || !pView->m_pViewData)
  29                 return;
  30         replacedlines[nViewLine] = pView->m_pViewData->GetData(nViewLine);
  31         if (bAddEmptyLine)
  32         {
  33                 addedlines.push_back(nViewLine + 1);
  34                 pView->AddEmptyViewLine(nViewLine);
  35         }
  36 }
  37
  38 void viewstate::Clear()
  39 {
  40         difflines.clear();
  41         linestates.clear();
  42         linelines.clear();
  43         linesEOL.clear();
  44         addedlines.clear();
  45
  46         removedlines.clear();
  47         replacedlines.clear();
  48 }
  49
  50 CUndo& CUndo::GetInstance()
  51 {
  52         static CUndo instance;
  53         return instance;
  54 }
  55
  56 CUndo::CUndo()
  57 {
  58         m_originalstate = 0;
  59         m_groupCount = 0;
  60 }
  61
  62 CUndo::~CUndo()
  63 {
  64 }
  65
  66 void CUndo::AddState(const allviewstate& allstate, POINT pt)
  67 {
  68         m_viewstates.push_back(allstate);
  69         m_caretpoints.push_back(pt);
  70 }
  71
  72 bool CUndo::Undo(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
  73 {
  74         if (!CanUndo())
  75                 return false;
  76
  77         if (m_groups.size() && m_groups.back() == m_caretpoints.size())
  78         {
  79                 m_groups.pop_back();
  80                 std::list<int>::size_type b = m_groups.back();
  81                 m_groups.pop_back();
  82                 while (b < m_caretpoints.size())
  83                         UndoOne(pLeft, pRight, pBottom);
  84         }
  85         else
  86                 UndoOne(pLeft, pRight, pBottom);
  87
  88         CBaseView * pActiveView = NULL;
  89
  90         if (pBottom && pBottom->IsTarget())
  91         {
  92                 pActiveView = pBottom;
  93         }
  94         else
  95         if (pRight && pRight->IsTarget())
  96         {
  97                 pActiveView = pRight;
  98         }
  99         else
 100         //if (pLeft && pLeft->IsTarget())
 101         {
 102                 pActiveView = pLeft;
 103         }
 104
 105
 106         if (pActiveView) {
 107                 pActiveView->ClearSelection();
 108                 pActiveView->BuildAllScreen2ViewVector();
 109                 pActiveView->RecalcAllVertScrollBars();
 110                 pActiveView->RecalcAllHorzScrollBars();
 111                 pActiveView->EnsureCaretVisible();
 112                 pActiveView->UpdateCaret();
 113                 bool bModified = m_viewstates.size() != m_originalstate;
 114                 if (pLeft)
 115                         pLeft->SetModified(bModified);
 116                 if (pRight)
 117                         pRight->SetModified(bModified);
 118                 if (pBottom)
 119                         pBottom->SetModified(bModified);
 120                 pActiveView->RefreshViews();
 121         }
 122
 123         if (m_viewstates.size() < m_originalstate)
 124                 // Can never get back to original state now
 125                 m_originalstate = (size_t)-1;
 126
 127         return true;
 128 }
 129
 130 void CUndo::UndoOne(CBaseView * pLeft, CBaseView * pRight, CBaseView * pBottom)
 131 {
 132         allviewstate allstate = m_viewstates.back();
 133         POINT pt = m_caretpoints.back();
 134
 135         Undo(allstate.left, pLeft, pt);
 136         Undo(allstate.right, pRight, pt);
 137         Undo(allstate.bottom, pBottom, pt);
 138
 139         m_viewstates.pop_back();
 140         m_caretpoints.pop_back();
 141 }
 142
 143 void CUndo::Undo(const viewstate& state, CBaseView * pView, const POINT& pt)
 144 {
 145         if (!pView)
 146                 return;
 147
 148         CViewData* viewData = pView->m_pViewData;
 149         if (!viewData)
 150                 return;
 151
 152         for (std::list<int>::const_reverse_iterator it = state.addedlines.rbegin(); it != state.addedlines.rend(); ++it)
 153         {
 154                 viewData->RemoveData(*it);
 155         }
 156         for (std::map<int, DWORD>::const_iterator it = state.linelines.begin(); it != state.linelines.end(); ++it)
 157         {
 158                 viewData->SetLineNumber(it->first, it->second);
 159         }
 160         for (std::map<int, DWORD>::const_iterator it = state.linestates.begin(); it != state.linestates.end(); ++it)
 161         {
 162                 viewData->SetState(it->first, (DiffStates)it->second);
 163         }
 164         for (std::map<int, EOL>::const_iterator it = state.linesEOL.begin(); it != state.linesEOL.end(); ++it)
 165         {
 166                 viewData->SetLineEnding(it->first, (EOL)it->second);
 167         }
 168         for (std::map<int, CString>::const_iterator it = state.difflines.begin(); it != state.difflines.end(); ++it)
 169         {
 170                 viewData->SetLine(it->first, it->second);
 171         }
 172         for (std::map<int, viewdata>::const_iterator it = state.removedlines.begin(); it != state.removedlines.end(); ++it)
 173         {
 174                 viewData->InsertData(it->first, it->second);
 175         }
 176         for (std::map<int, viewdata>::const_iterator it = state.replacedlines.begin(); it != state.replacedlines.end(); ++it)
 177         {
 178                 viewData->SetData(it->first, it->second);
 179         }
 180
 181         if (pView->IsTarget())
 182         {
 183                 pView->SetCaretViewPosition(pt);
 184                 pView->EnsureCaretVisible();
 185         }
 186
 187
 188 }
 189
 190 void CUndo::Clear()
 191 {
 192         m_viewstates.clear();
 193         m_caretpoints.clear();
 194         m_groups.clear();
 195         m_originalstate = 0;
 196         m_groupCount = 0;
 197 }