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 <KGlobalSettings>
  23
  24 #include <QHeaderView>
  25
  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) delete wid;
  78     }
  79     //delete m_delegate;
  80 }
  81
  82 void KeyframeEdit::addParameter(QDomElement e)
  83 {
  84     keyframe_list->blockSignals(true);
  85     m_params.append(e.cloneNode().toElement());
  86     QDomNode na = e.firstChildElement("name");
  87     QString paramName = i18n(na.toElement().text().toUtf8().data());
  88     int columnId = keyframe_list->columnCount();
  89     keyframe_list->insertColumn(columnId);
  90     keyframe_list->setHorizontalHeaderItem(columnId, new QTableWidgetItem(paramName));
  91     m_slidersLayout->addWidget(new QLabel(paramName), columnId, 0);
  92     QSlider *sl = new QSlider(Qt::Horizontal, this);
  93     sl->setRange(m_params.at(columnId).attribute("min").toInt(), m_params.at(columnId).attribute("max").toInt());
  94     connect(sl, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
  95     m_slidersLayout->addWidget(sl, columnId, 1);
  96
  97     QStringList frames = e.attribute("keyframes").split(";", QString::SkipEmptyParts);
  98     for (int i = 0; i < frames.count(); i++) {
  99         int frame = frames.at(i).section(':', 0, 0).toInt();
 100         bool found = false;
 101         int j;
 102         for (j = 0; j < keyframe_list->rowCount(); j++) {
 103             int currentPos;
 104             if (KdenliveSettings::frametimecode()) currentPos = keyframe_list->verticalHeaderItem(j)->text().toInt();
 105             else currentPos = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(j)->text());
 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             QString currentPos;
 117             if (KdenliveSettings::frametimecode()) currentPos = QString::number(frame);
 118             else currentPos = m_timecode.getTimecodeFromFrames(frame);
 119             keyframe_list->setVerticalHeaderItem(j, new QTableWidgetItem(currentPos));
 120             keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
 121             keyframe_list->resizeRowToContents(j);
 122         }
 123     }
 124     keyframe_list->resizeColumnsToContents();
 125     keyframe_list->blockSignals(false);
 126     slotAdjustKeyframeInfo(false);
 127 }
 128
 129 void KeyframeEdit::setupParam()
 130 {
 131     keyframe_list->blockSignals(true);
 132     keyframe_list->clear();
 133     int col = keyframe_list->columnCount();
 134     QDomNode na = m_params.at(0).firstChildElement("name");
 135     QString paramName = i18n(na.toElement().text().toUtf8().data());
 136     kDebug() << " INSERT COL: " << col << ", " << paramName;
 137     keyframe_list->insertColumn(col);
 138     keyframe_list->setHorizontalHeaderItem(col, new QTableWidgetItem(paramName));
 139     m_slidersLayout = new QGridLayout;
 140     m_slidersLayout->addWidget(new QLabel(paramName), 0, 0);
 141     QSlider *sl = new QSlider(Qt::Horizontal, this);
 142     sl->setRange(m_params.at(0).attribute("min").toInt(), m_params.at(0).attribute("max").toInt());
 143     connect(sl, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
 144     m_slidersLayout->addWidget(sl, 0, 1);
 145     param_sliders->setLayout(m_slidersLayout);
 146     keyframe_list->setSelectionBehavior(QAbstractItemView::SelectRows);
 147     keyframe_list->setSelectionMode(QAbstractItemView::SingleSelection);
 148
 149     QStringList frames = m_params.at(0).attribute("keyframes").split(";", QString::SkipEmptyParts);
 150     for (int i = 0; i < frames.count(); i++) {
 151         keyframe_list->insertRow(i);
 152         int currentpos = frames.at(i).section(':', 0, 0).toInt();
 153         QString framePos;
 154         if (KdenliveSettings::frametimecode()) framePos  = QString::number(currentpos);
 155         else framePos = m_timecode.getTimecodeFromFrames(currentpos);
 156         keyframe_list->setVerticalHeaderItem(i, new QTableWidgetItem(framePos));
 157         keyframe_list->setItem(i, col, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
 158         if ((m_active_keyframe > -1) && (m_active_keyframe == currentpos)) {
 159             keyframe_list->setCurrentCell(i, 0);
 160             keyframe_list->selectRow(i);
 161         }
 162         //item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
 163     }
 164     /*QTreeWidgetItem *first = keyframe_list->topLevelItem(0);
 165     if (first) keyframe_list->setCurrentItem(first);*/
 166     keyframe_list->blockSignals(false);
 167     //keyframe_list->setCurrentCell(0, 0);
 168     slotAdjustKeyframeInfo(false);
 169     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 170 }
 171
 172 void KeyframeEdit::slotDeleteKeyframe()
 173 {
 174     if (keyframe_list->rowCount() < 2) return;
 175     int col = keyframe_list->currentColumn();
 176     int row = keyframe_list->currentRow();
 177     keyframe_list->removeRow(keyframe_list->currentRow());
 178     row = qMin(row, keyframe_list->rowCount() - 1);
 179     keyframe_list->setCurrentCell(row, col);
 180     keyframe_list->selectRow(row);
 181     generateAllParams();
 182     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 183 }
 184
 185 void KeyframeEdit::slotAddKeyframe()
 186 {
 187     keyframe_list->blockSignals(true);
 188     QTableWidgetItem *item = keyframe_list->currentItem();
 189     int row = keyframe_list->currentRow();
 190     int col = keyframe_list->currentColumn();
 191     int newrow = row;
 192     int pos1;
 193     if (KdenliveSettings::frametimecode()) pos1 = keyframe_list->verticalHeaderItem(row)->text().toInt();
 194     else pos1 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row)->text());
 195
 196     int result;
 197     kDebug() << "// ADD KF: " << row << ", MAX: " << keyframe_list->rowCount() << ", POS: " << pos1;
 198     if (row < (keyframe_list->rowCount() - 1)) {
 199         int pos2;
 200         if (KdenliveSettings::frametimecode()) pos2 = keyframe_list->verticalHeaderItem(row + 1)->text().toInt();
 201         else pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row + 1)->text());
 202         result = pos1 + (pos2 - pos1) / 2;
 203         newrow++;
 204     } else if (row == 0) {
 205         if (pos1 == m_min) {
 206             result = m_max - 1;
 207             newrow++;
 208         } else {
 209             result = m_min;
 210         }
 211     } else {
 212         int pos2;
 213         if (KdenliveSettings::frametimecode()) pos2 = keyframe_list->verticalHeaderItem(row - 1)->text().toInt();
 214         else pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row - 1)->text());
 215         result = pos2 + (pos1 - pos2) / 2;
 216     }
 217
 218     keyframe_list->insertRow(newrow);
 219     QString currentPos;
 220     if (KdenliveSettings::frametimecode()) currentPos = QString::number(result);
 221     else currentPos = m_timecode.getTimecodeFromFrames(result);
 222     keyframe_list->setVerticalHeaderItem(newrow, new QTableWidgetItem(currentPos));
 223     for (int i = 0; i < keyframe_list->columnCount(); i++) {
 224         keyframe_list->setItem(newrow, i, new QTableWidgetItem(keyframe_list->item(item->row(), i)->text()));
 225     }
 226     keyframe_list->resizeRowsToContents();
 227     //keyframe_list->resizeRowToContents(newrow);
 228     slotAdjustKeyframeInfo();
 229     keyframe_list->blockSignals(false);
 230     generateAllParams();
 231     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 232     keyframe_list->setCurrentCell(newrow, col);
 233     keyframe_list->selectRow(newrow);
 234     //slotGenerateParams(newrow, 0);
 235 }
 236
 237 void KeyframeEdit::slotGenerateParams(int row, int column)
 238 {
 239     if (column == -1) {
 240         // position of keyframe changed
 241         QTableWidgetItem *item = keyframe_list->item(row, 0);
 242         if (item == NULL) return;
 243         QString val = keyframe_list->verticalHeaderItem(row)->text();
 244         int pos;
 245         if (KdenliveSettings::frametimecode()) pos = val.toInt();
 246         else pos = m_timecode.getFrameCount(val);
 247
 248         if (pos <= m_min) {
 249             pos = m_min;
 250             if (KdenliveSettings::frametimecode()) val = QString::number(pos);
 251             else val = m_timecode.getTimecodeFromFrames(pos);
 252         }
 253         if (pos > m_max) {
 254             pos = m_max;
 255             if (KdenliveSettings::frametimecode()) val = QString::number(pos);
 256             else val = m_timecode.getTimecodeFromFrames(pos);
 257         }
 258         if (val != keyframe_list->verticalHeaderItem(row)->text()) keyframe_list->verticalHeaderItem(row)->setText(val);
 259
 260         for (int col = 0; col < keyframe_list->horizontalHeader()->count(); col++) {
 261             item = keyframe_list->item(row, col);
 262             int v = item->text().toInt();
 263             if (v >= m_params.at(col).attribute("max").toInt()) item->setText(m_params.at(col).attribute("max"));
 264             if (v <= m_params.at(col).attribute("min").toInt()) item->setText(m_params.at(col).attribute("min"));
 265             QString keyframes;
 266             for (int i = 0; i < keyframe_list->rowCount(); i++) {
 267                 if (keyframe_list->item(i, col)) {
 268                     int pos3;
 269                     if (KdenliveSettings::frametimecode()) pos3 = keyframe_list->verticalHeaderItem(i)->text().toInt();
 270                     else pos3 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text());
 271                     keyframes.append(QString::number(pos3) + ':' + keyframe_list->item(i, col)->text() + ';');
 272                 }
 273             }
 274             m_params[col].setAttribute("keyframes", keyframes);
 275         }
 276
 277         emit parameterChanged();
 278         return;
 279
 280     }
 281     QTableWidgetItem *item = keyframe_list->item(row, column);
 282     if (item == NULL) return;
 283     QString val = keyframe_list->verticalHeaderItem(row)->text();
 284     ;
 285     int pos;
 286     if (KdenliveSettings::frametimecode()) pos = val.toInt();
 287     else pos = m_timecode.getFrameCount(val);
 288
 289     if (pos <= m_min) {
 290         pos = m_min;
 291         if (KdenliveSettings::frametimecode()) val = QString::number(pos);
 292         else val = m_timecode.getTimecodeFromFrames(pos);
 293     }
 294     if (pos > m_max) {
 295         pos = m_max;
 296         if (KdenliveSettings::frametimecode()) val = QString::number(pos);
 297         else val = m_timecode.getTimecodeFromFrames(pos);
 298     }
 299     /*QList<QTreeWidgetItem *> duplicates = keyframe_list->findItems(val, Qt::MatchExactly, 0);
 300     duplicates.removeAll(item);
 301     if (!duplicates.isEmpty()) {
 302         // Trying to insert a keyframe at existing value, revert it
 303         val = m_timecode.getTimecodeFromFrames(m_previousPos);
 304     }*/
 305     if (val != keyframe_list->verticalHeaderItem(row)->text()) keyframe_list->verticalHeaderItem(row)->setText(val);
 306
 307     int v = item->text().toInt();
 308     if (v >= m_params.at(column).attribute("max").toInt()) item->setText(m_params.at(column).attribute("max"));
 309     if (v <= m_params.at(column).attribute("min").toInt()) item->setText(m_params.at(column).attribute("min"));
 310     slotAdjustKeyframeInfo(false);
 311
 312     QString keyframes;
 313     for (int i = 0; i < keyframe_list->rowCount(); i++) {
 314         if (keyframe_list->item(i, column)) {
 315             int pos3;
 316             if (KdenliveSettings::frametimecode()) pos3 = keyframe_list->verticalHeaderItem(i)->text().toInt();
 317             else pos3 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text());
 318             keyframes.append(QString::number(pos3) + ':' + keyframe_list->item(i, column)->text() + ';');
 319         }
 320     }
 321     m_params[column].setAttribute("keyframes", keyframes);
 322     emit parameterChanged();
 323 }
 324
 325 void KeyframeEdit::generateAllParams()
 326 {
 327     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 328         QString keyframes;
 329         for (int i = 0; i < keyframe_list->rowCount(); i++) {
 330             if (keyframe_list->item(i, col)) {
 331                 int pos3;
 332                 if (KdenliveSettings::frametimecode()) pos3 = keyframe_list->verticalHeaderItem(i)->text().toInt();
 333                 else pos3 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text());
 334                 keyframes.append(QString::number(pos3) + ':' + keyframe_list->item(i, col)->text() + ';');
 335             }
 336         }
 337         m_params[col].setAttribute("keyframes", keyframes);
 338     }
 339     emit parameterChanged();
 340 }
 341
 342 const QString KeyframeEdit::getValue(const QString &name)
 343 {
 344     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 345         QDomNode na = m_params.at(col).firstChildElement("name");
 346         QString paramName = i18n(na.toElement().text().toUtf8().data());
 347         kDebug() << paramName << " == " << name;
 348         if (paramName == name) return m_params.at(col).attribute("keyframes");
 349     }
 350     return QString();
 351 }
 352
 353 void KeyframeEdit::slotAdjustKeyframeInfo(bool seek)
 354 {
 355     QTableWidgetItem *item = keyframe_list->currentItem();
 356     if (!item) return;
 357     int min = m_min;
 358     int max = m_max;
 359     QTableWidgetItem *above = keyframe_list->item(item->row() - 1, item->column());
 360     QTableWidgetItem *below = keyframe_list->item(item->row() + 1, item->column());
 361     if (above) {
 362         if (KdenliveSettings::frametimecode()) min = keyframe_list->verticalHeaderItem(above->row())->text().toInt();
 363         else min = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(above->row())->text());
 364         min++;
 365     }
 366     if (below) {
 367         if (KdenliveSettings::frametimecode()) max = keyframe_list->verticalHeaderItem(below->row())->text().toInt();
 368         else max = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(below->row())->text());
 369         max--;
 370     }
 371     keyframe_pos->blockSignals(true);
 372     keyframe_pos->setRange(min, max);
 373     int pos;
 374     if (KdenliveSettings::frametimecode()) pos = keyframe_list->verticalHeaderItem(item->row())->text().toInt();
 375     else pos = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(item->row())->text());
 376
 377     keyframe_pos->setValue(pos);
 378     keyframe_pos->blockSignals(false);
 379     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 380         QSlider *sl = static_cast <QSlider*>(m_slidersLayout->itemAtPosition(col, 1)->widget());
 381         if (!sl) continue;
 382         sl->blockSignals(true);
 383         sl->setValue(keyframe_list->item(item->row(), col)->text().toInt());
 384         sl->blockSignals(false);
 385     }
 386     if (KdenliveSettings::keyframeseek() && seek) emit seekToPos(keyframe_pos->value());
 387 }
 388
 389 void KeyframeEdit::slotAdjustKeyframePos(int value)
 390 {
 391     QTableWidgetItem *item = keyframe_list->currentItem();
 392     QString val;
 393     if (KdenliveSettings::frametimecode()) val = QString::number(value);
 394     else val = m_timecode.getTimecodeFromFrames(value);
 395     keyframe_list->verticalHeaderItem(item->row())->setText(val);
 396     slotGenerateParams(item->row(), -1);
 397     if (KdenliveSettings::keyframeseek()) emit seekToPos(value);
 398 }
 399
 400 void KeyframeEdit::slotAdjustKeyframeValue(int /*value*/)
 401 {
 402     QTableWidgetItem *item = keyframe_list->currentItem();
 403     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 404         QSlider *sl = static_cast <QSlider*>(m_slidersLayout->itemAtPosition(col, 1)->widget());
 405         if (!sl) continue;
 406         int val = sl->value();
 407         QTableWidgetItem *nitem = keyframe_list->item(item->row(), col);
 408         if (nitem->text().toInt() != val) {
 409             nitem->setText(QString::number(val));
 410         }
 411     }
 412     //keyframe_list->item(item->row() - 1, item->column());
 413
 414 }
 415
 416 void KeyframeEdit::slotSetSeeking(int state)
 417 {
 418     KdenliveSettings::setKeyframeseek(state == Qt::Checked);
 419 }
 420
 421
 422 /*void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
 423 {
 424     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0));
 425 }*/
 426