/*
 * 
 jFin, open source derivatives trade processing 
 *
 * 
Copyright (C) 2005-2011 Morgan Brown Consultancy Ltd. 
 *
 * 
This file is part of jFin. 
 *
 * 
 jFin is free software; you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option) any later
 * version. 
 *
 * 
 jFin 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 General Public
 * License for more details. 
 *
 * 
 You should have received a copy of the GNU General Public License along
 * with jFin; if not, write to the Free Software Foundation, Inc., 51 Franklin
 * St, Fifth Floor, Boston, MA 02110-1301 USA. 
 */

package com.mbc.jfin.schedule.impl;

import com.mbc.jfin.schedule.ScheduleException;
import org.joda.time.*;

/**
 * Created by IntelliJ IDEA.
 * User: dmb
 * Date: 24/04/2011
 * Time: 16:13
 */
public abstract class AbstractBaseScheduleGenerator implements ScheduleGenerator {
    protected int maxPeriods;

    protected AbstractBaseScheduleGenerator(int maxPeriods) {
        this.maxPeriods = maxPeriods;
    }

    protected ReadablePeriod multiplyPeriod(ReadablePeriod frequency, int periodCount) throws ScheduleException {

        if(frequency instanceof Months) {
            return Months.months(frequency.getValue(0)*periodCount);
        } else if(frequency instanceof Days) {
            return Days.days(frequency.getValue(0) * periodCount);
        } else if(frequency instanceof Weeks) {
            return Weeks.weeks(frequency.getValue(0) * periodCount);
        } else if(frequency instanceof Years) {
            return Years.years(frequency.getValue(0) * periodCount);
        } else {
            throw new ScheduleException("Unknown frequency type: "+frequency);
        }
    }
}