PeriodSet.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.Date;
import java.util.TreeSet;
import org.amic.util.date.CheckDate;
 
public class PeriodSet extends TreeSet<AccountReportPeriod> {
 
  private int year = 0, month = 0;
 
  public PeriodSet() {
    super(new PeriodComparator());
  }
 
  public AccountReportPeriod getPeriod(AccountReportPeriod period) {
    for(AccountReportPeriod p : this) {
      if(p.equals(period)) {
        return p;
      }
    }
    return null;
  }
 
  public void fillGapsTillNow(AccountReportItem item) {
    AccountReportPeriod arp = first();
    if(arp != null) {
      CheckDate cd = new CheckDate(arp.getYear(), arp.getMonth(), 1, 0, 0, 0);
      Date now = new Date();
      while(cd.compareMonthYear(now) == -1) {
        AccountReportPeriod a = new AccountReportPeriod(cd.getYear(), cd.getMonth());
        a.setValue(0);
        a.setItem(item);
        add(a);
        cd.addMonths(1);
      }
    }
  }
 
  public PeriodSet getCurrentPeriod(int count) {
    if(year == 0 || month == 0) {
      CheckDate cd = new CheckDate();
      cd.addMonths(-(count/2));
      year = cd.getYear();
      month = cd.getMonth();
    }
    PeriodSet ps = new PeriodSet();
    int c = 0;
    for(AccountReportPeriod arp : tailSet(new AccountReportPeriod(year, month))) {
      ps.add(arp);
      if(c++ > count - 1) break;
    }
    return ps;
  }
 
  public void incrementPeriod(int amount) {
    CheckDate cd = new CheckDate(year, month, 1, 0, 0, 0);
    cd.addMonths(amount);
    year = cd.getYear();
    month = cd.getMonth();
  }
}