using System;
using EvoCalculator.Core.Models.Calculation.Models.Prepared;

namespace EvoCalculator.Core.Calculation.Columns
{
    public class DateTempColumn : BaseColumn<DateTime>
    {
        public DateTempColumn(int count) : base(count)
        {
        }

        public void ComputeValues(PreparedValues preparedValues)
        {
            Values[0] = preparedValues.DogDate;
            Values[1] = Values[0];
            Values[2] = preparedValues.PaymentDateNew ?? Values[1].AddMonths(1);

            for (var i = 3; i < Values.Length; i++)
                Values[i] = preparedValues.PaymentDateNew == null
                    ? Values[1].AddMonths(i - 1)
                    : Values[2].AddMonths(i - 2);
        }
    }
}