Initial code check in continued

src/spectrum/WaterfallPanel.java

@@ -0,0 +1,287 @@
+/*******************************************************************************
+ *     SDR Trunk 
+ *     Copyright (C) 2014 Dennis Sheirer
+ * 
+ *     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 3 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, see <http://www.gnu.org/licenses/>
+ ******************************************************************************/
+package spectrum;
+
+import java.awt.Color;
+import java.awt.EventQueue;
+import java.awt.Graphics;
+import java.awt.Graphics2D;
+import java.awt.Image;
+import java.awt.Point;
+import java.awt.event.MouseEvent;
+import java.awt.event.MouseListener;
+import java.awt.geom.Line2D;
+import java.awt.image.ColorModel;
+import java.awt.image.MemoryImageSource;
+import java.text.DecimalFormat;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+import javax.swing.JPanel;
+
+import settings.ColorSetting;
+import settings.ColorSetting.ColorSettingName;
+import settings.Setting;
+import settings.SettingChangeListener;
+import buffer.FloatAveragingBuffer;
+import controller.ResourceManager;
+
+/**
+ * Produces a JPanel filled with a waterfall image derived from a continuous
+ * stream of received Discrete Fourier Transform (DFT) results from the
+ * DFTProcessor class.
+ */
+public class WaterfallPanel extends JPanel implements DFTResultsListener,
+													  Pausable,
+													  SettingChangeListener
+{
+    private static final long serialVersionUID = 1L;
+	private static DecimalFormat sCURSOR_FORMAT = new DecimalFormat( "000.00000" );
+	private static final String sSPECTRUM_CURSOR = "spectrum_cursor";
+	private static final String sPAUSED = "PAUSED";
+
+	private byte[] mPixels;
+    private int mFFTSize = 1024;
+    private int mImageHeight = 700;
+    private MemoryImageSource mMemoryImageSource;
+    private ColorModel mColorModel = WaterfallColorModel.getDefaultColorModel();
+	private Color mColorSpectrumCursor;
+    private Image mWaterfallImage;
+
+    float mAverage = 0.0f;
+    double mAlpha = 0.95d;
+    int mScalor = 85;
+    int temp = 0;
+
+	private Point mCursorLocation = new Point( 0, 0 );
+	private boolean mCursorVisible = false;
+	private int mCursorFrequency = 0;
+	private AtomicBoolean mPaused = new AtomicBoolean();
+	
+	private ResourceManager mResourceManager;
+    
+    private FloatAveragingBuffer mIntraBinAveragingBuffer = new FloatAveragingBuffer( 9 );
+
+	public WaterfallPanel( ResourceManager resourceManager )
+	{
+		super();
+
+		mResourceManager = resourceManager;
+		
+		mColorSpectrumCursor = getColor( ColorSettingName.SPECTRUM_CURSOR );
+
+		reset();
+	}
+	
+	public void dispose()
+	{
+		mResourceManager = null;
+		mMemoryImageSource = null;
+	}
+	
+	private void reset()
+	{
+		mPixels = new byte[ mFFTSize * mImageHeight ];
+
+		mMemoryImageSource = new MemoryImageSource( mFFTSize, 
+				mImageHeight,
+				mColorModel,
+				mPixels,
+				0,
+				mFFTSize );
+		
+        mMemoryImageSource.setAnimated( true );
+
+        mWaterfallImage = createImage( mMemoryImageSource );
+	}
+
+	/**
+	 * Pausable interface - pauses updates to the screen
+	 */
+	public void setPaused( boolean paused )
+	{
+		mPaused.set( paused );
+		
+		repaint();
+	}
+	
+	public boolean isPaused()
+	{
+		return mPaused.get();
+	}
+	
+	/**
+	 * Fetches a named color setting from the settings manager.  If the setting
+	 * doesn't exist, creates the setting using the defaultColor
+	 */
+	private Color getColor( ColorSettingName name )
+	{
+		ColorSetting setting = mResourceManager.getSettingsManager()
+				.getColorSetting( name );
+		
+		return setting.getColor();
+	}
+
+	/**
+	 * Monitors for setting changes.  Colors can be changed by external actions
+	 * and will automatically update in this class
+	 */
+	@Override
+    public void settingChanged( Setting setting )
+    {
+		if( setting instanceof ColorSetting )
+		{
+			ColorSetting colorSetting = (ColorSetting)setting;
+
+			if( ( (ColorSetting) setting ).getColorSettingName() == 
+					ColorSettingName.SPECTRUM_CURSOR )
+			{
+				mColorSpectrumCursor = colorSetting.getColor();
+			}
+		}
+    }
+
+	public void setCursorLocation( Point point )
+	{
+		mCursorLocation = point;
+		
+		repaint();
+	}
+	
+	public void setCursorFrequency( int frequency )
+	{
+		mCursorFrequency = frequency;
+	}
+	
+	public void setCursorVisible( boolean visible )
+	{
+		mCursorVisible = visible;
+		
+		repaint();
+	}
+
+	public void paintComponent( Graphics g )
+	{
+		g.drawImage( mWaterfallImage, 0, 0, getWidth(), mImageHeight, this );
+		
+    	Graphics2D graphics = (Graphics2D) g;
+
+    	if( mCursorVisible )
+    	{
+    		
+        	graphics.setColor( mColorSpectrumCursor );
+        	
+        	graphics.draw( new Line2D.Float( mCursorLocation.x, 
+        									 0, 
+        									 mCursorLocation.x, 
+        									 (float)(getSize().getHeight() ) ) );
+
+    		String frequency = sCURSOR_FORMAT.format( mCursorFrequency / 1000000.0D );
+
+    		graphics.drawString( frequency , 
+    							 mCursorLocation.x + 5, 
+    							 mCursorLocation.y );
+    		
+    	}
+    	
+    	if( mPaused.get() )
+    	{
+    		graphics.drawString( sPAUSED, 20, 20 ); 
+    	}
+    	
+		graphics.dispose();
+	}
+
+	@Override
+    public void receive( float[] update )
+    {
+		//If our FFT size changes, reset our pixel map and image source
+		if( mFFTSize != update.length )
+		{
+			mFFTSize = update.length;
+
+			reset();
+		}
+
+		//Move the pixels down a row to make room for the new results
+		System.arraycopy( mPixels, 0, 
+				mPixels, mFFTSize, mPixels.length - mFFTSize );
+		
+		//Convert the power spectrum bin values into a byte (0-255)
+		//value to use with the color mapping
+		
+		double averagingTotal = 0.0;
+		
+		for( int x = 0; x < update.length; x++ )
+		{
+			float value = update[ x ];
+			
+			averagingTotal += value;
+
+			//Average the values toward zero
+			float pixel = value - mAverage;
+
+			//If we're below the average, make the negative value a positive
+			if( pixel < 0 )
+			{
+//				pixel = -pixel;
+				pixel = 0.0f;
+			}
+
+			//Scale the value
+			pixel *= mScalor;
+
+//			mPixels[ x ] = (byte)pixel;
+			mPixels[ x ] = (byte)( mIntraBinAveragingBuffer.get( pixel ) );
+		}
+		
+		//Update the average value
+		mAverage = (float) averagingTotal / update.length;
+
+		if( !mPaused.get() )
+		{
+			EventQueue.invokeLater( new Runnable() 
+			{
+				@Override
+	            public void run()
+	            {
+					if( mMemoryImageSource != null )
+					{
+						mMemoryImageSource.newPixels( mPixels, mColorModel, 0, mFFTSize );
+					}
+	            }
+			} );
+		}
+    }
+	
+	public class Handler implements MouseListener
+	{
+		@Override
+        public void mouseClicked( MouseEvent e )
+        {
+			//Display context menu to reset the display reset();
+        }
+
+        public void mousePressed( MouseEvent e ) {}
+        public void mouseReleased( MouseEvent e ) {}
+        public void mouseEntered( MouseEvent e ) {}
+        public void mouseExited( MouseEvent e ){}
+	}
+
+	@Override
+    public void settingDeleted( Setting setting ) {}
+}