niftkQtAudioDataSourceService.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/
#ifndef niftkQtAudioDataSourceService_h
#define niftkQtAudioDataSourceService_h

#include <niftkIGIDataSource.h>
#include <niftkIGIDataSourceLocker.h>
#include <niftkIGILocalDataSourceI.h>

#include <QObject>
#include <QSet>
#include <QMutex>
#include <QString>
#include <QAudioInput>

// forward-decl
class QAudioDeviceInfo;
class QAudioFormat;
class QFile;

namespace niftk
{

class QtAudioDataSourceService
    : public QObject
    , public IGIDataSource
{

  Q_OBJECT

public:

  mitkClassMacroItkParent(QtAudioDataSourceService, IGIDataSource)
  mitkNewMacro3Param(QtAudioDataSourceService, QString, const IGIDataSourceProperties&, mitk::DataStorage::Pointer)

  virtual void StartPlayback(niftk::IGIDataType::IGITimeType firstTimeStamp,
                             niftk::IGIDataType::IGITimeType lastTimeStamp) override;

  void PlaybackData(niftk::IGIDataType::IGITimeType requestedTimeStamp) override;

  virtual void StopPlayback() override;

  virtual std::vector<IGIDataItemInfo> Update(const niftk::IGIDataType::IGITimeType& time) override;

  bool ProbeRecordedData(niftk::IGIDataType::IGITimeType* firstTimeStampInStore,
                         niftk::IGIDataType::IGITimeType* lastTimeStampInStore) override;

  virtual void StartRecording() override;

  virtual void StopRecording() override;

  virtual void SetProperties(const IGIDataSourceProperties& properties) override;

  virtual IGIDataSourceProperties GetProperties() const override;

protected:

  QtAudioDataSourceService(QString factoryName,
                               const IGIDataSourceProperties& properties,
                               mitk::DataStorage::Pointer dataStorage
                               );
  virtual ~QtAudioDataSourceService();

private slots:

  void OnStateChanged(QAudio::State state);
  void OnReadyRead();

private:

  QtAudioDataSourceService(const QtAudioDataSourceService&); // deliberately not implemented
  QtAudioDataSourceService& operator=(const QtAudioDataSourceService&); // deliberately not implemented

  void StartWAVFile();
  void FinishWAVFile();

  static niftk::IGIDataSourceLocker               s_Lock;
  QMutex                                          m_Lock;
  int                                             m_SourceNumber;
  niftk::IGIDataType::IGIIndexType                m_FrameId;

  QAudioInput*                                    m_InputDevice;
  QIODevice*                                      m_InputStream;      // we do not own this one!
  QFile*                                          m_OutputFile;
  QAudioDeviceInfo                                m_DeviceInfo;
  QAudioFormat                                    m_Inputformat;
  int                                             m_SegmentCounter;

}; // end class

} // end namespace

#endif