org.spearce.egit.core/src/org/spearce/egit/core/EclipseGitProgressTransformer.java

   1 /*******************************************************************************
   2  * Copyright (C) 2008, Shawn O. Pearce <spearce@spearce.org>
   3  *
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * See LICENSE for the full license text, also available.
   7  *******************************************************************************/
   8 package org.spearce.egit.core;
   9
  10 import org.eclipse.core.runtime.IProgressMonitor;
  11 import org.eclipse.core.runtime.SubProgressMonitor;
  12 import org.spearce.jgit.lib.ProgressMonitor;
  13
  14 /** Create a new Git to Eclipse progress monitor. */
  15 public class EclipseGitProgressTransformer implements ProgressMonitor {
  16         private final IProgressMonitor root;
  17
  18         private IProgressMonitor task;
  19
  20         private String msg;
  21
  22         private int lastWorked;
  23
  24         private int totalWork;
  25
  26         /**
  27          * Create a new progress monitor.
  28          *
  29          * @param eclipseMonitor
  30          *            the Eclipse monitor we update.
  31          */
  32         public EclipseGitProgressTransformer(final IProgressMonitor eclipseMonitor) {
  33                 root = eclipseMonitor;
  34         }
  35
  36         public void start(final int totalTasks) {
  37                 root.beginTask("", totalTasks * 1000);
  38         }
  39
  40         public void beginTask(final String name, final int total) {
  41                 endTask();
  42                 msg = name;
  43                 lastWorked = 0;
  44                 totalWork = total;
  45                 task = new SubProgressMonitor(root, 1000);
  46                 if (totalWork == UNKNOWN)
  47                         task.beginTask("", IProgressMonitor.UNKNOWN);
  48                 else
  49                         task.beginTask("", totalWork);
  50                 task.subTask(msg);
  51         }
  52
  53         public void update(final int work) {
  54                 if (task == null)
  55                         return;
  56
  57                 final int cmp = lastWorked + work;
  58                 if (lastWorked == UNKNOWN && cmp > 0) {
  59                         task.subTask(msg + ", " + cmp);
  60                 } else if (totalWork <= 0) {
  61                         // Do nothing to update the task.
  62                 } else if (cmp * 100 / totalWork != lastWorked * 100 / totalWork) {
  63                         final StringBuilder m = new StringBuilder();
  64                         m.append(msg);
  65                         m.append(": ");
  66                         while (m.length() < 25)
  67                                 m.append(' ');
  68
  69                         if (totalWork == UNKNOWN) {
  70                                 m.append(cmp);
  71                         } else {
  72                                 final String twstr = String.valueOf(totalWork);
  73                                 String cmpstr = String.valueOf(cmp);
  74                                 while (cmpstr.length() < twstr.length())
  75                                         cmpstr = " " + cmpstr;
  76                                 final int pcnt = (cmp * 100 / totalWork);
  77                                 if (pcnt < 100)
  78                                         m.append(' ');
  79                                 if (pcnt < 10)
  80                                         m.append(' ');
  81                                 m.append(pcnt);
  82                                 m.append("% (");
  83                                 m.append(cmpstr);
  84                                 m.append("/");
  85                                 m.append(twstr);
  86                                 m.append(")");
  87                         }
  88                         task.subTask(m.toString());
  89                 }
  90                 lastWorked = cmp;
  91                 task.worked(work);
  92         }
  93
  94         public void endTask() {
  95                 if (task != null) {
  96                         try {
  97                                 task.done();
  98                         } finally {
  99                                 task = null;
 100                         }
 101                 }
 102         }
 103
 104         public boolean isCancelled() {
 105                 if (task != null)
 106                         return task.isCanceled();
 107                 return root.isCanceled();
 108         }
 109 }