Expose metadata from every controlled component.

[sonitus.git] / src / main / java / net / pterodactylus / sonitus / data / sink / AudioSink.java

diff --git a/src/main/java/net/pterodactylus/sonitus/data/sink/AudioSink.java b/src/main/java/net/pterodactylus/sonitus/data/sink/AudioSink.java
index 72442f5..05253f3 100644 (file)
--- a/src/main/java/net/pterodactylus/sonitus/data/sink/AudioSink.java
+++ b/src/main/java/net/pterodactylus/sonitus/data/sink/AudioSink.java
@@ -20,6 +20,7 @@ package net.pterodactylus.sonitus.data.sink;
 import static javax.sound.sampled.FloatControl.Type.VOLUME;
 
 import java.io.IOException;
+import java.io.OutputStream;
 import java.util.Arrays;
 import java.util.List;
 import java.util.logging.Logger;
@@ -34,6 +35,8 @@ import net.pterodactylus.sonitus.data.Metadata;
 import net.pterodactylus.sonitus.data.Sink;
 import net.pterodactylus.sonitus.data.Source;
 import net.pterodactylus.sonitus.data.controller.Fader;
+import net.pterodactylus.sonitus.data.controller.Switch;
+import net.pterodactylus.sonitus.io.IntegralWriteOutputStream;
 
 import com.google.common.base.Preconditions;
 
@@ -51,22 +54,61 @@ public class AudioSink implements Sink {
        /** The volume fader. */
        private final Fader volumeFader;
 
+       /** The “mute” switch. */
+       private final Switch muteSwitch;
+
        /** The current metadata. */
        private Metadata metadata;
 
        /** The audio output. */
        private SourceDataLine sourceDataLine;
 
+       /** A buffered output stream to ensure correct writing to the source data line. */
+       private OutputStream sourceDataLineOutputStream = new IntegralWriteOutputStream(new OutputStream() {
+
+               @Override
+               public void write(int b) throws IOException {
+               }
+
+               @Override
+               public void write(byte[] b) throws IOException {
+                       write(b, 0, b.length);
+               }
+
+               @Override
+               public void write(byte[] b, int off, int len) throws IOException {
+                       if (sourceDataLine != null) {
+                               sourceDataLine.write(b, off, len);
+                       }
+               }
+       }, 1024);
+
        /** Creates a new audio sink. */
        public AudioSink() {
-               super();
-               volumeFader = new Fader() {
+               volumeFader = new Fader("Volume") {
 
                        @Override
-                       protected void valueSet(int value) {
+                       protected void valueSet(Double value) {
                                if (sourceDataLine != null) {
                                        FloatControl volumeControl = (FloatControl) sourceDataLine.getControl(VOLUME);
-                                       volumeControl.setValue(value * volumeControl.getMaximum() / (float) maximum());
+                                       volumeControl.setValue((float) (value * volumeControl.getMaximum()));
+                               }
+                       }
+               };
+               muteSwitch = new Switch("Mute") {
+
+                       private float previousValue;
+
+                       @Override
+                       protected void valueSet(Boolean value) {
+                               if (sourceDataLine != null) {
+                                       FloatControl volumeControl = (FloatControl) sourceDataLine.getControl(VOLUME);
+                                       if (value) {
+                                               previousValue = volumeControl.getValue();
+                                               volumeControl.setValue(0);
+                                       } else {
+                                               volumeControl.setValue(previousValue);
+                                       }
                                }
                        }
                };
@@ -77,8 +119,18 @@ public class AudioSink implements Sink {
        //
 
        @Override
-       public List<Controller> controllers() {
-               return Arrays.<Controller>asList(volumeFader);
+       public String name() {
+               return "Audio Output";
+       }
+
+       @Override
+       public Metadata metadata() {
+               return metadata;
+       }
+
+       @Override
+       public List<Controller<?>> controllers() {
+               return Arrays.<Controller<?>>asList(volumeFader, muteSwitch);
        }
 
        //
@@ -108,11 +160,12 @@ public class AudioSink implements Sink {
        @Override
        public void metadataUpdated(Metadata metadata) {
                logger.info(String.format("Now playing %s.", metadata));
+               this.metadata = metadata;
        }
 
        @Override
-       public void process(byte[] buffer) {
-               sourceDataLine.write(buffer, 0, buffer.length);
+       public void process(byte[] buffer) throws IOException {
+               sourceDataLineOutputStream.write(buffer);
                logger.finest(String.format("AudioSink: Wrote %d Bytes.", buffer.length));
        }