BudgetsTotalCalculator.java

/*
 * Copyright (c) 2012 Denis Solonenko.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 */

package ru.orangesoftware.financisto2.db;

import android.os.Handler;
import android.support.v4.util.LongSparseArray;
import android.util.Log;
import ru.orangesoftware.financisto2.model.*;
import ru.orangesoftware.financisto2.rates.ExchangeRate;
import ru.orangesoftware.financisto2.rates.ExchangeRateProvider;

import java.math.BigDecimal;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by IntelliJ IDEA.
 * User: solomonk
 * Date: 4/28/12
 * Time: 2:33 AM
 */
public class BudgetsTotalCalculator {

    private final DatabaseAdapter db;
    private final CategoryRepository categoryRepository;
    private final List<Budget> budgets;

    public BudgetsTotalCalculator(DatabaseAdapter db, CategoryRepository categoryRepository, List<Budget> budgets) {
        this.db = db;
        this.categoryRepository = categoryRepository;
        this.budgets = budgets;
    }

    public void updateBudgets(Handler handler) {
        long t0 = System.currentTimeMillis();
        try {
            LongSparseArray<Category> categories = categoryRepository.loadCategories().asIdMap();
            LongSparseArray<Project> projects = MyEntity.asMap(db.getAllProjectsList(true));
            for (final Budget b : budgets) {
                final long spent = db.fetchBudgetBalance(categories, projects, b);
                final String categoriesText = getChecked(categories, b.categories);
                final String projectsText = getChecked(projects, b.projects);
                b.spent = spent;
                if (handler != null) {
                    handler.post(new Runnable() {
                        @Override
                        public void run() {
                            b.updated = true;
                            b.spent = spent;
                            b.categoriesText = categoriesText;
                            b.projectsText = projectsText;
                        }
                    });
                }
            }
        } finally {
            long t1 = System.currentTimeMillis();
            Log.d("BUDGET UPDATE", (t1 - t0) + "ms");
        }
    }
    
    public Total[] calculateTotals() {
        Map<Currency, Total> totals = new HashMap<Currency, Total>();
        for (Budget b : budgets) {
            Currency c = b.getBudgetCurrency();
            Total total = totals.get(c);
            if (total == null) {
                total = new Total(c, true);
                totals.put(c, total);
            }
            total.amount += b.spent;
            total.balance += b.amount+b.spent;
        }
        Collection<Total> values = totals.values();
        return values.toArray(new Total[values.size()]);
    }

    public Total calculateTotalInHomeCurrency() {
        long t0 = System.currentTimeMillis();
        try {
            BigDecimal amount = BigDecimal.ZERO;
            BigDecimal balance = BigDecimal.ZERO;
            ExchangeRateProvider rates = db.getLatestRates();
            Currency homeCurrency = db.getHomeCurrency();
            for (Budget b : budgets) {
                Currency currency = b.getBudgetCurrency();
                ExchangeRate r = rates.getRate(currency, homeCurrency);
                if (r == ExchangeRate.NA) {
                    return new Total(homeCurrency, TotalError.lastRateError(currency));
                } else {
                    amount = amount.add(convert(r, b.spent));
                    balance = balance.add(convert(r, b.amount+b.spent));
                }
            }
            Total total = new Total(homeCurrency, true);
            total.amount = amount.longValue();
            total.balance = balance.longValue();
            return total;
        } finally {
            long t1 = System.currentTimeMillis();
            Log.d("BUDGET TOTALS", (t1 - t0) + "ms");
        }
    }

    private BigDecimal convert(ExchangeRate r, long spent) {
        return BigDecimal.valueOf(r.rate*spent);
    }

    private <T extends MyEntity> String getChecked(LongSparseArray<T> entities, String s) {
        long[] ids = MyEntity.splitIds(s);
        if (ids == null) {
            return null;
        }
        if (ids.length == 1) {
            MyEntity e = entities.get(ids[0]);
            if (e == null) {
                return null;
            } else {
                return e.title;
            }
        } else {
            StringBuilder sb = new StringBuilder();
            for (long id : ids) {
                MyEntity e = entities.get(id);
                if (e != null) {
                    if (sb.length() > 0) {
                        sb.append(",");
                    }
                    sb.append(e.title);
                }
            }
            if (sb.length() == 0) {
                return null;
            } else {
                return sb.toString();
            }
        }
    }


}