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);
  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);
  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 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(j)->text());
 104             if (frame == currentPos) {
 105                 keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
 106                 found = true;
 107                 break;
 108             } else if (currentPos > frame) {
 109                 break;
 110             }
 111         }
 112         if (!found) {
 113             keyframe_list->insertRow(j);
 114             keyframe_list->setVerticalHeaderItem(j, new QTableWidgetItem(m_timecode.getTimecodeFromFrames(frame)));
 115             keyframe_list->setItem(j, columnId, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
 116             keyframe_list->resizeRowToContents(j);
 117         }
 118     }
 119     keyframe_list->resizeColumnsToContents();
 120     keyframe_list->blockSignals(false);
 121     slotAdjustKeyframeInfo(false);
 122 }
 123
 124 void KeyframeEdit::setupParam()
 125 {
 126     keyframe_list->blockSignals(true);
 127     keyframe_list->clear();
 128     int col = keyframe_list->columnCount();
 129     QDomNode na = m_params.at(0).firstChildElement("name");
 130     QString paramName = i18n(na.toElement().text().toUtf8().data());
 131     kDebug() << " INSERT COL: " << col << ", " << paramName;
 132     keyframe_list->insertColumn(col);
 133     keyframe_list->setHorizontalHeaderItem(col, new QTableWidgetItem(paramName));
 134     m_slidersLayout = new QGridLayout;
 135     m_slidersLayout->addWidget(new QLabel(paramName), 0, 0);
 136     QSlider *sl = new QSlider(Qt::Horizontal, this);
 137     sl->setRange(m_params.at(0).attribute("min").toInt(), m_params.at(0).attribute("max").toInt());
 138     connect(sl, SIGNAL(valueChanged(int)), this, SLOT(slotAdjustKeyframeValue(int)));
 139     m_slidersLayout->addWidget(sl, 0, 1);
 140     param_sliders->setLayout(m_slidersLayout);
 141     keyframe_list->setSelectionBehavior(QAbstractItemView::SelectRows);
 142     keyframe_list->setSelectionMode(QAbstractItemView::SingleSelection);
 143
 144     QStringList frames = m_params.at(0).attribute("keyframes").split(";", QString::SkipEmptyParts);
 145     for (int i = 0; i < frames.count(); i++) {
 146         keyframe_list->insertRow(i);
 147         int currentpos = frames.at(i).section(':', 0, 0).toInt();
 148         QString framePos = m_timecode.getTimecodeFromFrames(currentpos);
 149         keyframe_list->setVerticalHeaderItem(i, new QTableWidgetItem(framePos));
 150         keyframe_list->setItem(i, col, new QTableWidgetItem(frames.at(i).section(':', 1, 1)));
 151         if ((m_active_keyframe > -1) && (m_active_keyframe == currentpos)) {
 152             keyframe_list->setCurrentCell(i, 0);
 153             keyframe_list->selectRow(i);
 154         }
 155         //item->setFlags(Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled);
 156     }
 157     /*QTreeWidgetItem *first = keyframe_list->topLevelItem(0);
 158     if (first) keyframe_list->setCurrentItem(first);*/
 159     keyframe_list->blockSignals(false);
 160     //keyframe_list->setCurrentCell(0, 0);
 161     //slotAdjustKeyframeInfo();
 162     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 163 }
 164
 165 void KeyframeEdit::slotDeleteKeyframe()
 166 {
 167     if (keyframe_list->rowCount() < 2) return;
 168     int col = keyframe_list->currentColumn();
 169     int row = keyframe_list->currentRow();
 170     keyframe_list->removeRow(keyframe_list->currentRow());
 171     row = qMin(row, keyframe_list->rowCount() - 1);
 172     keyframe_list->setCurrentCell(row, col);
 173     keyframe_list->selectRow(row);
 174     generateAllParams();
 175     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 176 }
 177
 178 void KeyframeEdit::slotAddKeyframe()
 179 {
 180     keyframe_list->blockSignals(true);
 181     QTableWidgetItem *item = keyframe_list->currentItem();
 182     int row = keyframe_list->currentRow();
 183     int col = keyframe_list->currentColumn();
 184     int newrow = row;
 185     int pos1 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row)->text());
 186     int result;
 187     kDebug() << "// ADD KF: " << row << ", MAX: " << keyframe_list->rowCount() << ", POS: " << pos1;
 188     if (row < (keyframe_list->rowCount() - 1)) {
 189         int pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row + 1)->text());
 190         result = pos1 + (pos2 - pos1) / 2;
 191         newrow++;
 192     } else if (row == 0) {
 193         if (pos1 == m_min) {
 194             result = m_max - 1;
 195             newrow++;
 196         } else {
 197             result = m_min;
 198         }
 199     } else {
 200         int pos2 = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(row - 1)->text());
 201         result = pos2 + (pos1 - pos2) / 2;
 202     }
 203
 204     keyframe_list->insertRow(newrow);
 205     keyframe_list->setVerticalHeaderItem(newrow, new QTableWidgetItem(m_timecode.getTimecodeFromFrames(result)));
 206     for (int i = 0; i < keyframe_list->columnCount(); i++) {
 207         keyframe_list->setItem(newrow, i, new QTableWidgetItem(keyframe_list->item(item->row(), i)->text()));
 208     }
 209     keyframe_list->resizeRowsToContents();
 210     //keyframe_list->resizeRowToContents(newrow);
 211     slotAdjustKeyframeInfo();
 212     keyframe_list->blockSignals(false);
 213     generateAllParams();
 214     button_delete->setEnabled(keyframe_list->rowCount() > 1);
 215     keyframe_list->setCurrentCell(newrow, col);
 216     keyframe_list->selectRow(newrow);
 217     //slotGenerateParams(newrow, 0);
 218 }
 219
 220 void KeyframeEdit::slotGenerateParams(int row, int column)
 221 {
 222     if (column == -1) {
 223         // position of keyframe changed
 224         QTableWidgetItem *item = keyframe_list->item(row, 0);
 225         if (item == NULL) return;
 226         QString val = keyframe_list->verticalHeaderItem(row)->text();
 227         int pos = m_timecode.getFrameCount(val);
 228         if (pos <= m_min) {
 229             pos = m_min;
 230             val = m_timecode.getTimecodeFromFrames(pos);
 231         }
 232         if (pos > m_max) {
 233             pos = m_max;
 234             val = m_timecode.getTimecodeFromFrames(pos);
 235         }
 236         if (val != keyframe_list->verticalHeaderItem(row)->text()) keyframe_list->verticalHeaderItem(row)->setText(val);
 237
 238         for (int col = 0; col < keyframe_list->horizontalHeader()->count(); col++) {
 239             item = keyframe_list->item(row, col);
 240             int v = item->text().toInt();
 241             if (v >= m_params.at(col).attribute("max").toInt()) item->setText(m_params.at(col).attribute("max"));
 242             if (v <= m_params.at(col).attribute("min").toInt()) item->setText(m_params.at(col).attribute("min"));
 243             QString keyframes;
 244             for (int i = 0; i < keyframe_list->rowCount(); i++) {
 245                 if (keyframe_list->item(i, col)) keyframes.append(QString::number(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text())) + ':' + keyframe_list->item(i, col)->text() + ';');
 246             }
 247             m_params[col].setAttribute("keyframes", keyframes);
 248         }
 249
 250         emit parameterChanged();
 251         return;
 252
 253     }
 254     QTableWidgetItem *item = keyframe_list->item(row, column);
 255     if (item == NULL) return;
 256     QString val = keyframe_list->verticalHeaderItem(row)->text();
 257     int pos = m_timecode.getFrameCount(val);
 258     if (pos <= m_min) {
 259         pos = m_min;
 260         val = m_timecode.getTimecodeFromFrames(pos);
 261     }
 262     if (pos > m_max) {
 263         pos = m_max;
 264         val = m_timecode.getTimecodeFromFrames(pos);
 265     }
 266     /*QList<QTreeWidgetItem *> duplicates = keyframe_list->findItems(val, Qt::MatchExactly, 0);
 267     duplicates.removeAll(item);
 268     if (!duplicates.isEmpty()) {
 269         // Trying to insert a keyframe at existing value, revert it
 270         val = m_timecode.getTimecodeFromFrames(m_previousPos);
 271     }*/
 272     if (val != keyframe_list->verticalHeaderItem(row)->text()) keyframe_list->verticalHeaderItem(row)->setText(val);
 273
 274     int v = item->text().toInt();
 275     if (v >= m_params.at(column).attribute("max").toInt()) item->setText(m_params.at(column).attribute("max"));
 276     if (v <= m_params.at(column).attribute("min").toInt()) item->setText(m_params.at(column).attribute("min"));
 277     slotAdjustKeyframeInfo(false);
 278
 279     QString keyframes;
 280     for (int i = 0; i < keyframe_list->rowCount(); i++) {
 281         if (keyframe_list->item(i, column)) keyframes.append(QString::number(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text())) + ':' + keyframe_list->item(i, column)->text() + ';');
 282     }
 283     m_params[column].setAttribute("keyframes", keyframes);
 284     emit parameterChanged();
 285 }
 286
 287 void KeyframeEdit::generateAllParams()
 288 {
 289     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 290         QString keyframes;
 291         for (int i = 0; i < keyframe_list->rowCount(); i++) {
 292             if (keyframe_list->item(i, col)) keyframes.append(QString::number(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(i)->text())) + ':' + keyframe_list->item(i, col)->text() + ';');
 293         }
 294         m_params[col].setAttribute("keyframes", keyframes);
 295     }
 296     emit parameterChanged();
 297 }
 298
 299 void KeyframeEdit::slotAdjustKeyframeInfo(bool seek)
 300 {
 301     QTableWidgetItem *item = keyframe_list->currentItem();
 302     if (!item) return;
 303     int min = m_min;
 304     int max = m_max;
 305     QTableWidgetItem *above = keyframe_list->item(item->row() - 1, item->column());
 306     QTableWidgetItem *below = keyframe_list->item(item->row() + 1, item->column());
 307     if (above) min = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(above->row())->text()) + 1;
 308     if (below) max = m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(below->row())->text()) - 1;
 309     keyframe_pos->blockSignals(true);
 310     keyframe_pos->setRange(min, max);
 311     keyframe_pos->setValue(m_timecode.getFrameCount(keyframe_list->verticalHeaderItem(item->row())->text()));
 312     keyframe_pos->blockSignals(false);
 313     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 314         QSlider *sl = static_cast <QSlider*>(m_slidersLayout->itemAtPosition(col, 1)->widget());
 315         if (!sl) continue;
 316         sl->blockSignals(true);
 317         sl->setValue(keyframe_list->item(item->row(), col)->text().toInt());
 318         sl->blockSignals(false);
 319     }
 320     if (KdenliveSettings::keyframeseek() && seek) emit seekToPos(keyframe_pos->value());
 321 }
 322
 323 void KeyframeEdit::slotAdjustKeyframePos(int value)
 324 {
 325     QTableWidgetItem *item = keyframe_list->currentItem();
 326     keyframe_list->verticalHeaderItem(item->row())->setText(m_timecode.getTimecodeFromFrames(value));
 327     slotGenerateParams(item->row(), -1);
 328     if (KdenliveSettings::keyframeseek()) emit seekToPos(value);
 329 }
 330
 331 void KeyframeEdit::slotAdjustKeyframeValue(int /*value*/)
 332 {
 333     QTableWidgetItem *item = keyframe_list->currentItem();
 334     for (int col = 0; col < keyframe_list->columnCount(); col++) {
 335         QSlider *sl = static_cast <QSlider*>(m_slidersLayout->itemAtPosition(col, 1)->widget());
 336         if (!sl) continue;
 337         int val = sl->value();
 338         QTableWidgetItem *nitem = keyframe_list->item(item->row(), col);
 339         if (nitem->text().toInt() != val) {
 340             nitem->setText(QString::number(val));
 341         }
 342     }
 343     //keyframe_list->item(item->row() - 1, item->column());
 344
 345 }
 346
 347 void KeyframeEdit::slotSetSeeking(int state)
 348 {
 349     KdenliveSettings::setKeyframeseek(state == Qt::Checked);
 350 }
 351
 352
 353 /*void KeyframeEdit::slotSaveCurrentParam(QTreeWidgetItem *item, int column)
 354 {
 355     if (item && column == 0) m_previousPos = m_timecode.getFrameCount(item->text(0));
 356 }*/
 357