Java tutorial
/* * This file is part of LibrePlan * * Copyright (C) 2011 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; import java.util.Date; import java.util.List; import java.util.Set; import org.apache.commons.lang3.StringUtils; import org.libreplan.business.labels.entities.Label; import org.libreplan.business.orders.entities.OrderElement; import org.libreplan.business.planner.entities.GenericResourceAllocation; import org.libreplan.business.planner.entities.ResourceAllocation; import org.libreplan.business.planner.entities.SpecificResourceAllocation; import org.libreplan.business.planner.entities.TaskElement; import org.libreplan.business.resources.entities.Criterion; import org.libreplan.business.resources.entities.Resource; import org.libreplan.web.common.components.finders.FilterPair; import org.libreplan.web.common.components.finders.TaskElementFilterEnum; import org.zkoss.ganttz.IPredicate; /** * Checks if {@link TaskElement} matches with the different filters. * * @author Manuel Rego Casasnovas <rego@igalia.com> */ public class TaskElementPredicate implements IPredicate { private List<FilterPair> filters; private Date startDate; private Date finishDate; private String name; private boolean ignoreLabelsInheritance; public TaskElementPredicate(List<FilterPair> filters, Date startDate, Date finishDate, String name, boolean ignoreLabelsInheritance) { this.filters = filters; this.startDate = startDate; this.finishDate = finishDate; this.name = name; this.ignoreLabelsInheritance = ignoreLabelsInheritance; } @Override public boolean accepts(Object object) { final TaskElement taskElement = (TaskElement) object; return accepts(taskElement) || accepts(taskElement.getAllChildren()); } private boolean accepts(TaskElement taskElement) { if (taskElement == null) { return false; } if (acceptFilters(taskElement) && acceptFiltersDates(taskElement) && acceptFilterName(taskElement)) { return true; } return false; } private boolean accepts(List<TaskElement> taskElements) { for (TaskElement taskElement : taskElements) { if (accepts(taskElement)) { return true; } } return false; } private boolean acceptFilters(TaskElement taskElement) { if ((filters == null) || (filters.isEmpty())) { return true; } for (FilterPair filter : filters) { if (!acceptFilter(filter, taskElement)) { return false; } } return true; } private boolean acceptFilter(FilterPair filter, TaskElement taskElement) { switch ((TaskElementFilterEnum) filter.getType()) { case Criterion: return acceptCriterion(filter, taskElement); case Resource: return acceptResource(filter, taskElement); case Label: return acceptLabel(filter, taskElement); } return false; } private boolean acceptCriterion(FilterPair filter, TaskElement taskElement) { Criterion filterCriterion = (Criterion) filter.getValue(); return existCriterionInTaskElementResourceAllocations(filterCriterion, taskElement); } private boolean existCriterionInTaskElementResourceAllocations(Criterion filterCriterion, TaskElement taskElement) { for (ResourceAllocation<?> each : taskElement.getAllResourceAllocations()) { if (acceptsCriterionInResourceAllocation(filterCriterion, each)) { return true; } } return false; } private boolean acceptsCriterionInResourceAllocation(Criterion filterCriterion, ResourceAllocation<?> resourceAllocation) { if (resourceAllocation instanceof GenericResourceAllocation) { Set<Criterion> criteria = ((GenericResourceAllocation) resourceAllocation).getCriterions(); for (Criterion criterion : criteria) { if (criterion.getId().equals(filterCriterion.getId())) { return true; } } } return false; } private boolean acceptResource(FilterPair filter, TaskElement taskElement) { Resource filterResource = (Resource) filter.getValue(); return existResourceInTaskElementResourceAllocations(filterResource, taskElement); } private boolean existResourceInTaskElementResourceAllocations(Resource filterResource, TaskElement taskElement) { for (ResourceAllocation<?> each : taskElement.getAllResourceAllocations()) { if (acceptsResourceInResourceAllocation(filterResource, each)) { return true; } } return false; } private boolean acceptsResourceInResourceAllocation(Resource filterResource, ResourceAllocation<?> resourceAllocation) { if (resourceAllocation instanceof SpecificResourceAllocation) { Resource resource = ((SpecificResourceAllocation) resourceAllocation).getResource(); if (resource.getId().equals(filterResource.getId())) { return true; } } return false; } private boolean acceptLabel(FilterPair filter, TaskElement taskElement) { Label filterLabel = (Label) filter.getValue(); return existLabelInTaskElement(filterLabel, taskElement); } private boolean existLabelInTaskElement(Label filterLabel, TaskElement taskElement) { OrderElement orderElement = taskElement.getOrderElement(); if (orderElement == null) { return false; } Set<Label> labels; if (ignoreLabelsInheritance) { labels = orderElement.getLabels(); } else { labels = orderElement.getAllLabels(); } for (Label label : labels) { if (label.getId().equals(filterLabel.getId())) { return true; } } return false; } private boolean acceptFiltersDates(TaskElement taskElement) { return (acceptStartDate(taskElement.getStartDate()) && (acceptFinishDate(taskElement.getEndDate()))); } private boolean acceptStartDate(Date initDate) { if ((initDate == null) && (startDate == null)) { return true; } return isInTheRangeFilterDates(initDate); } private boolean acceptFinishDate(Date deadLine) { if ((deadLine == null) && (finishDate == null)) { return true; } return isInTheRangeFilterDates(deadLine); } private boolean isInTheRangeFilterDates(Date date) { // Check if date is into interval between the startdate and finish date return (isGreaterToStartDate(date, startDate) && isLowerToFinishDate(date, finishDate)); } private boolean isGreaterToStartDate(Date date, Date startDate) { if (startDate == null) { return true; } if (date != null && (date.compareTo(startDate) >= 0)) { return true; } return false; } private boolean isLowerToFinishDate(Date date, Date finishDate) { if (finishDate == null) { return true; } if (date != null && (date.compareTo(finishDate) <= 0)) { return true; } return false; } private boolean acceptFilterName(TaskElement taskElement) { if (name == null) { return true; } if ((taskElement.getName() != null) && (StringUtils.containsIgnoreCase(taskElement.getName(), name))) { return true; } return false; } }