Files
oak-editor/ui/timelineheader.cpp
T
2018-10-13 13:08:20 +11:00

368 lines
11 KiB
C++

#include "timelineheader.h"
#include "panels/panels.h"
#include "panels/timeline.h"
#include "project/sequence.h"
#include "project/undo.h"
#include "panels/viewer.h"
#include "io/config.h"
#include <QPainter>
#include <QMouseEvent>
#include <QDebug>
#include <QScrollArea>
#include <QtMath>
#define CLICK_RANGE 5
#define PLAYHEAD_SIZE 6
#define LINE_MIN_PADDING 50
#define SUBLINE_MIN_PADDING 50 // TODO play with this
#define MARKER_SIZE 4
TimelineHeader::TimelineHeader(QWidget *parent) :
QWidget(parent),
dragging(false),
resizing_workarea(false),
zoom(1),
in_visible(0),
snapping(true),
fm(font()),
dragging_markers(false),
scroll(0)
{
height_actual = fm.height();
setCursor(Qt::ArrowCursor);
setMouseTracking(true);
setFocusPolicy(Qt::ClickFocus);
show_text(true);
}
void TimelineHeader::set_scroll(int s) {
scroll = s;
update();
}
long TimelineHeader::getHeaderFrameFromScreenPoint(int x) {
return getFrameFromScreenPoint(zoom, x + scroll) + in_visible;
}
int TimelineHeader::getHeaderScreenPointFromFrame(long frame) {
return getScreenPointFromFrame(zoom, frame - in_visible) - scroll;
}
void TimelineHeader::set_playhead(int mouse_x) {
long frame = getHeaderFrameFromScreenPoint(mouse_x);
if (snapping) panel_timeline->snap_to_timeline(&frame, false, true, true);
if (frame != sequence->playhead) {
panel_timeline->seek(frame);
}
}
void TimelineHeader::set_visible_in(long i) {
in_visible = i;
update();
}
void TimelineHeader::set_in_point(long new_in) {
long new_out = sequence->workarea_out;
if (new_out == new_in) {
new_in--;
} else if (new_out < new_in) {
new_out = sequence->getEndFrame();
}
undo_stack.push(new SetTimelineInOutCommand(sequence, true, new_in, new_out));
panel_timeline->repaint_timeline(false);
}
void TimelineHeader::set_out_point(long new_out) {
long new_in = sequence->workarea_in;
if (new_out == new_in) {
new_out++;
} else if (new_in > new_out) {
new_in = 0;
}
undo_stack.push(new SetTimelineInOutCommand(sequence, true, new_in, new_out));
panel_timeline->repaint_timeline(false);
}
void TimelineHeader::show_text(bool enable) {
text_enabled = enable;
if (enable) {
setFixedHeight(height_actual*2);
} else {
setFixedHeight(height_actual);
}
update();
}
void TimelineHeader::mousePressEvent(QMouseEvent* event) {
if (resizing_workarea) {
sequence_end = sequence->getEndFrame();
} else {
bool shift = (event->modifiers() & Qt::ShiftModifier);
bool clicked_on_marker = false;
for (int i=0;i<sequence->markers.size();i++) {
int marker_pos = getHeaderScreenPointFromFrame(sequence->markers.at(i).frame);
if (event->pos().x() > marker_pos - MARKER_SIZE && event->pos().x() < marker_pos + MARKER_SIZE) {
bool found = false;
for (int j=0;j<selected_markers.size();j++) {
if (selected_markers.at(j) == i) {
if (shift) {
selected_markers.removeAt(j);
}
found = true;
break;
}
}
if (!found) {
if (!shift) {
selected_markers.clear();
}
selected_markers.append(i);
}
clicked_on_marker = true;
update();
break;
}
}
if (clicked_on_marker) {
selected_marker_original_times.resize(selected_markers.size());
for (int i=0;i<selected_markers.size();i++) {
selected_marker_original_times[i] = sequence->markers.at(selected_markers.at(i)).frame;
}
drag_start = event->pos().x();
dragging_markers = true;
} else {
if (selected_markers.size() > 0) {
selected_markers.clear();
update();
}
set_playhead(event->pos().x());
}
}
dragging = true;
}
void TimelineHeader::mouseMoveEvent(QMouseEvent* event) {
if (dragging) {
if (resizing_workarea) {
long frame = getHeaderFrameFromScreenPoint(event->pos().x());
if (snapping) panel_timeline->snap_to_timeline(&frame, true, true, false);
if (resizing_workarea_in) {
temp_workarea_in = qMax(qMin(temp_workarea_out-1, frame), 0L);
} else {
temp_workarea_out = qMin(qMax(temp_workarea_in+1, frame), sequence_end);
}
panel_timeline->repaint_timeline(false);
} else if (dragging_markers) {
long frame_movement = getHeaderFrameFromScreenPoint(event->pos().x()) - getHeaderFrameFromScreenPoint(drag_start);
// snap markers
for (int i=0;i<selected_markers.size();i++) {
long fm = selected_marker_original_times.at(i) + frame_movement;
if (snapping && panel_timeline->snap_to_timeline(&fm, true, false, true)) {
frame_movement = fm - selected_marker_original_times.at(i);
break;
}
}
// validate markers (ensure none go below 0)
long validator;
for (int i=0;i<selected_markers.size();i++) {
validator = selected_marker_original_times.at(i) + frame_movement;
if (validator < 0) {
frame_movement -= validator;
}
}
// move markers
for (int i=0;i<selected_markers.size();i++) {
sequence->markers[selected_markers.at(i)].frame = selected_marker_original_times.at(i) + frame_movement;
}
panel_timeline->repaint_timeline(false);
} else {
set_playhead(event->pos().x());
}
} else {
resizing_workarea = false;
unsetCursor();
if (sequence != NULL && sequence->using_workarea) {
long min_frame = getHeaderFrameFromScreenPoint(event->pos().x() - CLICK_RANGE) - 1;
long max_frame = getHeaderFrameFromScreenPoint(event->pos().x() + CLICK_RANGE) + 1;
if (sequence->workarea_in > min_frame && sequence->workarea_in < max_frame) {
resizing_workarea = true;
resizing_workarea_in = true;
} else if (sequence->workarea_out > min_frame && sequence->workarea_out < max_frame) {
resizing_workarea = true;
resizing_workarea_in = false;
}
if (resizing_workarea) {
temp_workarea_in = sequence->workarea_in;
temp_workarea_out = sequence->workarea_out;
setCursor(Qt::SizeHorCursor);
}
}
}
}
void TimelineHeader::mouseReleaseEvent(QMouseEvent*) {
dragging = false;
if (resizing_workarea) {
undo_stack.push(new SetTimelineInOutCommand(sequence, true, temp_workarea_in, temp_workarea_out));
} else if (dragging_markers && selected_markers.size() > 0) {
bool moved = false;
ComboAction* ca = new ComboAction();
for (int i=0;i<selected_markers.size();i++) {
Marker* m = &sequence->markers[selected_markers.at(i)];
if (selected_marker_original_times.at(i) != m->frame) {
ca->append(new MoveMarkerAction(m, selected_marker_original_times.at(i), m->frame));
moved = true;
}
}
if (moved) {
undo_stack.push(ca);
} else {
delete ca;
}
}
resizing_workarea = false;
dragging = false;
dragging_markers = false;
panel_timeline->snapped = false;
panel_timeline->repaint_timeline(false);
}
void TimelineHeader::focusOutEvent(QFocusEvent*) {
selected_markers.clear();
update();
}
void TimelineHeader::update_zoom(double z) {
zoom = z;
update();
}
void TimelineHeader::delete_markers() {
if (selected_markers.size() > 0) {
DeleteMarkerAction* dma = new DeleteMarkerAction(sequence);
for (int i=0;i<selected_markers.size();i++) {
dma->markers.append(selected_markers.at(i));
}
undo_stack.push(dma);
panel_timeline->repaint_timeline(false);
}
}
void TimelineHeader::paintEvent(QPaintEvent*) {
if (sequence != NULL && zoom > 0) {
QPainter p(this);
int yoff = (text_enabled) ? height()/2 : 0;
double interval = sequence->frame_rate;
int textWidth = 0;
int lastTextBoundary = INT_MIN;
int i = 0;
int lastLineX = INT_MIN;
int sublineCount = 1;
int sublineTest = qRound(interval*zoom);
int sublineInterval = 1;
while (sublineTest > SUBLINE_MIN_PADDING
&& sublineInterval >= 1) {
sublineCount *= 2;
sublineInterval = (interval/sublineCount);
sublineTest = qRound(sublineInterval*zoom);
}
sublineCount = qMin(sublineCount, qRound(interval));
while (true) {
long frame = qRound(interval*i);
int lineX = qRound(frame*zoom) - scroll;
int next_lineX = qRound(qRound(interval*(i+1))*zoom) - scroll;
if (lineX > width()) break;
if (next_lineX > 0 && lineX > lastLineX+LINE_MIN_PADDING) {
// draw text
if (text_enabled && lineX-textWidth > lastTextBoundary) {
p.setPen(Qt::white);
QString timecode = frame_to_timecode(frame + in_visible, config.timecode_view, sequence->frame_rate);
int fullTextWidth = fm.width(timecode);
textWidth = fullTextWidth>>1;
int text_x = lineX-textWidth;
lastTextBoundary = lineX+textWidth;
if (lastTextBoundary >= 0) {
p.drawText(QRect(text_x, 0, fullTextWidth, yoff), timecode);
}
}
// draw line markers
p.setPen(Qt::gray);
p.drawLine(lineX, yoff, lineX, height());
lastLineX = lineX;
// draw sub-line markers
for (int j=1;j<sublineCount;j++) {
int sublineX = lineX+(qRound(j*interval/sublineCount)*zoom);
p.drawLine(sublineX, yoff, sublineX, yoff+(yoff>>1));
}
}
i++;
}
// draw in/out selection
int in_x;
if (sequence->using_workarea) {
in_x = getHeaderScreenPointFromFrame((resizing_workarea ? temp_workarea_in : sequence->workarea_in));
int out_x = getHeaderScreenPointFromFrame((resizing_workarea ? temp_workarea_out : sequence->workarea_out));
p.fillRect(QRect(in_x, 0, out_x-in_x, height()), QColor(0, 192, 255, 128));
p.setPen(Qt::white);
p.drawLine(in_x, 0, in_x, height());
p.drawLine(out_x, 0, out_x, height());
}
// draw markers
for (int i=0;i<sequence->markers.size();i++) {
const Marker& m = sequence->markers.at(i);
int marker_x = getHeaderScreenPointFromFrame(m.frame);
const QPoint points[5] = {
QPoint(marker_x, height()-1),
QPoint(marker_x + MARKER_SIZE, height() - MARKER_SIZE - 1),
QPoint(marker_x + MARKER_SIZE, yoff),
QPoint(marker_x - MARKER_SIZE, yoff),
QPoint(marker_x - MARKER_SIZE, height() - MARKER_SIZE - 1)
};
p.setPen(Qt::black);
bool selected = false;
for (int j=0;j<selected_markers.size();j++) {
if (selected_markers.at(j) == i) {
selected = true;
break;
}
}
if (selected) {
p.setBrush(QColor(208, 255, 208));
} else {
p.setBrush(QColor(128, 224, 128));
}
p.drawPolygon(points, 5);
}
// draw playhead triangle
in_x = getHeaderScreenPointFromFrame(sequence->playhead);
QPoint start(in_x, height()+2);
QPainterPath path;
path.moveTo(start);
path.lineTo(in_x-PLAYHEAD_SIZE, yoff);
path.lineTo(in_x+PLAYHEAD_SIZE, yoff);
path.lineTo(start);
p.fillPath(path, Qt::red);
}
}