require "flex_station_data/presenters/sample_csv"
require "flex_station_data/presenters/linear_regression/sample_regression_hash"

module FlexStationData
  module Presenters
    module LinearRegression
      class VerboseSampleCsv < Presenters::SampleCsv
        include Concerns::Presenter

        def sample_values
          [ *sample.values, sample.mean ]
        end

        def regressions_hash
          SampleRegressionHash.present(times, *sample_values, **options)
        end

        def regressions_csv
          regressions_hash.to_a.map(&:flatten)
        end

        def values_csv
          super + regressions_csv
        end
      end
    end
  end
end