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 <QMouseEvent>
  22 #include <QStylePainter>
  23 #include <QScrollBar>
  24
  25 #include <KDebug>
  26
  27 #include "definitions.h"
  28 #include "headertrack.h"
  29 #include "trackview.h"
  30 #include "clipitem.h"
  31 #include "transition.h"
  32 #include "kdenlivesettings.h"
  33 #include "clipmanager.h"
  34 #include "customruler.h"
  35 #include "kdenlivedoc.h"
  36 #include "mainwindow.h"
  37 #include "customtrackview.h"
  38
  39 TrackView::TrackView(KdenliveDoc *doc, QWidget *parent)
  40         : QWidget(parent), m_doc(doc), m_scale(1.0), m_projectTracks(0), m_currentZoom(4) {
  41
  42     view = new Ui::TimeLine_UI();
  43     view->setupUi(this);
  44
  45     m_scene = new QGraphicsScene();
  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     QHBoxLayout *layout = new QHBoxLayout;
  53     view->ruler_frame->setLayout(layout);
  54     int left_margin;
  55     int right_margin;
  56     layout->getContentsMargins(&left_margin, 0, &right_margin, 0);
  57     layout->setContentsMargins(left_margin, 0, right_margin, 0);
  58     layout->addWidget(m_ruler);
  59
  60     QHBoxLayout *tracksLayout = new QHBoxLayout;
  61     tracksLayout->setContentsMargins(0, 0, 0, 0);
  62     view->tracks_frame->setLayout(tracksLayout);
  63
  64     view->headers_area->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  65     view->headers_area->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
  66
  67     m_headersLayout = new QVBoxLayout;
  68     m_headersLayout->setContentsMargins(0, 0, 0, 0);
  69     m_headersLayout->setSpacing(0);
  70     view->headers_container->setLayout(m_headersLayout);
  71
  72     connect(view->headers_area->verticalScrollBar(), SIGNAL(valueChanged(int)), m_trackview->verticalScrollBar(), SLOT(setValue(int)));
  73
  74     tracksLayout->addWidget(m_trackview);
  75
  76     connect(m_trackview->verticalScrollBar(), SIGNAL(valueChanged(int)), view->headers_area->verticalScrollBar(), SLOT(setValue(int)));
  77     connect(m_trackview, SIGNAL(trackHeightChanged()), this, SLOT(slotRebuildTrackHeaders()));
  78
  79     parseDocument(doc->toXml());
  80
  81     connect(m_trackview, SIGNAL(cursorMoved(int, int)), m_ruler, SLOT(slotCursorMoved(int, int)));
  82     connect(m_trackview->horizontalScrollBar(), SIGNAL(valueChanged(int)), m_ruler, SLOT(slotMoveRuler(int)));
  83     connect(m_trackview, SIGNAL(mousePosition(int)), this, SIGNAL(mousePosition(int)));
  84     connect(m_trackview, SIGNAL(clipItemSelected(ClipItem*)), this, SLOT(slotClipItemSelected(ClipItem*)));
  85     connect(m_trackview, SIGNAL(transitionItemSelected(Transition*)), this, SLOT(slotTransitionItemSelected(Transition*)));
  86     slotChangeZoom(m_currentZoom);
  87 }
  88
  89 int TrackView::currentZoom() const {
  90     return m_currentZoom;
  91 }
  92
  93 int TrackView::duration() const {
  94     return m_trackview->duration();
  95 }
  96
  97 int TrackView::tracksNumber() const {
  98     return m_projectTracks;
  99 }
 100
 101 int TrackView::inPoint() const {
 102     return m_ruler->inPoint();
 103 }
 104
 105 int TrackView::outPoint() const {
 106     return m_ruler->outPoint();
 107 }
 108
 109 void TrackView::slotClipItemSelected(ClipItem*c) {
 110     emit clipItemSelected(c);
 111 }
 112
 113 void TrackView::slotTransitionItemSelected(Transition *t) {
 114     emit transitionItemSelected(t);
 115 }
 116
 117 void TrackView::setDuration(int dur) {
 118     m_trackview->setDuration(dur);
 119     m_ruler->setDuration(dur);
 120 }
 121
 122 void TrackView::parseDocument(QDomDocument doc) {
 123     int cursorPos = 0;
 124     //kDebug() << "//// DOCUMENT: " << doc.toString();
 125     QDomNode props = doc.elementsByTagName("properties").item(0);
 126     if (!props.isNull()) {
 127         cursorPos = props.toElement().attribute("timeline_position").toInt();
 128     }
 129
 130     // parse project tracks
 131     QDomNodeList tracks = doc.elementsByTagName("track");
 132     QDomNodeList playlists = doc.elementsByTagName("playlist");
 133     int duration = 300;
 134     m_projectTracks = tracks.count();
 135     int trackduration = 0;
 136     QDomElement e;
 137     QDomElement p;
 138     bool videotrack;
 139
 140     int pos = m_projectTracks - 1;
 141
 142     for (int i = 0; i < m_projectTracks; i++) {
 143         e = tracks.item(i).toElement();
 144         QString playlist_name = e.attribute("producer");
 145         if (playlist_name != "black_track" && playlist_name != "playlistmain") {
 146             // find playlist related to this track
 147             p = QDomElement();
 148             for (int j = 0; j < m_projectTracks; j++) {
 149                 p = playlists.item(j).toElement();
 150                 if (p.attribute("id") == playlist_name) break;
 151             }
 152             videotrack = (e.attribute("hide") != "video");
 153             trackduration = slotAddProjectTrack(pos, p, videotrack);
 154             pos--;
 155             kDebug() << " PRO DUR: " << trackduration << ", TRACK DUR: " << duration;
 156             if (trackduration > duration) duration = trackduration;
 157         } else {
 158                         // background black track
 159                         int black_clips = e.childNodes().count();
 160                         for (int i = 0; i < black_clips; i++)
 161                                 m_doc->loadingProgressed();
 162                         qApp->processEvents();
 163                         pos--;
 164                 }
 165     }
 166
 167     // parse transitions
 168     QDomNodeList transitions = doc.elementsByTagName("transition");
 169     int projectTransitions = transitions.count();
 170     kDebug() << "//////////// TIMELINE FOUND: " << projectTransitions << " transitions";
 171     for (int i = 0; i < projectTransitions; i++) {
 172         e = transitions.item(i).toElement();
 173         QDomNodeList transitionparams = e.childNodes();
 174         bool transitionAdd = true;
 175         int a_track = 0;
 176         int b_track = 0;
 177         QString mlt_service;
 178         for (int k = 0; k < transitionparams.count(); k++) {
 179             p = transitionparams.item(k).toElement();
 180             if (!p.isNull()) {
 181                 // do not add audio mixing transitions
 182                 if (p.attribute("name") == "internal_added" && p.text() == "237") {
 183                     transitionAdd = false;
 184                     kDebug() << "//  TRANSITRION " << i << " IS NOT VALID (INTERN ADDED)";
 185                     break;
 186                 } else if (p.attribute("name") == "a_track") a_track = p.text().toInt();
 187                 else if (p.attribute("name") == "b_track") b_track = m_projectTracks - 1 - p.text().toInt();
 188                 else if (p.attribute("name") == "mlt_service") mlt_service = p.text();
 189             }
 190         }
 191         if (transitionAdd) {
 192             // Transition should be added to the scene
 193             ItemInfo transitionInfo;
 194             transitionInfo.startPos = GenTime(e.attribute("in").toInt(), m_doc->fps());
 195             transitionInfo.endPos = GenTime(e.attribute("out").toInt(), m_doc->fps());
 196             transitionInfo.track = b_track;
 197             kDebug() << "///////////////   +++++++++++  ADDING TRANSITION ON TRACK: " << b_track << ", TOTAL TRKA: " << m_projectTracks;
 198             Transition *tr = new Transition(transitionInfo, a_track, m_scale, m_doc->fps(), QDomElement());
 199             m_scene->addItem(tr);
 200         }
 201     }
 202
 203
 204     m_trackview->setDuration(duration);
 205     kDebug() << "///////////  TOTAL PROJECT DURATION: " << duration;
 206     slotRebuildTrackHeaders();
 207     //m_trackview->setCursorPos(cursorPos);
 208     //m_scrollBox->setGeometry(0, 0, 300 * zoomFactor(), m_scrollArea->height());
 209 }
 210
 211 void TrackView::slotDeleteClip(int clipId) {
 212     m_trackview->deleteClip(clipId);
 213 }
 214
 215 void TrackView::setCursorPos(int pos) {
 216     m_trackview->setCursorPos(pos);
 217 }
 218
 219 void TrackView::moveCursorPos(int pos) {
 220     m_trackview->setCursorPos(pos, false);
 221 }
 222
 223 void TrackView::slotChangeZoom(int factor) {
 224
 225     m_ruler->setPixelPerMark(factor);
 226     m_scale = (double) FRAME_SIZE / m_ruler->comboScale[factor]; // m_ruler->comboScale[m_currentZoom] /
 227     m_currentZoom = factor;
 228     m_trackview->setScale(m_scale);
 229 }
 230
 231 int TrackView::fitZoom() const {
 232     int zoom = (int)((duration() + 20 / m_scale) * FRAME_SIZE / m_trackview->width());
 233     int i;
 234     for (i = 0; i < 13; i++)
 235         if (m_ruler->comboScale[i] > zoom) break;
 236     return i;
 237 }
 238
 239 const double TrackView::zoomFactor() const {
 240     return m_scale;
 241 }
 242
 243 const int TrackView::mapLocalToValue(int x) const {
 244     return (int)(x * zoomFactor());
 245 }
 246
 247 KdenliveDoc *TrackView::document() {
 248     return m_doc;
 249 }
 250
 251 void TrackView::refresh() {
 252     m_trackview->viewport()->update();
 253 }
 254
 255 void TrackView::slotRebuildTrackHeaders() {
 256     QList <TrackInfo> list = m_trackview->tracksList();
 257     QList<HeaderTrack *> widgets = this->findChildren<HeaderTrack *>();
 258     for (int i = 0; i < widgets.count(); i++)
 259         delete widgets.at(i);
 260     int max = list.count();
 261     for (int i = 0; i < max; i++) {
 262         HeaderTrack *header = new HeaderTrack(i, list.at(max - i - 1), this);
 263         connect(header, SIGNAL(switchTrackVideo(int)), m_trackview, SLOT(slotSwitchTrackVideo(int)));
 264         connect(header, SIGNAL(switchTrackAudio(int)), m_trackview, SLOT(slotSwitchTrackAudio(int)));
 265         m_headersLayout->addWidget(header);
 266     }
 267     view->headers_container->adjustSize();
 268 }
 269
 270 int TrackView::slotAddProjectTrack(int ix, QDomElement xml, bool videotrack) {
 271     TrackInfo info;
 272
 273     if (videotrack) {
 274         info.type = VIDEOTRACK;
 275         info.isMute = false;
 276         info.isBlind = false;
 277     } else {
 278         info.type = AUDIOTRACK;
 279         info.isMute = false;
 280         info.isBlind = false;
 281     }
 282
 283     m_trackview->addTrack(info);
 284
 285     int trackTop = KdenliveSettings::trackheight() * ix;
 286     // parse track
 287     int position = 0;
 288     for (QDomNode n = xml.firstChild(); !n.isNull(); n = n.nextSibling()) {
 289         QDomElement elem = n.toElement();
 290         if (elem.tagName() == "blank") {
 291             position += elem.attribute("length").toInt();
 292         } else if (elem.tagName() == "entry") {
 293                         m_doc->loadingProgressed();
 294                         qApp->processEvents();
 295             // Found a clip
 296             int in = elem.attribute("in").toInt();
 297             int id = elem.attribute("producer").toInt();
 298             DocClipBase *clip = m_doc->clipManager()->getClipById(id);
 299             if (clip != NULL) {
 300                 int out = elem.attribute("out").toInt();
 301
 302                 ItemInfo clipinfo;
 303                 clipinfo.startPos = GenTime(position, m_doc->fps());
 304                 clipinfo.endPos = clipinfo.startPos + GenTime(out, m_doc->fps());
 305                 clipinfo.track = ix;
 306                 kDebug() << "// INSERTING CLIP: " << in << "x" << out << ", track: " << ix << ", ID: " << id << ", SCALE: " << m_scale << ", FPS: " << m_doc->fps();
 307                 ClipItem *item = new ClipItem(clip, clipinfo, m_scale, m_doc->fps());
 308                 m_scene->addItem(item);
 309                 position += out;
 310
 311                 // parse clip effects
 312                 for (QDomNode n2 = elem.firstChild(); !n2.isNull(); n2 = n2.nextSibling()) {
 313                     QDomElement effect = n2.toElement();
 314                     if (effect.tagName() == "filter") {
 315                         // add effect to clip
 316                         QString effecttag;
 317                         QString effectindex;
 318                         // Get effect tag & index
 319                         for (QDomNode n3 = effect.firstChild(); !n3.isNull(); n3 = n3.nextSibling()) {
 320                             // parse effect parameters
 321                             QDomElement effectparam = n3.toElement();
 322                             if (effectparam.attribute("name") == "tag") {
 323                                 effecttag = effectparam.text();
 324                             }
 325                             if (effectparam.attribute("name") == "kdenlive_ix") {
 326                                 effectindex = effectparam.text();
 327                             }
 328                         }
 329
 330                         // get effect standard tags
 331                         QDomElement clipeffect = MainWindow::videoEffects.getEffectByTag(effecttag);
 332                         clipeffect.setAttribute("kdenlive_ix", effectindex);
 333                         QDomNodeList clipeffectparams = clipeffect.childNodes();
 334
 335                         // adjust effect parameters
 336                         for (QDomNode n3 = effect.firstChild(); !n3.isNull(); n3 = n3.nextSibling()) {
 337                             // parse effect parameters
 338                             QDomElement effectparam = n3.toElement();
 339                             QString paramname = effectparam.attribute("name");
 340                             QString paramvalue = effectparam.text();
 341
 342                             // try to find this parameter in the effect xml
 343                             QDomElement e;
 344                             for (int k = 0; k < clipeffectparams.count(); k++) {
 345                                 e = clipeffectparams.item(k).toElement();
 346                                 if (!e.isNull() && e.tagName() == "parameter" && e.attribute("name") == paramname) {
 347                                     e.setAttribute("value", paramvalue);
 348                                     break;
 349                                 }
 350                             }
 351                         }
 352                         item->addEffect(clipeffect, false);
 353                     }
 354                 }
 355
 356             } else kWarning() << "CANNOT INSERT CLIP " << id;
 357             //m_clipList.append(clip);
 358         }
 359     }
 360     //m_trackDuration = position;
 361
 362
 363     //documentTracks.insert(ix, track);
 364     kDebug() << "*************  ADD DOC TRACK " << ix << ", DURATION: " << position;
 365     return position;
 366     //track->show();
 367 }
 368
 369 QGraphicsScene *TrackView::projectScene() {
 370     return m_scene;
 371 }
 372
 373 CustomTrackView *TrackView::projectView() {
 374     return m_trackview;
 375 }
 376
 377 void TrackView::setEditMode(const QString & editMode) {
 378     m_editMode = editMode;
 379 }
 380
 381 const QString & TrackView::editMode() const {
 382     return m_editMode;
 383 }
 384
 385 #include "trackview.moc"