src/TortoiseProc/RevisionGraph/RevisionGraphDlgFunc.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2011 - TortoiseSVN
   4 // Copyright (C) 2012-2013 - TortoiseGit
   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 #include "stdafx.h"
  21 #include "TortoiseProc.h"
  22 #include <gdiplus.h>
  23 #include "Revisiongraphdlg.h"
  24 #include "MessageBox.h"
  25 #include "Git.h"
  26 #include "TempFile.h"
  27 #include "UnicodeUtils.h"
  28 #include "TGitPath.h"
  29 //#include "SVNInfo.h"
  30 //#include ".\revisiongraphwnd.h"
  31 //#include "CachedLogInfo.h"
  32 //#include "RevisionGraph/IRevisionGraphLayout.h"
  33 //#include "RevisionGraph/FullGraphBuilder.h"
  34 //#include "RevisionGraph/FullGraphFinalizer.h"
  35 //#include "RevisionGraph/VisibleGraphBuilder.h"
  36 //#include "RevisionGraph/StandardLayout.h"
  37 //#include "RevisionGraph/ShowWC.h"
  38 //#include "RevisionGraph/ShowWCModification.h"
  39
  40 #ifdef _DEBUG
  41 #define new DEBUG_NEW
  42 #undef THIS_FILE
  43 static char THIS_FILE[] = __FILE__;
  44 #endif
  45
  46 using namespace Gdiplus;
  47 using namespace ogdf;
  48
  49 void CRevisionGraphWnd::InitView()
  50 {
  51         m_bIsCanvasMove = false;
  52
  53         SetScrollbars();
  54 }
  55
  56 void CRevisionGraphWnd::BuildPreview()
  57 {
  58
  59         m_Preview.DeleteObject();
  60         if (!m_bShowOverview)
  61                 return;
  62
  63         // is there a point in drawing this at all?
  64
  65         int nodeCount = this->m_Graph.numberOfNodes();
  66         if ((nodeCount > REVGRAPH_PREVIEW_MAX_NODES) || (nodeCount == 0))
  67                 return;
  68
  69         float origZoom = m_fZoomFactor;
  70
  71         CRect clientRect = GetClientRect();
  72         CSize preViewSize (max (REVGRAPH_PREVIEW_WIDTH, clientRect.Width() / 4)
  73                                           ,max (REVGRAPH_PREVIEW_HEIGHT, clientRect.Height() / 4));
  74
  75         // zoom the graph so that it is completely visible in the window
  76         CRect graphRect = GetGraphRect();
  77         float horzfact = float(graphRect.Width())/float(preViewSize.cx);
  78         float vertfact = float(graphRect.Height())/float(preViewSize.cy);
  79         m_previewZoom = min (DEFAULT_ZOOM, 1.0f/(max(horzfact, vertfact)));
  80
  81         // make sure the preview window has a minimal size
  82
  83         m_previewWidth = (int)min (max (graphRect.Width() * m_previewZoom, 30), preViewSize.cx);
  84         m_previewHeight = (int)min (max (graphRect.Height() * m_previewZoom, 30), preViewSize.cy);
  85
  86         CClientDC ddc(this);
  87         CDC dc;
  88         if (!dc.CreateCompatibleDC(&ddc))
  89                 return;
  90
  91         m_Preview.CreateCompatibleBitmap(&ddc, m_previewWidth, m_previewHeight);
  92         HBITMAP oldbm = (HBITMAP)dc.SelectObject (m_Preview);
  93
  94         // paint the whole graph
  95         DoZoom (m_previewZoom, false);
  96         CRect rect (0, 0, m_previewWidth, m_previewHeight);
  97         GraphicsDevice dev;
  98         dev.pDC = &dc;
  99         DrawGraph(dev, rect, 0, 0, true);
 100
 101         // now we have a bitmap the size of the preview window
 102         dc.SelectObject(oldbm);
 103         dc.DeleteDC();
 104
 105         DoZoom (origZoom, false);
 106
 107 }
 108
 109 void CRevisionGraphWnd::SetScrollbar (int bar, int newPos, int clientMax, int graphMax)
 110 {
 111         SCROLLINFO ScrollInfo = {sizeof(SCROLLINFO), SIF_ALL};
 112         GetScrollInfo (bar, &ScrollInfo);
 113
 114         clientMax = max(1, clientMax);
 115         int oldHeight = ScrollInfo.nMax <= 0 ? clientMax : ScrollInfo.nMax;
 116         int newHeight = static_cast<int>(graphMax * m_fZoomFactor);
 117         int maxPos = max (0, newHeight - clientMax);
 118         int pos = min (maxPos, newPos >= 0
 119                                                  ? newPos
 120                                                  : ScrollInfo.nPos * newHeight / oldHeight);
 121
 122         ScrollInfo.nPos = pos;
 123         ScrollInfo.nMin = 0;
 124         ScrollInfo.nMax = newHeight;
 125         ScrollInfo.nPage = clientMax;
 126         ScrollInfo.nTrackPos = pos;
 127
 128         SetScrollInfo(bar, &ScrollInfo);
 129
 130 }
 131
 132 void CRevisionGraphWnd::SetScrollbars (int nVert, int nHorz)
 133 {
 134         CRect clientrect = GetClientRect();
 135         const CRect& pRect = GetGraphRect();
 136
 137         SetScrollbar (SB_VERT, nVert, clientrect.Height(), pRect.Height());
 138         SetScrollbar (SB_HORZ, nHorz, clientrect.Width(), pRect.Width());
 139 }
 140
 141 CRect CRevisionGraphWnd::GetGraphRect()
 142 {
 143         return m_GraphRect;
 144 }
 145
 146 CRect CRevisionGraphWnd::GetClientRect()
 147 {
 148         CRect clientRect;
 149         CWnd::GetClientRect (&clientRect);
 150         return clientRect;
 151 }
 152
 153 CRect CRevisionGraphWnd::GetWindowRect()
 154 {
 155         CRect windowRect;
 156         CWnd::GetWindowRect (&windowRect);
 157         return windowRect;
 158 }
 159
 160 CRect CRevisionGraphWnd::GetViewRect()
 161 {
 162         CRect result;
 163         result.UnionRect (GetClientRect(), GetGraphRect());
 164         return result;
 165 }
 166
 167 int CRevisionGraphWnd::GetEncoderClsid(const WCHAR* format, CLSID* pClsid)
 168 {
 169         UINT num = 0;           // number of image encoders
 170         UINT size = 0;           // size of the image encoder array in bytes
 171
 172         if (GetImageEncodersSize(&num, &size)!=Ok)
 173                 return -1;
 174         if(size == 0)
 175                 return -1;  // Failure
 176
 177         ImageCodecInfo* pImageCodecInfo = (ImageCodecInfo*)(malloc(size));
 178         if(pImageCodecInfo == NULL)
 179                 return -1;  // Failure
 180
 181         if (GetImageEncoders(num, size, pImageCodecInfo)==Ok)
 182         {
 183                 for(UINT j = 0; j < num; ++j)
 184                 {
 185                         if( wcscmp(pImageCodecInfo[j].MimeType, format) == 0 )
 186                         {
 187                                 *pClsid = pImageCodecInfo[j].Clsid;
 188                                 free(pImageCodecInfo);
 189                                 return j;  // Success
 190                         }
 191                 }
 192
 193         }
 194         free(pImageCodecInfo);
 195         return -1;  // Failure
 196 }
 197
 198 bool CRevisionGraphWnd::FetchRevisionData
 199         ( const CString& /*path*/
 200         , GitRev /*pegRevision*/
 201         , CProgressDlg* /*progress*/
 202         , ITaskbarList3 * /*pTaskbarList*/
 203         , HWND /*hWnd*/)
 204 {
 205         this->m_LogCache.ClearAllParent();
 206         this->m_logEntries.ClearAll();
 207         CString range;
 208         if (!m_ToRev.IsEmpty() && !m_FromRev.IsEmpty())
 209                 range.Format(_T("%s..%s"), g_Git.FixBranchName(m_FromRev), g_Git.FixBranchName(m_ToRev));
 210         else if (!m_ToRev.IsEmpty())
 211                 range = m_ToRev;
 212         else if (!m_FromRev.IsEmpty())
 213                 range = m_FromRev;
 214         this->m_logEntries.ParserFromLog(nullptr, -1, CGit::LOG_INFO_SIMPILFY_BY_DECORATION|(this->m_bCurrentBranch? 0: CGit::LOG_INFO_ALL_BRANCH), &range);
 215
 216         ReloadHashMap();
 217         this->m_Graph.clear();
 218
 219         CArray<node> nodes;
 220         GraphicsDevice dev;
 221         dev.pDC = this->GetDC();
 222         dev.graphics = Graphics::FromHDC(dev.pDC->m_hDC);
 223         dev.graphics->SetPageUnit (UnitPixel);
 224
 225         for (int i = 0; i < m_logEntries.size(); ++i)
 226         {
 227                 node nd;
 228                 nd = this->m_Graph.newNode();
 229                 nodes.Add(nd);
 230                 m_GraphAttr.width(nd)=100;
 231                 m_GraphAttr.height(nd)=20;
 232                 SetNodeRect(dev, &nd, m_logEntries[i], 0);
 233         }
 234
 235         for (int i = 0; i < m_logEntries.size(); ++i)
 236         {
 237                 GitRev rev=m_logEntries.GetGitRevAt(i);
 238                 for (int j = 0; j < rev.m_ParentHash.size(); ++j)
 239                 {
 240                         if(m_logEntries.m_HashMap.find(rev.m_ParentHash[j]) == m_logEntries.m_HashMap.end())
 241                         {
 242                                 TRACE(_T("Can't found parent node"));
 243                                 //new parent node as new node
 244                                 node nd;
 245                                 nd = this->m_Graph.newNode();
 246                                 m_Graph.newEdge(nodes[i], nd);
 247                                 m_logEntries.push_back(rev.m_ParentHash[j]);
 248                                 m_logEntries.m_HashMap[rev.m_ParentHash[j]] = (int)m_logEntries.size() -1;
 249                                 nodes.Add(nd);
 250                                 SetNodeRect(dev, &nd, rev.m_ParentHash[j], 0);
 251
 252                         }else
 253                         {
 254                                 TRACE(_T("edge %d - %d\n"),i, m_logEntries.m_HashMap[rev.m_ParentHash[j]]);
 255                                 m_Graph.newEdge(nodes[i], nodes[m_logEntries.m_HashMap[rev.m_ParentHash[j]]]);
 256                         }
 257                 }
 258         }
 259
 260         //this->m_OHL.layerDistance(30.0);
 261         //this->m_OHL.nodeDistance(25.0);
 262         //this->m_OHL.weightBalancing(0.8);
 263
 264         m_SugiyamLayout.call(m_GraphAttr);
 265
 266         node v;
 267         double xmax = 0;
 268         double ymax = 0;
 269         forall_nodes(v,m_Graph)
 270         {
 271                 double x = m_GraphAttr.x(v) + m_GraphAttr.width(v)/2;
 272                 double y = m_GraphAttr.y(v) + m_GraphAttr.height(v)/2;
 273                 if(x>xmax)
 274                         xmax = x;
 275                 if(y>ymax)
 276                         ymax = y;
 277         }
 278
 279         this->m_GraphRect.top=m_GraphRect.left=0;
 280         m_GraphRect.bottom = (LONG)ymax;
 281         m_GraphRect.right = (LONG)xmax;
 282
 283         return true;
 284 }
 285
 286 bool CRevisionGraphWnd::AnalyzeRevisionData()
 287 {
 288 #if 0
 289         CSyncPointer<const CFullGraph> fullGraph (m_state.GetFullGraph());
 290         if ((fullGraph.get() != NULL) && (fullGraph->GetNodeCount() > 0))
 291         {
 292                 // filter graph
 293
 294                 CSyncPointer<CAllRevisionGraphOptions> options (m_state.GetOptions());
 295                 options->Prepare();
 296
 297                 std::unique_ptr<CVisibleGraph> visibleGraph (new CVisibleGraph());
 298                 CVisibleGraphBuilder builder ( *fullGraph
 299                                                                          , *visibleGraph
 300                                                                          , options->GetCopyFilterOptions());
 301                 builder.Run();
 302                 options->GetModificationOptions().Apply (visibleGraph.get());
 303
 304                 index_t index = 0;
 305                 for (size_t i = 0, count = visibleGraph->GetRootCount(); i < count; ++i)
 306                         index = visibleGraph->GetRoot (i)->InitIndex (index);
 307
 308                 // layout nodes
 309
 310                 std::unique_ptr<CStandardLayout> newLayout
 311                         ( new CStandardLayout ( m_state.GetFullHistory()->GetCache()
 312                                                                 , visibleGraph.get()
 313                                                                 , m_state.GetFullHistory()->GetWCInfo()));
 314                 options->GetLayoutOptions().Apply (newLayout.get());
 315                 newLayout->Finalize();
 316
 317                 // switch state
 318
 319                 m_state.SetAnalysisResult (visibleGraph, newLayout);
 320         }
 321
 322         return m_state.GetNodes().get() != NULL;
 323 #endif
 324         return true;
 325 }
 326
 327 bool CRevisionGraphWnd::IsUpdateJobRunning() const
 328 {
 329         return (updateJob.get() != NULL) && !updateJob->IsDone();
 330 }
 331
 332 bool CRevisionGraphWnd::GetShowOverview() const
 333 {
 334         return m_bShowOverview;
 335 }
 336
 337 void CRevisionGraphWnd::SetShowOverview (bool value)
 338 {
 339         m_bShowOverview = value;
 340         if (m_bShowOverview)
 341                 BuildPreview();
 342 }
 343 #if 0
 344 void CRevisionGraphWnd::GetSelected
 345         ( const CVisibleGraphNode* node
 346         , bool head
 347         , CTGitPath& path
 348         , GitRev& rev
 349         , GitRev& peg)
 350 {
 351
 352         CString repoRoot = m_state.GetRepositoryRoot();
 353
 354         // get path and revision
 355
 356         path.SetFromSVN (repoRoot + CUnicodeUtils::GetUnicode (node->GetPath().GetPath().c_str()));
 357         rev = head ? GitRev::REV_HEAD : node->GetRevision();
 358
 359         // handle 'modified WC' node
 360
 361         if (node->GetClassification().Is (CNodeClassification::IS_MODIFIED_WC))
 362         {
 363                 path.SetFromUnknown (m_sPath);
 364                 rev = GitRev::REV_WC;
 365
 366                 // don't set peg, if we aren't the first node
 367                 // (i.e. would not be valid for node1)
 368
 369                 if (node == m_SelectedEntry1)
 370                         peg = GitRev::REV_WC;
 371         }
 372         else
 373         {
 374                 // set head, if still necessary
 375
 376                 if (head && !peg.IsValid())
 377                         peg = node->GetRevision();
 378         }
 379
 380 }
 381 #endif
 382
 383 CString CRevisionGraphWnd::GetFriendRefName(ogdf::node v)
 384 {
 385         if(v == NULL)
 386                 return CString();
 387         CGitHash hash = this->m_logEntries[v->index()];
 388         if(this->m_HashMap.find(hash) == m_HashMap.end())
 389                 return hash.ToString();
 390         else if(this->m_HashMap[hash].size() == 0)
 391                 return hash.ToString();
 392         else if(this->m_HashMap[hash][0].IsEmpty())
 393                 return hash.ToString();
 394         else
 395                 return m_HashMap[hash][0];
 396
 397 }
 398
 399 void CRevisionGraphWnd::CompareRevs(bool bHead)
 400 {
 401         ASSERT(m_SelectedEntry1 != NULL);
 402         ASSERT(bHead || m_SelectedEntry2 != NULL);
 403
 404 //      bool alternativeTool = !!(GetAsyncKeyState(VK_SHIFT) & 0x8000);
 405
 406         CString sCmd;
 407
 408         sCmd.Format(_T("/command:showcompare %s /revision1:%s /revision2:%s"),
 409                         this->m_sPath.IsEmpty() ?  _T("") : (_T("/path:\"") + this->m_sPath + _T("\"")),
 410                         GetFriendRefName(m_SelectedEntry1),
 411                         bHead ?  _T("HEAD"): GetFriendRefName(m_SelectedEntry2));
 412
 413         CAppUtils::RunTortoiseGitProc(sCmd);
 414
 415 }
 416
 417 void CRevisionGraphWnd::UnifiedDiffRevs(bool bHead)
 418 {
 419         ASSERT(m_SelectedEntry1 != NULL);
 420         ASSERT(bHead || m_SelectedEntry2 != NULL);
 421
 422         bool alternativeTool = !!(GetAsyncKeyState(VK_SHIFT) & 0x8000);
 423         CAppUtils::StartShowUnifiedDiff(m_hWnd, CString(), GetFriendRefName(m_SelectedEntry1), CString(),
 424                 bHead? _T("HEAD"):GetFriendRefName(m_SelectedEntry2),
 425                 alternativeTool);
 426
 427 }
 428
 429 void CRevisionGraphWnd::DoZoom (float fZoomFactor, bool updateScrollbars)
 430 {
 431         float oldzoom = m_fZoomFactor;
 432         m_fZoomFactor = fZoomFactor;
 433
 434         m_nFontSize = max(1, int(DEFAULT_ZOOM_FONT * fZoomFactor));
 435         if (m_nFontSize < SMALL_ZOOM_FONT_THRESHOLD)
 436                 m_nFontSize = min (SMALL_ZOOM_FONT_THRESHOLD, int(SMALL_ZOOM_FONT * fZoomFactor));
 437
 438         for (int i = 0; i < MAXFONTS; ++i)
 439         {
 440                 if (m_apFonts[i] != NULL)
 441                 {
 442                         m_apFonts[i]->DeleteObject();
 443                         delete m_apFonts[i];
 444                 }
 445                 m_apFonts[i] = NULL;
 446         }
 447
 448         if (updateScrollbars)
 449         {
 450                 SCROLLINFO si1 = {sizeof(SCROLLINFO), SIF_ALL};
 451                 GetScrollInfo(SB_VERT, &si1);
 452                 SCROLLINFO si2 = {sizeof(SCROLLINFO), SIF_ALL};
 453                 GetScrollInfo(SB_HORZ, &si2);
 454
 455                 InitView();
 456
 457                 si1.nPos = int(float(si1.nPos)*m_fZoomFactor/oldzoom);
 458                 si2.nPos = int(float(si2.nPos)*m_fZoomFactor/oldzoom);
 459                 SetScrollPos (SB_VERT, si1.nPos);
 460                 SetScrollPos (SB_HORZ, si2.nPos);
 461         }
 462
 463         Invalidate (FALSE);
 464 }