You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

335 lines
6.8 KiB

/*
* Copyright (C) 2009,2010,2012,2014 by Jonathan Naylor G4KLX
*
* 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.
*/
#include "DVToolWriterThread.h"
#include "DVToolWriterApp.h"
#include "MessageData.h"
#include "HeaderData.h"
CDVToolWriterThread::CDVToolWriterThread() :
wxThread(wxTHREAD_JOINABLE),
m_soundcard(NULL),
m_dongle(NULL),
m_encoder(),
m_stopped(true),
m_reader(NULL),
m_writer(NULL),
m_audioBuffer(DSTAR_AUDIO_BLOCK_SIZE * 30U),
m_count(0U),
m_killed(false)
{
}
CDVToolWriterThread::~CDVToolWriterThread()
{
}
void* CDVToolWriterThread::Entry()
{
// Wait here until we have the essentials to run
while (!m_killed && (m_dongle == NULL || m_soundcard == NULL))
Sleep(500UL); // 1/2 sec
if (m_killed)
return NULL;
m_dongle->Create();
m_dongle->SetPriority(100U);
m_dongle->Run();
while (!m_killed) {
if (m_writer != NULL && m_reader != NULL)
processFile();
if (m_writer != NULL && m_reader == NULL)
processSoundcard();
Sleep(500UL); // 1/2 sec
}
m_soundcard->close();
delete m_soundcard;
m_dongle->kill();
return NULL;
}
void CDVToolWriterThread::kill()
{
m_killed = true;
}
void CDVToolWriterThread::encodeCallback(const unsigned char* ambe, unsigned int length)
{
if (m_writer == NULL)
return;
unsigned char frame[DV_FRAME_LENGTH_BYTES];
::memcpy(frame, ambe, VOICE_FRAME_LENGTH_BYTES);
if (m_count == 0U)
::memcpy(frame + VOICE_FRAME_LENGTH_BYTES, DATA_SYNC_BYTES, DATA_FRAME_LENGTH_BYTES);
else
m_encoder.getData(frame + VOICE_FRAME_LENGTH_BYTES);
m_count++;
if (m_count >= 21U)
m_count = 0U;
m_writer->writeFrame(frame, DV_FRAME_LENGTH_BYTES);
}
bool CDVToolWriterThread::processFile(CWAVFileReader* reader, CHeaderData* header, const wxString& text, IFileWriter* writer)
{
wxASSERT(reader != NULL);
wxASSERT(header != NULL);
wxASSERT(writer != NULL);
wxLogMessage(wxT("Processing file - %s"), reader->getFilename().c_str());
if (!m_stopped) {
wxLogError(wxT("Already processing a file"));
return false;
}
if (m_dongle == NULL) {
wxLogError(wxT("No Dongle has been set"));
return false;
}
if (m_reader != NULL) {
m_reader->close();
delete m_reader;
}
if (m_writer != NULL) {
m_writer->close();
delete m_writer;
}
wxLogMessage(wxT("Writing file - %s"), writer->getFilename().c_str());
bool ret = writer->open();
if (!ret) {
wxLogError(wxT("Cannot open the file for writing"));
return false;
}
writer->writeHeader(*header);
m_encoder.reset();
m_encoder.setMessageData(text);
m_encoder.setHeaderData(*header);
delete header;
m_reader = reader;
m_writer = writer;
return true;
}
bool CDVToolWriterThread::processAudio(CHeaderData* header, const wxString& text, IFileWriter* writer)
{
wxASSERT(header != NULL);
wxASSERT(writer != NULL);
if (!m_stopped) {
wxLogError(wxT("Already processing a file"));
return false;
}
if (m_dongle == NULL) {
wxLogError(wxT("No Dongle has been set"));
return false;
}
if (m_writer != NULL) {
m_writer->close();
delete m_writer;
}
wxLogMessage(wxT("Writing file - %s"), writer->getFilename().c_str());
bool ret = writer->open();
if (!ret) {
wxLogError(wxT("Cannot open the file for writing"));
return false;
}
writer->writeHeader(*header);
m_encoder.reset();
m_encoder.setMessageData(text);
m_encoder.setHeaderData(*header);
delete header;
m_writer = writer;
return true;
}
void CDVToolWriterThread::processAudio()
{
m_stopped = true;
}
void CDVToolWriterThread::setSoundCard(CSoundCardReaderWriter* soundcard)
{
wxASSERT(soundcard != NULL);
if (!m_stopped) {
soundcard->close();
delete soundcard;
return;
}
if (m_soundcard != NULL) {
m_soundcard->close();
delete m_soundcard;
}
m_soundcard = soundcard;
}
void CDVToolWriterThread::setDongle(CDongleThread* dongle)
{
wxASSERT(dongle != NULL);
if (!m_stopped) {
dongle->kill();
return;
}
if (m_dongle != NULL)
m_dongle->kill();
m_dongle = dongle;
}
void CDVToolWriterThread::processFile()
{
m_stopped = false;
m_count = 0U;
m_dongle->setEncode();
for (;;) {
if (m_killed) {
m_dongle->setIdle();
m_reader->close();
m_writer->close();
delete m_reader;
delete m_writer;
m_reader = NULL;
m_writer = NULL;
return;
}
unsigned int space = m_dongle->getEncodeSpace();
if (space >= DSTAR_RADIO_BLOCK_SIZE) {
wxFloat32 audio[DSTAR_RADIO_BLOCK_SIZE];
::memset(audio, 0x00U, DSTAR_RADIO_BLOCK_SIZE * sizeof(wxFloat32));
unsigned int n = m_reader->read(audio, DSTAR_RADIO_BLOCK_SIZE);
m_dongle->writeEncode(audio, DSTAR_RADIO_BLOCK_SIZE);
// End of file?
if (n != DSTAR_RADIO_BLOCK_SIZE)
break;
}
Sleep(FRAME_TIME_MS);
}
Sleep(FRAME_TIME_MS * 30U);
// Release the input file
m_reader->close();
delete m_reader;
m_reader = NULL;
// Release the output file
m_writer->close();
delete m_writer;
m_writer = NULL;
::wxGetApp().showFinished();
wxLogMessage(wxT("Finished processing the file"));
m_stopped = true;
m_dongle->setIdle();
}
void CDVToolWriterThread::processSoundcard()
{
m_stopped = false;
m_count = 0U;
m_dongle->setEncode();
for (;;) {
if (m_killed) {
m_dongle->setIdle();
m_writer->close();
delete m_writer;
m_writer = NULL;
return;
}
if (m_stopped)
break;
unsigned int size = m_audioBuffer.dataSpace();
if (size >= DSTAR_RADIO_BLOCK_SIZE) {
wxFloat32 audio[DSTAR_RADIO_BLOCK_SIZE];
unsigned int n = m_audioBuffer.getData(audio, DSTAR_RADIO_BLOCK_SIZE);
m_dongle->writeEncode(audio, n);
}
Sleep(FRAME_TIME_MS / 4U);
}
Sleep(FRAME_TIME_MS * 30U);
// Release the output file
m_writer->close();
delete m_writer;
m_writer = NULL;
wxLogMessage(wxT("Finished processing from the sound card"));
m_stopped = true;
m_dongle->setIdle();
}
void CDVToolWriterThread::callback(const wxFloat32* input, wxFloat32* output, unsigned int nSamples, int id)
{
if (m_writer != NULL && m_reader == NULL)
m_audioBuffer.addData(input, nSamples);
}

Powered by TurnKey Linux.