org.eclipse.mylyn.sandbox.ui/src/org/eclipse/mylyn/internal/sandbox/ui/planner/TaskActivityLabelProvider.java

/*******************************************************************************
 * Copyright (c) 2004, 2008 Tasktop Technologies and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Tasktop Technologies - initial API and implementation
 *******************************************************************************/

package org.eclipse.mylyn.internal.sandbox.ui.planner;

import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.viewers.IColorProvider;
import org.eclipse.jface.viewers.IFontProvider;
import org.eclipse.jface.viewers.ILabelDecorator;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.mylyn.commons.core.DateUtil;
import org.eclipse.mylyn.commons.core.StatusHandler;
import org.eclipse.mylyn.internal.provisional.commons.ui.CommonFonts;
import org.eclipse.mylyn.internal.tasks.core.AbstractTask;
import org.eclipse.mylyn.internal.tasks.core.ScheduledTaskContainer;
import org.eclipse.mylyn.internal.tasks.ui.TasksUiPlugin;
import org.eclipse.mylyn.tasks.core.ITask;
import org.eclipse.mylyn.tasks.core.ITaskActivityManager;
import org.eclipse.mylyn.tasks.core.ITaskContainer;
import org.eclipse.mylyn.tasks.core.ITask.PriorityLevel;
import org.eclipse.mylyn.tasks.ui.TaskElementLabelProvider;
import org.eclipse.mylyn.tasks.ui.TasksUiImages;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.graphics.Font;
import org.eclipse.swt.graphics.Image;

/**
 * @author Rob Elves
 */
public class TaskActivityLabelProvider extends TaskElementLabelProvider implements ITableLabelProvider, IColorProvider,
		IFontProvider {

	private static final String UNITS_HOURS = " hours";

	private static final String NO_MINUTES = "0 minutes";

	private final ITaskActivityManager activityManager;

	private Color categoryBackgroundColor;

	private final ITreeContentProvider contentProvider;

	public TaskActivityLabelProvider(ILabelProvider provider, ILabelDecorator decorator, Color parentBacground,
			ITreeContentProvider contentProvider) {
		super(true);
		this.categoryBackgroundColor = parentBacground;
		this.activityManager = TasksUiPlugin.getTaskActivityManager();
		this.contentProvider = contentProvider;
	}

	public Image getColumnImage(Object element, int columnIndex) {
		if (columnIndex == 0) {
			if (element instanceof ScheduledTaskContainer) {
				super.getImage(element);
			} else {
				return super.getImage(element);
			}
		} else if (columnIndex == 1) {
			if (element instanceof ITask) {
				ITask taskElement = (ITask) element;
				return TasksUiImages.getImageForPriority(PriorityLevel.fromString(taskElement.getPriority()));
			}
		}
		return null;
	}

	public String getColumnText(Object element, int columnIndex) {
		if (element instanceof ITask) {
			AbstractTask task = (AbstractTask) element;
			switch (columnIndex) {
			case 2:
				if (task != null) {
					return task.getSummary();
				}
			case 3:
				ScheduledTaskContainer container = (ScheduledTaskContainer) contentProvider.getParent(task);
				if (container != null) {
					return DateUtil.getFormattedDurationShort(activityManager.getElapsedTime(task,
							container.getDateRange().getStartDate(), container.getDateRange().getEndDate()));
				} else {
					return "na";
				}
			case 4:
				return task.getEstimatedTimeHours() + UNITS_HOURS;
			case 5:
				if (task.getScheduledForDate() != null) {
					return task.getScheduledForDate().toString();//DateFormat.getDateInstance(DateFormat.MEDIUM).format(task.getScheduledForDate());
				} else {
					return "";
				}
//			case 6:
//				if (activityDelegate.getStart() > 0
//						&& activityDelegate.getDateRangeContainer().getElapsed(activityDelegate) > 0) {
//					return DateFormat.getDateInstance(DateFormat.MEDIUM).format(activityDelegate.getStart());
//				} else {
//					return "";
//				}
			}
		} else if (element instanceof ScheduledTaskContainer) {
			ScheduledTaskContainer taskCategory = (ScheduledTaskContainer) element;
			switch (columnIndex) {
			case 2:
				if (taskCategory.isPresent()) {
					return taskCategory.getSummary() + " - Today";
				}

				return taskCategory.getSummary();
			case 3:
				String elapsedTimeString = NO_MINUTES;
				try {

					long elapsed = 0;
					for (Object o : contentProvider.getChildren(taskCategory)) {
						AbstractTask task = (AbstractTask) o;
						elapsed += activityManager.getElapsedTime(task, taskCategory.getStart(), taskCategory.getEnd());
					}

					elapsedTimeString = DateUtil.getFormattedDurationShort(elapsed);
					if (elapsedTimeString.equals("")) {
						elapsedTimeString = NO_MINUTES;
					}
				} catch (RuntimeException e) {
					StatusHandler.log(new Status(IStatus.ERROR, TasksUiPlugin.ID_PLUGIN,
							"Could not format elapsed time", e));
				}
				return elapsedTimeString;
			case 4:
				long estimated = 0;
				for (Object o : contentProvider.getChildren(taskCategory)) {
					if (o instanceof AbstractTask) {
						estimated += ((AbstractTask) o).getEstimatedTimeHours();
					}
				}
				return estimated + UNITS_HOURS;
			}
		}
		return null;
	}

	@Override
	public Color getBackground(Object element) {
		if (element instanceof ITaskContainer && !(element instanceof ITask)) {
			return categoryBackgroundColor;
		} else {
			return super.getBackground(element);
		}
	}

	@Override
	public Font getFont(Object element) {
		if (element instanceof ScheduledTaskContainer) {
			ScheduledTaskContainer container = (ScheduledTaskContainer) element;
			if (container.isPresent()) {
				return CommonFonts.BOLD;
			}
		} else if (element instanceof ITask) {
			AbstractTask task = (AbstractTask) element;
			return super.getFont(task);
		}
		return super.getFont(element);
	}

	public void setCategoryBackgroundColor(Color categoryBackgroundColor) {
		this.categoryBackgroundColor = categoryBackgroundColor;
	}
}