]> git.sesse.net Git - nageru/blobdiff - clip_list.cpp
Show interpolated frames in the live window.
[nageru] / clip_list.cpp
index 2599dcc2d246cbbeceabd00ba48d940192cd1b0f..75f4a9f46b6be66e0f72523528600262aab20350 100644 (file)
 #include "mainwindow.h"
 
-#include "clip_list.h"
-#include "ui_mainwindow.h"
-
+#include <math.h>
 #include <string>
 #include <vector>
 
+#include "clip_list.h"
+#include "timebase.h"
+#include "ui_mainwindow.h"
+
 using namespace std;
 
+string pts_to_string(int64_t pts)
+{
+       int64_t t = lrint((pts / double(TIMEBASE)) * 1e3);  // In milliseconds.
+       int ms = t % 1000;
+       t /= 1000;
+       int sec = t % 60;
+       t /= 60;
+       int min = t % 60;
+       t /= 60;
+       int hour = t;
+
+       char buf[256];
+       snprintf(buf, sizeof(buf), "%d:%02d:%02d.%03d", hour, min, sec, ms);
+       return buf;
+}
+
+string duration_to_string(int64_t pts_diff)
+{
+       int64_t t = lrint((pts_diff / double(TIMEBASE)) * 1e3);  // In milliseconds.
+       int ms = t % 1000;
+       t /= 1000;
+       int sec = t % 60;
+       t /= 60;
+       int min = t;
+
+       char buf[256];
+       snprintf(buf, sizeof(buf), "%d:%02d.%03d", min, sec, ms);
+       return buf;
+}
+
 int ClipList::rowCount(const QModelIndex &parent) const {
        if (parent.isValid()) return 0;
        return clips.size();
 }
 
+int PlayList::rowCount(const QModelIndex &parent) const {
+       if (parent.isValid()) return 0;
+       return clips.size();
+}
+
 int ClipList::columnCount(const QModelIndex &parent) const {
        if (parent.isValid()) return 0;
-       return Column::NUM_COLUMNS;
+       return int(Column::NUM_COLUMNS);
+}
+
+int PlayList::columnCount(const QModelIndex &parent) const {
+       if (parent.isValid()) return 0;
+       return int(Column::NUM_COLUMNS);
 }
 
 QVariant ClipList::data(const QModelIndex &parent, int role) const {
        if (!parent.isValid())
                return QVariant();
-       if (role != Qt::DisplayRole)
+       const int row = parent.row(), column = parent.column();
+       if (size_t(row) >= clips.size())
+               return QVariant();
+
+       if (role == Qt::TextAlignmentRole) {
+               switch (Column(column)) {
+               case Column::IN:
+               case Column::OUT:
+               case Column::DURATION:
+                       return Qt::AlignRight + Qt::AlignVCenter;
+               default:
+                       return Qt::AlignLeft + Qt::AlignVCenter;
+               }
+       }
+
+       if (role != Qt::DisplayRole && role != Qt::EditRole)
                return QVariant();
 
+       switch (Column(column)) {
+       case Column::IN:
+               return QString::fromStdString(pts_to_string(clips[row].pts_in));
+       case Column::OUT:
+               if (clips[row].pts_out >= 0) {
+                       return QString::fromStdString(pts_to_string(clips[row].pts_out));
+               } else {
+                       return QVariant();
+               }
+       case Column::DURATION:
+               if (clips[row].pts_out >= 0) {
+                       return QString::fromStdString(duration_to_string(clips[row].pts_out - clips[row].pts_in));
+               } else {
+                       return QVariant();
+               }
+       case Column::CAMERA_1:
+       case Column::CAMERA_2:
+       case Column::CAMERA_3:
+       case Column::CAMERA_4: {
+               unsigned stream_idx = column - int(Column::CAMERA_1);
+               return QString::fromStdString(clips[row].descriptions[stream_idx]);
+       }
+       default:
+               return "";
+       }
+}
+
+QVariant PlayList::data(const QModelIndex &parent, int role) const {
+       if (!parent.isValid())
+               return QVariant();
        const int row = parent.row(), column = parent.column();
        if (size_t(row) >= clips.size())
                return QVariant();
 
-       switch (column) {
-       case 0:
-               return qlonglong(clips[row].pts_in);
-       case 1:
+       if (role == Qt::TextAlignmentRole) {
+               switch (Column(column)) {
+               case Column::PLAYING:
+                       return Qt::AlignCenter;
+               case Column::IN:
+               case Column::OUT:
+               case Column::DURATION:
+                       return Qt::AlignRight + Qt::AlignVCenter;
+               case Column::CAMERA:
+                       return Qt::AlignCenter;
+               default:
+                       return Qt::AlignLeft + Qt::AlignVCenter;
+               }
+       }
+
+       if (role != Qt::DisplayRole && role != Qt::EditRole)
+               return QVariant();
+
+       switch (Column(column)) {
+       case Column::PLAYING:
+               return (row == currently_playing_index) ? "→" : "";
+       case Column::IN:
+               return QString::fromStdString(pts_to_string(clips[row].pts_in));
+       case Column::OUT:
                if (clips[row].pts_out >= 0) {
-                       return qlonglong(clips[row].pts_out);
+                       return QString::fromStdString(pts_to_string(clips[row].pts_out));
                } else {
                        return QVariant();
                }
-       case 2:
+       case Column::DURATION:
                if (clips[row].pts_out >= 0) {
-                       return qlonglong(clips[row].pts_out - clips[row].pts_in);
+                       return QString::fromStdString(duration_to_string(clips[row].pts_out - clips[row].pts_in));
                } else {
                        return QVariant();
                }
+       case Column::CAMERA:
+               return qlonglong(clips[row].stream_idx + 1);
+       case Column::DESCRIPTION:
+               return QString::fromStdString(clips[row].descriptions[clips[row].stream_idx]);
        default:
-               return QVariant();
+               return "";
        }
 }
 
@@ -54,7 +165,7 @@ QVariant ClipList::headerData(int section, Qt::Orientation orientation, int role
        if (orientation != Qt::Horizontal)
                return QVariant();
 
-       switch (section) {
+       switch (Column(section)) {
        case Column::IN:
                return "In";
        case Column::OUT:
@@ -74,6 +185,122 @@ QVariant ClipList::headerData(int section, Qt::Orientation orientation, int role
        }
 }
 
+QVariant PlayList::headerData(int section, Qt::Orientation orientation, int role) const {
+       if (role != Qt::DisplayRole)
+               return QVariant();
+       if (orientation != Qt::Horizontal)
+               return QVariant();
+
+       switch (Column(section)) {
+       case Column::PLAYING:
+               return "";
+       case Column::IN:
+               return "In";
+       case Column::OUT:
+               return "Out";
+       case Column::DURATION:
+               return "Duration";
+       case Column::CAMERA:
+               return "Camera";
+       case Column::DESCRIPTION:
+               return "Description";
+       default:
+               return "";
+       }
+}
+
+Qt::ItemFlags ClipList::flags(const QModelIndex &index) const
+{
+       if (!index.isValid())
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+       const int row = index.row(), column = index.column();
+       if (size_t(row) >= clips.size())
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+
+       switch (Column(column)) {
+       case Column::CAMERA_1:
+       case Column::CAMERA_2:
+       case Column::CAMERA_3:
+       case Column::CAMERA_4:
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsDragEnabled;
+       default:
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+       }
+}
+
+Qt::ItemFlags PlayList::flags(const QModelIndex &index) const
+{
+       if (!index.isValid())
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+       const int row = index.row(), column = index.column();
+       if (size_t(row) >= clips.size())
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+
+       switch (Column(column)) {
+       case Column::DESCRIPTION:
+       case Column::CAMERA:
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
+       default:
+               return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+       }
+}
+
+bool ClipList::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+       if (!index.isValid() || role != Qt::EditRole) {
+               return false;
+       }
+
+       const int row = index.row(), column = index.column();
+       if (size_t(row) >= clips.size())
+               return false;
+
+       switch (Column(column)) {
+       case Column::CAMERA_1:
+       case Column::CAMERA_2:
+       case Column::CAMERA_3:
+       case Column::CAMERA_4: {
+               unsigned stream_idx = column - int(Column::CAMERA_1);
+               clips[row].descriptions[stream_idx] = value.toString().toStdString();
+               emit_data_changed(row);
+               return true;
+       }
+       default:
+               return false;
+       }
+}
+
+bool PlayList::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+       if (!index.isValid() || role != Qt::EditRole) {
+               return false;
+       }
+
+       const int row = index.row(), column = index.column();
+       if (size_t(row) >= clips.size())
+               return false;
+
+       switch (Column(column)) {
+       case Column::DESCRIPTION:
+               clips[row].descriptions[clips[row].stream_idx] = value.toString().toStdString();
+               emit_data_changed(row);
+               return true;
+       case Column::CAMERA: {
+               bool ok;
+               int camera_idx = value.toInt(&ok);
+               if (!ok || camera_idx < 1 || camera_idx > NUM_CAMERAS) {
+                       return false;
+               }
+               clips[row].stream_idx = camera_idx - 1;
+               emit_data_changed(row);
+               return true;
+       }
+       default:
+               return false;
+       }
+}
+
 void ClipList::add_clip(const Clip &clip)
 {
        beginInsertRows(QModelIndex(), clips.size(), clips.size());
@@ -81,7 +308,61 @@ void ClipList::add_clip(const Clip &clip)
        endInsertRows();
 }
 
+void PlayList::add_clip(const Clip &clip)
+{
+       beginInsertRows(QModelIndex(), clips.size(), clips.size());
+       clips.push_back(clip);
+       endInsertRows();
+}
+
+void PlayList::duplicate_clips(size_t first, size_t last)
+{
+       beginInsertRows(QModelIndex(), first, last);
+       clips.insert(clips.begin() + first, clips.begin() + first, clips.begin() + last + 1);
+       endInsertRows();
+}
+
+void PlayList::erase_clips(size_t first, size_t last)
+{
+       beginRemoveRows(QModelIndex(), first, last);
+       clips.erase(clips.begin() + first, clips.begin() + last + 1);
+       endRemoveRows();
+}
+
+void PlayList::move_clips(size_t first, size_t last, int delta)
+{
+       if (delta == -1) {
+               beginMoveRows(QModelIndex(), first, last, QModelIndex(), first - 1);
+               rotate(clips.begin() + first - 1, clips.begin() + first, clips.begin() + last + 1);
+       } else {
+               beginMoveRows(QModelIndex(), first, last, QModelIndex(), first + (last-first+1) + 1);
+               first = clips.size() - first - 1;
+               last = clips.size() - last - 1;
+               rotate(clips.rbegin() + last - 1, clips.rbegin() + last, clips.rbegin() + first + 1);
+       }
+       endMoveRows();
+}
+
 void ClipList::emit_data_changed(size_t row)
 {
-       emit dataChanged(index(row, 0), index(row, 6));
+       emit dataChanged(index(row, 0), index(row, int(Column::NUM_COLUMNS)));
+}
+
+void PlayList::emit_data_changed(size_t row)
+{
+       emit dataChanged(index(row, 0), index(row, int(Column::NUM_COLUMNS)));
+}
+
+void PlayList::set_currently_playing(int index)
+{
+       int old_index = currently_playing_index;
+       if (index != old_index) {
+               currently_playing_index = index;
+               if (old_index != -1) {
+                       emit_data_changed(old_index);
+               }
+               if (index != -1) {
+                       emit_data_changed(index);
+               }
+       }
 }