AccountReport.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.report;
 
import java.util.ArrayList;
import java.util.List;
import org.turro.string.Strings;
import org.turro.financials.db.FinancialsPU;
import org.turro.jpa.Dao;
 
public class AccountReport {
 
  private String name;
  private List<AccountReportItem> children;
  private List<AccountReportFormula> formulas;
  private boolean showTotal;
  private Dao dao;
  private List<String> accounts;
  private PeriodSet existingPeriods;
 
  public AccountReport(String name) {
    this.name = name;
    children = new ArrayList<AccountReportItem>();
    formulas = new ArrayList<AccountReportFormula>();
  }
 
  public AccountReportItem addItem(String label) {
    return addItem(new AccountReportItem(label));
  }
 
  public AccountReportItem addItem(AccountReportItem item) {
    item.setReport(this);
    item.setParent(null);
    children.add(item);
    return item;
  }
 
  public AccountReportFormula addFormula(String label) {
    return addFormula(new AccountReportFormula(label));
  }
 
  public AccountReportFormula addFormula(AccountReportFormula formula) {
    formula.setReport(this);
    formulas.add(formula);
    return formula;
  }
 
  public List<AccountReportItem> getChildren() {
    return children;
  }
 
  public List<AccountReportFormula> getFormulas() {
    return formulas;
  }
 
  public String getName() {
    return name;
  }
 
  public void setName(String name) {
    this.name = name;
  }
 
  public boolean isShowTotal() {
    return showTotal;
  }
 
  public void setShowTotal(boolean showTotal) {
    this.showTotal = showTotal;
  }
 
  public PeriodSet getExistingPeriods() {
    return existingPeriods;
  }
 
  public AccountReportItem getItem(String label) {
    if(!Strings.isBlank(label)) {
      for(AccountReportItem ari : children) {
        if(label.equals(ari.getLabel())) {
          return ari;
        }
        AccountReportItem ari2 = ari.getItem(label);
        if(ari2 != null) {
          return ari2;
        }
      }
    }
    return null;
  }
 
  public AccountReportFormula getFormula(String label) {
    if(!Strings.isBlank(label)) {
      for(AccountReportFormula arf : formulas) {
        if(label.equals(arf.getLabel())) {
          return arf;
        }
      }
    }
    return null;
  }
 
  public Dao getDao() {
    if(dao == null) {
      dao = new FinancialsPU();
    }
    return dao;
  }
 
  public List<String> getAccounts() {
    if(accounts == null) {
      accounts = getDao().getResultList("select a.id from Account as a");
    }
    return accounts;
  }
 
  public void loadData() {
    existingPeriods = new PeriodSet();
    for(AccountReportItem ari : children) {
      ari.loadData(existingPeriods);
    }
    existingPeriods.fillGapsTillNow(null);
    for(AccountReportItem ari : children) {
      ari.regularizeMode(existingPeriods);
    }
  }
 
  public AccountReportItem getTotalItem() {
    AccountReportItem total = new AccountReportItem(name);
    total.setMode(null);
    total.setParent(null);
    total.setReport(this);
    total.setTotalItem(true);
    for(AccountReportPeriod period : existingPeriods) {
      AccountReportPeriod arp = new AccountReportPeriod(period.getYear(), period.getMonth());
      arp.setItem(total);
      for(AccountReportItem item : children) {
        arp.setValue(arp.getValue() + item.getAccumulatedFor(period));
      }
      total.getPeriods().add(arp);
    }
    return total;
  }
 
}