Java tutorial
/* * This file is part of LibrePlan * * Copyright (C) 2012 Igalia, S.L. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package org.libreplan.web.planner.adaptplanning; import static org.libreplan.web.I18nHelper._; import java.util.Date; import java.util.List; import org.joda.time.LocalDate; import org.libreplan.business.advance.bootstrap.PredefinedAdvancedTypes; import org.libreplan.business.advance.entities.AdvanceMeasurement; import org.libreplan.business.advance.entities.AdvanceType; import org.libreplan.business.advance.entities.DirectAdvanceAssignment; import org.libreplan.business.advance.exceptions.DuplicateAdvanceAssignmentForOrderElementException; import org.libreplan.business.advance.exceptions.DuplicateValueTrueReportGlobalAdvanceException; import org.libreplan.business.orders.entities.OrderElement; import org.libreplan.business.planner.entities.PositionConstraintType; import org.libreplan.business.planner.entities.ResourceAllocation; import org.libreplan.business.planner.entities.Task; import org.libreplan.business.planner.entities.TaskElement; import org.libreplan.business.workingday.IntraDayDate; import org.libreplan.web.planner.order.PlanningStateCreator.PlanningState; import org.springframework.beans.factory.config.BeanDefinition; import org.springframework.context.annotation.Scope; import org.springframework.stereotype.Component; import org.zkoss.ganttz.Planner; import org.zkoss.ganttz.TaskComponent; import org.zkoss.ganttz.extensions.IContext; import org.zkoss.ganttz.util.LongOperationFeedback; import org.zkoss.ganttz.util.LongOperationFeedback.ILongOperation; /** * @author Manuel Rego Casasnovas <rego@igalia.com> */ @Component @Scope(BeanDefinition.SCOPE_PROTOTYPE) public class AdaptPlanningCommand implements IAdaptPlanningCommand { private PlanningState planningState; @Override public String getName() { return _("Adapt planning according to timesheets"); } @Override public void doAction(final IContext<TaskElement> context) { LongOperationFeedback.execute(context.getRelativeTo(), new ILongOperation() { @Override public String getName() { return _("Adapting planning according to timesheets"); } @Override public void doAction() throws Exception { adaptPlanning(context); } }); } private void adaptPlanning(IContext<TaskElement> context) { List<TaskElement> taskElements = planningState.getRootTask().getAllChildren(); for (TaskElement taskElement : taskElements) { // Only adapt task leafs if (!taskElement.isTask()) { continue; } OrderElement orderElement = taskElement.getOrderElement(); // Reset status to allow move the task if needed while adapting the // planning taskElement.setUpdatedFromTimesheets(false); if (orderElement.hasTimesheetsReportingHours()) { setStartDateAndConstraint(taskElement, orderElement.getFirstTimesheetDate()); Date lastTimesheetDate = orderElement.getLastTimesheetDate(); setEndDateIfNeeded(taskElement, lastTimesheetDate); if (orderElement.isFinishedTimesheets()) { setEndDate(taskElement, lastTimesheetDate); addTimesheetsProgress(orderElement, lastTimesheetDate); removeResourceAllocationsBeyondEndDate(taskElement); } else { removeTimesheetsProgressIfAny(orderElement); } taskElement.setUpdatedFromTimesheets(true); } else { removeTimesheetsProgressIfAny(orderElement); } } for (TaskElement taskElement : taskElements) { if (taskElement.isUpdatedFromTimesheets()) { updateTask(context, taskElement); } } ((Planner) context.getRelativeTo()).invalidate(); context.reloadCharts(); } private void removeResourceAllocationsBeyondEndDate(TaskElement taskElement) { LocalDate endDate = taskElement.getEndAsLocalDate(); for (ResourceAllocation<?> resourceAllocation : taskElement.getAllResourceAllocations()) { resourceAllocation.removeDayAssignmentsBeyondDate(endDate); } } private void setStartDateAndConstraint(TaskElement taskElement, Date startDate) { taskElement.setStartDate(startDate); setStartInFixedDateConstarint(taskElement, startDate); } private void setStartInFixedDateConstarint(TaskElement taskElement, Date startDate) { if (taskElement.isTask()) { Task task = (Task) taskElement; task.getPositionConstraint().update(PositionConstraintType.START_IN_FIXED_DATE, IntraDayDate.startOfDay(LocalDate.fromDateFields(startDate))); } } private void setEndDateIfNeeded(TaskElement taskElement, Date endDate) { if (taskElement.getEndDate().compareTo(endDate) <= 0) { setEndDate(taskElement, endDate); } } private void setEndDate(TaskElement taskElement, Date endDate) { taskElement.setEndDate(LocalDate.fromDateFields(endDate).plusDays(1).toDateTimeAtStartOfDay().toDate()); } private void addTimesheetsProgress(OrderElement orderElement, Date progressDate) { AdvanceType timesheetsAdvanceType = getTimesheetsAdvanceType(); DirectAdvanceAssignment timesheetsAdvanceAssignment = orderElement .getDirectAdvanceAssignmentByType(timesheetsAdvanceType); if (timesheetsAdvanceAssignment == null) { timesheetsAdvanceAssignment = DirectAdvanceAssignment.create(false, timesheetsAdvanceType.getDefaultMaxValue()); timesheetsAdvanceAssignment.setAdvanceType(timesheetsAdvanceType); try { orderElement.addAdvanceAssignment(timesheetsAdvanceAssignment); } catch (DuplicateValueTrueReportGlobalAdvanceException e) { // This shouldn't happen as the new advanceAssignment is not // marked as spread yet throw new RuntimeException(e); } catch (DuplicateAdvanceAssignmentForOrderElementException e) { // If the same type already exists in other element we don't do // anything return; } } DirectAdvanceAssignment spreadAdvanceAssignment = orderElement.getReportGlobalAdvanceAssignment(); if (spreadAdvanceAssignment != null) { spreadAdvanceAssignment.setReportGlobalAdvance(false); } timesheetsAdvanceAssignment.setReportGlobalAdvance(true); timesheetsAdvanceAssignment.resetAdvanceMeasurements(AdvanceMeasurement .create(LocalDate.fromDateFields(progressDate), timesheetsAdvanceType.getDefaultMaxValue())); } private AdvanceType getTimesheetsAdvanceType() { return PredefinedAdvancedTypes.TIMESHEETS.getType(); } private void removeTimesheetsProgressIfAny(OrderElement orderElement) { DirectAdvanceAssignment timesheetsAdvanceAssignment = orderElement .getDirectAdvanceAssignmentByType(getTimesheetsAdvanceType()); if (timesheetsAdvanceAssignment != null) { orderElement.removeAdvanceAssignment(timesheetsAdvanceAssignment); } } private void updateTask(IContext<TaskElement> context, TaskElement taskElement) { taskElement.updateAdvancePercentageFromOrderElement(); Planner planner = (Planner) context.getRelativeTo(); org.zkoss.ganttz.data.Task task = context.getMapper().findAssociatedBean(taskElement); task.firePropertyChangeForTaskDates(); TaskComponent taskComponent = planner.getTaskComponentRelatedTo(task); if (taskComponent != null) { taskComponent.updateTooltipText(); taskComponent.updateProperties(); taskComponent.invalidate(); } context.recalculatePosition(taskElement); } @Override public String getImage() { return "/common/img/ico_adapt_planning.png"; } @Override public boolean isDisabled() { return false; } @Override public void setState(PlanningState planningState) { this.planningState = planningState; } @Override public boolean isPlannerCommand() { return true; } }