git.sesse.net Git - kdenlive/blob - src/keyframeedit.cpp

   1 /***************************************************************************
   2                           geomeytrval.cpp  -  description
   3                              -------------------
   4     begin                : 03 Aug 2008
   5     copyright            : (C) 2008 by Marco Gittler
   6     email                : g.marco@freenet.de
   7  ***************************************************************************/
   8
   9 /***************************************************************************
  10  *                                                                         *
  11  *   This program is free software; you can redistribute it and/or modify  *
  12  *   it under the terms of the GNU General Public License as published by  *
  13  *   the Free Software Foundation; either version 2 of the License, or     *
  14  *   (at your option) any later version.                                   *
  15  *                                                                         *
  16  ***************************************************************************/
  17
  18 #include "keyframeedit.h"
  19 #include "kdenlivesettings.h"
  20
  21 #include <KDebug>
  22 #include <KGlobalSettings>
  23
  24 #include <QHeaderView>
  25
  26
  27 KeyframeEdit::KeyframeEdit(QDomElement e, int maxFrame, int minVal, int maxVal, Timecode tc, QWidget* parent) :
  28         QWidget(parent),
  29         m_param(e),
  30         m_max(maxFrame),
  31         m_minVal(minVal),
  32         m_maxVal(maxVal),
  33         m_timecode(tc),
  34         m_previousPos(0)
  35 {
  36     m_ui.setupUi(this);
  37     m_ui.keyframe_list->setFont(KGlobalSettings::generalFont());
  38     m_ui.keyframe_list->setHeaderLabels(QStringList() << i18n("Position") << i18n("Value"));
  39     //setResizeMode(1, QHeaderView::Interactive);
  40     m_ui.button_add->setIcon(KIcon("document-new"));
  41     m_ui.button_delete->setIcon(KIcon("edit-delete"));
  42     connect(m_ui.keyframe_list, SIGNAL(itemSelectionChanged()/*itemClicked(QTreeWidgetItem *, int)*/), this, SLOT(slotAdjustKeyframeInfo()));
  43     setupParam();
  44     m_ui.keyframe_list->header()->resizeSections(QHeaderView::ResizeToContents);
  45     connect(m_ui.button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteKeyframe()));
  46     connect(m_ui.button_add, SIGNAL(clicked()), this, SLOT(slotAddKeyframe()));
  47     connect(m_ui.keyframe_list, SIGNAL(itemChanged(QTreeWidgetItem *, int)), this, SLOT(slotGenerateParams(QTreeWidgetItem *, int)));
  48     connect(m_ui.keyframe_list, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotSaveCurrentParam(QTreeWidgetItem *, int)));
  49     connect(m_ui.keyframe_pos, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
  50     m_ui.keyframe_pos->setPageStep(1);
  51     m_ui.keyframe_list->setItemDelegate(new KeyItemDelegate(minVal, maxVal));
  52 }
  53
  54 void KeyframeEdit::setupParam(QDomElement e)
  55 {
  56     if (!e.isNull()) m_param = e;
  57     m_ui.keyframe_list->clear();
  58     QStringList frames = m_param.attribute("keyframes").split(";", QString::SkipEmptyParts);
  59     for (int i = 0; i < frames.count(); i++) {
  60         QString framePos = m_timecode.getTimecodeFromFrames(frames.at(i).section(':', 0, 0).toInt());
  61         QTreeWidgetItem *item = new QTreeWidgetItem(QStringList() << framePos << frames.at(i).section(':', 1, 1));
  62         item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
  63         m_ui.keyframe_list->addTopLevelItem(item);
  64     }
  65     QTreeWidgetItem *first = m_ui.keyframe_list->topLevelItem(0);
  66     if (first) m_ui.keyframe_list->setCurrentItem(first);
  67     slotAdjustKeyframeInfo();
  68     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
  69 }
  70
  71 void KeyframeEdit::slotDeleteKeyframe()
  72 {
  73     if (m_ui.keyframe_list->topLevelItemCount() < 3) return;
  74     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
  75     if (item) {
  76         delete item;
  77         slotGenerateParams();
  78     }
  79     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
  80 }
  81
  82 void KeyframeEdit::slotAddKeyframe()
  83 {
  84     m_ui.keyframe_list->blockSignals(true);
  85     int pos2;
  86     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
  87     if (item == NULL) return;
  88     int ix = m_ui.keyframe_list->indexOfTopLevelItem(item);
  89     int pos1 = m_timecode.getFrameCount(item->text(0), m_timecode.fps());
  90     QTreeWidgetItem *below = m_ui.keyframe_list->topLevelItem(ix + 1);
  91     if (below == NULL) below = m_ui.keyframe_list->topLevelItem(ix - 1);
  92     if (below == NULL) {
  93         if (pos1 == 0) pos2 = m_max;
  94         else pos2 = 0;
  95     } else {
  96         pos2 = m_timecode.getFrameCount(below->text(0), m_timecode.fps());
  97     }
  98
  99     int result = (pos1 + pos2) / 2;
 100     if (result > pos1) ix++;
 101     QTreeWidgetItem *newItem = new QTreeWidgetItem(QStringList() << m_timecode.getTimecodeFromFrames(result) << item->text(1));
 102     newItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
 103     m_ui.keyframe_list->insertTopLevelItem(ix, newItem);
 104     m_ui.keyframe_list->setCurrentItem(newItem);
 105     slotAdjustKeyframeInfo();
 106     m_ui.keyframe_list->blockSignals(false);
 107     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
 108     slotGenerateParams();
 109 }
 110
 111 void KeyframeEdit::slotGenerateParams(QTreeWidgetItem *item, int column)
 112 {
 113     if (item) {
 114         if (column == 0) {
 115             QString val = item->text(0);
 116             int pos = m_timecode.getFrameCount(val, m_timecode.fps());
 117             if (pos <= 0) {
 118                 pos = 0;
 119                 val = m_timecode.getTimecodeFromFrames(pos);
 120             }
 121             if (pos > m_max) {
 122                 pos = m_max;
 123                 val = m_timecode.getTimecodeFromFrames(pos);
 124             }
 125             QList<QTreeWidgetItem *> duplicates = m_ui.keyframe_list->findItems(val, Qt::MatchExactly, 0);
 126             duplicates.removeAll(item);
 127             if (!duplicates.isEmpty()) {
 128                 // Trying to insert a keyframe at existing value, revert it
 129                 val = m_timecode.getTimecodeFromFrames(m_previousPos);
 130             }
 131             if (val != item->text(0)) item->setText(0, val);
 132         }
 133         if (column == 1) {
 134             if (item->text(1).toInt() >= m_param.attribute("max").toInt()) item->setText(1, m_param.attribute("max"));
 135             if (item->text(1).toInt() <= m_param.attribute("min").toInt()) item->setText(1, m_param.attribute("min"));
 136         }
 137     }
 138     QString keyframes;
 139     for (int i = 0; i < m_ui.keyframe_list->topLevelItemCount(); i++) {
 140         QTreeWidgetItem *item = m_ui.keyframe_list->topLevelItem(i);
 141         keyframes.append(QString::number(m_timecode.getFrameCount(item->text(0), m_timecode.fps())) + ':' + item->text(1) + ';');
 142     }
 143     m_param.setAttribute("keyframes", keyframes);
 144     emit parameterChanged();
 145 }
 146
 147 void KeyframeEdit::slotAdjustKeyframeInfo()
 148 {
 149     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
 150     if (!item) return;
 151     int min = 0;
 152     int max = m_max;
 153     QTreeWidgetItem *above = m_ui.keyframe_list->itemAbove(item);
 154     QTreeWidgetItem *below = m_ui.keyframe_list->itemBelow(item);
 155     if (above) min = m_timecode.getFrameCount(above->text(0), m_timecode.fps()) + 1;
 156     if (below) max = m_timecode.getFrameCount(below->text(0), m_timecode.fps()) - 1;
 157     m_ui.keyframe_pos->blockSignals(true);
 158     m_ui.keyframe_pos->setRange(min, max);
 159     m_ui.keyframe_pos->setValue(m_timecode.getFrameCount(item->text(0), m_timecode.fps()));
 160     m_ui.keyframe_pos->blockSignals(false);
 161 }
 162
 163 void KeyframeEdit::slotAdjustKeyframeValue(int value)
 164 {
 165     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
 166     item->setText(0, m_timecode.getTimecodeFromFrames(value));
 167 }
 168
 169 void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
 170 {
 171     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0), m_timecode.fps());
 172 }
 173