from __future__ import division

from fractions import Fraction

import av

from .common import TestCase, fate_suite


try:
    long
except NameError:
    long = int


class TestAudioProbe(TestCase):
    def setUp(self):
        self.file = av.open(fate_suite('aac/latm_stereo_to_51.ts'))

    def test_container_probing(self):
        self.assertEqual(str(self.file.format), "<av.ContainerFormat 'mpegts'>")
        self.assertEqual(self.file.format.name, 'mpegts')
        self.assertEqual(self.file.format.long_name, "MPEG-TS (MPEG-2 Transport Stream)")
        self.assertEqual(self.file.size, 207740)

        # This is a little odd, but on OS X with FFmpeg we get a different value.
        self.assertIn(self.file.bit_rate, (269558, 270494))

        self.assertEqual(len(self.file.streams), 1)
        self.assertEqual(self.file.start_time, long(1400000))
        self.assertEqual(self.file.metadata, {})

    def test_stream_probing(self):
        stream = self.file.streams[0]

        # actual stream properties
        self.assertEqual(stream.average_rate, None)
        self.assertEqual(stream.r_rate, None)
        self.assertEqual(stream.duration, 554880)
        self.assertEqual(stream.frames, 0)
        self.assertEqual(stream.id, 256)
        self.assertEqual(stream.index, 0)
        self.assertEqual(stream.language, 'eng')
        self.assertEqual(stream.metadata, {
            'language': 'eng',
        })
        self.assertEqual(stream.profile, 'LC')
        self.assertEqual(stream.start_time, 126000)
        self.assertEqual(stream.time_base, Fraction(1, 90000))
        self.assertEqual(stream.type, 'audio')

        # codec properties
        self.assertEqual(stream.name, 'aac_latm')
        self.assertEqual(stream.long_name, 'AAC LATM (Advanced Audio Coding LATM syntax)')

        # codec context properties
        self.assertEqual(stream.bit_rate, None)
        self.assertEqual(stream.channels, 2)
        self.assertEqual(stream.format.bits, 32)
        self.assertEqual(stream.format.name, 'fltp')
        self.assertEqual(stream.layout.name, 'stereo')
        self.assertEqual(stream.max_bit_rate, None)
        self.assertEqual(stream.rate, 48000)


class TestDataProbe(TestCase):
    def setUp(self):
        self.file = av.open(fate_suite('mxf/track_01_v02.mxf'))

    def test_container_probing(self):
        self.assertEqual(str(self.file.format), "<av.ContainerFormat 'mxf'>")
        self.assertEqual(self.file.format.name, 'mxf')
        self.assertEqual(self.file.format.long_name, 'MXF (Material eXchange Format)')
        self.assertEqual(self.file.size, 1453153)

        self.assertEqual(self.file.bit_rate, 8 * self.file.size * av.time_base // self.file.duration)
        self.assertEqual(self.file.duration, 417083)
        self.assertEqual(len(self.file.streams), 4)
        self.assertEqual(self.file.metadata, {
            'application_platform': 'AAFSDK (MacOS X)',
            'comment_Comments': 'example comment',
            'comment_UNC Path': '/Users/mark/Desktop/dnxhr_tracknames_export.aaf',
            'company_name': 'Avid Technology, Inc.',
            'generation_uid': 'b6bcfcab-70ff-7331-c592-233869de11d2',
            'material_package_name': 'Example.new.04',
            'material_package_umid': '0x060A2B340101010101010F001300000057E19D16BA8202DB060E2B347F7F2A80',
            'modification_date': '2016-09-20T20:33:26.000000Z',
            'product_name': 'Avid Media Composer 8.6.3.43955',
            'product_uid': 'acfbf03a-4f42-a231-d0b7-c06ecd3d4ad7',
            'product_version': 'Unknown version',
            'project_name': 'UHD',
            'uid': '4482d537-4203-ea40-9e4e-08a22900dd39',
        })

    def test_stream_probing(self):
        stream = self.file.streams[0]

        # actual stream properties
        self.assertEqual(stream.average_rate, None)
        self.assertEqual(stream.r_rate, None)
        self.assertEqual(stream.duration, 37537)
        self.assertEqual(stream.frames, 0)
        self.assertEqual(stream.id, 1)
        self.assertEqual(stream.index, 0)
        self.assertEqual(stream.language, None)
        self.assertEqual(stream.metadata, {
            'data_type': 'video',
            'file_package_umid': '0x060A2B340101010101010F001300000057E19D16BA8302DB060E2B347F7F2A80',
            'track_name': 'Base',
        })
        self.assertEqual(stream.profile, None)
        self.assertEqual(stream.start_time, 0)
        self.assertEqual(stream.time_base, Fraction(1, 90000))
        self.assertEqual(stream.type, 'data')

        # codec properties
        self.assertEqual(stream.name, None)
        self.assertEqual(stream.long_name, None)


class TestSubtitleProbe(TestCase):
    def setUp(self):
        self.file = av.open(fate_suite('sub/MovText_capability_tester.mp4'))

    def test_container_probing(self):
        self.assertEqual(str(self.file.format), "<av.ContainerFormat 'mov,mp4,m4a,3gp,3g2,mj2'>")
        self.assertEqual(self.file.format.name, 'mov,mp4,m4a,3gp,3g2,mj2')
        self.assertEqual(self.file.format.long_name, 'QuickTime / MOV')
        self.assertEqual(self.file.size, 825)

        self.assertEqual(self.file.bit_rate, 8 * self.file.size * av.time_base // self.file.duration)
        self.assertEqual(self.file.duration, 8140000)
        self.assertEqual(len(self.file.streams), 1)
        self.assertEqual(self.file.metadata, {
            'compatible_brands': 'isom',
            'creation_time': '2012-07-04T05:10:41.000000Z',
            'major_brand': 'isom',
            'minor_version': '1',
        })

    def test_stream_probing(self):
        stream = self.file.streams[0]

        # actual stream properties
        self.assertEqual(stream.average_rate, None)
        self.assertEqual(stream.duration, 8140)
        self.assertEqual(stream.frames, 6)
        self.assertEqual(stream.id, 1)
        self.assertEqual(stream.index, 0)
        self.assertEqual(stream.language, 'und')
        self.assertEqual(stream.metadata, {
            'creation_time': '2012-07-04T05:10:41.000000Z',
            'handler_name': 'reference.srt - Imported with GPAC 0.4.6-DEV-rev4019',
            'language': 'und'
        })
        self.assertEqual(stream.profile, None)
        self.assertEqual(stream.start_time, None)
        self.assertEqual(stream.time_base, Fraction(1, 1000))
        self.assertEqual(stream.type, 'subtitle')

        # codec properties
        self.assertEqual(stream.name, 'mov_text')
        self.assertEqual(stream.long_name, '3GPP Timed Text subtitle')


class TestVideoProbe(TestCase):
    def setUp(self):
        self.file = av.open(fate_suite('mpeg2/mpeg2_field_encoding.ts'))

    def test_container_probing(self):
        self.assertEqual(str(self.file.format), "<av.ContainerFormat 'mpegts'>")
        self.assertEqual(self.file.format.name, 'mpegts')
        self.assertEqual(self.file.format.long_name, "MPEG-TS (MPEG-2 Transport Stream)")
        self.assertEqual(self.file.size, 800000)

        # This is a little odd, but on OS X with FFmpeg we get a different value.
        self.assertIn(self.file.duration, (1620000, 1580000))

        self.assertEqual(self.file.bit_rate, 8 * self.file.size * av.time_base // self.file.duration)
        self.assertEqual(len(self.file.streams), 1)
        self.assertEqual(self.file.start_time, long(22953408322))
        self.assertEqual(self.file.metadata, {})

    def test_stream_probing(self):
        stream = self.file.streams[0]

        # actual stream properties
        self.assertEqual(stream.average_rate, Fraction(25, 1))
        self.assertEqual(stream.duration, 145800)
        self.assertEqual(stream.frames, 0)
        self.assertEqual(stream.id, 4131)
        self.assertEqual(stream.index, 0)
        self.assertEqual(stream.language, None)
        self.assertEqual(stream.metadata, {})
        self.assertEqual(stream.profile, 'Simple')
        self.assertEqual(stream.start_time, 2065806749)
        self.assertEqual(stream.time_base, Fraction(1, 90000))
        self.assertEqual(stream.type, 'video')

        # codec properties
        self.assertEqual(stream.long_name, 'MPEG-2 video')
        self.assertEqual(stream.name, 'mpeg2video')

        # codec context properties
        self.assertEqual(stream.bit_rate, 3364800)
        self.assertEqual(stream.display_aspect_ratio, Fraction(4, 3))
        self.assertEqual(stream.format.name, 'yuv420p')
        self.assertFalse(stream.has_b_frames)
        self.assertEqual(stream.gop_size, 12)
        self.assertEqual(stream.height, 576)
        self.assertEqual(stream.max_bit_rate, None)
        self.assertEqual(stream.sample_aspect_ratio, Fraction(16, 15))
        self.assertEqual(stream.width, 720)

        # For some reason, these behave differently on OS X (@mikeboers) and
        # Ubuntu (Travis). We think it is FFmpeg, but haven't been able to
        # confirm.
        self.assertIn(stream.coded_width, (720, 0))
        self.assertIn(stream.coded_height, (576, 0))
