mmif-python 1.0.5__tar.gz → 1.0.7__tar.gz

{mmif-python-1.0.5/mmif_python.egg-info → mmif-python-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mmif-python
-Version: 1.0.5
+Version: 1.0.7
 Summary: Python implementation of MultiMedia Interchange Format specification. (https://mmif.clams.ai)
 Home-page: https://mmif.clams.ai
 Author: Brandeis Lab for Linguistics and Computation

mmif-python-1.0.7/VERSION

@@ -0,0 +1 @@
+1.0.7

{mmif-python-1.0.5 → mmif-python-1.0.7}/mmif/utils/video_document_helper.py

@@ -1,20 +1,24 @@
+import importlib
+import warnings
 from typing import List, Union, Tuple
 
-import numpy as np
-from PIL import Image
-
+import mmif
 from mmif import Annotation, Document, Mmif
 from mmif.vocabulary import DocumentTypes
 
-try:
-    import cv2
-    import ffmpeg
-    import PIL
-except ImportError as e:
-    raise ImportError(
-        f"Optional package {e.name} not found. You might want to install Computer-Vision dependencies by running `pip install mmif-python[cv]`")
+for cv_dep in ('cv2', 'ffmpeg', 'PIL'):
+    try:
+        importlib.__import__(cv_dep)
+    except ImportError as e:
+        warnings.warn(f"Optional package \"{e.name}\" is not found. "
+                      f"You might want to install Computer-Vision dependencies "
+                      f"by running `pip install mmif-python[cv]=={mmif.__version__}`")
+
 
 FPS_DOCPROP_KEY = 'fps'
+DURATION_DOCPROP_KEY = 'duration'
+DURATIONUNIT_DOCPROP_KEY = 'durationTimeUnit'
+                           
 UNIT_NORMALIZATION = {
     'm': 'millisecond',
     'ms': 'millisecond',
@@ -33,12 +37,15 @@ UNIT_NORMALIZATION = {
 }
 
 
-def capture(vd: Document) -> cv2.VideoCapture:
+def capture(vd: Document):
+    import cv2  # pytype: disable=import-error
     if vd is None or vd.at_type != DocumentTypes.VideoDocument:
         raise ValueError(f'The document does not exist.')
 
     v = cv2.VideoCapture(vd.location_path())
     vd.add_property(FPS_DOCPROP_KEY, v.get(cv2.CAP_PROP_FPS))
+    vd.add_property(DURATION_DOCPROP_KEY, v.get(cv2.CAP_PROP_FRAME_COUNT))
+    vd.add_property(DURATIONUNIT_DOCPROP_KEY, 'frames')
     return v
 
 
@@ -55,7 +62,7 @@ def get_framerate(vd: Document) -> float:
     return vd.get_property(FPS_DOCPROP_KEY)
 
 
-def extract_frames_as_images(vd: Document, framenums: List[int], as_PIL: bool = False) -> List[Union[np.ndarray, PIL.Image.Image]]:
+def extract_frames_as_images(vd: Document, framenums: List[int], as_PIL: bool = False):
     """
     Extracts frames from a video document as a list of numpy arrays.
     Use `sample_frames` function in this module to get the list of frame numbers first. 
@@ -65,6 +72,9 @@ def extract_frames_as_images(vd: Document, framenums: List[int], as_PIL: bool =
     :param as_PIL: use PIL.Image instead of numpy.ndarray
     :return: frames as a list of numpy arrays or PIL.Image objects
     """
+    import cv2  # pytype: disable=import-error
+    if as_PIL:
+        from PIL import Image
     frames = []
     video = capture(vd)
     for framenum in framenums:
@@ -77,15 +87,19 @@ def extract_frames_as_images(vd: Document, framenums: List[int], as_PIL: bool =
     return frames
 
 
-def extract_mid_frame(mmif: Mmif, tf: Annotation, as_PIL: bool = False) -> Union[np.ndarray, PIL.Image.Image]:
+def get_mid_framenum(mmif: Mmif, tf: Annotation):
     """
     Extracts the middle frame from a video document
     """
     timeunit = get_annotation_property(mmif, tf, 'timeUnit')
     vd = mmif[get_annotation_property(mmif, tf, 'document')]
     fps = get_framerate(vd)
-    midframe = sum(convert(float(tf.get_property(timepoint_propkey)), timeunit, 'frame', fps) for timepoint_propkey in ('start', 'end')) // 2
-    return extract_frames_as_images(vd, [midframe], as_PIL=as_PIL)[0]
+    return sum(convert(float(tf.get_property(timepoint_propkey)), timeunit, 'frame', fps) for timepoint_propkey in ('start', 'end')) // 2
+
+
+def extract_mid_frame(mmif: Mmif, tf: Annotation, as_PIL: bool = False):
+    vd = mmif[get_annotation_property(mmif, tf, 'document')]
+    return extract_frames_as_images(vd, [get_mid_framenum(mmif, tf)], as_PIL=as_PIL)[0]
 
 
 def sample_frames(start_frame: int, end_frame: int, sample_ratio: int = 1) -> List[int]:

mmif-python-1.0.7/mmif/ver/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "1.0.7"
+__specver__ = "1.0.0"

{mmif-python-1.0.5 → mmif-python-1.0.7/mmif_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mmif-python
-Version: 1.0.5
+Version: 1.0.7
 Summary: Python implementation of MultiMedia Interchange Format specification. (https://mmif.clams.ai)
 Home-page: https://mmif.clams.ai
 Author: Brandeis Lab for Linguistics and Computation

mmif-python-1.0.5/VERSION

@@ -1 +0,0 @@
-1.0.5

mmif-python-1.0.5/mmif/ver/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "1.0.5"
-__specver__ = "1.0.0"