package shtick.math;

//////////////////////////////////////////////////////
//**************************************************//
//*               FourierSeries Class              *//
//*                                                *//
//* Version 1.2.1                                  *//
//* Author: Sean M. Cox                            *//
//* Organization: The Shtick                       *//
//* http://www.smcox.com/                          *//
//*                                                *//
//* Base class for a Fourier Series. Translates a  *//
//* string of data into its component harmonics    *//
//* Using a basis created by the FourierBasis      *//
//* class passed in the constructor.               *//
//*                                                *//
//**************************************************//
//////////////////////////////////////////////////////

public class FourierSeries {
  double[] toTranslate;
  double[] sineSeries= new double[10];
  double[] cosineSeries= new double[10];
  double[][] sineBasis;
  double[][] cosineBasis;

  boolean  translated= false;

  public FourierSeries(double[] data){
    toTranslate=data;
  }

  public FourierSeries(double[] data, FourierBasis fb){
    toTranslate=data;
    sineBasis=fb.getSineBasis();
    cosineBasis=fb.getCosineBasis();
    sineSeries=new double[sineBasis.length];
    cosineSeries=new double[cosineBasis.length];
  }

  public boolean translate(){
    if (toTranslate.length > 0){
      double tally=0;
      for (int i=0;i<toTranslate.length;i++){
        tally+=toTranslate[i];
      }
      cosineSeries[0]=(double)(tally/toTranslate.length);
      for (int h=1;h<sineSeries.length;h++){
        tally=0;
        for (int i=0;i<toTranslate.length;i++){
          tally+=(long)toTranslate[i]*cosineBasis[h][i];//Math.cos(h*2*Math.PI*(i+.5)/toTranslate.length);
        }
        cosineSeries[h]=((double)(int)(10*2*tally/toTranslate.length+.5))/10;
        tally=0;
        for (int i=0;i<toTranslate.length;i++){
          tally+=toTranslate[i]*sineBasis[h][i];//Math.sin(h*2*Math.PI*(i+.5)/toTranslate.length);
        }
        sineSeries[h]=((double)(int)(10*2*tally/toTranslate.length+.5))/10;
      }
      translated=true;
    }
    return translated;
  }

  public boolean isTranslated(){
    return translated;
  }

  public double[] getCosineSeries(){
    return cosineSeries;
  }

  public double[] getSineSeries(){
    return sineSeries;
  }
}