PortfolioDisagreementSet.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 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.financials.model.document;
 
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.TreeMap;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.Contract;
import org.turro.financials.entity.Document;
import org.turro.financials.entity.DocumentLine;
import org.turro.financials.entity.DocumentRelation;
import org.turro.financials.entity.DocumentWorkflow;
import org.turro.financials.model.business.CompanyWrapper;
import org.turro.jpa.Dao;
import org.turro.util.CompareUtil;
 
public class PortfolioDisagreementSet extends TreeMap<Long, PortfolioDisagreement> {
 
  private Dao dao;
 
  public PortfolioDisagreementSet() {
    fillSet();
    checkBackwards();
  }
 
  public PortfolioDisagreementSet(Contract contract) {
    fillSet(contract);
    checkBackwards();
  }
 
  private void fillSet() {
    Date closingDate = CompanyWrapper.getCompanyClosingDate();
    dao = new FinancialsPU();
    List<Object[]> l = dao.getResultList(
            "select dr, dw from DocumentRelation dr, DocumentWorkflow dw " +
            "where dr.ancestor.documentDefinition.id = dw.ancestor.id " +
            "and dr.descendant.documentDefinition.id = dw.descendant.id " +
            "and dr.ancestor.contract.contractDefinition.id = dw.contractDefinition.id " +
            "and dr.ancestor.documentDate >= ?",
            new Object[] { closingDate }
            );
    for(Object[] o : l) {
      checkRelation((DocumentRelation) o[0], (DocumentWorkflow) o[1]);
    }
  }
 
  private void fillSet(Contract contract) {
    Date closingDate = CompanyWrapper.getCompanyClosingDate();
    dao = new FinancialsPU();
    List<Object[]> l = dao.getResultList(
            "select dr, dw from DocumentRelation dr, DocumentWorkflow dw " +
            "where dr.ancestor.documentDefinition.id = dw.ancestor.id " +
            "and dr.descendant.documentDefinition.id = dw.descendant.id " +
            "and dr.ancestor.contract.contractDefinition.id = dw.contractDefinition.id " +
            "and dr.ancestor.contract.id = ? " +
            "and dr.ancestor.documentDate >= ?",
            new Object[] { contract.getId(), closingDate }
            );
    for(Object[] o : l) {
      checkRelation((DocumentRelation) o[0], (DocumentWorkflow) o[1]);
    }
  }
 
  private void checkRelation(DocumentRelation dr, DocumentWorkflow dw) {
    double descendantAmount = 0.0;
    if(CompareUtil.compare(dr.getAncestor().isDraft(), dr.getDescendant().isDraft()) != 0) {
      return;
    }
    for(DocumentLine dl : dr.getDescendant().getDocumentLines()) {
      if(dl.getLineType() != null && dl.getLineType().getId() == dw.getLineType().getId()) {
        descendantAmount += dl.getAmount();
      }
    }
    if(CompareUtil.compare(dr.getAncestor().getTotalAmount(), descendantAmount) != 0) {
      PortfolioDisagreement pd = get(dr.getAncestor().getId());
      if(pd != null) {
        if(pd.fullfillsWith(descendantAmount)) {
          remove(dr.getAncestor().getId());
        } else {
          pd.addAmount(descendantAmount);
        }
      } else {
        put(dr.getAncestor().getId(), new PortfolioDisagreement(dr.getAncestor(), descendantAmount));
      }
    }
  }
 
  private void checkBackwards() {
    Iterator<PortfolioDisagreement> it = values().iterator();
    while(it.hasNext()) {
      if(checBackwardRelation(it.next())) {
        it.remove();
      }
    }
  }
 
  private boolean checBackwardRelation(PortfolioDisagreement pd) {
    double siblingsAmount = 0.0;
    for(DocumentRelation dr : pd.getDocument().getDescendants()) {
      // OK, now backward: siblings
      for(DocumentRelation drdoc : dr.getDescendant().getAncestors()) {
        Document doc = drdoc.getAncestor();
        siblingsAmount += doc.getTotalAmount();
      }
    }
    return pd.fullfillsBackwardWith(siblingsAmount);
  }
 
}