git.sesse.net Git - kdenlive/blob - src/trackview.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Jean-Baptiste Mardelle (jb@kdenlive.org)        *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA          *
  18  ***************************************************************************/
  19
  20
  21 #include <QScrollBar>
  22
  23 #include <KDebug>
  24 #include <KMessageBox>
  25
  26 #include "definitions.h"
  27 #include "headertrack.h"
  28 #include "trackview.h"
  29 #include "clipitem.h"
  30 #include "transition.h"
  31 #include "kdenlivesettings.h"
  32 #include "clipmanager.h"
  33 #include "customruler.h"
  34 #include "kdenlivedoc.h"
  35 #include "mainwindow.h"
  36 #include "customtrackview.h"
  37
  38
  39 TrackView::TrackView(KdenliveDoc *doc, QWidget *parent)
  40         : QWidget(parent), m_doc(doc), m_scale(1.0), m_projectTracks(0) {
  41
  42     view = new Ui::TimeLine_UI();
  43     view->setupUi(this);
  44
  45     m_scene = new CustomTrackScene(doc);
  46     m_trackview = new CustomTrackView(doc, m_scene, parent);
  47     m_trackview->scale(1, 1);
  48     m_trackview->setAlignment(Qt::AlignLeft | Qt::AlignTop);
  49     //m_scene->addRect(QRectF(0, 0, 100, 100), QPen(), QBrush(Qt::red));
  50
  51     m_ruler = new CustomRuler(doc->timecode(), m_trackview);
  52     connect(m_ruler, SIGNAL(zoneMoved(int, int)), this, SIGNAL(zoneMoved(int, int)));
  53     QHBoxLayout *layout = new QHBoxLayout;
  54     view->ruler_frame->setLayout(layout);
  55     int left_margin;
  56     int right_margin;
  57     layout->getContentsMargins(&left_margin, 0, &right_margin, 0);
  58     layout->setContentsMargins(left_margin, 0, right_margin, 0);
  59     layout->addWidget(m_ruler);
  60
  61     QHBoxLayout *tracksLayout = new QHBoxLayout;
  62     tracksLayout->setContentsMargins(0, 0, 0, 0);
  63     view->tracks_frame->setLayout(tracksLayout);
  64
  65     view->headers_area->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  66     view->headers_area->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  67
  68     m_headersLayout = new QVBoxLayout;
  69     m_headersLayout->setContentsMargins(0, 0, 0, 0);
  70     m_headersLayout->setSpacing(0);
  71     view->headers_container->setLayout(m_headersLayout);
  72
  73     connect(view->headers_area->verticalScrollBar(), SIGNAL(valueChanged(int)), m_trackview->verticalScrollBar(), SLOT(setValue(int)));
  74
  75     tracksLayout->addWidget(m_trackview);
  76
  77     connect(m_trackview->verticalScrollBar(), SIGNAL(valueChanged(int)), view->headers_area->verticalScrollBar(), SLOT(setValue(int)));
  78     connect(m_trackview, SIGNAL(trackHeightChanged()), this, SLOT(slotRebuildTrackHeaders()));
  79
  80     parseDocument(m_doc->toXml());
  81
  82     connect(m_trackview, SIGNAL(cursorMoved(int, int)), m_ruler, SLOT(slotCursorMoved(int, int)));
  83     connect(m_trackview->horizontalScrollBar(), SIGNAL(valueChanged(int)), m_ruler, SLOT(slotMoveRuler(int)));
  84     connect(m_trackview, SIGNAL(mousePosition(int)), this, SIGNAL(mousePosition(int)));
  85     connect(m_trackview, SIGNAL(transitionItemSelected(Transition*)), this, SLOT(slotTransitionItemSelected(Transition*)));
  86     slotChangeZoom(m_doc->zoom());
  87 }
  88
  89
  90 int TrackView::duration() const {
  91     return m_trackview->duration();
  92 }
  93
  94 int TrackView::tracksNumber() const {
  95     return m_projectTracks;
  96 }
  97
  98 int TrackView::inPoint() const {
  99     return m_ruler->inPoint();
 100 }
 101
 102 int TrackView::outPoint() const {
 103     return m_ruler->outPoint();
 104 }
 105
 106 void TrackView::slotSetZone(QPoint p) {
 107     m_ruler->setZone(p);
 108 }
 109
 110 void TrackView::slotTransitionItemSelected(Transition *t) {
 111     emit transitionItemSelected(t);
 112 }
 113
 114 void TrackView::setDuration(int dur) {
 115     m_trackview->setDuration(dur);
 116     m_ruler->setDuration(dur);
 117 }
 118
 119 void TrackView::parseDocument(QDomDocument doc) {
 120     int cursorPos = 0;
 121     m_documentErrors = QString();
 122     // kDebug() << "//// DOCUMENT: " << doc.toString();
 123     QDomNode props = doc.elementsByTagName("properties").item(0);
 124     if (!props.isNull()) {
 125         cursorPos = props.toElement().attribute("timeline_position").toInt();
 126     }
 127
 128     // parse project tracks
 129     QDomNodeList tracks = doc.elementsByTagName("track");
 130     QDomNodeList playlists = doc.elementsByTagName("playlist");
 131     int duration = 300;
 132     m_projectTracks = tracks.count();
 133     int trackduration = 0;
 134     QDomElement e;
 135     QDomElement p;
 136     bool videotrack;
 137
 138     int pos = m_projectTracks - 1;
 139
 140
 141     for (int i = 0; i < m_projectTracks; i++) {
 142         e = tracks.item(i).toElement();
 143         QString playlist_name = e.attribute("producer");
 144         if (playlist_name != "black_track" && playlist_name != "playlistmain") {
 145             // find playlist related to this track
 146             p = QDomElement();
 147             for (int j = 0; j < m_projectTracks; j++) {
 148                 p = playlists.item(j).toElement();
 149                 if (p.attribute("id") == playlist_name) break;
 150             }
 151             videotrack = (e.attribute("hide") != "video");
 152             trackduration = slotAddProjectTrack(pos, p, videotrack);
 153             pos--;
 154             //kDebug() << " PRO DUR: " << trackduration << ", TRACK DUR: " << duration;
 155             if (trackduration > duration) duration = trackduration;
 156         } else {
 157             // background black track
 158             for (int j = 0; j < m_projectTracks; j++) {
 159                 p = playlists.item(j).toElement();
 160                 if (p.attribute("id") == playlist_name) break;
 161             }
 162             int black_clips = p.childNodes().count();
 163             for (int i = 0; i < black_clips; i++)
 164                 m_doc->loadingProgressed();
 165             qApp->processEvents();
 166             pos--;
 167         }
 168     }
 169
 170     // parse transitions
 171     QDomNodeList transitions = doc.elementsByTagName("transition");
 172     int projectTransitions = transitions.count();
 173     //kDebug() << "//////////// TIMELINE FOUND: " << projectTransitions << " transitions";
 174     for (int i = 0; i < projectTransitions; i++) {
 175         e = transitions.item(i).toElement();
 176         QDomNodeList transitionparams = e.childNodes();
 177         bool transitionAdd = true;
 178         int a_track = 0;
 179         int b_track = 0;
 180         QString mlt_service;
 181         for (int k = 0; k < transitionparams.count(); k++) {
 182             p = transitionparams.item(k).toElement();
 183             if (!p.isNull()) {
 184                 QString paramName = p.attribute("name");
 185                 // do not add audio mixing transitions
 186                 if (paramName == "internal_added" && p.text() == "237") {
 187                     transitionAdd = false;
 188                     //kDebug() << "//  TRANSITRION " << i << " IS NOT VALID (INTERN ADDED)";
 189                     break;
 190                 } else if (paramName == "a_track") a_track = p.text().toInt();
 191                 else if (paramName == "b_track") b_track = m_projectTracks - 1 - p.text().toInt();
 192                 else if (paramName == "mlt_service") mlt_service = p.text();
 193             }
 194         }
 195         if (transitionAdd) {
 196             // Transition should be added to the scene
 197             ItemInfo transitionInfo;
 198             QDomElement base = MainWindow::transitions.getEffectByTag(mlt_service, QString());
 199
 200             for (int k = 0; k < transitionparams.count(); k++) {
 201                 p = transitionparams.item(k).toElement();
 202                 if (!p.isNull()) {
 203                     QString paramName = p.attribute("name");
 204                     QString paramValue = p.text();
 205
 206                     QDomNodeList params = base.elementsByTagName("parameter");
 207                     if (paramName != "a_track" && paramName != "b_track") for (int i = 0; i < params.count(); i++) {
 208                             QDomElement e = params.item(i).toElement();
 209                             if (!e.isNull() && e.attribute("tag") == paramName) {
 210                                 if (e.attribute("type") == "double") {
 211                                     QString factor = e.attribute("factor", "1");
 212                                     if (factor != "1") {
 213                                         double val = paramValue.toDouble() * factor.toDouble();
 214                                         paramValue = QString::number(val);
 215                                     }
 216                                 }
 217                                 e.setAttribute("value", paramValue);
 218                                 break;
 219                             }
 220                         }
 221                 }
 222             }
 223
 224             /*QDomDocument doc;
 225             doc.appendChild(doc.importNode(base, true));
 226             kDebug() << "///////  TRANSITION XML: "<< doc.toString();*/
 227
 228             transitionInfo.startPos = GenTime(e.attribute("in").toInt(), m_doc->fps());
 229             transitionInfo.endPos = GenTime(e.attribute("out").toInt(), m_doc->fps());
 230             transitionInfo.track = b_track;
 231             //kDebug() << "///////////////   +++++++++++  ADDING TRANSITION ON TRACK: " << b_track << ", TOTAL TRKA: " << m_projectTracks;
 232             Transition *tr = new Transition(transitionInfo, a_track, m_doc->fps(), base);
 233             m_scene->addItem(tr);
 234         }
 235     }
 236
 237     // Add guides
 238     QDomNodeList guides = doc.elementsByTagName("guide");
 239     for (int i = 0; i < guides.count(); i++) {
 240         e = guides.item(i).toElement();
 241         const QString comment = e.attribute("comment");
 242         const GenTime pos = GenTime(e.attribute("time").toDouble());
 243         m_trackview->addGuide(pos, comment);
 244     }
 245
 246     m_trackview->setDuration(duration);
 247     kDebug() << "///////////  TOTAL PROJECT DURATION: " << duration;
 248     slotRebuildTrackHeaders();
 249     if (!m_documentErrors.isNull()) KMessageBox::sorry(this, m_documentErrors);
 250     //m_trackview->setCursorPos(cursorPos);
 251     //m_scrollBox->setGeometry(0, 0, 300 * zoomFactor(), m_scrollArea->height());
 252 }
 253
 254 void TrackView::slotDeleteClip(const QString &clipId) {
 255     m_trackview->deleteClip(clipId);
 256 }
 257
 258 void TrackView::setCursorPos(int pos) {
 259     m_trackview->setCursorPos(pos);
 260 }
 261
 262 void TrackView::moveCursorPos(int pos) {
 263     m_trackview->setCursorPos(pos, false);
 264 }
 265
 266 void TrackView::slotChangeZoom(int factor) {
 267     m_doc->setZoom(factor);
 268     m_ruler->setPixelPerMark(factor);
 269     m_scale = (double) FRAME_SIZE / m_ruler->comboScale[factor]; // m_ruler->comboScale[m_currentZoom] /
 270     m_trackview->setScale(m_scale);
 271 }
 272
 273 int TrackView::fitZoom() const {
 274     int zoom = (int)((duration() + 20 / m_scale) * FRAME_SIZE / m_trackview->width());
 275     int i;
 276     for (i = 0; i < 13; i++)
 277         if (m_ruler->comboScale[i] > zoom) break;
 278     return i;
 279 }
 280
 281 const double TrackView::zoomFactor() const {
 282     return m_scale;
 283 }
 284
 285 const int TrackView::mapLocalToValue(int x) const {
 286     return (int)(x * zoomFactor());
 287 }
 288
 289 KdenliveDoc *TrackView::document() {
 290     return m_doc;
 291 }
 292
 293 void TrackView::refresh() {
 294     m_trackview->viewport()->update();
 295 }
 296
 297 void TrackView::slotRebuildTrackHeaders() {
 298     QList <TrackInfo> list = m_trackview->tracksList();
 299     QList<HeaderTrack *> widgets = this->findChildren<HeaderTrack *>();
 300     for (int i = 0; i < widgets.count(); i++)
 301         delete widgets.at(i);
 302     int max = list.count();
 303     for (int i = 0; i < max; i++) {
 304         HeaderTrack *header = new HeaderTrack(i, list.at(max - i - 1), this);
 305         connect(header, SIGNAL(switchTrackVideo(int)), m_trackview, SLOT(slotSwitchTrackVideo(int)));
 306         connect(header, SIGNAL(switchTrackAudio(int)), m_trackview, SLOT(slotSwitchTrackAudio(int)));
 307         /*
 308         //TODO: re-enable when add / remove track is implemented
 309         connect(header, SIGNAL(deleteTrack(int)), m_trackview, SLOT(slotDeleteTrack(int)));
 310         connect(header, SIGNAL(insertTrack(int)), m_trackview, SLOT(slotInsertTrack(int)));*/
 311         m_headersLayout->addWidget(header);
 312     }
 313     view->headers_container->adjustSize();
 314 }
 315
 316 int TrackView::slotAddProjectTrack(int ix, QDomElement xml, bool videotrack) {
 317     TrackInfo info;
 318
 319     if (videotrack) {
 320         info.type = VIDEOTRACK;
 321         info.isMute = false;
 322         info.isBlind = false;
 323     } else {
 324         info.type = AUDIOTRACK;
 325         info.isMute = false;
 326         info.isBlind = false;
 327     }
 328
 329     m_trackview->addTrack(info);
 330
 331     int trackTop = KdenliveSettings::trackheight() * ix;
 332     // parse track
 333     int position = 0;
 334     for (QDomNode n = xml.firstChild(); !n.isNull(); n = n.nextSibling()) {
 335         QDomElement elem = n.toElement();
 336         if (elem.tagName() == "blank") {
 337             position += elem.attribute("length").toInt();
 338         } else if (elem.tagName() == "entry") {
 339             m_doc->loadingProgressed();
 340             qApp->processEvents();
 341             // Found a clip
 342             int in = elem.attribute("in").toInt();
 343             QString idString = elem.attribute("producer");
 344             QString id = idString;
 345             bool hasSpeedAttribute = false;
 346             double speed;
 347             if (idString.startsWith("slowmotion")) {
 348                 hasSpeedAttribute = true;
 349                 id = idString.section(":", 1, 1);
 350                 speed = idString.section(":", 2, 2).toDouble();
 351             }
 352             DocClipBase *clip = m_doc->clipManager()->getClipById(id);
 353             if (clip != NULL) {
 354                 int out = elem.attribute("out").toInt();
 355
 356                 ItemInfo clipinfo;
 357                 clipinfo.startPos = GenTime(position, m_doc->fps());
 358                 clipinfo.endPos = clipinfo.startPos + GenTime(out - in + 1, m_doc->fps());
 359                 clipinfo.cropStart = GenTime(in, m_doc->fps());
 360                 clipinfo.track = ix;
 361                 //kDebug() << "// INSERTING CLIP: " << in << "x" << out << ", track: " << ix << ", ID: " << id << ", SCALE: " << m_scale << ", FPS: " << m_doc->fps();
 362                 ClipItem *item = new ClipItem(clip, clipinfo, m_doc->fps(), false);
 363                 if (hasSpeedAttribute) item->setSpeed(speed);
 364                 m_scene->addItem(item);
 365                 clip->addReference();
 366                 position += (out - in + 1);
 367
 368                 // parse clip effects
 369                 QDomNodeList effects = elem.childNodes();
 370                 for (int ix = 0; ix < effects.count(); ix++) {
 371                     QDomElement effect = effects.at(ix).toElement();
 372                     if (effect.tagName() == "filter") {
 373                         kDebug() << " * * * * * * * * * * ** CLIP EFF FND  * * * * * * * * * * *";
 374                         // add effect to clip
 375                         QString effecttag;
 376                         QString effectid;
 377                         QString effectindex;
 378                         // Get effect tag & index
 379                         for (QDomNode n3 = effect.firstChild(); !n3.isNull(); n3 = n3.nextSibling()) {
 380                             // parse effect parameters
 381                             QDomElement effectparam = n3.toElement();
 382                             if (effectparam.attribute("name") == "tag") {
 383                                 effecttag = effectparam.text();
 384                             } else if (effectparam.attribute("name") == "kdenlive_id") {
 385                                 effectid = effectparam.text();
 386                             } else if (effectparam.attribute("name") == "kdenlive_ix") {
 387                                 effectindex = effectparam.text();
 388                             }
 389                         }
 390
 391                         // get effect standard tags
 392                         QDomElement clipeffect = MainWindow::customEffects.getEffectByTag(QString(), effectid);
 393                         if (clipeffect.isNull()) clipeffect = MainWindow::videoEffects.getEffectByTag(effecttag, effectid);
 394                         if (clipeffect.isNull()) clipeffect = MainWindow::audioEffects.getEffectByTag(effecttag, effectid);
 395                         if (clipeffect.isNull()) {
 396                             kDebug() << "///  WARNING, EFFECT: " << effecttag << ": " << effectid << " not found, removing it from project";
 397                             m_documentErrors.append(i18n("Effect %1:%2 not found in MLT, it was removed from this project\n", effecttag, effectid));
 398                             elem.removeChild(effects.at(ix));
 399                             ix--;
 400                         } else {
 401                             QDomElement currenteffect = clipeffect.cloneNode().toElement();
 402                             currenteffect.setAttribute("kdenlive_ix", effectindex);
 403                             QDomNodeList clipeffectparams = currenteffect.childNodes();
 404
 405                             if (MainWindow::videoEffects.hasKeyFrames(currenteffect)) {
 406                                 kDebug() << " * * * * * * * * * * ** CLIP EFF WITH KFR FND  * * * * * * * * * * *";
 407                                 // effect is key-framable, read all effects to retrieve keyframes
 408                                 double factor;
 409                                 QString starttag;
 410                                 QString endtag;
 411                                 QDomNodeList params = currenteffect.elementsByTagName("parameter");
 412                                 for (int i = 0; i < params.count(); i++) {
 413                                     QDomElement e = params.item(i).toElement();
 414                                     if (e.attribute("type") == "keyframe") {
 415                                         starttag = e.attribute("starttag", "start");
 416                                         endtag = e.attribute("endtag", "end");
 417                                         factor = e.attribute("factor", "1").toDouble();
 418                                         break;
 419                                     }
 420                                 }
 421                                 QString keyframes;
 422                                 int effectin = effect.attribute("in").toInt();
 423                                 int effectout = effect.attribute("out").toInt();
 424                                 double startvalue;
 425                                 double endvalue;
 426                                 for (QDomNode n3 = effect.firstChild(); !n3.isNull(); n3 = n3.nextSibling()) {
 427                                     // parse effect parameters
 428                                     QDomElement effectparam = n3.toElement();
 429                                     if (effectparam.attribute("name") == starttag)
 430                                         startvalue = effectparam.text().toDouble() * factor;
 431                                     if (effectparam.attribute("name") == endtag)
 432                                         endvalue = effectparam.text().toDouble() * factor;
 433                                 }
 434                                 // add first keyframe
 435                                 keyframes.append(QString::number(effectin) + ":" + QString::number(startvalue) + ";" + QString::number(effectout) + ":" + QString::number(endvalue) + ";");
 436                                 QDomNode lastParsedEffect;
 437                                 ix++;
 438                                 QDomNode n2 = effects.at(ix);
 439                                 bool continueParsing = true;
 440                                 for (; !n2.isNull() && continueParsing; n2 = n2.nextSibling()) {
 441                                     // parse all effects
 442                                     QDomElement kfreffect = n2.toElement();
 443                                     int effectout = kfreffect.attribute("out").toInt();
 444
 445                                     for (QDomNode n4 = kfreffect.firstChild(); !n4.isNull(); n4 = n4.nextSibling()) {
 446                                         // parse effect parameters
 447                                         QDomElement subeffectparam = n4.toElement();
 448                                         if (subeffectparam.attribute("name") == "kdenlive_ix" && subeffectparam.text() != effectindex) {
 449                                             //We are not in the same effect, stop parsing
 450                                             lastParsedEffect = n2.previousSibling();
 451                                             continueParsing = false;
 452                                             break;
 453                                         } else if (subeffectparam.attribute("name") == endtag) {
 454                                             endvalue = subeffectparam.text().toDouble() * factor;
 455                                             break;
 456                                         }
 457                                     }
 458                                     if (continueParsing) keyframes.append(QString::number(effectout) + ":" + QString::number(endvalue) + ";");
 459                                     ix++;
 460                                 }
 461
 462                                 params = currenteffect.elementsByTagName("parameter");
 463                                 for (int i = 0; i < params.count(); i++) {
 464                                     QDomElement e = params.item(i).toElement();
 465                                     if (e.attribute("type") == "keyframe") e.setAttribute("keyframes", keyframes);
 466                                 }
 467                                 if (!continueParsing) {
 468                                     n2 = lastParsedEffect;
 469                                 }
 470                             }
 471
 472                             // adjust effect parameters
 473                             for (QDomNode n3 = effect.firstChild(); !n3.isNull(); n3 = n3.nextSibling()) {
 474                                 // parse effect parameters
 475                                 QDomElement effectparam = n3.toElement();
 476                                 QString paramname = effectparam.attribute("name");
 477                                 QString paramvalue = effectparam.text();
 478
 479                                 // try to find this parameter in the effect xml
 480                                 QDomElement e;
 481                                 for (int k = 0; k < clipeffectparams.count(); k++) {
 482                                     e = clipeffectparams.item(k).toElement();
 483                                     if (!e.isNull() && e.tagName() == "parameter" && e.attribute("name") == paramname) {
 484                                         e.setAttribute("value", paramvalue);
 485                                         break;
 486                                     }
 487                                 }
 488                             }
 489                             item->addEffect(currenteffect, false);
 490                             item->effectsCounter();
 491                         }
 492                     }
 493                 }
 494             } else kWarning() << "CANNOT INSERT CLIP " << id;
 495             //m_clipList.append(clip);
 496         }
 497     }
 498     //m_trackDuration = position;
 499
 500
 501     //documentTracks.insert(ix, track);
 502     kDebug() << "*************  ADD DOC TRACK " << ix << ", DURATION: " << position;
 503     return position;
 504     //track->show();
 505 }
 506
 507 QGraphicsScene *TrackView::projectScene() {
 508     return m_scene;
 509 }
 510
 511 CustomTrackView *TrackView::projectView() {
 512     return m_trackview;
 513 }
 514
 515 void TrackView::setEditMode(const QString & editMode) {
 516     m_editMode = editMode;
 517 }
 518
 519 const QString & TrackView::editMode() const {
 520     return m_editMode;
 521 }
 522
 523
 524
 525 #include "trackview.moc"