import java.awt.*;

import java.applet.*;

import sun.audio.*;

import graph.*;



// @(#)fourier.java	1.15 96/07/15



/*

 * Copyright (c) 1996 Manfred Thole

 *      thole@nst.ing.tu-bs.de

 *

 * Modified by Tom Huber, huber@gac.edu

 * Revision Date: 20-Sep-96

 *    This requires the graph2d package from Leigh Brookshaw

 *       http://www.sci.usq.edu.au/staff/leighb/graph

 *

 * This program 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.

 *

 * 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

 * General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along

 * with this program; if not, write to the Free Software Foundation, Inc.,

 * 675 Mass Ave, Cambridge, MA 02139, USA.

 *

 *

 */



/**

    Fourier synthesis.

    @author Manfred Thole, thole@nst.ing.tu-bs.de

    @version 96/07/15

    @

    @ Modified by Tom Huber, huber@gac.edu

    @ version 20-Sep-96

    @ To Compile, this requires the graph2d package from Leigh Brookshaw

    @    http://www.sci.usq.edu.au/staff/leighb/graph

    @

  */



public class fourier extends Applet {

  // Max. |ak|,|bk|

  final static int MAX_AB = 50;

  // "k_max"

  final static int MAX_ANZ = 14;

  // a_k = a[k]/INT_MULT

  final static double INT_MULT = 50.0;

  // time base

  final static double TIM_MULT = 40.0/Math.PI;

  // Audio??

  final static boolean AUDIO = true;

  // Maximum Number of Waves

  final static int MAX_WAVES = 3;



  private double acoeff [][] = new double[MAX_WAVES][MAX_ANZ];

  private double bcoeff [][] = new double[MAX_WAVES][MAX_ANZ];



  private Button la_a [] = new Button[MAX_ANZ];

  private Button la_b [] = new Button[MAX_ANZ];



  private TextField ta_a [] = new TextField[MAX_ANZ];

  private TextField ta_b [] = new TextField[MAX_ANZ];



  private TextField CosExpr = new TextField();

  private TextField SinExpr = new TextField();



  private Scrollbar s_a [] = new Scrollbar[MAX_ANZ];

  private Scrollbar s_b [] = new Scrollbar[MAX_ANZ];



  Panel panel1, panel2 ;    // Sub-containers for all this stuff.



  GridBagLayout gridbag = new GridBagLayout();



  private CheckboxGroup WaveNumber;

  private Checkbox Wave1;

  private Checkbox Wave2;

  private Checkbox Wave3;



  private CheckboxGroup AudioGroup;

  private Checkbox AudioOn;

  private Checkbox AudioOff;





  private Image bg;

  private Graphics bgg;



  private String sin_name;

  private String cos_name;



  private byte sound[] = new byte[40];

  private java.io.InputStream soundStream;

  private boolean soundOn;



  private int iy[] = new int[81];



  private int Wnum=0;   // Wave Number



  int height,width;





  public void init() {

    width = size().width;

    height = size().height;

    setBackground(Color.lightGray);

    resize(width,height);



    sin_name = "Sin: ";

    cos_name = "Cos: ";



    panel1 = new Panel();

    panel1.setLayout(gridbag);



    CosExpr = new TextField(13);

    SinExpr = new TextField(13);



    CosExpr.setForeground(Color.blue);

    CosExpr.setBackground(Color.white);

    SinExpr.setForeground(Color.blue);

    SinExpr.setBackground(Color.white);



    constrain(panel1, new Label(cos_name), 0, 0, 1, 1);

    constrain(panel1, new Label(sin_name), 3, 0, 1, 1);

    constrain(panel1, CosExpr, 1,0,2,1);

    constrain(panel1, SinExpr, 4,0,2,1);



    for (int i=0; i < MAX_ANZ; i++)

      {

      if (i<10)

  	   la_a[i] = new Button("a" + i + ":  ");

      else

  	   la_a[i] = new Button("a" + i + ":");

      ta_a[i] = new TextField(10);

      ta_a[i].setText(""+acoeff[Wnum][i]);

      ta_a[i].setForeground(Color.blue);

      ta_a[i].setBackground(Color.white);

      ta_a[i].setEditable(true);

	s_a[i] = new Scrollbar(Scrollbar.HORIZONTAL,(int)(acoeff[Wnum][i]*INT_MULT),1,-MAX_AB,MAX_AB);

      s_a[i].setBackground(Color.white);

      constrain(panel1, la_a[i], 0, i+1, 1, 1);

      constrain(panel1, ta_a[i], 1, i+1, 1, 1);

      constrain(panel1, s_a[i],  2, i+1, 1, 1, GridBagConstraints.HORIZONTAL,

          GridBagConstraints.NORTHWEST, 1.0, 0.0, 0, 0, 0, 10, 40, 0);

	if ( i == 0 )

	  {

	    constrain(panel1, new Label(""), 3, 1, 1, 1);

	  }

	else

	  {

	    if (i<10)

	       la_b[i] = new Button("b" + i + ":  ");

          else

	       la_b[i] = new Button("b" + i + ":");

          ta_b[i] = new TextField(10);

          ta_b[i].setText(""+bcoeff[Wnum][i]);

          ta_b[i].setForeground(Color.blue);

          ta_b[i].setBackground(Color.white);

          ta_b[i].setEditable(true);

	    s_b[i] = new Scrollbar(Scrollbar.HORIZONTAL,(int)(bcoeff[Wnum][i]*INT_MULT),1,-MAX_AB,MAX_AB);

          s_b[i].setBackground(Color.white);

          constrain(panel1, la_b[i], 3, i+1, 1, 1);

          constrain(panel1, ta_b[i], 4, i+1, 1, 1);

          constrain(panel1, s_b[i],  5, i+1, 1, 1, GridBagConstraints.HORIZONTAL,

          GridBagConstraints.NORTHWEST, 1.0, 0.0, 0, 0, 0, 10, 40, 0);

//          constrain(panel1, s_b[i],  5, i+1, 1, 1,0,5,0,5);

	  }

      }

    WaveNumber = new CheckboxGroup();

    add(Wave1 = new Checkbox("Wave1",WaveNumber,true));

    add(Wave2 = new Checkbox("Wave2",WaveNumber,false));

    add(Wave3 = new Checkbox("Wave3",WaveNumber,false));



    AudioGroup = new CheckboxGroup();

    add(AudioOn = new Checkbox("Audio On",AudioGroup,true));

    add(AudioOff = new Checkbox("Audio Off",AudioGroup,false));



    panel2 = new Panel();

    panel2.setLayout(gridbag);

    //

    // Create a Panel to contain all the components along the

    // left hand side of the window.  Use a GridBagLayout for it.

    constrain(panel2,Wave1,0,0,1,1);

    constrain(panel2,Wave2,0,1,1,1);

    constrain(panel2,Wave3,0,2,1,1);

    constrain(panel2, new Label(""), 0, 3, 1, 1);

    constrain(panel2,AudioOn,0,4,1,1);

    constrain(panel2,AudioOff,0,5,1,1);

//    constrain(panel2, new Label("  Sample:"), 0, 6, 1, 1);

//    constrain(panel2, new Label("  (-1^(x-1))/x"), 0, 7, 1, 1);



    // Finally, use a GridBagLayout to arrange the panels themselves

    this.setLayout(gridbag);

    // And add the panels to the toplevel window

    constrain(this, panel1, 0, 0, 1, 1, GridBagConstraints.NONE,

        GridBagConstraints.NORTHWEST, 0.0, 0.0, 205, 10, 0, 0);

    constrain(this, panel2, 0, 0, 1, 1, GridBagConstraints.NONE,

        GridBagConstraints.NORTHWEST, 0.0, 0.0, 10, 380, 0, 0);



//    bg = this.createImage(320,200);

    bg = this.createImage(width,height);

    bgg = bg.getGraphics();

    bgg.setColor(Color.lightGray);

    bgg.fillRect(0, 0, width, height);

    bgg.setColor(Color.black);

    bgg.fillRect(0, 0, 320, 200);

    bgg.setColor( new Color(0, 75, 0) );

    for (int i = 0; i < 320; i+=20)

         bgg.drawLine(i, 0, i, 200);

    for (int i = 100; i < 200; i+=30)

   	   bgg.drawLine(0, i, 320, i);

    for (int i = 100; i > 0; i-=30)

   	   bgg.drawLine(0, i, 320, i);

    bgg.setColor(Color.green);

    bgg.drawLine(0, 100, 320, 100);

    bgg.drawLine(160, 0, 160, 200);



    CalculatePlot();

    if ( AUDIO )  {

      soundStream = new ContinuousAudioDataStream(new AudioData(sound));

      soundOn = true;

    }

  }





  private final void CalculatePlot() {

    if ( AUDIO && soundOn )

      AudioPlayer.player.stop(soundStream);



    // One period, TIM_MULT dependent! This needs clean up...

    for (int i = 0; i < 80; i++)

      {

	double y = 0;

	y += acoeff[Wnum][0]/2.0;

	for (int j = 1; j < MAX_ANZ; j++)

	  {

	    y += acoeff[Wnum][j]*Math.cos(j*i/TIM_MULT);

	    y += bcoeff[Wnum][j]*Math.sin(j*i/TIM_MULT);

	  }

	iy[i] = (int) (100-y*30);

	if ( AUDIO && ( i % 2 != 0 ) )

	  sound[i/2] = int2ulaw((int)(y*2000));

      } // for (i...

      iy[80] = iy[0];



      if ( AUDIO && soundOn )

         AudioPlayer.player.start(soundStream);

  }



  public boolean handleEvent(Event e) {

//    this.showStatus("ID: "+e.id+"   Event: " + e.target);



    switch (e.id) { // See pg 114 of Java in a Nutshell for other event types



    case Event.ACTION_EVENT:



      if (e.target instanceof TextField) {

        for (int i=0; i < MAX_ANZ; i++)  {

  	    if ( e.target.equals(ta_a[i]) )  {

            ParseFunction function = new ParseFunction( ((TextField)e.target).getText() );



            if(!function.parse()) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               return true;

            }

            try {

               double value = function.getResult();

  	         acoeff[Wnum][i] = value;

               acoeff[Wnum][i] = Math.min(acoeff[Wnum][i],1.0);

               acoeff[Wnum][i] = Math.max(acoeff[Wnum][i],-1.0);

            } catch(Exception err) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               acoeff[Wnum][i] = 0.0;

            }

              CosExpr.setText("");

              SinExpr.setText("");

  	        ta_a[i].setText(""+acoeff[Wnum][i]);

	        s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

	        break;

	    }  // if ( e.target.equals(ta_a[i]) )



  	    else if ( e.target.equals(ta_b[i]) )  {

            ParseFunction function = new ParseFunction( ((TextField)e.target).getText() );



            if(!function.parse()) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               return true;

            }

            try {

               double value = function.getResult();

  	         bcoeff[Wnum][i] = value;

               bcoeff[Wnum][i] = Math.min(bcoeff[Wnum][i],1.0);

               bcoeff[Wnum][i] = Math.max(bcoeff[Wnum][i],-1.0);

            } catch(Exception err) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               bcoeff[Wnum][i] = 0.0;

            }

              CosExpr.setText("");

              SinExpr.setText("");

  	        ta_b[i].setText(""+bcoeff[Wnum][i]);

	        s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

	        break;

	    }  // else if ( e.target.equals(ta_b[i]) )



        } // for (int i=0; i < MAX_ANZ; i++)



        if (e.target.equals(CosExpr)) {

            ParseFunction function = new ParseFunction( ((TextField)e.target).getText() );



            if(!function.parse()) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               return true;

            }

            for (int i=0; i < MAX_ANZ; i++) {

              try {

               double X = (double)(i);

               double value = function.getResult(X);

  	         acoeff[Wnum][i] = value;

               acoeff[Wnum][i] = Math.min(acoeff[Wnum][i],1.0);

               acoeff[Wnum][i] = Math.max(acoeff[Wnum][i],-1.0);

              } catch(Exception err) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               acoeff[Wnum][i] = 0.0;

              }

  	        ta_a[i].setText(""+acoeff[Wnum][i]);

	        s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

            }

        }



        else if (e.target.equals(SinExpr)) {

            ParseFunction function = new ParseFunction( ((TextField)e.target).getText() );



            if(!function.parse()) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               return true;

            }

            for (int i=1; i < MAX_ANZ; i++) {

              try {

               double X = (double)(i);

               double value = function.getResult(X);

  	         bcoeff[Wnum][i] = value;

               bcoeff[Wnum][i] = Math.min(bcoeff[Wnum][i],1.0);

               bcoeff[Wnum][i] = Math.max(bcoeff[Wnum][i],-1.0);

              } catch(Exception err) {

               this.showStatus("Failed to parse function!");

               System.out.println("Failed to parse function!");

               bcoeff[Wnum][i] = 0.0;

              }

  	        ta_b[i].setText(""+bcoeff[Wnum][i]);

	        s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

            }

        }

      repaint();

      return true;

      } //  if (e.target instanceof TextField)



      if (e.target instanceof Button) {

       for (int i=0; i < MAX_ANZ; i++)  {

  	   if ( e.target.equals(la_a[i]) )  {

//             this.showStatus("Zeroing "+e.target);

  	       acoeff[Wnum][i] = 0.0;

             CosExpr.setText("");

             SinExpr.setText("");

	       ta_a[i].setText(""+acoeff[Wnum][i]);

	       s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

             repaint();

             return true;

	   }  // if ( e.target.equals(ta_a[i]) )

  	   else if ( e.target.equals(la_b[i]) )  {

           if (i > 0) {

//             this.showStatus("Zeroing "+e.target);

  	       bcoeff[Wnum][i] = 0.0;

             CosExpr.setText("");

             SinExpr.setText("");

	       ta_b[i].setText(""+bcoeff[Wnum][i]);

	       s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

             repaint();

             return true;

           } // if (i > 0)

	   }  // else if ( e.target.equals(ta_b[i]) )

        } // for (int i=0; i < MAX_ANZ; i++)

      } // if (e.target instanceof Button)





      else if (e.target.equals(Wave1)) {

         Wnum = 0;

         CosExpr.setText("");

         SinExpr.setText("");

         this.showStatus("Using Wave Number 1");

         for (int i=0; i < MAX_ANZ; i++)  {

  	      ta_a[i].setText(""+acoeff[Wnum][i]);

	      s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

            if (i > 0) {

     	         ta_b[i].setText(""+bcoeff[Wnum][i]);

	         s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

            }

         }

         repaint();

         return true;



      }

      else if (e.target.equals(Wave2)) {

         Wnum = 1;

         CosExpr.setText("");

         SinExpr.setText("");

         this.showStatus("Using Wave Number 2");

         for (int i=0; i < MAX_ANZ; i++)  {

  	      ta_a[i].setText(""+acoeff[Wnum][i]);

	      s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

            if (i > 0) {

     	         ta_b[i].setText(""+bcoeff[Wnum][i]);

	         s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

            }

         }

         repaint();

         return true;



      }

      else if (e.target.equals(Wave3)) {

         Wnum = 2;

         CosExpr.setText("");

         SinExpr.setText("");

         this.showStatus("Using Wave Number 3");

         for (int i=0; i < MAX_ANZ; i++)  {

  	      ta_a[i].setText(""+acoeff[Wnum][i]);

	      s_a[i].setValue( (int)(acoeff[Wnum][i]*INT_MULT) );

            if (i > 0) {

     	         ta_b[i].setText(""+bcoeff[Wnum][i]);

	         s_b[i].setValue( (int)(bcoeff[Wnum][i]*INT_MULT) );

            }

         }

         repaint();

         return true;



      }

      else if (e.target.equals(AudioOff)) {

         if ( soundOn ) {

   	      AudioPlayer.player.stop(soundStream);

	      showStatus("Audio OFF");

            soundOn = false;

         }

         return true;

      }



      else if (e.target.equals(AudioOn)) {

         if ( !soundOn ) {

  	      AudioPlayer.player.start(soundStream);

	      showStatus("Audio ON");

            soundOn = true;

         }

         return true;



      }



    case Event.SCROLL_ABSOLUTE:

    case Event.SCROLL_LINE_UP:

    case Event.SCROLL_PAGE_UP:

    case Event.SCROLL_LINE_DOWN:

    case Event.SCROLL_PAGE_DOWN:



      for (int i=0; i < MAX_ANZ; i++) {

   	   if ( e.target.equals(s_a[i]) )

	   {

           CosExpr.setText("");

           SinExpr.setText("");

           int value = ((Scrollbar)e.target).getValue();

	     acoeff[Wnum][i] = (double)(value) / INT_MULT;

	     ta_a[i].setText(""+acoeff[Wnum][i]);

	     break;

	   }

	   else if ( ( i != 0 ) &&  ( e.target.equals(s_b[i]) ) )

	   {

           CosExpr.setText("");

           SinExpr.setText("");

           int value = ((Scrollbar)e.target).getValue();

	     bcoeff[Wnum][i] = (double)(value) / INT_MULT;

	     ta_b[i].setText(""+bcoeff[Wnum][i]);

	     break;

	   }

	} // for (int i=0...

      repaint();

      return true;

    case Event.WINDOW_DESTROY:

      destroy();

      return true;





    default:

       return super.handleEvent(e);



    } // switch ...



  }



  public void paint(Graphics g) {

    CalculatePlot();

    g.drawImage(bg, 0, 0, null);

    g.clipRect(0, 0, 320, 200);  // Keeps drawing out of area



    g.setColor(Color.red);

    for (int i = 0; i < 320; i+=80) {

      for (int j = 0; j < 80; j++) {

	      g.drawLine(i+j, iy[j], i+j+1, iy[j+1]);

      }

    }





  }



  public void stop() {

     if ( AUDIO && soundOn )

	AudioPlayer.player.stop(soundStream);

  }



  public void start() {

     if ( AUDIO && soundOn )

	AudioPlayer.player.start(soundStream);

  }



  public void destroy() {

    if ( AUDIO )

      AudioPlayer.player.stop(soundStream);

    for (int i=0; i < MAX_ANZ; i++)

      {

      for (int j=0; j<MAX_WAVES; j++) {

	   acoeff[j][i] = 0.0;

	   bcoeff[j][i] = 0.0;

      }

	la_a[i] = null;

	la_b[i] = null;

	s_a[i] = null;

	s_b[i] = null;

      iy = null;

      }

      Wnum = 0;

      removeAll();

  }



  public static byte int2ulaw(int ch) {

    int mask;



    if (ch < 0) {

      ch = -ch;

      mask = 0x7f;

    }

    else {

      mask = 0xff;

    }

    //

    if (ch < 32) {

      ch = 0xF0 | 15 - (ch/2);

    }

    else

      if (ch < 96) {

	ch = 0xE0 | 15 - (ch-32)/4;

      }

      else

	if (ch < 224) {

	  ch = 0xD0 | 15 - (ch-96)/8;

	}

	else

	  if (ch < 480) {

	    ch = 0xC0 | 15 - (ch-224)/16;

	  }

	  else

	    if (ch < 992 ) {

	      ch = 0xB0 | 15 - (ch-480)/32;

	    }

	    else

	      if (ch < 2016) {

		ch = 0xA0 | 15 - (ch-992)/64;

	      }

	      else

		if (ch < 4064) {

		  ch = 0x90 | 15 - (ch-2016)/128;

		}

		else

		  if (ch < 8160) {

		    ch = 0x80 | 15 - (ch-4064)/256;

		  }

		  else {

		    ch = 0x80;

		  }

    return (byte)(mask & ch);

  }





  public String getAppletInfo()

    {

      return "fourier.class 96/07/15, Copyright (C) 1996 Manfred Thole\n" +

	     "This program is free software; you can redistribute it and/or modify it\n" +

             "under the terms of the GNU General Public License as published by the\n" +

             "Free Software Foundation; either version 2 of the License, or (at your\n" +

             "option) any later version.\n" +

             "\n" +

             "This program is distributed in the hope that it will be useful, but\n" +

             "WITHOUT ANY WARRANTY; without even the implied warranty of\n" +

             "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n" +

             "General Public License for more details.\n" +

             "\n" +

             "You should have received a copy of the GNU General Public License along\n" +

             "with this program; if not, write to the Free Software Foundation, Inc.,\n" +

             "675 Mass Ave, Cambridge, MA 02139, USA.\n";

    }



  public String[][] getParameterInfo() {

    String[][] info = { { "sin_name", "String", "Sine: / Sine: / ..." },

			{ "cos_name", "String", "Cosine: / Cosine: / ..." }

		      };

    return info;

  }



   public void constrain(Container container, Component component,

                  int grid_x, int grid_y, int grid_width, int grid_height,

                  int fill, int anchor, double weight_x, double weight_y,

                  int top, int left, int bottom, int right, int ipad_x,

                  int ipad_y)

    {

        GridBagConstraints c = new GridBagConstraints();

        c.gridx = grid_x; c.gridy = grid_y;

        c.gridwidth = grid_width; c.gridheight = grid_height;

        c.fill = fill; c.anchor = anchor;

        c.ipadx = ipad_x; c.ipady = ipad_y;

        c.weightx = weight_x; c.weighty = weight_y;

        if (top+bottom+left+right > 0)

            c.insets = new Insets(top, left, bottom, right);



        ((GridBagLayout)container.getLayout()).setConstraints(component, c);

        container.add(component);

    }



    public void constrain(Container container, Component component,

                  int grid_x, int grid_y, int grid_width, int grid_height,

                  int fill, int anchor, double weight_x, double weight_y,

                  int top, int left, int bottom, int right) {

       constrain(container, component, grid_x, grid_y, grid_width,

                  grid_height, fill, anchor, weight_x, weight_y,

                  top, left, bottom, right, 0, 0);

    }



    public void constrain(Container container, Component component,

                  int grid_x, int grid_y, int grid_width, int grid_height) {

        constrain(container, component, grid_x, grid_y,

              grid_width, grid_height, GridBagConstraints.NONE,

              GridBagConstraints.NORTHWEST, 0.0, 0.0, 0, 0, 0, 0, 0, 0);

    }



    public void constrain(Container container, Component component,

                  int grid_x, int grid_y, int grid_width, int grid_height,

                  int top, int left, int bottom, int right) {

        constrain(container, component, grid_x, grid_y,

              grid_width, grid_height, GridBagConstraints.NONE,

              GridBagConstraints.NORTHWEST,

              0.0, 0.0, top, left, bottom, right, 0, 0);

    }





}