|
packagecom.test;
importjava.io.ByteArrayInputStream;
importjava.io.ByteArrayOutputStream;
importjava.io.File;
importjava.io.FileOutputStream;
importjava.io.IOException;
importjava.io.SequenceInputStream;
importjavax.servlet.http.HttpServletRequest;
importjavax.sound.sampled.AudioFileFormat;
importjavax.sound.sampled.AudioInputStream;
importjavax.sound.sampled.AudioSystem;
importorg.apache.commons.logging.Log;
importorg.apache.commons.logging.LogFactory;
importorg.eclipse.jetty.io.Connection;
importorg.eclipse.jetty.server.Server;
importorg.eclipse.jetty.server.nio.SelectChannelConnector;
importorg.eclipse.jetty.websocket.WebSocket;
importorg.eclipse.jetty.websocket.WebSocketHandler;
publicclass
TestRecordServlet extendsServer
{
privatefinal
static
Log LOG = LogFactory.getLog( TestRecordServlet.class);
publicTestRecordServlet(intport)
{
SelectChannelConnector
connector = newSelectChannelConnector();
connector.setPort(port);
addConnector(connector);
WebSocketHandler
wsHandler = newWebSocketHandler()
{
publicWebSocket
doWebSocketConnect(HttpServletRequest request, String protocol) {
returnnew
FaceDetectWebSocket();
}
};
setHandler(wsHandler);
}
/**
*
Simple innerclass that is used to handle websocket connections.
*
*
@author jos
*/
privatestatic
class
FaceDetectWebSocket implementsWebSocket,
WebSocket.OnBinaryMessage,
WebSocket.OnTextMessage {
privateString
currentCommand ="";
privateConnection
connection;
//private
FaceDetection faceDetection = new FaceDetection();
publicFaceDetectWebSocket()
{
super();
}
/**
*
On open we set the connection locally, and enable
*
binary support
*/
publicvoid
onOpen(Connection connection) {
this.connection
= connection;
this.connection.setMaxBinaryMessageSize(1024*
512);
}
/**
*
Cleanup if needed. Not used for this example
*/
publicvoid
onClose(intcode,
String message) {}
/**
*
When we receive a binary message we assume it is an image. We then run this
*
image through our face detection algorithm and send back the response.
*/
publicvoid
onMessage(byte[]
data, intoffset,
intlength)
{
if(currentCommand.equals("start"))
{
try{
//
The temporary file that contains our captured audio stream
File
f = newFile("out.wav");
//
if the file already exists we append it.
if(f.exists())
{
LOG.info("Adding
received block to existing file.");
//
two clips are used to concat the data
AudioInputStream
clip1 = AudioSystem.getAudioInputStream(f);
AudioInputStream
clip2 = AudioSystem.getAudioInputStream(newByteArrayInputStream(data));
//
use a sequenceinput to cat them together
AudioInputStream
appendedFiles =
newAudioInputStream(
newSequenceInputStream(clip1,
clip2),
clip1.getFormat(),
clip1.getFrameLength()
+ clip2.getFrameLength());
//
write out the output to a temporary file
AudioSystem.write(appendedFiles,
AudioFileFormat.Type.WAVE,
newFile("out2.wav"));
//
rename the files and delete the old one
File
f1 = newFile("out.wav");
File
f2 = newFile("out2.wav");
f1.delete();
f2.renameTo(newFile("out.wav"));
}else{
LOG.info("Starting
new recording.");
FileOutputStream
fOut = newFileOutputStream("out.wav",true);
fOut.write(data);
fOut.close();
}
}catch(Exception
e) {
LOG.error("sss:"+
e );
}
}
}
publicvoid
onMessage(String data) {
if(data.startsWith("start"))
{
//
before we start we cleanup anything left over
//cleanup();
currentCommand
= "start";
}elseif
(data.startsWith("stop"))
{
currentCommand
= "stop";
}elseif
(data.startsWith("clear"))
{
//
just remove the current recording
//cleanup();
}elseif
(data.startsWith("analyze"))
{
}
}
}
/**
*
Start the server on port 999
*/
publicstatic
void
main(String[] args) throwsException
{
TestRecordServlet
server = newTestRecordServlet(8080);
server.start();
server.join();
}
}
|
