]> git.sesse.net Git - kdenlive/blob - src/keyframeedit.cpp
Prevent possible segfault:
[kdenlive] / src / keyframeedit.cpp
1 /***************************************************************************
2                           keyframedit.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 "doubleparameterwidget.h"
20 #include "positionedit.h"
21 #include "kdenlivesettings.h"
22
23 #include <KDebug>
24 #include <KGlobalSettings>
25
26 #include <QHeaderView>
27 #include <QButtonGroup>
28 #include <QRadioButton>
29
30 KeyframeEdit::KeyframeEdit(QDomElement e, int minFrame, int maxFrame, Timecode tc, int activeKeyframe, QWidget* parent) :
31         QWidget(parent),
32         m_min(minFrame),
33         m_max(maxFrame),
34         m_timecode(tc)
35 {
36     setupUi(this);
37     if (m_max == -1) {
38         // special case: keyframe for tracks, do not allow keyframes
39         widgetTable->setHidden(true);
40     }
41     keyframe_list->setFont(KGlobalSettings::generalFont());
42     buttonSeek->setChecked(KdenliveSettings::keyframeseek());
43     connect(buttonSeek, SIGNAL(toggled(bool)), this, SLOT(slotSetSeeking(bool)));
44
45     buttonKeyframes->setIcon(KIcon("chronometer"));
46     button_add->setIcon(KIcon("list-add"));
47     button_add->setToolTip(i18n("Add keyframe"));
48     button_delete->setIcon(KIcon("list-remove"));
49     button_delete->setToolTip(i18n("Delete keyframe"));
50     buttonResetKeyframe->setIcon(KIcon("edit-undo"));
51     buttonSeek->setIcon(KIcon("insert-link"));
52     connect(keyframe_list, SIGNAL(itemSelectionChanged()), this, SLOT(slotAdjustKeyframeInfo()));
53     connect(keyframe_list, SIGNAL(cellChanged(int, int)), this, SLOT(slotGenerateParams(int, int)));
54
55     m_position = new PositionEdit(i18n("Position"), 0, 0, 1, tc, widgetTable);
56     ((QGridLayout*)widgetTable->layout())->addWidget(m_position, 3, 0, 1, -1);
57
58     m_showButtons = new QButtonGroup(this);
59     m_slidersLayout = new QGridLayout(param_sliders);
60     keyframe_list->setSelectionBehavior(QAbstractItemView::SelectRows);
61     keyframe_list->setSelectionMode(QAbstractItemView::SingleSelection);
62     addParameter(e, activeKeyframe);
63     keyframe_list->resizeRowsToContents();
64
65     //keyframe_list->horizontalHeader()->resizeSections(QHeaderView::ResizeToContents);
66     connect(button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteKeyframe()));
67     connect(button_add, SIGNAL(clicked()), this, SLOT(slotAddKeyframe()));
68     connect(buttonKeyframes, SIGNAL(clicked()), this, SLOT(slotKeyframeMode()));
69     connect(buttonResetKeyframe, SIGNAL(clicked()), this, SLOT(slotResetKeyframe()));
70     connect(m_position, SIGNAL(parameterChanged(int)), this, SLOT(slotAdjustKeyframePos(int)));
71     connect(m_showButtons, SIGNAL(buttonClicked(int)), this, SLOT(slotUpdateVisibleParameter(int)));
72
73     //connect(keyframe_list, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotSaveCurrentParam(QTreeWidgetItem *, int)));
74
75     if (!keyframe_list->currentItem()) {
76         keyframe_list->setCurrentCell(0, 0);
77         keyframe_list->selectRow(0);
78     }
79     // ensure the keyframe list shows at least 3 lines
80     keyframe_list->setMinimumHeight(QFontInfo(keyframe_list->font()).pixelSize() * 9);
81
82     // Do not show keyframe table if only one keyframe exists at the beginning
83     if (keyframe_list->rowCount() < 2 && getPos(0) == m_min && m_max != -1)
84         widgetTable->setHidden(true);
85     else
86         buttonKeyframes->setHidden(true);
87 }
88
89 KeyframeEdit::~KeyframeEdit()
90 {
91     keyframe_list->blockSignals(true);
92     keyframe_list->clear();
93     QLayoutItem *child;
94     while ((child = m_slidersLayout->takeAt(0)) != 0) {
95         QWidget *wid = child->widget();
96         delete child;
97         if (wid)
98             delete wid;
99     }
100 }
101
102 void KeyframeEdit::addParameter(QDomElement e, int activeKeyframe)
103 {
104     keyframe_list->blockSignals(true);
105     m_params.append(e.cloneNode().toElement());
106
107     QDomElement na = e.firstChildElement("name");
108     QString paramName = i18n(na.text().toUtf8().data());
109     QDomElement commentElem = e.firstChildElement("comment");
110     QString comment;
111     if (!commentElem.isNull())
112         comment = i18n(commentElem.text().toUtf8().data());
113
114     int columnId = keyframe_list->columnCount();
115     keyframe_list->insertColumn(columnId);
116     keyframe_list->setHorizontalHeaderItem(columnId, new QTableWidgetItem(paramName));
117
118     DoubleParameterWidget *doubleparam = new DoubleParameterWidget(paramName, 0,
119             m_params.at(columnId).attribute("min").toInt(), m_params.at(columnId).attribute("max").toInt(),
120             m_params.at(columnId).attribute("default").toInt(), comment, m_params.at(columnId).attribute("suffix"), this);
121     connect(doubleparam, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
122     connect(this, SIGNAL(showComments(bool)), doubleparam, SLOT(slotShowComment(bool)));
123     m_slidersLayout->addWidget(doubleparam, columnId, 0);
124
125     QRadioButton *radio = new QRadioButton(this);
126     radio->setToolTip(i18n("Show %1 in timeline").arg(paramName));
127     m_showButtons->addButton(radio, columnId);
128     if (e.attribute("intimeline") == "1")
129         radio->setChecked(true);
130
131     QVBoxLayout *radioLayout = new QVBoxLayout;
132     radioLayout->addWidget(radio, 0, Qt::AlignTop);
133     m_slidersLayout->addLayout(radioLayout, columnId, 1);
134
135     QStringList frames = e.attribute("keyframes").split(";", QString::SkipEmptyParts);
136     for (int i = 0; i < frames.count(); i++) {
137         int frame = frames.at(i).section(':', 0, 0).toInt();
138         bool found = false;
139         int j;
140         for (j = 0; j < keyframe_list->rowCount(); j++) {
141             int currentPos = getPos(j);
142             if (frame == currentPos) {
143                 keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
144                 found = true;
145                 break;
146             } else if (currentPos > frame) {
147                 break;
148             }
149         }
150         if (!found) {
151             keyframe_list->insertRow(j);
152             keyframe_list->setVerticalHeaderItem(j, new QTableWidgetItem(getPosString(frame)));
153             keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
154             keyframe_list->resizeRowToContents(j);
155         }
156         if ((activeKeyframe > -1) && (activeKeyframe == frame)) {
157             keyframe_list->setCurrentCell(i, columnId);
158             keyframe_list->selectRow(i);
159         }
160     }
161     keyframe_list->resizeColumnsToContents();
162     keyframe_list->blockSignals(false);
163     slotAdjustKeyframeInfo(false);
164     button_delete->setEnabled(keyframe_list->rowCount() > 1);
165 }
166
167 void KeyframeEdit::slotDeleteKeyframe()
168 {
169     if (keyframe_list->rowCount() < 2)
170         return;
171     int col = keyframe_list->currentColumn();
172     int row = keyframe_list->currentRow();
173     keyframe_list->removeRow(keyframe_list->currentRow());
174     row = qMin(row, keyframe_list->rowCount() - 1);
175     keyframe_list->setCurrentCell(row, col);
176     keyframe_list->selectRow(row);
177     generateAllParams();
178
179     bool disable = keyframe_list->rowCount() < 2;
180     button_delete->setEnabled(!disable);
181     disable &= getPos(0) == m_min;
182     widgetTable->setHidden(disable);
183     buttonKeyframes->setHidden(!disable);
184 }
185
186 void KeyframeEdit::slotAddKeyframe()
187 {
188     keyframe_list->blockSignals(true);
189     QTableWidgetItem *item = keyframe_list->currentItem();
190     int row = keyframe_list->currentRow();
191     int col = keyframe_list->currentColumn();
192     int newrow = row;
193     int pos1 = getPos(row);
194
195     int result;
196     if (row < (keyframe_list->rowCount() - 1)) {
197         result = pos1 + (getPos(row + 1) - pos1) / 2;
198         newrow++;
199     } else if (row == 0) {
200         if (pos1 == m_min) {
201             result = m_max - 1;
202             newrow++;
203         } else {
204             result = m_min;
205         }
206     } else {
207         int pos2 = getPos(row - 1);
208         result = pos2 + (pos1 - pos2) / 2;
209     }
210
211     keyframe_list->insertRow(newrow);
212     keyframe_list->setVerticalHeaderItem(newrow, new QTableWidgetItem(getPosString(result)));
213     for (int i = 0; i < keyframe_list->columnCount(); i++)
214         keyframe_list->setItem(newrow, i, new QTableWidgetItem(keyframe_list->item(item->row(), i)->text()));
215
216     keyframe_list->resizeRowsToContents();
217     slotAdjustKeyframeInfo();
218     keyframe_list->blockSignals(false);
219     generateAllParams();
220     button_delete->setEnabled(keyframe_list->rowCount() > 1);
221     keyframe_list->setCurrentCell(newrow, col);
222     keyframe_list->selectRow(newrow);
223     //slotGenerateParams(newrow, 0);
224 }
225
226 void KeyframeEdit::slotGenerateParams(int row, int column)
227 {
228     if (column == -1) {
229         // position of keyframe changed
230         QTableWidgetItem *item = keyframe_list->item(row, 0);
231         if (item == NULL)
232             return;
233
234         int pos = getPos(row);
235         if (pos <= m_min)
236             pos = m_min;
237         if (m_max != -1 && pos > m_max)
238             pos = m_max;
239         QString val = getPosString(pos);
240         if (val != keyframe_list->verticalHeaderItem(row)->text())
241             keyframe_list->verticalHeaderItem(row)->setText(val);
242
243         for (int col = 0; col < keyframe_list->horizontalHeader()->count(); col++) {
244             item = keyframe_list->item(row, col);
245             int v = item->text().toInt();
246             if (v >= m_params.at(col).attribute("max").toInt())
247                 item->setText(m_params.at(col).attribute("max"));
248             if (v <= m_params.at(col).attribute("min").toInt())
249                 item->setText(m_params.at(col).attribute("min"));
250             QString keyframes;
251             for (int i = 0; i < keyframe_list->rowCount(); i++) {
252                 if (keyframe_list->item(i, col))
253                     keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, col)->text() + ';');
254             }
255             m_params[col].setAttribute("keyframes", keyframes);
256         }
257
258         emit parameterChanged();
259         return;
260
261     }
262     QTableWidgetItem *item = keyframe_list->item(row, column);
263     if (item == NULL)
264         return;
265
266     int pos = getPos(row);
267     if (pos <= m_min)
268         pos = m_min;
269     if (m_max != -1 && pos > m_max)
270         pos = m_max;
271     /*QList<QTreeWidgetItem *> duplicates = keyframe_list->findItems(val, Qt::MatchExactly, 0);
272     duplicates.removeAll(item);
273     if (!duplicates.isEmpty()) {
274         // Trying to insert a keyframe at existing value, revert it
275         val = m_timecode.getTimecodeFromFrames(m_previousPos);
276     }*/
277     QString val = getPosString(pos);
278     if (val != keyframe_list->verticalHeaderItem(row)->text())
279         keyframe_list->verticalHeaderItem(row)->setText(val);
280
281     int v = item->text().toInt();
282     if (v >= m_params.at(column).attribute("max").toInt())
283         item->setText(m_params.at(column).attribute("max"));
284     if (v <= m_params.at(column).attribute("min").toInt())
285         item->setText(m_params.at(column).attribute("min"));
286     slotAdjustKeyframeInfo(false);
287
288     QString keyframes;
289     for (int i = 0; i < keyframe_list->rowCount(); i++) {
290         if (keyframe_list->item(i, column))
291             keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, column)->text() + ';');
292     }
293     m_params[column].setAttribute("keyframes", keyframes);
294     emit parameterChanged();
295 }
296
297 void KeyframeEdit::generateAllParams()
298 {
299     for (int col = 0; col < keyframe_list->columnCount(); col++) {
300         QString keyframes;
301         for (int i = 0; i < keyframe_list->rowCount(); i++) {
302             if (keyframe_list->item(i, col))
303                 keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, col)->text() + ';');
304         }
305         m_params[col].setAttribute("keyframes", keyframes);
306     }
307     emit parameterChanged();
308 }
309
310 const QString KeyframeEdit::getValue(const QString &name)
311 {
312     for (int col = 0; col < keyframe_list->columnCount(); col++) {
313         QDomNode na = m_params.at(col).firstChildElement("name");
314         QString paramName = i18n(na.toElement().text().toUtf8().data());
315         if (paramName == name)
316             return m_params.at(col).attribute("keyframes");
317     }
318     return QString();
319 }
320
321 void KeyframeEdit::slotAdjustKeyframeInfo(bool seek)
322 {
323     QTableWidgetItem *item = keyframe_list->currentItem();
324     if (!item)
325         return;
326     int min = m_min;
327     int max = m_max;
328     QTableWidgetItem *above = keyframe_list->item(item->row() - 1, item->column());
329     QTableWidgetItem *below = keyframe_list->item(item->row() + 1, item->column());
330     if (above)
331         min = getPos(above->row()) + 1;
332     if (below)
333         max = getPos(below->row()) - 1;
334
335     m_position->blockSignals(true);
336     m_position->setRange(min, max);
337     m_position->setPosition(getPos(item->row()));
338     m_position->blockSignals(false);
339
340     for (int col = 0; col < keyframe_list->columnCount(); col++) {
341         DoubleParameterWidget *doubleparam = static_cast <DoubleParameterWidget*>(m_slidersLayout->itemAtPosition(col, 0)->widget());
342         if (!doubleparam)
343             continue;
344         doubleparam->blockSignals(true);
345         if (keyframe_list->item(item->row(), col)) {
346             doubleparam->setValue(keyframe_list->item(item->row(), col)->text().toInt());
347         } else {
348             kDebug() << "Null pointer exception caught: http://www.kdenlive.org/mantis/view.php?id=1771";
349         }
350         doubleparam->blockSignals(false);
351     }
352     if (KdenliveSettings::keyframeseek() && seek)
353         emit seekToPos(m_position->getPosition() - m_min);
354 }
355
356 void KeyframeEdit::slotAdjustKeyframePos(int value)
357 {
358     QTableWidgetItem *item = keyframe_list->currentItem();
359     keyframe_list->verticalHeaderItem(item->row())->setText(getPosString(value));
360     slotGenerateParams(item->row(), -1);
361     if (KdenliveSettings::keyframeseek())
362         emit seekToPos(value - m_min);
363 }
364
365 void KeyframeEdit::slotAdjustKeyframeValue(int value)
366 {
367     Q_UNUSED(value);
368
369     QTableWidgetItem *item = keyframe_list->currentItem();
370     for (int col = 0; col < keyframe_list->columnCount(); col++) {
371         DoubleParameterWidget *doubleparam = static_cast <DoubleParameterWidget*>(m_slidersLayout->itemAtPosition(col, 0)->widget());
372         if (!doubleparam)
373             continue;
374         int val = doubleparam->getValue();
375         QTableWidgetItem *nitem = keyframe_list->item(item->row(), col);
376         if (nitem && nitem->text().toInt() != val)
377             nitem->setText(QString::number(val));
378     }
379     //keyframe_list->item(item->row() - 1, item->column());
380
381 }
382
383 int KeyframeEdit::getPos(int row)
384 {
385     if (KdenliveSettings::frametimecode())
386         return keyframe_list->verticalHeaderItem(row)->text().toInt();
387     else
388         return m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row)->text());
389 }
390
391 QString KeyframeEdit::getPosString(int pos)
392 {
393     if (KdenliveSettings::frametimecode())
394         return QString::number(pos);
395     else
396         return m_timecode.getTimecodeFromFrames(pos);
397 }
398
399 void KeyframeEdit::slotSetSeeking(bool seek)
400 {
401     KdenliveSettings::setKeyframeseek(seek);
402 }
403
404 void KeyframeEdit::updateTimecodeFormat()
405 {
406     for (int row = 0; row < keyframe_list->rowCount(); ++row) {
407         QString pos = keyframe_list->verticalHeaderItem(row)->text();
408         if (KdenliveSettings::frametimecode())
409             keyframe_list->verticalHeaderItem(row)->setText(QString::number(m_timecode.getFrameCount(pos)));
410         else
411             keyframe_list->verticalHeaderItem(row)->setText(m_timecode.getTimecodeFromFrames(pos.toInt()));
412     }
413
414     m_position->updateTimecodeFormat();
415 }
416
417 void KeyframeEdit::slotKeyframeMode()
418 {
419     widgetTable->setHidden(false);
420     buttonKeyframes->setHidden(true);
421     slotAddKeyframe();
422 }
423
424 void KeyframeEdit::slotResetKeyframe()
425 {
426     for (int col = 0; col < keyframe_list->columnCount(); ++col) {
427         DoubleParameterWidget *doubleparam = static_cast<DoubleParameterWidget*>(m_slidersLayout->itemAtPosition(col, 0)->widget());
428         if (doubleparam)
429             doubleparam->slotReset();
430     }
431 }
432
433 void KeyframeEdit::slotUpdateVisibleParameter(int id, bool update)
434 {
435     for (int i = 0; i < m_params.count(); ++i)
436         m_params[i].setAttribute("intimeline", (i == id ? "1" : "0"));
437     if (update) emit parameterChanged();
438 }
439
440 bool KeyframeEdit::isVisibleParam(const QString& name)
441 {
442     for (int col = 0; col < keyframe_list->columnCount(); ++col) {
443         QDomNode na = m_params.at(col).firstChildElement("name");
444         QString paramName = i18n(na.toElement().text().toUtf8().data());
445         if (paramName == name)
446             return m_params.at(col).attribute("intimeline") == "1";
447     }
448     return false;
449 }
450
451 void KeyframeEdit::checkVisibleParam()
452 {
453     if (m_params.count() == 0)
454         return;
455
456     foreach(QDomElement elem, m_params) {
457         if (elem.attribute("intimeline") == "1")
458             return;
459     }
460
461     slotUpdateVisibleParameter(0, false);
462     QRadioButton *radio = static_cast<QRadioButton*>(m_slidersLayout->itemAtPosition(0, 1)->widget());
463     if (radio)
464         radio->setChecked(true);
465 }
466
467 #include "keyframeedit.moc"