]> git.sesse.net Git - kdenlive/blob - src/keyframeedit.cpp
Updated 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 #include <KGlobalSettings>
23
24 #include <QHeaderView>
25
26
27 KeyframeEdit::KeyframeEdit(QDomElement e, int maxFrame, int minVal, int maxVal, Timecode tc, const QString paramName, QWidget* parent) :
28         QWidget(parent),
29         m_max(maxFrame),
30         m_minVal(minVal),
31         m_maxVal(maxVal),
32         m_timecode(tc),
33         m_previousPos(0)
34 {
35     setupUi(this);
36     m_params.append(e);
37     keyframe_list->setFont(KGlobalSettings::generalFont());
38     
39     //keyframe_list->setHorizontalHeaderLabels(QStringList() << (paramName.isEmpty() ? i18n("Value") : paramName));
40     //setResizeMode(1, QHeaderView::Interactive);
41     button_add->setIcon(KIcon("list-add"));
42     button_add->setToolTip(i18n("Add keyframe"));
43     button_delete->setIcon(KIcon("list-remove"));
44     button_delete->setToolTip(i18n("Delete keyframe"));
45     connect(keyframe_list, SIGNAL(itemSelectionChanged()/*itemClicked(QTreeWidgetItem *, int)*/), this, SLOT(slotAdjustKeyframeInfo()));
46     keyframe_val->setRange(m_minVal, m_maxVal);
47     setupParam();
48     //keyframe_list->sortItems(0);
49     keyframe_list->resizeRowsToContents();
50     keyframe_list->verticalHeader()->setResizeMode(QHeaderView::Fixed);
51
52     //keyframe_list->horizontalHeader()->resizeSections(QHeaderView::ResizeToContents);
53     connect(button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteKeyframe()));
54     connect(button_add, SIGNAL(clicked()), this, SLOT(slotAddKeyframe()));
55     connect(keyframe_list, SIGNAL(cellChanged(int, int)), this, SLOT(slotGenerateParams(int, int)));
56     //connect(keyframe_list, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotSaveCurrentParam(QTreeWidgetItem *, int)));
57     connect(keyframe_pos, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframePos(int)));
58     connect(keyframe_val, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
59     keyframe_pos->setPageStep(1);
60     /*m_delegate = new KeyItemDelegate(minVal, maxVal);
61     keyframe_list->setItemDelegate(m_delegate);*/
62 }
63
64 KeyframeEdit::~KeyframeEdit()
65 {
66     keyframe_list->blockSignals(true);
67     keyframe_list->clear();
68     //delete m_delegate;
69 }
70
71 void KeyframeEdit::addParameter(QDomElement e) {
72     keyframe_list->blockSignals(true);
73     QDomNode na = e.firstChildElement("name");
74     QString paramName = i18n(na.toElement().text().toUtf8().data());
75     int columnId = keyframe_list->columnCount();
76     keyframe_list->insertColumn(columnId);
77     keyframe_list->setHorizontalHeaderItem(columnId, new QTableWidgetItem(paramName));
78     m_params.append(e);
79     QStringList frames = e.attribute("keyframes").split(";", QString::SkipEmptyParts);
80     for (int i = 0; i < frames.count(); i++) {
81         int frame = frames.at(i).section(':', 0, 0).toInt();
82         bool found = false;
83         int j;
84         for (j = 0; j < keyframe_list->rowCount(); j++) {
85             int currentPos = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(j)->text());
86             if (frame == currentPos) {
87                 keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
88                 found = true;
89                 break;
90             }
91             else if (currentPos > frame) {
92                 break;
93             }
94         }
95         if (!found) {
96             //int newRow = keyframe_list->rowCount();
97             keyframe_list->insertRow(j);
98             keyframe_list->setVerticalHeaderItem(j, new QTableWidgetItem(m_timecode.getTimecodeFromFrames(frame)));
99             keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
100             keyframe_list->resizeRowToContents(j);
101         }
102     }
103     keyframe_list->blockSignals(false);
104 }
105
106 void KeyframeEdit::setupParam()
107 {
108     keyframe_list->blockSignals(true);
109     keyframe_list->clear();
110     int col = keyframe_list->columnCount();
111     QDomNode na = m_params.at(0).firstChildElement("name");
112     QString paramName = i18n(na.toElement().text().toUtf8().data());
113     kDebug()<<" INSERT COL: "<<col<<", "<<paramName;
114     keyframe_list->insertColumn(col);
115     keyframe_list->setHorizontalHeaderItem(col, new QTableWidgetItem(paramName));
116     QStringList frames = m_params.at(0).attribute("keyframes").split(";", QString::SkipEmptyParts);
117     for (int i = 0; i < frames.count(); i++) {
118         keyframe_list->insertRow(i);
119         QString framePos = m_timecode.getTimecodeFromFrames(frames.at(i).section(':', 0, 0).toInt());
120         keyframe_list->setVerticalHeaderItem(i, new QTableWidgetItem(framePos));
121         keyframe_list->setItem(i, col, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
122         //item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
123     }
124     /*QTreeWidgetItem *first = keyframe_list->topLevelItem(0);
125     if (first) keyframe_list->setCurrentItem(first);*/
126     keyframe_list->blockSignals(false);
127     keyframe_list->setCurrentCell(0, 0);
128     //slotAdjustKeyframeInfo();
129     button_delete->setEnabled(keyframe_list->rowCount() > 1);
130 }
131
132 void KeyframeEdit::slotDeleteKeyframe()
133 {
134     if (keyframe_list->rowCount() < 2) return;
135     int col = keyframe_list->currentColumn();
136     int row = keyframe_list->currentRow();
137     keyframe_list->removeRow(keyframe_list->currentRow());
138     row = qMin(row, keyframe_list->rowCount() - 1);
139     keyframe_list->setCurrentCell(row, col);
140     slotGenerateParams(row, col);
141     button_delete->setEnabled(keyframe_list->rowCount() > 1);
142 }
143
144 void KeyframeEdit::slotAddKeyframe()
145 {
146     keyframe_list->blockSignals(true);
147     int pos2;
148     QTableWidgetItem *item = keyframe_list->currentItem();
149     int row = keyframe_list->currentRow();
150     int col = keyframe_list->currentColumn();
151     int newrow = row;
152     int pos1 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row)->text());
153     kDebug()<<"// ADD KF: "<<row<<", MAX: "<<keyframe_list->rowCount()<<", POS: "<<pos1;
154     if (row < (keyframe_list->rowCount() - 1)) {
155         pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row + 1)->text());
156         newrow++;
157     }
158     else if (row == 0) {
159         if (pos1 == 0) {
160             pos2 = m_max * 2;
161             newrow++;
162         }
163         else {
164             pos2 = 0;
165             pos1 = 0;
166         }
167     }
168     else pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row - 1)->text());
169
170     int result = (pos1 + pos2) / 2;
171     
172     keyframe_list->insertRow(newrow);
173     keyframe_list->setVerticalHeaderItem(newrow, new QTableWidgetItem(m_timecode.getTimecodeFromFrames(result)));
174     keyframe_list->setItem(newrow, keyframe_list->currentColumn(), new QTableWidgetItem(item->text()));
175     keyframe_list->resizeRowToContents(newrow);
176     slotAdjustKeyframeInfo();
177     keyframe_list->blockSignals(false);
178     slotGenerateParams(newrow, keyframe_list->currentColumn());
179     button_delete->setEnabled(keyframe_list->rowCount() > 1);
180     keyframe_list->setCurrentCell(newrow, col);
181     //slotGenerateParams(newrow, 0);
182 }
183
184 void KeyframeEdit::slotGenerateParams(int row, int column)
185 {
186     QTableWidgetItem *item = keyframe_list->item(row, column);
187     if (item == NULL) return;
188     QString val = keyframe_list->verticalHeaderItem(row)->text();
189     int pos = m_timecode.getFrameCount(val);
190     if (pos <= 0) {
191         pos = 0;
192         val = m_timecode.getTimecodeFromFrames(pos);
193     }
194     if (pos > m_max) {
195         pos = m_max;
196         val = m_timecode.getTimecodeFromFrames(pos);
197     }
198             /*QList<QTreeWidgetItem *> duplicates = keyframe_list->findItems(val, Qt::MatchExactly, 0);
199             duplicates.removeAll(item);
200             if (!duplicates.isEmpty()) {
201                 // Trying to insert a keyframe at existing value, revert it
202                 val = m_timecode.getTimecodeFromFrames(m_previousPos);
203             }*/
204     if (val != keyframe_list->verticalHeaderItem(row)->text()) keyframe_list->verticalHeaderItem(row)->setText(val);
205
206     int v = item->text().toInt();
207     if (v >= m_params.at(column).attribute("max").toInt()) item->setText(m_params.at(column).attribute("max"));
208     if (v <= m_params.at(column).attribute("min").toInt()) item->setText(m_params.at(column).attribute("min"));
209     slotAdjustKeyframeInfo();
210
211     QString keyframes;
212     for (int i = 0; i < keyframe_list->rowCount(); i++) {
213         keyframes.append(QString::number(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text())) + ':' + keyframe_list->item(i, 0)->text() + ';');
214     }
215     m_params[column].setAttribute("keyframes", keyframes);
216     emit parameterChanged();
217 }
218
219 void KeyframeEdit::slotAdjustKeyframeInfo()
220 {
221     QTableWidgetItem *item = keyframe_list->currentItem();
222     if (!item) return;
223     int min = 0;
224     int max = m_max;
225     QTableWidgetItem *above = keyframe_list->item(item->row() - 1, item->column());
226     QTableWidgetItem *below = keyframe_list->item(item->row() + 1, item->column());
227     if (above) min = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(above->row())->text()) + 1;
228     if (below) max = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(below->row())->text()) - 1;
229     keyframe_pos->blockSignals(true);
230     keyframe_pos->setRange(min, max);
231     keyframe_pos->setValue(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(item->row())->text()));
232     keyframe_pos->blockSignals(false);
233     keyframe_val->blockSignals(true);
234     keyframe_val->setValue(item->text().toInt());
235     keyframe_val->blockSignals(false);
236 }
237
238 void KeyframeEdit::slotAdjustKeyframePos(int value)
239 {
240     QTableWidgetItem *item = keyframe_list->currentItem();
241     keyframe_list->verticalHeaderItem(item->row())->setText(m_timecode.getTimecodeFromFrames(value));
242     slotGenerateParams(item->row(), item->column());
243 }
244
245 void KeyframeEdit::slotAdjustKeyframeValue(int value)
246 {
247     QTableWidgetItem *item = keyframe_list->currentItem();
248     item->setText(QString::number(value));
249 }
250
251 /*void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
252 {
253     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0));
254 }*/
255