org.eclipse.mylyn.sandbox.ui/src/org/eclipse/mylyn/internal/sandbox/ui/planner/TaskPlannerLabelProvider.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:
 *     Ken Sueda - initial API and implementation
 *     Tasktop Technologies - improvements
 *******************************************************************************/

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

import java.text.DateFormat;
import java.util.Calendar;
import java.util.Date;

import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.viewers.IColorProvider;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.mylyn.commons.core.DateUtil;
import org.eclipse.mylyn.commons.core.StatusHandler;
import org.eclipse.mylyn.internal.tasks.core.AbstractTask;
import org.eclipse.mylyn.internal.tasks.ui.TasksUiPlugin;
import org.eclipse.mylyn.tasks.core.IRepositoryElement;
import org.eclipse.mylyn.tasks.core.ITask;
import org.eclipse.mylyn.tasks.core.ITaskContainer;
import org.eclipse.mylyn.tasks.ui.TaskElementLabelProvider;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.graphics.Image;

/**
 * @author Ken Sueda
 * @author Rob Elves
 */
public class TaskPlannerLabelProvider extends TaskElementLabelProvider implements ITableLabelProvider, IColorProvider {

	private final Calendar startDate;

	private final Calendar endDate;

	private final TreeViewer viewer;

	public TaskPlannerLabelProvider(TreeViewer viewer, Date startDate, Date endDate) {
		super(true);
		this.viewer = viewer;
		this.startDate = Calendar.getInstance();
		this.startDate.setTime(startDate);
		this.endDate = Calendar.getInstance();
		this.endDate.setTime(endDate);
	}

	private final TaskElementLabelProvider taskListLabelProvider = new TaskElementLabelProvider(true);

	public Image getColumnImage(Object element, int columnIndex) {
		if (columnIndex == 0) {
			return super.getImage(element);
		} else {
			return null;
		}
	}

	public String getColumnText(Object element, int columnIndex) {
		try {
			if (element instanceof AbstractTask) {
				AbstractTask task = (AbstractTask) element;
				switch (columnIndex) {
				case 1:
					return task.getPriority();
				case 2:
					return task.getSummary();
				case 3:
					return DateUtil.getFormattedDurationShort(TasksUiPlugin.getTaskActivityManager().getElapsedTime(
							task, startDate, endDate));
				case 4:
					return task.getEstimatedTimeHours() + " hours";
				case 5:
					Date creationDate = task.getCreationDate();
					if (creationDate != null) {
						return DateFormat.getDateInstance(DateFormat.MEDIUM).format(creationDate);
					} else {
						StatusHandler.log(new Status(IStatus.WARNING, TasksUiPlugin.ID_PLUGIN,
								"Task has no creation date: " + task.getSummary()));
						return "[unknown]";
					}
				case 6:
					if (task.getCompletionDate() != null) {
						return DateFormat.getDateInstance(DateFormat.MEDIUM).format(task.getCompletionDate());
					} else {
						return "";
					}
				}
			} else if (element instanceof ITaskContainer) {
				IRepositoryElement container = (IRepositoryElement) element;
				switch (columnIndex) {
				case 1:
					return null;
				case 2:
					return container.getSummary();
				case 3: {
					ITreeContentProvider contentProvider = ((ITreeContentProvider) viewer.getContentProvider());
					long duration = 0;
					for (Object o : contentProvider.getChildren(container)) {
						if (o instanceof ITask) {
							duration += TasksUiPlugin.getTaskActivityManager().getElapsedTime((ITask) o, startDate,
									endDate);
						}
					}
					return DateUtil.getFormattedDurationShort(duration);
				}
				case 4: {
					ITreeContentProvider contentProvider = ((ITreeContentProvider) viewer.getContentProvider());
					long estimated = 0;
					for (Object o : contentProvider.getChildren(container)) {
						if (o instanceof AbstractTask) {
							estimated += ((AbstractTask) o).getEstimatedTimeHours();
						}
					}
					return estimated + " hours";
				}
				case 5:
					return null;
				case 6:
					return null;
				}
			}
		} catch (RuntimeException e) {
			StatusHandler.log(new Status(IStatus.ERROR, TasksUiPlugin.ID_PLUGIN,
					"Could not produce completed task label", e));
			return "";
		}
		return null;
	}

	@Override
	public Color getForeground(Object element) {
		return taskListLabelProvider.getForeground(element);
	}

	@Override
	public Color getBackground(Object element) {
		return taskListLabelProvider.getBackground(element);
	}

}