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