bserp-www/src/main/java/org/turro/erp/task/usage/WorkloadGantt.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2012 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * 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.turro.erp.task.usage;
 
import java.util.HashMap;
import java.util.List;
import org.turro.erp.db.ErpPU;
import org.turro.erp.entity.HumanResource;
import org.turro.erp.entity.OrderItem;
import org.turro.erp.entity.RequiredUsage;
import org.turro.erp.entity.Resource;
import org.turro.erp.task.Workload;
import org.turro.erp.task.WorkloadSet;
import org.turro.jpa.Dao;
import org.turro.zkoss.svg.GanttResource;
import org.turro.zkoss.svg.GanttResourceData;
import org.turro.zkoss.svg.GanttResourceItem;
import org.zkoss.zk.ui.ext.AfterCompose;
 
public class WorkloadGantt extends GanttResource implements AfterCompose {
 
  private String selectItem;
  private double scale = 1.0;
 
  public WorkloadGantt() {
  }
 
  public String getSelectItem() {
    return selectItem;
  }
 
  public void setSelectItem(String selectItem) {
    this.selectItem = selectItem;
  }
 
  public double getScale() {
    return scale;
  }
 
  public void setScale(double scale) {
    this.scale = scale;
  }
  
  public void reload() {
    loadData();
    repaint();
  }
  
  @Override
  public void afterCompose() {
    loadData();
    repaint();
  }
 
  public void repaint() {
    setVisible(false);
    getChildren().clear();
    clear();
    startComposing(selectItem, null, null, scale);
    generateContent(null);
    setVisible(true);
  }
  
  private void loadData() {
    setData(fillModel());
  }
  
  private GanttResourceData fillModel() {
    Dao dao = new ErpPU();
    List<HumanResource> humanResources = (List<HumanResource>) dao.getResultList("select x from HumanResource x");
    List<Resource> resources = (List<Resource>) dao.getResultList("select x from Resource x");
    HashMap<Long, GanttResourceItem> map = new HashMap<Long, GanttResourceItem>();
    for(HumanResource humanResource : humanResources) {
      if(humanResource.isActive()) {
        GanttResourceItem gri = new GanttResourceItem(1, "" + humanResource.getId(), humanResource.getName());
        map.put(humanResource.getId() + 1000000000000L, gri);
      }
    }
    for(Resource resource : resources) {
      if(resource.isActive()) {
        GanttResourceItem gri = new GanttResourceItem(2, "" + resource.getId(), resource.getName());
        map.put(resource.getId() + 2000000000000L, gri);
      }
    }
    WorkloadSet wls = new WorkloadSet((HumanResource) null, true);
    for(Workload wl : wls) {
      OrderItem oi = wl.getOrderItem();
      HumanResource humanResource = oi.getHumanResource();
      Resource resource = oi.getRelated() != null ? oi.getRelated().getResource() : null;
      if(humanResource != null && humanResource.isActive()) {
        GanttResourceItem gri = map.get(humanResource.getId() + 1000000000000L);
        if(gri != null) {
          gri.setExpected(Math.max(oi.getExpectedUnits() - oi.getRealUnits(), 0.0d) + gri.getExpected());
        }
      }
      if(resource != null && resource.isActive()) {
        oi = oi.getRelated();
        GanttResourceItem gri = map.get(resource.getId() + 2000000000000L);
        if(gri != null) {
          gri.setExpected(Math.max(oi.getExpectedUnits() - oi.getRealUnits(), 0.0d) + gri.getExpected());
        }
      }
    }
    wls = new WorkloadSet((HumanResource) null, false);
    for(Workload wl : wls) {
      RequiredUsage ru = wl.getRequiredUsage();
      for(HumanResource humanResource : humanResources) {
        if(humanResource != null && humanResource.isActive() && ru.fits(humanResource)) {
          GanttResourceItem gri = map.get(humanResource.getId() + 1000000000000L);
          if(gri != null) {
            gri.setVariable(Math.max((ru.getUnits() * ru.getDedication()) - getExpectedUnits(ru), 0.0d) + gri.getVariable());
          }
        } 
      }
      Resource resource = ru.getResource();
      if(resource != null && resource.isActive()) {
        GanttResourceItem gri = map.get(resource.getId() + 2000000000000L);
        if(gri != null) {
          gri.setVariable(Math.max((ru.getUnits()) - getRelatedExpectedUnits(ru), 0.0d) + gri.getVariable());
        }
      }
    }
 
//    for(HumanResource humanResource : (List<HumanResource>) dao.getResultList("select x from HumanResource x")) {
//      if(humanResource.isActive()) {
//        WorkloadSet wls = new WorkloadSet(humanResource, true);
//        double value = 0.0d;
//        for(Workload wl : wls) {
//          OrderItem oi = wl.getOrderItem();
//          value += Math.max(oi.getExpectedUnits() - oi.getRealUnits(), 0.0d);
//        }
//        GanttResourceItem gri = map.get(humanResource);
//        if(gri == null) {
//          gri = new GanttResourceItem(1, humanResource.getId(), humanResource.getName());
//          map.put(humanResource, gri);
//        }
//        gri.setExpected(value);
//        wls = new WorkloadSet(humanResource, false);
//        value = 0.0d;
//        for(Workload wl : wls) {
//          RequiredUsage ru = wl.getRequiredUsage();
//          value += Math.max((ru.getUnits() * ru.getDedication()) - getExpectedUnits(ru), 0.0d);
//        }
//        gri = map.get(humanResource);
//        if(gri == null) {
//          gri = new GanttResourceItem(1, humanResource.getId(), humanResource.getName());
//          map.put(humanResource, gri);
//        }
//        gri.setVariable(value);
//      }
//    }
//    for(Resource resource : (List<Resource>) dao.getResultList("select x from Resource x")) {
//      if(resource.isActive()) {
//        WorkloadSet wls = new WorkloadSet(resource, true);
//        double value = 0.0d;
//        for(Workload wl : wls) {
//          OrderItem oi = wl.getOrderItem();
//          value += Math.max(oi.getExpectedUnits() - oi.getRealUnits(), 0.0d);
//        }
//        GanttResourceItem gri = map.get(resource);
//        if(gri == null) {
//          gri = new GanttResourceItem(2, resource.getId(), resource.getName());
//          map.put(resource, gri);
//        }
//        gri.setExpected(value);
//        wls = new WorkloadSet(resource, false);
//        value = 0.0d;
//        for(Workload wl : wls) {
//          RequiredUsage ru = wl.getRequiredUsage();
//          value += Math.max((ru.getUnits() * ru.getDedication()) - getExpectedUnits(ru), 0.0d);
//        }
//        gri = map.get(resource);
//        if(gri == null) {
//          gri = new GanttResourceItem(2, resource.getId(), resource.getName());
//          map.put(resource, gri);
//        }
//        gri.setVariable(value);
//      }
//    }
    GanttResourceData grd = new GanttResourceData();
    for(GanttResourceItem gri : map.values()) {
      grd.add(gri);
      gri.setSet(grd);
    }
    return grd;
  }
 
  private double getExpectedUnits(RequiredUsage ru) {
    double q = 0.0d;
    for(OrderItem oi : ru.getOrderItems()) {
      q += oi.getUnits() * oi.getDedication();
    }
    return q;
  }
 
  private double getRelatedExpectedUnits(RequiredUsage ru) {
    double q = 0.0d;
    for(OrderItem oi : ru.getOrderItems()) {
      if(oi.getRelated() != null) {
        q += oi.getRelated().getUnits() * oi.getRelated().getDedication();
      }
    }
    return q;
  }
 
}