← Back to branch summary

~ubuntu-branches/ubuntu/trusty/pitivi/trusty

« back to all changes in this revision

Viewing changes to pitivi/timeline/timeline_undo.py

* New upstream pre-release:
  + debian/control:
    - Update dependencies.
* debian/control:
  + Update Standards-Version to 3.8.2.

Show diffs side-by-side

added added

removed removed

Lines of Context:
pitivi/timeline/timeline_undo.py
 
1
# PiTiVi , Non-linear video editor
 
2
#
 
3
#       pitivi/timeline/timeline_undo.py
 
4
#
 
5
# Copyright (c) 2009, Alessandro Decina <alessandro.d@gmail.com>
 
6
#
 
7
# This program is free software; you can redistribute it and/or
 
8
# modify it under the terms of the GNU Lesser General Public
 
9
# License as published by the Free Software Foundation; either
 
10
# version 2.1 of the License, or (at your option) any later version.
 
11
#
 
12
# This program is distributed in the hope that it will be useful,
 
13
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
15
# Lesser General Public License for more details.
 
16
#
 
17
# You should have received a copy of the GNU Lesser General Public
 
18
# License along with this program; if not, write to the
 
19
# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 
20
# Boston, MA 02111-1307, USA.
 
21
 
 
22
from pitivi.signalinterface import Signallable
 
23
from pitivi.utils import PropertyChangeTracker
 
24
from pitivi.undo import UndoableAction
 
25
 
 
26
class TimelineObjectPropertyChangeTracker(PropertyChangeTracker):
 
27
    # no out-point
 
28
    property_names = ["start", "duration", "in-point",
 
29
            "media-duration", "priority", "selected"]
 
30
 
 
31
    _disabled = False
 
32
 
 
33
    def connectToObject(self, obj):
 
34
        PropertyChangeTracker.connectToObject(self, obj)
 
35
        self.timeline = obj.timeline
 
36
        self.timeline.connect("disable-updates", self._timelineDisableUpdatesCb)
 
37
 
 
38
    def disconnectFromObject(self, obj):
 
39
        self.timeline.disconnect_by_func(self._timelineDisableUpdatesCb)
 
40
        PropertyChangeTracker.disconnectFromObject(self, obj)
 
41
 
 
42
    def _timelineDisableUpdatesCb(self, timeline, disabled):
 
43
        if self._disabled and not disabled:
 
44
            self._disabled = disabled
 
45
            properties = self._takeCurrentSnapshot(self.obj)
 
46
            for property_name, property_value in properties.iteritems():
 
47
                old_value = self.properties[property_name]
 
48
                if old_value != property_value:
 
49
                    self._propertyChangedCb(self.obj, property_value, property_name)
 
50
        else:
 
51
            self._disabled = disabled
 
52
 
 
53
    def _propertyChangedCb(self, timeline_object, value, property_name):
 
54
        if not self._disabled:
 
55
            PropertyChangeTracker._propertyChangedCb(self,
 
56
                    timeline_object, value, property_name)
 
57
 
 
58
class KeyframeChangeTracker(Signallable):
 
59
    __signals__ = {
 
60
        "keyframe-moved": ["keyframe"]
 
61
    }
 
62
 
 
63
    def __init__(self):
 
64
        self.keyframes = None
 
65
        self.obj = None
 
66
 
 
67
    def connectToObject(self, obj):
 
68
        self.obj = obj
 
69
        self.keyframes = self._takeCurrentSnapshot(obj)
 
70
        obj.connect("keyframe-added", self._keyframeAddedCb)
 
71
        obj.connect("keyframe-removed", self._keyframeRemovedCb)
 
72
        obj.connect("keyframe-moved", self._keyframeMovedCb)
 
73
 
 
74
    def _takeCurrentSnapshot(self, obj):
 
75
        keyframes = {}
 
76
        for keyframe in self.obj.getKeyframes():
 
77
            keyframes[keyframe] = self._getKeyframeSnapshot(keyframe)
 
78
 
 
79
        return keyframes
 
80
 
 
81
    def disconnectFromObject(self, obj):
 
82
        self.obj = None
 
83
        obj.disconnect_by_func(self._keyframeMovedCb)
 
84
 
 
85
    def _keyframeAddedCb(self, interpolator, keyframe):
 
86
        self.keyframes[keyframe] = self._getKeyframeSnapshot(keyframe)
 
87
 
 
88
    def _keyframeRemovedCb(self, interpolator, keyframe):
 
89
        pass
 
90
 
 
91
    def _keyframeMovedCb(self, interpolator, keyframe):
 
92
        old_snapshot = self.keyframes[keyframe]
 
93
        new_snapshot = self._getKeyframeSnapshot(keyframe)
 
94
        self.keyframes[keyframe] = new_snapshot
 
95
 
 
96
        self.emit("keyframe-moved", interpolator,
 
97
                keyframe, old_snapshot, new_snapshot)
 
98
 
 
99
    def _getKeyframeSnapshot(self, keyframe):
 
100
        return (keyframe.mode, keyframe.time, keyframe.value)
 
101
 
 
102
class TimelineObjectPropertyChanged(UndoableAction):
 
103
    def __init__(self, timeline_object, property_name, old_value, new_value):
 
104
        self.timeline_object = timeline_object
 
105
        self.property_name = property_name
 
106
        self.old_value = old_value
 
107
        self.new_value = new_value
 
108
 
 
109
    def do(self):
 
110
        setattr(self.timeline_object,
 
111
                self.property_name.replace("-", "_"), self.new_value)
 
112
        self._done()
 
113
 
 
114
    def undo(self):
 
115
        setattr(self.timeline_object,
 
116
                self.property_name.replace("-", "_"), self.old_value)
 
117
        self._undone()
 
118
 
 
119
class TimelineObjectAdded(UndoableAction):
 
120
    def __init__(self, timeline, timeline_object):
 
121
        self.timeline = timeline
 
122
        self.timeline_object = timeline_object
 
123
        self.tracks = dict((track_object, track_object.track)
 
124
                for track_object in timeline_object.track_objects)
 
125
 
 
126
    def do(self):
 
127
        for track_object, track in self.tracks.iteritems():
 
128
            track.addTrackObject(track_object)
 
129
 
 
130
        self.timeline.addTimelineObject(self.timeline_object)
 
131
        self._done()
 
132
 
 
133
    def undo(self):
 
134
        self.timeline.removeTimelineObject(self.timeline_object, deep=True)
 
135
        self._undone()
 
136
 
 
137
class TimelineObjectRemoved(UndoableAction):
 
138
    def __init__(self, timeline, timeline_object):
 
139
        self.timeline = timeline
 
140
        self.timeline_object = timeline_object
 
141
        self.tracks = dict((track_object, track_object.track)
 
142
                for track_object in timeline_object.track_objects)
 
143
 
 
144
    def do(self):
 
145
        self.timeline.removeTimelineObject(self.timeline_object, deep=True)
 
146
        self._done()
 
147
 
 
148
    def undo(self):
 
149
        for track_object, track in self.tracks.iteritems():
 
150
            track.addTrackObject(track_object)
 
151
 
 
152
        self.timeline.addTimelineObject(self.timeline_object)
 
153
        self._undone()
 
154
 
 
155
class InterpolatorKeyframeAdded(UndoableAction):
 
156
    def __init__(self, track_object, keyframe):
 
157
        self.track_object = track_object
 
158
        self.keyframe = keyframe
 
159
 
 
160
    def do(self):
 
161
        self.track_object.newKeyframe(self.keyframe)
 
162
        self._done()
 
163
 
 
164
    def undo(self):
 
165
        self.track_object.removeKeyframe(self.keyframe)
 
166
        self._undone()
 
167
 
 
168
class InterpolatorKeyframeRemoved(UndoableAction):
 
169
    def __init__(self, track_object, keyframe):
 
170
        self.track_object = track_object
 
171
        self.keyframe = keyframe
 
172
 
 
173
    def do(self):
 
174
        self.track_object.removeKeyframe(self.keyframe)
 
175
        self._undone()
 
176
 
 
177
    def undo(self):
 
178
        self.track_object.newKeyframe(self.keyframe.time,
 
179
                self.keyframe.value, self.keyframe.mode)
 
180
        self._done()
 
181
 
 
182
class InterpolatorKeyframeChanged(UndoableAction):
 
183
    def __init__(self, track_object, keyframe, old_snapshot, new_snapshot):
 
184
        self.track_object = track_object
 
185
        self.keyframe = keyframe
 
186
        self.old_snapshot = old_snapshot
 
187
        self.new_snapshot = new_snapshot
 
188
 
 
189
    def do(self):
 
190
        self._setSnapshot(self.new_snapshot)
 
191
        self._done()
 
192
 
 
193
    def undo(self):
 
194
        self._setSnapshot(self.old_snapshot)
 
195
        self._undone()
 
196
 
 
197
    def _setSnapshot(self, snapshot):
 
198
        mode, time, value = snapshot
 
199
        self.keyframe.setMode(mode)
 
200
        self.keyframe.setTime(time)
 
201
        self.keyframe.setValue(value)
 
202
 
 
203
class TimelineLogObserver(object):
 
204
    timelinePropertyChangedAction = TimelineObjectPropertyChanged
 
205
    timelineObjectAddedAction = TimelineObjectAdded
 
206
    timelineObjectRemovedAction = TimelineObjectRemoved
 
207
    interpolatorKeyframeAddedAction = InterpolatorKeyframeAdded
 
208
    interpolatorKeyframeRemovedAction = InterpolatorKeyframeRemoved
 
209
    interpolatorKeyframeChangedAction = InterpolatorKeyframeChanged
 
210
 
 
211
    def __init__(self, log):
 
212
        self.log = log
 
213
        self.timeline_object_property_trackers = {}
 
214
        self.interpolator_keyframe_trackers = {}
 
215
 
 
216
    def startObserving(self, timeline):
 
217
        self._connectToTimeline(timeline)
 
218
        for timeline_object in timeline.timeline_objects:
 
219
            self._connectToTimelineObject(timeline_object)
 
220
            for track_object in timeline_object.track_objects:
 
221
                self._connectToTrackObject(track_object)
 
222
 
 
223
    def stopObserving(self, timeline):
 
224
        self._disconnectFromTimeline(timeline)
 
225
        for timeline_object in timeline.timeline_objects:
 
226
            self._disconnectFromTimelineObject(timeline_object)
 
227
            for track_object in timeline_object.track_objects:
 
228
                self._disconnectFromTrackObject(track_object)
 
229
 
 
230
    def _connectToTimeline(self, timeline):
 
231
        timeline.connect("timeline-object-added", self._timelineObjectAddedCb)
 
232
        timeline.connect("timeline-object-removed", self._timelineObjectRemovedCb)
 
233
 
 
234
    def _disconnectFromTimeline(self, timeline):
 
235
        timeline.disconnect_by_func(self._timelineObjectAddedCb)
 
236
        timeline.disconnect_by_func(self._timelineObjectRemovedCb)
 
237
 
 
238
    def _connectToTimelineObject(self, timeline_object):
 
239
        tracker = TimelineObjectPropertyChangeTracker()
 
240
        tracker.connectToObject(timeline_object)
 
241
        for property_name in tracker.property_names:
 
242
            tracker.connect(property_name + "-changed",
 
243
                    self._timelineObjectPropertyChangedCb, property_name)
 
244
        self.timeline_object_property_trackers[timeline_object] = tracker
 
245
 
 
246
        timeline_object.connect("track-object-added", self._timelineObjectTrackObjectAddedCb)
 
247
        timeline_object.connect("track-object-removed", self._timelineObjectTrackObjectRemovedCb)
 
248
        for obj in timeline_object.track_objects:
 
249
            self._connectToTrackObject(obj)
 
250
 
 
251
    def _disconnectFromTimelineObject(self, timeline_object):
 
252
        tracker = self.timeline_object_property_trackers.pop(timeline_object)
 
253
        tracker.disconnectFromObject(timeline_object)
 
254
        tracker.disconnect_by_func(self._timelineObjectPropertyChangedCb)
 
255
 
 
256
    def _connectToTrackObject(self, track_object):
 
257
        for prop, interpolator in track_object.getInterpolators().itervalues():
 
258
            self._connectToInterpolator(interpolator)
 
259
 
 
260
    def _disconnectFromTrackObject(self, track_object):
 
261
        for prop, interpolator in track_object.getInterpolators().itervalues():
 
262
            self._disconnectFromInterpolator(interpolator)
 
263
 
 
264
    def _connectToInterpolator(self, interpolator):
 
265
        interpolator.connect("keyframe-added", self._interpolatorKeyframeAddedCb)
 
266
        interpolator.connect("keyframe-removed",
 
267
                self._interpolatorKeyframeRemovedCb)
 
268
 
 
269
        tracker = KeyframeChangeTracker()
 
270
        tracker.connectToObject(interpolator)
 
271
        tracker.connect("keyframe-moved", self._interpolatorKeyframeMovedCb)
 
272
        self.interpolator_keyframe_trackers[interpolator] = tracker
 
273
 
 
274
    def _disconnectFromInterpolator(self, interpolator):
 
275
        tracker = self.interpolator_keyframe_trackers.pop(interpolator)
 
276
        tracker.disconnectFromObject(interpolator)
 
277
        tracker.disconnect_by_func(self._interpolatorKeyframeMovedCb)
 
278
 
 
279
    def _timelineObjectAddedCb(self, timeline, timeline_object):
 
280
        self._connectToTimelineObject(timeline_object)
 
281
        action = self.timelineObjectAddedAction(timeline, timeline_object)
 
282
        self.log.push(action)
 
283
 
 
284
    def _timelineObjectRemovedCb(self, timeline, timeline_object):
 
285
        self._disconnectFromTimelineObject(timeline_object)
 
286
        action = self.timelineObjectRemovedAction(timeline, timeline_object)
 
287
        self.log.push(action)
 
288
 
 
289
    def _timelineObjectPropertyChangedCb(self, tracker, timeline_object,
 
290
            old_value, new_value, property_name):
 
291
        action = self.timelinePropertyChangedAction(timeline_object,
 
292
                property_name, old_value, new_value)
 
293
        self.log.push(action)
 
294
 
 
295
    def _timelineObjectTrackObjectAddedCb(self, timeline_object, track_object):
 
296
        self._connectToTrackObject(track_object)
 
297
 
 
298
    def _timelineObjectTrackObjectRemovedCb(self, timeline_object,
 
299
            track_object):
 
300
        self._disconnectFromTrackObject(track_object)
 
301
 
 
302
    def _interpolatorKeyframeAddedCb(self, track_object, keyframe):
 
303
        action = self.interpolatorKeyframeAddedAction(track_object, keyframe)
 
304
        self.log.push(action)
 
305
 
 
306
    def _interpolatorKeyframeRemovedCb(self, track_object, keyframe):
 
307
        action = self.interpolatorKeyframeRemovedAction(track_object, keyframe)
 
308
        self.log.push(action)
 
309
 
 
310
    def _interpolatorKeyframeMovedCb(self, tracker, track_object,
 
311
            keyframe, old_snapshot, new_snapshot):
 
312
        action = self.interpolatorKeyframeChangedAction(track_object,
 
313
                keyframe, old_snapshot, new_snapshot)
 
314
        self.log.push(action)