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