class TestSourceList(TestCase):
def setUp(self):
self.sourcelist = SourceList()
def testAddUriDiscoveryOk(self):
"""
Test the simple case of adding an uri.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
# mock discovery-done
self.sourcelist.discoverer.emit("discovery-done", uri, factory)
self.failUnlessEqual(len(self.sourcelist.getSources()), 1)
# can't add again
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
def testAddUriDiscoveryOkSourceGone(self):
"""
Test that we don't explode if discoverer finishes analyzing a source
that in the meantime was removed.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.sourcelist.removeUri(uri)
self.sourcelist.discoverer.emit("discovery-done", uri, factory)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# this shouldn't fail since we removed the factory before the discovery
# was complete
self.sourcelist.addUri(uri)
def testAddUriDiscoveryErrorSourceGone(self):
"""
Same as the test above, but testing the discovery-error handler.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.sourcelist.removeUri(uri)
self.sourcelist.discoverer.emit("discovery-error", uri,
"error", "verbose debug")
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# this shouldn't fail since we removed the factory before the discovery
# was complete
self.sourcelist.addUri(uri)
def testAddUriDiscoveryError(self):
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
# mock discovery-done
self.sourcelist.discoverer.emit("discovery-error", uri,
"error", "verbose debug")
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# there was an error, the factory wasn't added so this shouldn't raise
self.sourcelist.addUri(uri)
class TestSourceList(TestCase):
def setUp(self):
self.sourcelist = SourceList()
def testAddUriDiscoveryOk(self):
"""
Test the simple case of adding an uri.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
# mock discovery-done
self.sourcelist.discoverer.emit("discovery-done", uri, factory)
self.failUnlessEqual(len(self.sourcelist.getSources()), 1)
# can't add again
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
def testAddUriDiscoveryOkSourceGone(self):
"""
Test that we don't explode if discoverer finishes analyzing a source
that in the meantime was removed.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.sourcelist.removeUri(uri)
self.sourcelist.discoverer.emit("discovery-done", uri, factory)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# this shouldn't fail since we removed the factory before the discovery
# was complete
self.sourcelist.addUri(uri)
def testAddUriDiscoveryErrorSourceGone(self):
"""
Same as the test above, but testing the discovery-error handler.
"""
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.sourcelist.removeUri(uri)
self.sourcelist.discoverer.emit("discovery-error", uri, "error",
"verbose debug")
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# this shouldn't fail since we removed the factory before the discovery
# was complete
self.sourcelist.addUri(uri)
def testAddUriDiscoveryError(self):
uri = "file:///ciao"
factory = FileSourceFactory(uri)
self.sourcelist.addUri(uri)
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
self.failUnlessRaises(SourceListError, self.sourcelist.addUri, uri)
# mock discovery-done
self.sourcelist.discoverer.emit("discovery-error", uri, "error",
"verbose debug")
self.failUnlessEqual(len(self.sourcelist.getSources()), 0)
# there was an error, the factory wasn't added so this shouldn't raise
self.sourcelist.addUri(uri)
class Project(Signallable, Loggable):
"""The base class for PiTiVi projects
@ivar name: The name of the project
@type name: C{str}
@ivar description: A description of the project
@type description: C{str}
@ivar sources: The sources used by this project
@type sources: L{SourceList}
@ivar timeline: The timeline
@type timeline: L{Timeline}
@ivar pipeline: The timeline's pipeline
@type pipeline: L{Pipeline}
@ivar factory: The timeline factory
@type factory: L{TimelineSourceFactory}
@ivar format: The format under which the project is currently stored.
@type format: L{FormatterClass}
@ivar loaded: Whether the project is fully loaded or not.
@type loaded: C{bool}
Signals:
- C{loaded} : The project is now fully loaded.
"""
__signals__ = {
"settings-changed" : ['old', 'new'],
"project-changed" : [],
}
def __init__(self, name="", uri=None, **kwargs):
"""
@param name: the name of the project
@param uri: the uri of the project
"""
Loggable.__init__(self)
self.log("name:%s, uri:%s", name, uri)
self.name = name
self.settings = None
self.description = ""
self.uri = uri
self.urichanged = False
self.format = None
self.sources = SourceList()
self.sources.connect("source-added", self._sourceAddedCb)
self.sources.connect("source-removed", self._sourceRemovedCb)
self._dirty = False
self.timeline = Timeline()
self.factory = TimelineSourceFactory(self.timeline)
self.pipeline = Pipeline()
self.view_action = ViewAction()
self.view_action.addProducers(self.factory)
self.seeker = Seeker(80)
self.settings = ExportSettings()
self._videocaps = self.settings.getVideoCaps()
def release(self):
self.pipeline.release()
self.pipeline = None
#{ Settings methods
def getSettings(self):
"""
return the currently configured settings.
"""
self.debug("self.settings %s", self.settings)
return self.settings
def setSettings(self, settings):
"""
Sets the given settings as the project's settings.
@param settings: The new settings for the project.
@type settings: ExportSettings
"""
assert settings
self.log("Setting %s as the project's settings", settings)
oldsettings = self.settings
self.settings = settings
self._projectSettingsChanged()
self.emit('settings-changed', oldsettings, settings)
#}
#{ Save and Load features
def setModificationState(self, state):
self._dirty = state
if state:
self.emit('project-changed')
def hasUnsavedModifications(self):
return self._dirty
def _projectSettingsChanged(self):
settings = self.getSettings()
self._videocaps = settings.getVideoCaps()
if self.timeline:
self.timeline.updateVideoCaps(self._videocaps)
for fact in self.sources.getSources():
fact.setFilterCaps(self._videocaps)
if self.pipeline.getState() != gst.STATE_NULL:
self.pipeline.stop()
self.pipeline.pause()
def _sourceAddedCb(self, sourcelist, factory):
factory.setFilterCaps(self._videocaps)
def _sourceRemovedCb(self, sourclist, uri, factory):
self.timeline.removeFactory(factory)
class Project(Signallable, Loggable):
"""The base class for PiTiVi projects
@ivar name: The name of the project
@type name: C{str}
@ivar description: A description of the project
@type description: C{str}
@ivar sources: The sources used by this project
@type sources: L{SourceList}
@ivar timeline: The timeline
@type timeline: L{Timeline}
@ivar pipeline: The timeline's pipeline
@type pipeline: L{Pipeline}
@ivar factory: The timeline factory
@type factory: L{TimelineSourceFactory}
@ivar format: The format under which the project is currently stored.
@type format: L{FormatterClass}
@ivar loaded: Whether the project is fully loaded or not.
@type loaded: C{bool}
Signals:
- C{loaded} : The project is now fully loaded.
"""
__signals__ = {
"settings-changed" : ['old', 'new'],
}
def __init__(self, name="", uri=None, **kwargs):
"""
name : the name of the project
uri : the uri of the project
"""
Loggable.__init__(self)
self.log("name:%s, uri:%s", name, uri)
self.name = name
self.settings = None
self.description = ""
self.uri = uri
self.urichanged = False
self.format = None
self.sources = SourceList()
self.sources.connect("source-added", self._sourceAddedCb)
self.sources.connect("source-removed", self._sourceRemovedCb)
self._dirty = False
self.timeline = Timeline()
self.factory = TimelineSourceFactory(self.timeline)
self.pipeline = Pipeline()
self.view_action = ViewAction()
self.view_action.addProducers(self.factory)
self.seeker = Seeker(80)
self.getCapsFromSettings()
def getCapsFromSettings(self):
settings = self.getSettings()
#formatstr = "video/x-raw-rgb,width=(int)%d,height=(int)%d;"\
# "video/x-raw-yuv,width=(int)%d,height=(int)%d"
#capstr = formatstr % (
# settings.videowidth,
# settings.videoheight,
# settings.videowidth,
# settings.videoheight)
#self._videocaps = gst.Caps(capstr)
self._videocaps = settings.getVideoCaps()
def release(self):
self.pipeline.release()
self.pipeline = None
#{ Settings methods
def getSettings(self):
"""
return the currently configured settings.
If no setting have been explicitely set, some smart settings will be
chosen.
"""
self.debug("self.settings %s", self.settings)
return self.settings or self.getAutoSettings()
def setSettings(self, settings):
"""
Sets the given settings as the project's settings.
If settings is None, the current settings will be unset
"""
self.log("Setting %s as the project's settings", settings)
oldsettings = self.settings
self.settings = settings
self._projectSettingsChanged()
self.emit('settings-changed', oldsettings, settings)
def unsetSettings(self, unused_settings):
""" Remove the currently configured settings."""
self.setSettings(None)
def getAutoSettings(self):
"""
Computes and returns smart settings for the project.
If the project only has one source, it will be that source's settings.
If it has more than one, it will return the largest setting that suits
all contained sources.
"""
settings = ExportSettings()
if not self.timeline:
self.warning("project doesn't have a timeline, returning default settings")
return settings
# FIXME: this is ugly, but rendering for now assumes at most one audio
# and one video tracks
have_audio = have_video = False
for track in self.timeline.tracks:
if isinstance(track.stream, VideoStream) and track.duration != 0:
have_video = True
elif isinstance(track.stream, AudioStream) and track.duration != 0:
have_audio = True
if not have_audio:
settings.aencoder = None
if not have_video:
settings.vencoder = None
return settings
#}
#{ Save and Load features
def save(self, location=None, overwrite=False):
"""
Save the project to the given location.
@param location: The location to write to. If not specified, the
current project location will be used (if set).
@type location: C{URI}
@param overwrite: Whether to overwrite existing location.
@type overwrite: C{bool}
"""
# import here to break circular import
from pitivi.formatters.format import save_project
from pitivi.formatters.base import FormatterError
self.log("saving...")
location = location or self.uri
if location == None:
raise FormatterError("Location unknown")
save_project(self, location or self.uri, self.format,
overwrite)
self.uri = location
def setModificationState(self, state):
self._dirty = state
def hasUnsavedModifications(self):
return self._dirty
def _projectSettingsChanged(self):
self.getCapsFromSettings()
for fact in self.sources.getSources():
fact.setFilterCaps(self._videocaps)
if self.pipeline.getState() != gst.STATE_NULL:
self.pipeline.stop()
self.pipeline.pause()
def _sourceAddedCb(self, sourcelist, factory):
factory.setFilterCaps(self._videocaps)
def _sourceRemovedCb(self, sourclist, uri, factory):
self.timeline.removeFactory(factory)
class Project(Signallable, Loggable):
"""The base class for PiTiVi projects
@ivar name: The name of the project
@type name: C{str}
@ivar description: A description of the project
@type description: C{str}
@ivar sources: The sources used by this project
@type sources: L{SourceList}
@ivar timeline: The timeline
@type timeline: L{Timeline}
@ivar pipeline: The timeline's pipeline
@type pipeline: L{Pipeline}
@ivar factory: The timeline factory
@type factory: L{TimelineSourceFactory}
@ivar format: The format under which the project is currently stored.
@type format: L{FormatterClass}
@ivar loaded: Whether the project is fully loaded or not.
@type loaded: C{bool}
Signals:
- C{loaded} : The project is now fully loaded.
"""
__signals__ = {
"settings-changed": ['old', 'new'],
"project-changed": [],
}
def __init__(self, name="", uri=None, **kwargs):
"""
@param name: the name of the project
@param uri: the uri of the project
"""
Loggable.__init__(self)
self.log("name:%s, uri:%s", name, uri)
self.name = name
self.settings = None
self.description = ""
self.uri = uri
self.urichanged = False
self.format = None
self.sources = SourceList()
self.sources.connect("source-added", self._sourceAddedCb)
self.sources.connect("source-removed", self._sourceRemovedCb)
self._dirty = False
self.timeline = Timeline()
self.factory = TimelineSourceFactory(self.timeline)
self.pipeline = Pipeline()
self.view_action = ViewAction()
self.view_action.addProducers(self.factory)
self.seeker = Seeker(80)
self.settings = ExportSettings()
self._videocaps = self.settings.getVideoCaps()
def release(self):
self.pipeline.release()
self.pipeline = None
#{ Settings methods
def getSettings(self):
"""
return the currently configured settings.
"""
self.debug("self.settings %s", self.settings)
return self.settings
def setSettings(self, settings):
"""
Sets the given settings as the project's settings.
@param settings: The new settings for the project.
@type settings: ExportSettings
"""
assert settings
self.log("Setting %s as the project's settings", settings)
oldsettings = self.settings
self.settings = settings
self._projectSettingsChanged()
self.emit('settings-changed', oldsettings, settings)
#}
#{ Save and Load features
def setModificationState(self, state):
self._dirty = state
if state:
self.emit('project-changed')
def hasUnsavedModifications(self):
return self._dirty
def _projectSettingsChanged(self):
settings = self.getSettings()
self._videocaps = settings.getVideoCaps()
if self.timeline:
self.timeline.updateVideoCaps(self._videocaps)
for fact in self.sources.getSources():
fact.setFilterCaps(self._videocaps)
if self.pipeline.getState() != gst.STATE_NULL:
self.pipeline.stop()
self.pipeline.pause()
def _sourceAddedCb(self, sourcelist, factory):
factory.setFilterCaps(self._videocaps)
def _sourceRemovedCb(self, sourclist, uri, factory):
self.timeline.removeFactory(factory)
class Project(Signallable, Loggable):
"""The base class for PiTiVi projects
@ivar name: The name of the project
@type name: C{str}
@ivar description: A description of the project
@type description: C{str}
@ivar sources: The sources used by this project
@type sources: L{SourceList}
@ivar timeline: The timeline
@type timeline: L{Timeline}
@ivar pipeline: The timeline's pipeline
@type pipeline: L{Pipeline}
@ivar factory: The timeline factory
@type factory: L{TimelineSourceFactory}
@ivar format: The format under which the project is currently stored.
@type format: L{FormatterClass}
@ivar loaded: Whether the project is fully loaded or not.
@type loaded: C{bool}
Signals:
- C{loaded} : The project is now fully loaded.
"""
__signals__ = {
"settings-changed": ['old', 'new'],
"project-changed": [],
}
def __init__(self, name="", uri=None, **kwargs):
"""
name : the name of the project
uri : the uri of the project
"""
Loggable.__init__(self)
self.log("name:%s, uri:%s", name, uri)
self.name = name
self.settings = None
self.description = ""
self.uri = uri
self.urichanged = False
self.format = None
self.sources = SourceList()
self.sources.connect("source-added", self._sourceAddedCb)
self.sources.connect("source-removed", self._sourceRemovedCb)
self._dirty = False
self.timeline = Timeline()
self.factory = TimelineSourceFactory(self.timeline)
self.pipeline = Pipeline()
self.view_action = ViewAction()
self.view_action.addProducers(self.factory)
self.seeker = Seeker(80)
settings = self.getSettings()
self._videocaps = settings.getVideoCaps()
def release(self):
self.pipeline.release()
self.pipeline = None
#{ Settings methods
def getSettings(self):
"""
return the currently configured settings.
If no setting have been explicitely set, some smart settings will be
chosen.
"""
self.debug("self.settings %s", self.settings)
return self.settings or self.getAutoSettings()
def setSettings(self, settings):
"""
Sets the given settings as the project's settings.
If settings is None, the current settings will be unset
"""
self.log("Setting %s as the project's settings", settings)
oldsettings = self.settings
self.settings = settings
self._projectSettingsChanged()
self.emit('settings-changed', oldsettings, settings)
def getAutoSettings(self):
"""
Computes and returns smart settings for the project.
If the project only has one source, it will be that source's settings.
If it has more than one, it will return the largest setting that suits
all contained sources.
"""
settings = ExportSettings()
if not self.timeline:
self.warning(
"project doesn't have a timeline, returning default settings")
return settings
# FIXME: this is ugly, but rendering for now assumes at most one audio
# and one video tracks
have_audio = have_video = False
for track in self.timeline.tracks:
if isinstance(track.stream, VideoStream) and track.duration != 0:
have_video = True
elif isinstance(track.stream, AudioStream) and track.duration != 0:
have_audio = True
if not have_audio:
settings.aencoder = None
if not have_video:
settings.vencoder = None
return settings
#}
#{ Save and Load features
def save(self, location=None, overwrite=False):
"""
Save the project to the given location.
@param location: The location to write to. If not specified, the
current project location will be used (if set).
@type location: C{URI}
@param overwrite: Whether to overwrite existing location.
@type overwrite: C{bool}
"""
# import here to break circular import
from pitivi.formatters.format import save_project
from pitivi.formatters.base import FormatterError
self.log("saving...")
location = location or self.uri
if location == None:
raise FormatterError("Location unknown")
save_project(self, location or self.uri, self.format, overwrite)
self.uri = location
def setModificationState(self, state):
self._dirty = state
if state:
self.emit('project-changed')
def hasUnsavedModifications(self):
return self._dirty
def _projectSettingsChanged(self):
settings = self.getSettings()
self._videocaps = settings.getVideoCaps()
for fact in self.sources.getSources():
fact.setFilterCaps(self._videocaps)
if self.pipeline.getState() != gst.STATE_NULL:
self.pipeline.stop()
self.pipeline.pause()
def _sourceAddedCb(self, sourcelist, factory):
factory.setFilterCaps(self._videocaps)
def _sourceRemovedCb(self, sourclist, uri, factory):
self.timeline.removeFactory(factory)
