# ##### BEGIN GPL LICENSE BLOCK #####
#
#  This program is free software; you can redistribute it and/or
#  modify it under the terms of the GNU General Public License
#  as published by the Free Software Foundation; either version 2
#  of the License, or (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software Foundation,
#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ##### END GPL LICENSE BLOCK #####

# <pep8 compliant>
import bpy


class CLIP_OT_apply_follow_track(bpy.types.Operator):
    bl_idname = "clip.apply_follow_track"
    bl_label = "Apply Follow Track"
    bl_options = {'UNDO', 'REGISTER'}

    @classmethod
    def poll(cls, context):
        if context.space_data.type != 'CLIP_EDITOR':
            return False

        sc = context.space_data
        clip = sc.clip

        return clip and clip.tracking.active_track and context.active_object

    def execute(self, context):
        ob = context.active_object
        sc = context.space_data
        clip = sc.clip
        track = clip.tracking.active_track
        constraint = None

        for con in ob.constraints:
            if con.type == 'FOLLOW_TRACK':
                constraint = con
                break

        if constraint is None:
            constraint = ob.constraints.new(type='FOLLOW_TRACK')

        constraint.clip = sc.clip
        constraint.track = track.name
        constraint.reference = 'TRACK'

        return {'FINISHED'}


class CLIP_OT_bundles_to_mesh(bpy.types.Operator):
    bl_idname = "clip.bundles_to_mesh"
    bl_label = "Bundles to Mesh"
    bl_options = {'UNDO', 'REGISTER'}

    @classmethod
    def poll(cls, context):
        if context.space_data.type != 'CLIP_EDITOR':
            return False

        sc = context.space_data
        clip = sc.clip

        return clip

    def execute(self, context):
        sc = context.space_data
        clip = sc.clip

        mesh = bpy.data.meshes.new(name="Bundles")
        for track in clip.tracking.tracks:
            if track.has_bundle:
                mesh.vertices.add(1)
                mesh.vertices[-1].co = track.bundle

        ob = bpy.data.objects.new(name="Bundles", object_data=mesh)

        bpy.context.scene.objects.link(ob)

        return {'FINISHED'}


class CLIP_HT_header(bpy.types.Header):
    bl_space_type = 'CLIP_EDITOR'

    def draw(self, context):
        layout = self.layout

        sc = context.space_data
        clip = sc.clip

        row = layout.row(align=True)
        row.template_header()

        if context.area.show_menus:
            sub = row.row(align=True)
            sub.menu("CLIP_MT_view")
            sub.menu("CLIP_MT_clip")

            if clip:
                sub.menu("CLIP_MT_select")
                sub.menu("CLIP_MT_track")

        layout.template_ID(sc, "clip")
        layout.template_running_jobs()

        if clip:
            r = clip.tracking.reconstruction

            if r.is_reconstructed:
                layout.label(text="Average solve error: %.4f"  % (r.average_error))


class CLIP_PT_tools(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'TOOLS'
    bl_label = "Tools"

    def draw(self, context):
        layout = self.layout
        clip = context.space_data.clip

        if clip:
            settings = clip.tracking.settings

            col = layout.column(align=True)
            col.label(text="Transform:")
            col.operator("transform.translate")
            col.operator("transform.resize")

            col = layout.column(align=True)
            col.label(text="Marker:")
            col.operator("clip.add_marker_move")
            col.operator("clip.delete_track")
            col.operator("clip.delete_marker")

            col = layout.column(align=True)
            col.label(text="2D tracking:")
            row = col.row(align=True)

            op = row.operator("clip.track_markers", text="", icon='FRAME_PREV')
            op.backwards = True
            op = row.operator("clip.track_markers", text="", \
                 icon='PLAY_REVERSE')
            op.backwards = True
            op.sequence = True
            op = row.operator("clip.track_markers", text="", icon='PLAY')
            op.sequence = True
            row.operator("clip.track_markers", text="", icon='FRAME_NEXT')

            col = layout.column(align=True)
            op = col.operator("clip.clear_track_path", \
                 text="Clear Remained Path")
            op.action = 'REMAINED'

            op = col.operator("clip.clear_track_path", text="Clear Path Up To")
            op.action = 'UPTO'

            op = col.operator("clip.clear_track_path", text="Clear Track Path")
            op.action = 'ALL'

            layout.operator("clip.join_tracks")
            layout.operator("clip.detect_features")
            layout.operator("clip.apply_follow_track")

            col = layout.column(align=True)
            col.label(text="Reconstruction:")

            col.prop(settings, "keyframe1")
            col.prop(settings, "keyframe2")

            col = layout.column(align=True)
            col.operator("clip.solve_camera")
            col.operator("clip.clear_reconstruction")

            col = layout.column(align=True)
            col.operator("clip.bundles_to_mesh")

            col = layout.column(align=True)
            col.label(text="Scene Orientation:")
            col.operator("clip.set_floor")
            col.operator("clip.set_origin")

            row = col.row()
            row.operator("clip.set_axis", text="Set X Axis").axis = 'X'
            row.operator("clip.set_axis", text="Set Y Axis").axis = 'Y'

            col = layout.column()
            col.prop(settings, "distance")
            col.operator("clip.set_scale")
        else:
            layout.operator('clip.open')


class CLIP_PT_track(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "Track"

    @classmethod
    def poll(cls, context):
        sc = context.space_data
        clip = sc.clip

        return clip and clip.tracking.active_track

    def draw(self, context):
        layout = self.layout
        sc = context.space_data
        clip = context.space_data.clip
        act_track = clip.tracking.active_track

        layout.prop(act_track, "name")
        layout.prop(act_track, "locked")

        row = layout.row(align=True)
        label = bpy.types.CLIP_MT_track_color_presets.bl_label
        row.menu('CLIP_MT_track_color_presets', text=label)
        row.menu('CLIP_MT_track_color_specials', text="", icon="DOWNARROW_HLT")
        row.operator("clip.track_color_preset_add", text="", icon="ZOOMIN")
        op = row.operator("clip.track_color_preset_add", \
            text="", icon="ZOOMOUT")
        op.remove_active = True

        row = layout.row()
        row.prop(act_track, "use_custom_color")
        if act_track.use_custom_color:
            row.prop(act_track, "color", text="")

        layout.template_track(sc, "scopes")

        row = layout.row()
        row.prop(act_track, "use_red_channel", text="Red")
        row.prop(act_track, "use_green_channel", text="Green")
        row.prop(act_track, "use_blue_channel", text="Blue")


class CLIP_PT_track_settings(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "Tracking Settings"
    bl_options = {'DEFAULT_CLOSED'}

    @classmethod
    def poll(cls, context):
        sc = context.space_data

        return sc.clip

    def draw(self, context):
        layout = self.layout
        clip = context.space_data.clip
        settings = clip.tracking.settings

        layout.label(text="2D tracking:")
        layout.prop(settings, "speed")
        layout.prop(settings, "use_frames_limit")

        row = layout.row()
        row.active = settings.use_frames_limit
        row.prop(settings, "frames_limit")


class CLIP_PT_tracking_camera(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "Camera Data"
    bl_options = {'DEFAULT_CLOSED'}

    @classmethod
    def poll(cls, context):
        sc = context.space_data

        return sc.clip

    def draw(self, context):
        layout = self.layout

        sc = context.space_data
        clip = sc.clip

        row = layout.row(align=True)
        label = bpy.types.CLIP_MT_camera_presets.bl_label
        row.menu('CLIP_MT_camera_presets', text=label)
        row.operator("clip.camera_preset_add", text="", icon="ZOOMIN")
        op = row.operator("clip.camera_preset_add", text="", icon="ZOOMOUT")
        op.remove_active = True

        layout.label(text="Sensor:")
        row = layout.row(align=True)
        row.prop(clip.tracking.camera, "sensor_width", text="X")
        row.prop(clip.tracking.camera, "sensor_height", text="Y")

        row = layout.row(align=True)
        sub = row.split(percentage=0.65)
        sub.prop(clip.tracking.camera, "focal_length")
        sub.prop(clip.tracking.camera, "units", text="")

        col = layout.column()
        col.label(text="Principal Point")
        row = col.row()
        row.prop(clip.tracking.camera, "principal", text="")
        col.operator("clip.set_center_principal", text="Center")

        col = layout.column(align=True)
        col.label(text="Undistortion:")
        col.prop(clip.tracking.camera, "k1")
        col.prop(clip.tracking.camera, "k2")
        col.prop(clip.tracking.camera, "k3")


class CLIP_PT_display(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "Display"
    bl_options = {'DEFAULT_CLOSED'}

    def draw(self, context):
        layout = self.layout
        sc = context.space_data

        row = layout.row()
        row.prop(sc, "show_marker_pattern", text="Pattern")
        row.prop(sc, "show_marker_search", text="Search")

        row = layout.row()
        row.prop(sc, "show_track_path", text="Path")
        sub = row.column()
        sub.active = sc.show_track_path
        sub.prop(sc, "path_length", text="Length")

        row = layout.row()
        row.prop(sc, "show_disabled", text="Disabled")
        row.prop(sc, "show_bundles", text="Bundles")

        row = layout.row()
        row.prop(sc, "show_names", text="Names")
        row.prop(sc, "show_grid", text="Grid")

        row = layout.row()
        row.prop(sc, "show_tiny_markers", text="Tiny Markers")
        row.prop(sc, "show_stable", text="Stable")

        layout.prop(sc, "lock_selection")
        layout.prop(sc, "use_mute_footage")

        clip = sc.clip
        if clip:
            layout.label(text="Display Aspect:")
            layout.prop(clip, "display_aspect", text="")


class CLIP_PT_stabilization(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "2D Stabilization"
    bl_options = {'DEFAULT_CLOSED'}

    @classmethod
    def poll(cls, context):
        sc = context.space_data

        return sc.clip

    def draw_header(self, context):
        sc = context.space_data
        tracking = sc.clip.tracking
        stab = tracking.stabilization

        self.layout.prop(stab, "use_2d_stabilization", text="")

    def draw(self, context):
        layout = self.layout
        sc = context.space_data
        tracking = sc.clip.tracking
        stab = tracking.stabilization

        layout.active = stab.use_2d_stabilization


        row = layout.row()
        row.template_list(stab, "tracks", stab, "active_track_index", rows=3)

        sub = row.column(align=True)
        sub.operator("clip.stabilize_2d_add", icon='ZOOMIN', text="")
        sub.operator("clip.stabilize_2d_remove", icon='ZOOMOUT', text="")
        sub.menu('CLIP_MT_stabilize_2d_specials', text="", icon="DOWNARROW_HLT")

        layout.prop(stab, "influence_location")

        layout.prop(stab, "use_autoscale")
        row = layout.row()
        row.active = stab.use_autoscale
        row.prop(stab, "influence_scale")


class CLIP_PT_footage(bpy.types.Panel):
    bl_space_type = 'CLIP_EDITOR'
    bl_region_type = 'UI'
    bl_label = "Footage Settings"
    bl_options = {'DEFAULT_CLOSED'}

    @classmethod
    def poll(cls, context):
        sc = context.space_data

        return sc.clip

    def draw(self, context):
        layout = self.layout

        sc = context.space_data
        clip = sc.clip

        if clip:
            layout.template_movieclip(sc, "clip", sc.clip_user, compact=True)
        else:
            layout.operator("clip.open", icon='FILESEL')


class CLIP_MT_view(bpy.types.Menu):
    bl_label = "View"

    def draw(self, context):
        layout = self.layout

        layout.operator("clip.properties", icon='MENU_PANEL')
        layout.operator("clip.tools", icon='MENU_PANEL')
        layout.separator()

        layout.operator("clip.view_selected")
        layout.operator("clip.view_all")

        layout.separator()
        layout.operator("clip.view_zoom_in")
        layout.operator("clip.view_zoom_out")

        layout.separator()

        ratios = [[1, 8], [1, 4], [1, 2], [1, 1], [2, 1], [4, 1], [8, 1]]

        for a, b in ratios:
            text = "Zoom %d:%d" % (a, b)
            layout.operator("clip.view_zoom_ratio", text=text).ratio = a / b

        layout.separator()
        layout.operator("screen.area_dupli")
        layout.operator("screen.screen_full_area")


class CLIP_MT_clip(bpy.types.Menu):
    bl_label = "Clip"

    def draw(self, context):
        layout = self.layout

        sc = context.space_data
        clip = sc.clip

        if clip:
            layout.operator("clip.reload")

        layout.operator("clip.open")


class CLIP_MT_track(bpy.types.Menu):
    bl_label = "Track"

    def draw(self, context):
        layout = self.layout

        layout.operator("clip.set_origin")

        layout.separator()
        layout.operator("clip.clear_reconstruction")
        layout.operator("clip.solve_camera")

        layout.separator()
        op = layout.operator("clip.clear_track_path", \
             text="Clear Remained Path")
        op.action = 'REMAINED'

        op = layout.operator("clip.clear_track_path", \
             text="Clear Path Up To")
        op.action = 'UPTO'

        op = layout.operator("clip.clear_track_path", \
             text="Clear Track Path")
        op.action = 'ALL'

        layout.separator()
        op = layout.operator("clip.track_markers", \
            text="Track Frame Backwards")
        op.backwards = True

        op = layout.operator("clip.track_markers", text="Track Backwards")
        op.backwards = True
        op.sequence = True

        op = layout.operator("clip.track_markers", text="Track Forwards")
        op.sequence = True
        layout.operator("clip.track_markers", text="Track Frame Forwards")

        layout.separator()
        layout.operator("clip.delete_track")
        layout.operator("clip.delete_marker")

        layout.separator()
        layout.operator("clip.add_marker_move")

        layout.separator()
        layout.menu("CLIP_MT_track_visibility")
        layout.menu("CLIP_MT_track_transform")


class CLIP_MT_track_visibility(bpy.types.Menu):
    bl_label = "Show/Hide"

    def draw(self, context):
        layout = self.layout

        layout.operator("clip.hide_tracks_clear", text="Show Hidden")
        layout.operator("clip.hide_tracks", text="Hide Selected")

        op = layout.operator("clip.hide_tracks", text="Hide Unselected")
        op.unselected = True


class CLIP_MT_track_transform(bpy.types.Menu):
    bl_label = "Transform"

    def draw(self, context):
        layout = self.layout

        layout.operator("transform.translate")
        layout.operator("transform.resize")


class CLIP_MT_select(bpy.types.Menu):
    bl_label = "Select"

    def draw(self, context):
        layout = self.layout

        sc = context.space_data

        layout.menu("CLIP_MT_select_grouped")
        layout.operator("clip.select_border")
        layout.operator("clip.select_circle")
        layout.operator("clip.select_all", text="Select/Deselect all")
        layout.operator("clip.select_all", text="Inverse").action = 'INVERT'


class CLIP_MT_select_grouped(bpy.types.Menu):
    bl_label = "Select Grouped"

    def draw(self, context):
        layout = self.layout

        sc = context.space_data

        op = layout.operator("clip.select_grouped", text="Select Keyframed")
        op.group = 'KEYFRAMED'

        op = layout.operator("clip.select_grouped", text="Select Estimated")
        op.group = 'ESTIMATED'

        op = layout.operator("clip.select_grouped", text="Select Tracked")
        op.group = 'TRACKED'

        op = layout.operator("clip.select_grouped", text="Select Locked")
        op.group = 'LOCKED'

        op = layout.operator("clip.select_grouped", text="Select Disabled")
        op.group = 'DISABLED'

        op = layout.operator("clip.select_grouped", text="Select Failed")
        op.group = 'FAILED'

        op = layout.operator("clip.select_grouped", text="Select by Color")
        op.group = 'COLOR'


class CLIP_MT_tracking_specials(bpy.types.Menu):
    bl_label = "Specials"

    @classmethod
    def poll(cls, context):
        return context.space_data.clip

    def draw(self, context):
        layout = self.layout

        op = layout.operator("clip.disable_markers", text="Enable Markers")
        op.action = 'ENABLE'

        op = layout.operator("clip.disable_markers", text="Disable markers")
        op.action = 'DISABLE'

        layout.separator()
        layout.operator("clip.set_origin")

        layout.separator()
        layout.operator("clip.hide_tracks")
        layout.operator("clip.hide_tracks_clear", text="Show Tracks")

        layout.separator()
        op = layout.operator("clip.lock_tracks", text="Lock Tracks")
        op.action = 'LOCK'

        op = layout.operator("clip.lock_tracks", text="Unlock Tracks")
        op.action = 'UNLOCK'


class CLIP_MT_camera_presets(bpy.types.Menu):
    bl_label = "Camera Presets"
    preset_subdir = "tracking_camera"
    preset_operator = "script.execute_preset"
    draw = bpy.types.Menu.draw_preset


class CLIP_MT_track_color_presets(bpy.types.Menu):
    bl_label = "Color Presets"
    preset_subdir = "tracking_track_color"
    preset_operator = "script.execute_preset"
    draw = bpy.types.Menu.draw_preset


class CLIP_MT_track_color_specials(bpy.types.Menu):
    bl_label = "Track Color Specials"

    def draw(self, context):
        layout = self.layout

        layout.operator('clip.track_copy_color', icon='COPY_ID')


class CLIP_MT_stabilize_2d_specials(bpy.types.Menu):
    bl_label = "Track Color Specials"

    def draw(self, context):
        layout = self.layout

        layout.operator('clip.stabilize_2d_select')


if __name__ == "__main__":  # only for live edit.
    bpy.utils.register_module(__name__)