rtp_player_dialog.h

Go to the documentation of this file.

 
#ifndef RTP_PLAYER_DIALOG_H
#define RTP_PLAYER_DIALOG_H
 
#include "config.h"
 
#include <mutex>
 
#include "ui/rtp_stream.h"
 
#include "wireshark_dialog.h"
#include "rtp_audio_stream.h"
 
#include <QWidget>
#include <QMap>
#include <QMultiHash>
#include <QTreeWidgetItem>
#include <QMetaType>
#include <ui/qt/widgets/qcustomplot.h>
 
#ifdef QT_MULTIMEDIA_LIB
# if (QT_VERSION >= QT_VERSION_CHECK(6, 0, 0))
# include <QAudioDevice>
# else
# include <QAudioDeviceInfo>
# endif
#endif
 
namespace Ui {
class RtpPlayerDialog;
}
 
class QCPItemStraightLine;
class QDialogButtonBox;
class QMenu;
class RtpAudioStream;
class QCPAxisTicker;
class QCPAxisTickerDateTime;
 
typedef enum {
    save_audio_none,    
    save_audio_au,      
    save_audio_wav      
} save_audio_t;
 
typedef enum {
    save_payload_none,  
    save_payload_data   
} save_payload_t;
 
typedef enum {
    save_mode_from_cursor,  
    save_mode_sync_stream,  
    save_mode_sync_file     
} save_mode_t;
 
class RtpBaseDialog : public WiresharkDialog
{
    Q_OBJECT
protected:
    explicit RtpBaseDialog(QWidget &parent, CaptureFile &cf) : WiresharkDialog(parent, cf) {}
 
#ifdef QT_MULTIMEDIA_LIB
public slots:
    virtual void rtpAnalysisReplace() = 0;
 
    virtual void rtpAnalysisAdd() = 0;
 
    virtual void rtpAnalysisRemove() = 0;
#endif // QT_MULTIMEDIA_LIB
};
 
class RtpPlayerDialog : public RtpBaseDialog
{
    Q_OBJECT
#ifdef QT_MULTIMEDIA_LIB
    Q_PROPERTY(QString currentOutputDeviceName READ currentOutputDeviceName)
#endif
 
public:
    static RtpPlayerDialog *openRtpPlayerDialog(QWidget &parent, CaptureFile &cf, QObject *packet_list, bool capture_running);
 
    RtpPlayerDialog(RtpPlayerDialog &other) = delete;
 
    void operator=(const RtpPlayerDialog &) = delete;
 
    static QToolButton *addPlayerButton(QDialogButtonBox *button_box, QDialog *dialog);
 
#ifdef QT_MULTIMEDIA_LIB
    void accept();
 
    void reject();
 
    void setMarkers();
 
    void replaceRtpStreams(QVector<rtpstream_id_t *> stream_ids);
 
    void addRtpStreams(QVector<rtpstream_id_t *> stream_ids);
 
    void removeRtpStreams(QVector<rtpstream_id_t *> stream_ids);
 
signals:
    // Tells the packet list to redraw. An alternative might be to add a
    // cf_packet_marked callback to file.[ch] but that's synchronous and
    // might incur too much overhead.
    void packetsMarked();
 
    void updateFilter(QString filter, bool force = false);
 
    void goToPacket(int packet_num);
 
    void rtpAnalysisDialogReplaceRtpStreams(QVector<rtpstream_id_t *> stream_infos);
 
    void rtpAnalysisDialogAddRtpStreams(QVector<rtpstream_id_t *> stream_infos);
 
    void rtpAnalysisDialogRemoveRtpStreams(QVector<rtpstream_id_t *> stream_infos);
 
public slots:
    void rtpAnalysisReplace();
 
    void rtpAnalysisAdd();
 
    void rtpAnalysisRemove();
 
#endif
protected:
    explicit RtpPlayerDialog(QWidget &parent, CaptureFile &cf, bool capture_running);
#ifdef QT_MULTIMEDIA_LIB
    ~RtpPlayerDialog();
 
    virtual void showEvent(QShowEvent *);
 
    void contextMenuEvent(QContextMenuEvent *event);
 
    bool eventFilter(QObject *obj, QEvent *event);
 
private slots:
    void retapPackets();
 
    void captureEvent(CaptureEvent e);
 
    void rescanPackets(bool rescale_axes = false);
 
    void createPlot(bool rescale_axes = false);
 
    void updateWidgets();
 
    void itemEntered(QTreeWidgetItem *item, int column);
 
    void mouseMovePlot(QMouseEvent *event);
 
    void mouseMoveUpdate();
 
    void showGraphContextMenu(const QPoint &pos);
 
    void graphClicked(QMouseEvent *event);
 
    void graphDoubleClicked(QMouseEvent *event);
 
    void plotClicked(QCPAbstractPlottable *plottable, int dataIndex, QMouseEvent *event);
 
    void updateHintLabel();
 
    void resetXAxis();
 
    void updateGraphs();
 
    void playFinished(RtpAudioStream *stream, QAudio::Error error);
 
    void setPlayPosition(double secs);
 
    void setPlaybackError(const QString playback_error);
 
    void changeAudioRoutingOnItem(QTreeWidgetItem *ti, AudioRouting new_audio_routing);
 
    void changeAudioRouting(AudioRouting new_audio_routing);
 
    void invertAudioMutingOnItem(QTreeWidgetItem *ti);
 
    void on_playButton_clicked();
 
    void on_pauseButton_clicked();
 
    void on_stopButton_clicked();
 
    void on_actionReset_triggered();
 
    void on_actionZoomIn_triggered();
 
    void on_actionZoomOut_triggered();
 
    void on_actionMoveLeft10_triggered();
 
    void on_actionMoveRight10_triggered();
 
    void on_actionMoveLeft1_triggered();
 
    void on_actionMoveRight1_triggered();
 
    void on_actionGoToPacket_triggered();
 
    void on_actionGoToSetupPacketPlot_triggered();
 
    void on_actionGoToSetupPacketTree_triggered();
 
    void on_actionRemoveStream_triggered();
 
    void on_actionAudioRoutingP_triggered();
 
    void on_actionAudioRoutingL_triggered();
 
    void on_actionAudioRoutingLR_triggered();
 
    void on_actionAudioRoutingR_triggered();
 
    void on_actionAudioRoutingMute_triggered();
 
    void on_actionAudioRoutingUnmute_triggered();
 
    void on_actionAudioRoutingMuteInvert_triggered();
 
    void on_streamTreeWidget_itemSelectionChanged();
 
    void on_streamTreeWidget_itemDoubleClicked(QTreeWidgetItem *item, const int column);
 
    void on_outputDeviceComboBox_currentTextChanged(const QString &);
 
    void on_outputAudioRate_currentTextChanged(const QString &rate_string);
 
    void on_jitterSpinBox_valueChanged(double);
 
    void on_timingComboBox_currentIndexChanged(int);
 
    void on_todCheckBox_toggled(bool checked);
 
    void on_visualSRSpinBox_editingFinished();
 
    void on_buttonBox_helpRequested();
 
    void on_actionSelectAll_triggered();
 
    void on_actionSelectInvert_triggered();
 
    void on_actionSelectNone_triggered();
 
    void outputNotify();
 
    void on_actionPlay_triggered();
 
    void on_actionStop_triggered();
 
    void on_actionSaveAudioFromCursor_triggered();
 
    void on_actionSaveAudioSyncStream_triggered();
 
    void on_actionSaveAudioSyncFile_triggered();
 
    void on_actionSavePayload_triggered();
 
    void on_actionSelectInaudible_triggered();
 
    void on_actionDeselectInaudible_triggered();
 
    void on_actionPrepareFilter_triggered();
 
    void on_actionReadCapture_triggered();
 
#if (QT_VERSION >= QT_VERSION_CHECK(6, 0, 0))
    void sinkStateChanged();
#endif
 
#endif
private:
    static RtpPlayerDialog *pinstance_;
 
    static std::mutex init_mutex_;
 
    static std::mutex run_mutex_;
 
#ifdef QT_MULTIMEDIA_LIB
    Ui::RtpPlayerDialog *ui;
 
    QMenu *graph_ctx_menu_;
 
    QMenu *list_ctx_menu_;
 
    double first_stream_rel_start_time_;
 
    double first_stream_abs_start_time_;
 
    double first_stream_rel_stop_time_;
 
    double streams_length_;
 
    double start_marker_time_;
 
    double start_marker_time_play_;
 
    QCPItemStraightLine *cur_play_pos_;
 
    QCPItemStraightLine *start_marker_pos_;
 
    QString playback_error_;
 
    QSharedPointer<QCPAxisTicker> number_ticker_;
 
    QSharedPointer<QCPAxisTickerDateTime> datetime_ticker_;
 
    bool stereo_available_;
 
    QList<RtpAudioStream *> playing_streams_;
 
#if (QT_VERSION >= QT_VERSION_CHECK(6, 0, 0))
    QAudioSink *marker_stream_;
 
    QTimer notify_timer_;
 
    qint64 notify_timer_start_diff_;
#else
    QAudioOutput *marker_stream_;
#endif
    quint32 marker_stream_requested_out_rate_;
 
    QTreeWidgetItem *last_ti_;
 
    bool listener_removed_;
 
    QPushButton *read_btn_;
 
    QToolButton *inaudible_btn_;
 
    QToolButton *analyze_btn_;
 
    QPushButton *prepare_btn_;
 
    QPushButton *export_btn_;
 
    QMultiHash<unsigned, RtpAudioStream *> stream_hash_;
 
    bool block_redraw_;
 
    int lock_ui_;
 
    bool read_capture_enabled_;
 
    double silence_skipped_time_;
 
    QTimer *mouse_update_timer_;
 
    QPoint mouse_pos_;
 
//  const QString streamKey(const rtpstream_info_t *rtpstream);
//  const QString streamKey(const packet_info *pinfo, const struct _rtp_info *rtpinfo);
 
    // Tap callbacks
//  static void tapReset(void *tapinfo_ptr);
 
    static tap_packet_status tapPacket(void *tapinfo_ptr, packet_info *pinfo, epan_dissect_t *, const void *rtpinfo_ptr, tap_flags_t flags);
 
    static void tapDraw(void *tapinfo_ptr);
 
    void addPacket(packet_info *pinfo, const struct _rtp_info *rtpinfo);
 
    void zoomXAxis(bool in);
 
    void panXAxis(int x_pixels);
 
    const QString getFormatedTime(double f_time);
 
    const QString getFormatedHoveredTime();
 
    int getHoveredPacket();
 
    QString currentOutputDeviceName();
 
    double getStartPlayMarker();
 
    void drawStartPlayMarker();
 
    void setStartPlayMarker(double new_time);
 
    void updateStartStopTime(rtpstream_info_t *rtpstream, bool is_first);
 
    void formatAudioRouting(QTreeWidgetItem *ti, AudioRouting audio_routing);
 
    bool isStereoAvailable();
 
#if (QT_VERSION >= QT_VERSION_CHECK(6, 0, 0))
    QAudioSink *getSilenceAudioOutput();
 
    QAudioDevice getCurrentDeviceInfo();
#else
    QAudioOutput *getSilenceAudioOutput();
 
    QAudioDeviceInfo getCurrentDeviceInfo();
#endif
 
    QTreeWidgetItem *findItemByCoords(QPoint point);
 
    QTreeWidgetItem *findItem(QCPAbstractPlottable *plottable);
 
    void handleItemHighlight(QTreeWidgetItem *ti, bool scroll);
 
    void highlightItem(QTreeWidgetItem *ti, bool highlight);
 
    void invertSelection();
 
    void handleGoToSetupPacket(QTreeWidgetItem *ti);
 
    void addSingleRtpStream(rtpstream_id_t *id);
 
    void removeRow(QTreeWidgetItem *ti);
 
    void fillAudioRateMenu();
 
    void cleanupMarkerStream();
 
    qint64 saveAudioHeaderAU(QFile *save_file, quint32 channels, unsigned audio_rate);
 
    qint64 saveAudioHeaderWAV(QFile *save_file, quint32 channels, unsigned audio_rate, qint64 samples);
 
    bool writeAudioSilenceSamples(QFile *out_file, qint64 samples, int stream_count);
 
    bool writeAudioStreamsSamples(QFile *out_file, QVector<RtpAudioStream *> streams, bool big_endian);
 
    save_audio_t selectFileAudioFormatAndName(QString *file_path);
 
    save_payload_t selectFilePayloadFormatAndName(QString *file_path);
 
    QVector<RtpAudioStream *>getSelectedAudibleNonmutedAudioStreams();
 
    void saveAudio(save_mode_t save_mode);
 
    void savePayload();
 
    void lockUI();
 
    void unlockUI();
 
    void selectInaudible(bool select);
 
    QVector<rtpstream_id_t *>getSelectedRtpStreamIDs();
 
    void fillTappedColumns();
 
#else // QT_MULTIMEDIA_LIB
private:
    Ui::RtpPlayerDialog *ui;
#endif // QT_MULTIMEDIA_LIB
};
 
#endif // RTP_PLAYER_DIALOG_H