private int getTotalWork(
      CheckWizardPage options, IPlottingSystem system, final IFunctionService funcService) {

    try {
      int ret = 1;
      if (options.is(PersistWizardConstants.ORIGINAL_DATA)) {
        Collection<ITrace> traces = system.getTraces();
        ret += traces != null ? traces.size() : 0;
      }
      if (options.is(PersistWizardConstants.IMAGE_HIST)) {
        final IToolPageSystem tsystem = (IToolPageSystem) system.getAdapter(IToolPageSystem.class);
        final IToolPage tool = tsystem.getActiveTool();
        if (tool != null
            && tool.getToolId().equals("org.dawb.workbench.plotting.tools.imageCompareTool")) {
          final Map<String, IDataset> data = (Map<String, IDataset>) tool.getToolData();
          if (data != null && !data.isEmpty()) {
            ret += data.size();
          }
        }
      }

      final ITrace trace = system.getTraces().iterator().next();
      if (options.is(PersistWizardConstants.MASK) && trace instanceof IImageTrace) {
        IImageTrace image = (IImageTrace) trace;
        final String name = options.getString(PersistWizardConstants.MASK);
        if (image.getMask() != null) {
          ret += 1;
        }
      }

      final Collection<IRegion> regions = system.getRegions();
      if (options.is(PersistWizardConstants.REGIONS) && regions != null && !regions.isEmpty()) {
        ret += regions.size();
      }

      if (options.is(PersistWizardConstants.DIFF_META)) {
        if (trace != null && trace instanceof IImageTrace && trace.getData() != null) {
          ret += 1;
        }
      }

      if (options.is(PersistWizardConstants.FUNCTIONS)) {
        if (funcService != null) {
          Map<String, IFunction> functions = funcService.getFunctions();
          if (functions != null) ret += functions.size();
        }
      }

      return ret;
    } catch (Exception ne) {
      logger.error("Cannot estimate work, assuming 100 things to do!", ne);
      return 100;
    }
  }
示例#2
0
  public static IROI runConicPeakFit(
      final IProgressMonitor monitor,
      Display display,
      final IPlottingSystem plotter,
      IImageTrace t,
      IParametricROI roi,
      IParametricROI[] innerOuter,
      int nPoints) {

    if (roi == null) return null;

    final ProgressMonitorWrapper mon = new ProgressMonitorWrapper(monitor);
    monitor.subTask("Find POIs near initial ellipse");
    Dataset image = (Dataset) t.getData();
    BooleanDataset mask = (BooleanDataset) t.getMask();
    PolylineROI points;
    monitor.subTask("Fit POIs");

    points = PeakFittingEllipseFinder.findPointsOnConic(image, mask, roi, innerOuter, nPoints, mon);

    if (monitor.isCanceled()) return null;

    if (points == null) return null;

    if (roi instanceof EllipticalROI) {
      if (points.getNumberOfPoints() < 3) {
        throw new IllegalArgumentException("Could not find enough points to trim");
      }

      monitor.subTask("Trim POIs");
      EllipticalFitROI efroi = PowderRingsUtils.fitAndTrimOutliers(mon, points, 5, false);
      logger.debug("Found {}...", efroi);
      monitor.subTask("");
      return efroi;
    }

    return points;
  }