]> git.sesse.net Git - kdenlive/blob - src/keyframeedit.cpp
http://www.kdenlive.org/mantis/view.php?id=1771
[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 "doubleparameterwidget.h"
20 #include "kdenlivesettings.h"
21
22 #include <KDebug>
23 #include <KGlobalSettings>
24
25 #include <QHeaderView>
26
27 KeyframeEdit::KeyframeEdit(QDomElement e, int minFrame, int maxFrame, int minVal, int maxVal, Timecode tc, int active_keyframe, QWidget* parent) :
28         QWidget(parent),
29         m_min(minFrame),
30         m_max(maxFrame),
31         m_minVal(minVal),
32         m_maxVal(maxVal),
33         m_timecode(tc),
34         m_previousPos(0),
35         m_active_keyframe(active_keyframe)
36 {
37     kDebug() << " / / / /MODIFIED KFR: " << m_active_keyframe;
38     setupUi(this);
39     m_params.append(e.cloneNode().toElement());
40     keyframe_list->setFont(KGlobalSettings::generalFont());
41     keyframe_seek->setChecked(KdenliveSettings::keyframeseek());
42     connect(keyframe_seek, SIGNAL(stateChanged(int)), this, SLOT(slotSetSeeking(int)));
43
44     button_add->setIcon(KIcon("list-add"));
45     button_add->setToolTip(i18n("Add keyframe"));
46     button_delete->setIcon(KIcon("list-remove"));
47     button_delete->setToolTip(i18n("Delete keyframe"));
48     connect(keyframe_list, SIGNAL(itemSelectionChanged()), this, SLOT(slotAdjustKeyframeInfo()));
49     connect(keyframe_list, SIGNAL(cellChanged(int, int)), this, SLOT(slotGenerateParams(int, int)));
50     setupParam();
51
52     keyframe_list->resizeRowsToContents();
53     keyframe_list->resizeColumnsToContents();
54     //keyframe_list->horizontalHeader()->resizeSections(QHeaderView::ResizeToContents);
55     connect(button_delete, SIGNAL(clicked()), this, SLOT(slotDeleteKeyframe()));
56     connect(button_add, SIGNAL(clicked()), this, SLOT(slotAddKeyframe()));
57     //connect(keyframe_list, SIGNAL(itemDoubleClicked(QTreeWidgetItem *, int)), this, SLOT(slotSaveCurrentParam(QTreeWidgetItem *, int)));
58     connect(keyframe_pos, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframePos(int)));
59     //connect(keyframe_val, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
60     keyframe_pos->setPageStep(1);
61     if (!keyframe_list->currentItem()) {
62         keyframe_list->setCurrentCell(0, 0);
63         keyframe_list->selectRow(0);
64     }
65     /*m_delegate = new KeyItemDelegate(minVal, maxVal);
66     keyframe_list->setItemDelegate(m_delegate);*/
67 }
68
69 KeyframeEdit::~KeyframeEdit()
70 {
71     keyframe_list->blockSignals(true);
72     keyframe_list->clear();
73     QLayoutItem *child;
74     while ((child = m_slidersLayout->takeAt(0)) != 0) {
75         QWidget *wid = child->widget();
76         delete child;
77         if (wid)
78             delete wid;
79     }
80     //delete m_delegate;
81 }
82
83 void KeyframeEdit::addParameter(QDomElement e)
84 {
85     keyframe_list->blockSignals(true);
86     m_params.append(e.cloneNode().toElement());
87     QDomNode na = e.firstChildElement("name");
88     QString paramName = i18n(na.toElement().text().toUtf8().data());
89     int columnId = keyframe_list->columnCount();
90     keyframe_list->insertColumn(columnId);
91     keyframe_list->setHorizontalHeaderItem(columnId, new QTableWidgetItem(paramName));
92
93     DoubleParameterWidget *doubleparam = new DoubleParameterWidget(paramName, 0,
94                                                                    m_params.at(columnId).attribute("min").toInt(), m_params.at(columnId).attribute("max").toInt(),
95                                                                    m_params.at(columnId).attribute("default").toInt(), m_params.at(columnId).attribute("suffix"), this);
96     connect(doubleparam, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
97     m_slidersLayout->addWidget(doubleparam, columnId, 0);
98
99     QStringList frames = e.attribute("keyframes").split(";", QString::SkipEmptyParts);
100     for (int i = 0; i < frames.count(); i++) {
101         int frame = frames.at(i).section(':', 0, 0).toInt();
102         bool found = false;
103         int j;
104         for (j = 0; j < keyframe_list->rowCount(); j++) {
105             int currentPos = getPos(j);
106             if (frame == currentPos) {
107                 keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
108                 found = true;
109                 break;
110             } else if (currentPos > frame) {
111                 break;
112             }
113         }
114         if (!found) {
115             keyframe_list->insertRow(j);
116             keyframe_list->setVerticalHeaderItem(j, new QTableWidgetItem(getPosString(frame)));
117             keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
118             keyframe_list->resizeRowToContents(j);
119         }
120     }
121     keyframe_list->resizeColumnsToContents();
122     keyframe_list->blockSignals(false);
123     slotAdjustKeyframeInfo(false);
124 }
125
126 void KeyframeEdit::setupParam()
127 {
128     keyframe_list->blockSignals(true);
129     keyframe_list->clear();
130     int col = keyframe_list->columnCount();
131     QDomNode na = m_params.at(0).firstChildElement("name");
132     QString paramName = i18n(na.toElement().text().toUtf8().data());
133     kDebug() << " INSERT COL: " << col << ", " << paramName;
134     keyframe_list->insertColumn(col);
135     keyframe_list->setHorizontalHeaderItem(col, new QTableWidgetItem(paramName));
136     m_slidersLayout = new QGridLayout(param_sliders);
137
138     DoubleParameterWidget *doubleparam = new DoubleParameterWidget(paramName, 0,
139                                                                    m_params.at(0).attribute("min").toInt(), m_params.at(0).attribute("max").toInt(),
140                                                                    m_params.at(0).attribute("default").toInt(), m_params.at(0).attribute("suffix"), this);
141     connect(doubleparam, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
142     m_slidersLayout->addWidget(doubleparam, 0, 0);
143
144     keyframe_list->setSelectionBehavior(QAbstractItemView::SelectRows);
145     keyframe_list->setSelectionMode(QAbstractItemView::SingleSelection);
146
147     QStringList frames = m_params.at(0).attribute("keyframes").split(";", QString::SkipEmptyParts);
148     for (int i = 0; i < frames.count(); i++) {
149         keyframe_list->insertRow(i);
150         int currentpos = frames.at(i).section(':', 0, 0).toInt();
151         keyframe_list->setVerticalHeaderItem(i, new QTableWidgetItem(getPosString(currentpos)));
152         keyframe_list->setItem(i, col, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
153         if ((m_active_keyframe > -1) && (m_active_keyframe == currentpos)) {
154             keyframe_list->setCurrentCell(i, 0);
155             keyframe_list->selectRow(i);
156         }
157         //item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
158     }
159     /*QTreeWidgetItem *first = keyframe_list->topLevelItem(0);
160     if (first) keyframe_list->setCurrentItem(first);*/
161     keyframe_list->blockSignals(false);
162     //keyframe_list->setCurrentCell(0, 0);
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     button_delete->setEnabled(keyframe_list->rowCount() > 1);
179 }
180
181 void KeyframeEdit::slotAddKeyframe()
182 {
183     keyframe_list->blockSignals(true);
184     QTableWidgetItem *item = keyframe_list->currentItem();
185     int row = keyframe_list->currentRow();
186     int col = keyframe_list->currentColumn();
187     int newrow = row;
188     int pos1 = getPos(row);
189
190     int result;
191     kDebug() << "// ADD KF: " << row << ", MAX: " << keyframe_list->rowCount() << ", POS: " << pos1;
192     if (row < (keyframe_list->rowCount() - 1)) {
193         result = pos1 + (getPos(row + 1) - pos1) / 2;
194         newrow++;
195     } else if (row == 0) {
196         if (pos1 == m_min) {
197             result = m_max - 1;
198             newrow++;
199         } else {
200             result = m_min;
201         }
202     } else {
203         int pos2 = getPos(row - 1);
204         result = pos2 + (pos1 - pos2) / 2;
205     }
206
207     keyframe_list->insertRow(newrow);
208     keyframe_list->setVerticalHeaderItem(newrow, new QTableWidgetItem(getPosString(result)));
209     for (int i = 0; i < keyframe_list->columnCount(); i++)
210         keyframe_list->setItem(newrow, i, new QTableWidgetItem(keyframe_list->item(item->row(), i)->text()));
211
212     keyframe_list->resizeRowsToContents();
213     //keyframe_list->resizeRowToContents(newrow);
214     slotAdjustKeyframeInfo();
215     keyframe_list->blockSignals(false);
216     generateAllParams();
217     button_delete->setEnabled(keyframe_list->rowCount() > 1);
218     keyframe_list->setCurrentCell(newrow, col);
219     keyframe_list->selectRow(newrow);
220     //slotGenerateParams(newrow, 0);
221 }
222
223 void KeyframeEdit::slotGenerateParams(int row, int column)
224 {
225     if (column == -1) {
226         // position of keyframe changed
227         QTableWidgetItem *item = keyframe_list->item(row, 0);
228         if (item == NULL)
229             return;
230
231         int pos = getPos(row);
232         if (pos <= m_min)
233             pos = m_min;
234         if (pos > m_max)
235             pos = m_max;
236         QString val = getPosString(pos);
237         if (val != keyframe_list->verticalHeaderItem(row)->text())
238             keyframe_list->verticalHeaderItem(row)->setText(val);
239
240         for (int col = 0; col < keyframe_list->horizontalHeader()->count(); col++) {
241             item = keyframe_list->item(row, col);
242             int v = item->text().toInt();
243             if (v >= m_params.at(col).attribute("max").toInt())
244                 item->setText(m_params.at(col).attribute("max"));
245             if (v <= m_params.at(col).attribute("min").toInt())
246                 item->setText(m_params.at(col).attribute("min"));
247             QString keyframes;
248             for (int i = 0; i < keyframe_list->rowCount(); i++) {
249                 if (keyframe_list->item(i, col))
250                     keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, col)->text() + ';');
251             }
252             m_params[col].setAttribute("keyframes", keyframes);
253         }
254
255         emit parameterChanged();
256         return;
257
258     }
259     QTableWidgetItem *item = keyframe_list->item(row, column);
260     if (item == NULL)
261         return;
262
263     int pos = getPos(row);
264     if (pos <= m_min)
265         pos = m_min;
266     if (pos > m_max)
267         pos = m_max;
268     /*QList<QTreeWidgetItem *> duplicates = keyframe_list->findItems(val, Qt::MatchExactly, 0);
269     duplicates.removeAll(item);
270     if (!duplicates.isEmpty()) {
271         // Trying to insert a keyframe at existing value, revert it
272         val = m_timecode.getTimecodeFromFrames(m_previousPos);
273     }*/
274     QString val = getPosString(pos);
275     if (val != keyframe_list->verticalHeaderItem(row)->text())
276         keyframe_list->verticalHeaderItem(row)->setText(val);
277
278     int v = item->text().toInt();
279     if (v >= m_params.at(column).attribute("max").toInt())
280         item->setText(m_params.at(column).attribute("max"));
281     if (v <= m_params.at(column).attribute("min").toInt())
282         item->setText(m_params.at(column).attribute("min"));
283     slotAdjustKeyframeInfo(false);
284
285     QString keyframes;
286     for (int i = 0; i < keyframe_list->rowCount(); i++) {
287         if (keyframe_list->item(i, column))
288             keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, column)->text() + ';');
289     }
290     m_params[column].setAttribute("keyframes", keyframes);
291     emit parameterChanged();
292 }
293
294 void KeyframeEdit::generateAllParams()
295 {
296     for (int col = 0; col < keyframe_list->columnCount(); col++) {
297         QString keyframes;
298         for (int i = 0; i < keyframe_list->rowCount(); i++) {
299             if (keyframe_list->item(i, col))
300                 keyframes.append(QString::number(getPos(i)) + ':' + keyframe_list->item(i, col)->text() + ';');
301         }
302         m_params[col].setAttribute("keyframes", keyframes);
303     }
304     emit parameterChanged();
305 }
306
307 const QString KeyframeEdit::getValue(const QString &name)
308 {
309     for (int col = 0; col < keyframe_list->columnCount(); col++) {
310         QDomNode na = m_params.at(col).firstChildElement("name");
311         QString paramName = i18n(na.toElement().text().toUtf8().data());
312         kDebug() << paramName << " == " << name;
313         if (paramName == name)
314             return m_params.at(col).attribute("keyframes");
315     }
316     return QString();
317 }
318
319 void KeyframeEdit::slotAdjustKeyframeInfo(bool seek)
320 {
321     QTableWidgetItem *item = keyframe_list->currentItem();
322     if (!item)
323         return;
324     int min = m_min;
325     int max = m_max;
326     QTableWidgetItem *above = keyframe_list->item(item->row() - 1, item->column());
327     QTableWidgetItem *below = keyframe_list->item(item->row() + 1, item->column());
328     if (above)
329         min = getPos(above->row()) + 1;
330     if (below)
331         max = getPos(below->row()) - 1;
332
333     keyframe_pos->blockSignals(true);
334     keyframe_pos->setRange(min, max);
335     keyframe_pos->setValue(getPos(item->row()));
336     keyframe_pos->blockSignals(false);
337     for (int col = 0; col < keyframe_list->columnCount(); col++) {
338         DoubleParameterWidget *doubleparam = static_cast <DoubleParameterWidget*>(m_slidersLayout->itemAtPosition(col, 0)->widget());
339         if (!doubleparam)
340             continue;
341         doubleparam->blockSignals(true);
342         if (keyframe_list->item(item->row(), col)) {
343             doubleparam->setValue(keyframe_list->item(item->row(), col)->text().toInt());
344         } else {
345             kDebug() << "Null pointer exception caught: http://www.kdenlive.org/mantis/view.php?id=1771";
346         }
347         doubleparam->blockSignals(false);
348     }
349     if (KdenliveSettings::keyframeseek() && seek)
350         emit seekToPos(keyframe_pos->value() - m_min);
351 }
352
353 void KeyframeEdit::slotAdjustKeyframePos(int value)
354 {
355     QTableWidgetItem *item = keyframe_list->currentItem();
356     keyframe_list->verticalHeaderItem(item->row())->setText(getPosString(value));
357     slotGenerateParams(item->row(), -1);
358     if (KdenliveSettings::keyframeseek())
359         emit seekToPos(value - m_min);
360 }
361
362 void KeyframeEdit::slotAdjustKeyframeValue(int /*value*/)
363 {
364     QTableWidgetItem *item = keyframe_list->currentItem();
365     for (int col = 0; col < keyframe_list->columnCount(); col++) {
366         DoubleParameterWidget *doubleparam = static_cast <DoubleParameterWidget*>(m_slidersLayout->itemAtPosition(col, 0)->widget());
367         if (!doubleparam)
368             continue;
369         int val = doubleparam->getValue();
370         QTableWidgetItem *nitem = keyframe_list->item(item->row(), col);
371         if (nitem->text().toInt() != val)
372             nitem->setText(QString::number(val));
373     }
374     //keyframe_list->item(item->row() - 1, item->column());
375
376 }
377
378 int KeyframeEdit::getPos(int row)
379 {
380     if (KdenliveSettings::frametimecode())
381         return keyframe_list->verticalHeaderItem(row)->text().toInt();
382     else
383         return m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row)->text());
384 }
385
386 QString KeyframeEdit::getPosString(int pos)
387 {
388     if (KdenliveSettings::frametimecode())
389         return QString::number(pos);
390     else
391         return m_timecode.getTimecodeFromFrames(pos);
392 }
393
394 void KeyframeEdit::slotSetSeeking(int state)
395 {
396     KdenliveSettings::setKeyframeseek(state == Qt::Checked);
397 }
398
399 void KeyframeEdit::updateTimecodeFormat()
400 {
401     for (int row = 0; row < keyframe_list->rowCount(); ++row) {
402         QString pos = keyframe_list->verticalHeaderItem(row)->text();
403         if (KdenliveSettings::frametimecode())
404             keyframe_list->verticalHeaderItem(row)->setText(QString::number(m_timecode.getFrameCount(pos)));
405         else
406             keyframe_list->verticalHeaderItem(row)->setText(m_timecode.getTimecodeFromFrames(pos.toInt()));
407     }
408 }
409
410 /*void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
411 {
412     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0));
413 }*/
414