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 <QHeaderView>
  23
  24
  25 KeyframeEdit::KeyframeEdit(QDomElement e, int max, Timecode tc, QWidget* parent) :
  26         QWidget(parent),
  27         m_param(e),
  28         m_max(max),
  29         m_timecode(tc),
  30         m_previousPos(0)
  31 {
  32     m_ui.setupUi(this);
  33     m_ui.keyframe_list->setHeaderLabels(QStringList() << i18n("Position") << i18n("Value"));
  34     //setResizeMode(1, QHeaderView::Interactive);
  35     m_ui.button_add->setIcon(KIcon("document-new"));
  36     m_ui.button_delete->setIcon(KIcon("edit-delete"));
  37     connect(m_ui.keyframe_list, SIGNAL(itemSelectionChanged()/*itemClicked(QTreeWidgetItem *, int)*/), this, SLOT(slotAdjustKeyframeInfo()));
  38     setupParam();
  39     m_ui.keyframe_list->header()->resizeSections(QHeaderView::ResizeToContents);
  40     connect(m_ui.button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteKeyframe()));
  41     connect(m_ui.button_add, SIGNAL(clicked()), this, SLOT(slotAddKeyframe()));
  42     connect(m_ui.keyframe_list, SIGNAL(itemChanged(QTreeWidgetItem *, int)), this, SLOT(slotGenerateParams(QTreeWidgetItem *, int)));
  43     connect(m_ui.keyframe_list, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotSaveCurrentParam(QTreeWidgetItem *, int)));
  44     connect(m_ui.keyframe_pos, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
  45     m_ui.keyframe_pos->setPageStep(1);
  46 }
  47
  48 void KeyframeEdit::setupParam(QDomElement e)
  49 {
  50     if (!e.isNull()) m_param = e;
  51     m_ui.keyframe_list->clear();
  52     QStringList frames = m_param.attribute("keyframes").split(";", QString::SkipEmptyParts);
  53     for (int i = 0; i < frames.count(); i++) {
  54         QString framePos = m_timecode.getTimecodeFromFrames(frames.at(i).section(':', 0, 0).toInt());
  55         QTreeWidgetItem *item = new QTreeWidgetItem(QStringList() << framePos << frames.at(i).section(':', 1, 1));
  56         item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
  57         m_ui.keyframe_list->addTopLevelItem(item);
  58     }
  59     QTreeWidgetItem *first = m_ui.keyframe_list->topLevelItem(0);
  60     if (first) m_ui.keyframe_list->setCurrentItem(first);
  61     slotAdjustKeyframeInfo();
  62     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
  63 }
  64
  65 void KeyframeEdit::slotDeleteKeyframe()
  66 {
  67     if (m_ui.keyframe_list->topLevelItemCount() < 3) return;
  68     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
  69     if (item) {
  70         delete item;
  71         slotGenerateParams();
  72     }
  73     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
  74 }
  75
  76 void KeyframeEdit::slotAddKeyframe()
  77 {
  78     m_ui.keyframe_list->blockSignals(true);
  79     int pos2;
  80     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
  81     if (item == NULL) return;
  82     int ix = m_ui.keyframe_list->indexOfTopLevelItem(item);
  83     int pos1 = m_timecode.getFrameCount(item->text(0), m_timecode.fps());
  84     QTreeWidgetItem *below = m_ui.keyframe_list->topLevelItem(ix + 1);
  85     if (below == NULL) below = m_ui.keyframe_list->topLevelItem(ix - 1);
  86     if (below == NULL) {
  87         if (pos1 == 0) pos2 = m_max;
  88         else pos2 = 0;
  89     } else {
  90         pos2 = m_timecode.getFrameCount(below->text(0), m_timecode.fps());
  91     }
  92
  93     int result = (pos1 + pos2) / 2;
  94     if (result > pos1) ix++;
  95     QTreeWidgetItem *newItem = new QTreeWidgetItem(QStringList() << m_timecode.getTimecodeFromFrames(result) << item->text(1));
  96     newItem->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
  97     m_ui.keyframe_list->insertTopLevelItem(ix, newItem);
  98     m_ui.keyframe_list->setCurrentItem(newItem);
  99     slotAdjustKeyframeInfo();
 100     m_ui.keyframe_list->blockSignals(false);
 101     m_ui.button_delete->setEnabled(m_ui.keyframe_list->topLevelItemCount() > 2);
 102     slotGenerateParams();
 103 }
 104
 105 void KeyframeEdit::slotGenerateParams(QTreeWidgetItem *item, int column)
 106 {
 107     if (item) {
 108         if (column == 0) {
 109             QString val = item->text(0);
 110             int pos = m_timecode.getFrameCount(val, m_timecode.fps());
 111             if (pos <= 0) {
 112                 pos = 0;
 113                 val = m_timecode.getTimecodeFromFrames(pos);
 114             }
 115             if (pos > m_max) {
 116                 pos = m_max;
 117                 val = m_timecode.getTimecodeFromFrames(pos);
 118             }
 119             QList<QTreeWidgetItem *> duplicates = m_ui.keyframe_list->findItems(val, Qt::MatchExactly, 0);
 120             duplicates.removeAll(item);
 121             if (!duplicates.isEmpty()) {
 122                 // Trying to insert a keyframe at existing value, revert it
 123                 val = m_timecode.getTimecodeFromFrames(m_previousPos);
 124             }
 125             if (val != item->text(0)) item->setText(0, val);
 126         }
 127         if (column == 1) {
 128             if (item->text(1).toInt() >= m_param.attribute("max").toInt()) item->setText(1, m_param.attribute("max"));
 129             if (item->text(1).toInt() <= m_param.attribute("min").toInt()) item->setText(1, m_param.attribute("min"));
 130         }
 131     }
 132     QString keyframes;
 133     for (int i = 0; i < m_ui.keyframe_list->topLevelItemCount(); i++) {
 134         QTreeWidgetItem *item = m_ui.keyframe_list->topLevelItem(i);
 135         keyframes.append(QString::number(m_timecode.getFrameCount(item->text(0), m_timecode.fps())) + ':' + item->text(1) + ';');
 136     }
 137     m_param.setAttribute("keyframes", keyframes);
 138     emit parameterChanged();
 139 }
 140
 141 void KeyframeEdit::slotAdjustKeyframeInfo()
 142 {
 143     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
 144     if (!item) return;
 145     int min = 0;
 146     int max = m_max;
 147     QTreeWidgetItem *above = m_ui.keyframe_list->itemAbove(item);
 148     QTreeWidgetItem *below = m_ui.keyframe_list->itemBelow(item);
 149     if (above) min = m_timecode.getFrameCount(above->text(0), m_timecode.fps()) + 1;
 150     if (below) max = m_timecode.getFrameCount(below->text(0), m_timecode.fps()) - 1;
 151     m_ui.keyframe_pos->blockSignals(true);
 152     m_ui.keyframe_pos->setRange(min, max);
 153     m_ui.keyframe_pos->setValue(m_timecode.getFrameCount(item->text(0), m_timecode.fps()));
 154     m_ui.keyframe_pos->blockSignals(false);
 155 }
 156
 157 void KeyframeEdit::slotAdjustKeyframeValue(int value)
 158 {
 159     QTreeWidgetItem *item = m_ui.keyframe_list->currentItem();
 160     item->setText(0, m_timecode.getTimecodeFromFrames(value));
 161 }
 162
 163 void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
 164 {
 165     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0), m_timecode.fps());
 166 }
 167