RegisterAssuredBalance.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2013 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.register;
 
import org.turro.financials.entity.Document;
import org.turro.financials.entity.Register;
import org.turro.financials.entity.RegisterEntry;
import org.turro.financials.model.document.DocumentAmounts;
import org.turro.math.Round;
 
public class RegisterAssuredBalance {
 
  private EntrySet entrySet;
  private Document document;
  private Register register;
  private DocumentAmounts amounts;
  private double balance;
  private int digits;
 
  public RegisterAssuredBalance(EntrySet entrySet, Document document) {
    this.entrySet = entrySet;
    this.document = document;
    this.digits = document.getCurrency().getDefaultFractionDigits();
    amounts = document.getAmounts();
  }
 
  public RegisterAssuredBalance(Register register) {
    this.register = register;
    this.document = register.getDocument();
    this.digits = document.getCurrency().getDefaultFractionDigits();
    amounts = document.getAmounts();
  }
  
  public void normalize() {
    checkTotal();
    checkBalance();
  }
  
  private void checkTotal() {
    RegisterEntry registerEntry = getMaxEntry("^4[0134][01].*");
    if(registerEntry != null && canRepairTotal(registerEntry)) {
      setTotal(registerEntry, amounts.getTotal());      
    }
  }
  
  private void checkBalance() {
    balance = 0.0d;
    if(entrySet != null) {
      for(EntryWrapper ew : entrySet) {
        balance += ew.getEntry().getDebit() - ew.getEntry().getCredit();
      }
    } else {
      for(RegisterEntry re : register.getRegisterEntries()) {
        balance += (re.getDebit() - re.getCredit());
      }
    }
    if(new Round(Math.abs(balance)).decimals(digits).value() == 0.01d) {
      if(!increaseAmount(getMaxEntry("^[76].*"), balance)) {
        increaseAmount(getMaxEntry("^4[0134]09.*"), balance);
      }
    }
  }
 
  private boolean canRepairTotal(RegisterEntry registerEntry) {
    double total = amounts.getTotal(),
           debit = registerEntry.getDebit(), credit = registerEntry.getCredit();
    return (new Round(Math.abs(total - debit)).decimals(digits).value() == 0.01d || 
            new Round(Math.abs(total - credit)).decimals(digits).value() == 0.01d);
  }
  
  private double debit() {
    double debit = 0.0d;
    if(entrySet != null) {
      for(EntryWrapper ew : entrySet) {
        RegisterEntry re = ew.getEntry();
        debit += re.getDebit();
      }
    } else {
      for(RegisterEntry re : register.getRegisterEntries()) {
        debit += re.getDebit();
      }
    }
    return debit;
  }
 
  private double credit() {
    double credit = 0.0d;
    if(entrySet != null) {
      for(EntryWrapper ew : entrySet) {
        RegisterEntry re = ew.getEntry();
        credit += re.getCredit();
      }
    } else {
      for(RegisterEntry re : register.getRegisterEntries()) {
        credit += re.getCredit();
      }
    }
    return credit;
  }
  
  private RegisterEntry getMaxEntry(String accountExpr) {
    double amount = 0.0d;
    RegisterEntry result = null;
    if(entrySet != null) {
      for(EntryWrapper ew : entrySet) {
        RegisterEntry re = ew.getEntry();
        if(re.getAccount().getId().matches(accountExpr) && (Math.abs(re.getCredit() + re.getDebit())) > amount) {
          result = re;
          amount = Math.abs(re.getCredit() + re.getDebit());
        }
      }
    } else {
      for(RegisterEntry re : register.getRegisterEntries()) {
        if(re.getAccount().getId().matches(accountExpr) && (Math.abs(re.getCredit() + re.getDebit())) > amount) {
          result = re;
          amount = Math.abs(re.getCredit() + re.getDebit());
        }
      }
    }
    return result;
  }
 
  private void setTotal(RegisterEntry entry, double total) {
    double debit = entry.getDebit(), credit = entry.getCredit();
    if(new Round(Math.abs(total - debit)).decimals(digits).value() == 0.01d) {
      entry.setDebit(total);
    } else if(new Round(Math.abs(total - credit)).decimals(digits).value() == 0.01d) {
      entry.setCredit(total);
    }
  }
 
  private boolean increaseAmount(RegisterEntry entry, double amount) {
    boolean done = false;
    if(entry != null) {
      double debit = entry.getDebit(), credit = entry.getCredit();
      if(new Round(Math.abs(debit)).decimals(digits).value() > 0.00d) {
        entry.setDebit(entry.getDebit() - amount);
        done = true;
      } else if(new Round(Math.abs(credit)).decimals(digits).value() > 0.00d) {
        entry.setCredit(entry.getCredit() + amount);
        done = true;
      }
    }
    return done;
  }
 
}