Skip to content
Snippets Groups Projects
ui.py 59.5 KiB
Newer Older
  • Learn to ignore specific revisions
  • Vilem Duha's avatar
    Vilem Duha committed
                ui_props.turn_off = False
                self.exit_modal()
                ui_props.draw_tooltip = False
                return {'CANCELLED'}
    
            if ui_props.down_up == 'UPLOAD':
    
                ui_props.mouse_x = 0
                ui_props.mouse_y = self.region.height
    
                mx = event.mouse_x
                my = event.mouse_y
    
                ui_props.draw_tooltip = True
    
                # only generate tooltip once in a while
                if (
                        event.type == 'LEFTMOUSE' or event.type == 'RIGHTMOUSE') and event.value == 'RELEASE' or event.type == 'ENTER' or ui_props.tooltip == '':
                    ao = bpy.context.active_object
                    if ui_props.asset_type == 'MODEL' and ao != None \
                            or ui_props.asset_type == 'MATERIAL' and ao != None and ao.active_material != None \
                            or ui_props.asset_type == 'BRUSH':
                        export_data, upload_data, eval_path_computing, eval_path_state, eval_path, props = upload.get_upload_data(
                            self,
                            context,
                            ui_props.asset_type)
                        ui_props.tooltip = search.generate_tooltip(upload_data)
    
                return {'PASS_THROUGH'}
    
            # TODO add one more condition here to take less performance.
            r = self.region
            s = bpy.context.scene
            sr = s.get('search results')
    
            # If there aren't any results, we need no interaction(yet)
            if sr is None:
                return {'PASS_THROUGH'}
            if len(sr) - ui_props.scrolloffset < (ui_props.wcount * ui_props.hcount) + 10:
                self.search_more()
            if event.type == 'WHEELUPMOUSE' or event.type == 'WHEELDOWNMOUSE' or event.type == 'TRACKPADPAN':
                # scrolling
                mx = event.mouse_region_x
                my = event.mouse_region_y
    
                if ui_props.dragging and not mouse_in_asset_bar(mx, my):  # and my < r.height - ui_props.bar_height \
                    # and mx > 0 and mx < r.width and my > 0:
                    sprops = bpy.context.scene.blenderkit_models
                    if event.type == 'WHEELUPMOUSE':
                        sprops.offset_rotation_amount += sprops.offset_rotation_step
                    elif event.type == 'WHEELDOWNMOUSE':
                        sprops.offset_rotation_amount -= sprops.offset_rotation_step
    
                    #### TODO - this snapping code below is 3x in this file.... refactor it.
                    ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = mouse_raycast(
                        context, mx, my)
    
                    # MODELS can be dragged on scene floor
                    if not ui_props.has_hit and ui_props.asset_type == 'MODEL':
                        ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = floor_raycast(
                            context,
                            mx, my)
    
                    return {'RUNNING_MODAL'}
    
                if not mouse_in_asset_bar(mx, my):
                    return {'PASS_THROUGH'}
    
                # note - TRACKPADPAN is unsupported in blender by now.
                # if event.type == 'TRACKPADPAN' :
                #     print(dir(event))
                #     print(event.value, event.oskey, event.)
                if (event.type == 'WHEELDOWNMOUSE') and len(sr) - ui_props.scrolloffset > ui_props.wcount:
                    ui_props.scrolloffset += 1
    
                if event.type == 'WHEELUPMOUSE' and ui_props.scrolloffset > 0:
                    ui_props.scrolloffset -= 1
                return {'RUNNING_MODAL'}
            if event.type == 'MOUSEMOVE':  # Apply
    
                r = self.region
                mx = event.mouse_region_x
                my = event.mouse_region_y
    
                ui_props.mouse_x = mx
                ui_props.mouse_y = my
    
                if ui_props.dragging_rating or ui_props.rating_menu_on:
                    res = interact_rating(r, mx, my, event)
                    if res == True:
                        return {'RUNNING_MODAL'}
    
                if ui_props.drag_init:
                    ui_props.drag_length += 1
                    if ui_props.drag_length > 0:
                        ui_props.dragging = True
                        ui_props.drag_init = False
    
                if not (ui_props.dragging and mouse_in_region(r, mx, my)) and not mouse_in_asset_bar(mx, my):
                    ui_props.active_index = -3
                    ui_props.draw_tooltip = False
                    bpy.context.window.cursor_set("DEFAULT")
                    return {'PASS_THROUGH'}
    
                sr = bpy.context.scene['search results']
    
                if not ui_props.dragging:
                    bpy.context.window.cursor_set("DEFAULT")
    
                    if sr != None and ui_props.wcount * ui_props.hcount > len(sr) and ui_props.scrolloffset > 0:
                        ui_props.scrolloffset = 0
    
                    asset_search_index = get_asset_under_mouse(mx, my)
                    ui_props.active_index = asset_search_index
                    if asset_search_index > -1:
    
                        asset_data = sr[asset_search_index]
                        ui_props.draw_tooltip = True
    
                        ui_props.tooltip = asset_data['tooltip']
                    else:
                        ui_props.draw_tooltip = False
    
                    if mx > ui_props.bar_x + ui_props.bar_width - 50 and len(sr) - ui_props.scrolloffset > (
                            ui_props.wcount * ui_props.hcount):
                        ui_props.active_index = -1
                        return {'RUNNING_MODAL'}
                    if mx < ui_props.bar_x + 50 and ui_props.scrolloffset > 0:
                        ui_props.active_index = -2
                        return {'RUNNING_MODAL'}
    
                else:
                    result = False
                    if ui_props.dragging and not mouse_in_asset_bar(mx, my) and mouse_in_region(r, mx, my):
                        ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = mouse_raycast(
                            context, mx, my)
                        # MODELS can be dragged on scene floor
                        if not ui_props.has_hit and ui_props.asset_type == 'MODEL':
                            ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = floor_raycast(
                                context,
                                mx, my)
                    if ui_props.has_hit and ui_props.asset_type == 'MODEL':
                        # this condition is here to fix a bug for a scene submitted by a user, so this situation shouldn't
                        # happen anymore, but there might exists scenes which have this problem for some reason.
    
                        if ui_props.active_index < len(sr) and ui_props.active_index > -1:
    
    Vilem Duha's avatar
    Vilem Duha committed
                            ui_props.draw_snapped_bounds = True
                            active_mod = sr[ui_props.active_index]
                            ui_props.snapped_bbox_min = Vector(active_mod['bbox_min'])
                            ui_props.snapped_bbox_max = Vector(active_mod['bbox_max'])
    
                    else:
                        ui_props.draw_snapped_bounds = False
                        ui_props.draw_drag_image = True
                return {'RUNNING_MODAL'}
    
            if event.type == 'LEFTMOUSE':
    
                r = self.region
                mx = event.mouse_x - r.x
                my = event.mouse_y - r.y
    
                ui_props = context.scene.blenderkitUI
                if event.value == 'PRESS' and ui_props.active_index > -1:
                    if ui_props.asset_type == 'MODEL' or ui_props.asset_type == 'MATERIAL':
                        ui_props.drag_init = True
                        bpy.context.window.cursor_set("NONE")
                        ui_props.draw_tooltip = False
                        ui_props.drag_length = 0
    
                if ui_props.rating_on:
                    res = interact_rating(r, mx, my, event)
                    if res:
                        return {'RUNNING_MODAL'}
    
                if not ui_props.dragging and not mouse_in_asset_bar(mx, my):
                    return {'PASS_THROUGH'}
    
                # this can happen by switching result asset types - length of search result changes
                if ui_props.scrolloffset > 0 and (ui_props.wcount * ui_props.hcount) > len(sr) - ui_props.scrolloffset:
                    ui_props.scrolloffset = len(sr) - (ui_props.wcount * ui_props.hcount)
    
                if event.value == 'RELEASE':  # Confirm
                    ui_props.drag_init = False
    
                    # scroll by a whole page
                    if mx > ui_props.bar_x + ui_props.bar_width - 50 and len(
                            sr) - ui_props.scrolloffset > ui_props.wcount * ui_props.hcount:
                        ui_props.scrolloffset = min(
                            ui_props.scrolloffset + (ui_props.wcount * ui_props.hcount),
                            len(sr) - ui_props.wcount * ui_props.hcount)
                        return {'RUNNING_MODAL'}
                    if mx < ui_props.bar_x + 50 and ui_props.scrolloffset > 0:
                        ui_props.scrolloffset = max(0, ui_props.scrolloffset - ui_props.wcount * ui_props.hcount)
                        return {'RUNNING_MODAL'}
    
                    # Drag-drop interaction
                    if ui_props.dragging and mouse_in_region(r, mx, my):
                        asset_search_index = ui_props.active_index
                        # raycast here
                        ui_props.active_index = -3
    
                        if ui_props.asset_type == 'MODEL':
    
                            ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = mouse_raycast(
                                context, mx, my)
    
                            # MODELS can be dragged on scene floor
                            if not ui_props.has_hit and ui_props.asset_type == 'MODEL':
                                ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = floor_raycast(
                                    context,
                                    mx, my)
    
                            if not ui_props.has_hit:
                                return {'RUNNING_MODAL'}
    
                            target_object = ''
                            if object is not None:
                                target_object = object.name
                            target_slot = ''
    
                        if ui_props.asset_type == 'MATERIAL':
                            ui_props.has_hit, ui_props.snapped_location, ui_props.snapped_normal, ui_props.snapped_rotation, face_index, object, matrix = mouse_raycast(
                                context, mx, my)
    
                            if not ui_props.has_hit:
                                # this is last attempt to get object under mouse - for curves and other objects than mesh.
                                ui_props.dragging = False
                                sel = utils.selection_get()
                                bpy.ops.view3d.select(location=(event.mouse_region_x, event.mouse_region_y))
                                sel1 = utils.selection_get()
                                if sel[0] != sel1[0] and sel1[0].type != 'MESH':
                                    object = sel1[0]
                                    target_slot = sel1[0].active_material_index
                                    ui_props.has_hit = True
                                utils.selection_set(sel)
    
                            if not ui_props.has_hit:
                                return {'RUNNING_MODAL'}
    
                            else:
                                # first, test if object can have material applied.
                                if object is not None and not object.is_library_indirect:
                                    target_object = object.name
                                    # create final mesh to extract correct material slot
    
                                    depsgraph = bpy.context.evaluated_depsgraph_get()
    
                                    object_eval = object.evaluated_get(depsgraph)
                                    temp_mesh = object_eval.to_mesh()
    
    Vilem Duha's avatar
    Vilem Duha committed
                                    target_slot = temp_mesh.polygons[face_index].material_index
    
                                    object_eval.to_mesh_clear()
    
    Vilem Duha's avatar
    Vilem Duha committed
                                else:
                                    self.report({'WARNING'}, "Invalid or library object as input:")
                                    target_object = ''
                                    target_slot = ''
    
                    # Click interaction
                    else:
                        asset_search_index = get_asset_under_mouse(mx, my)
    
                        if ui_props.asset_type in ('MATERIAL',
                                                   'MODEL'):  # this was meant for particles, commenting for now or ui_props.asset_type == 'MODEL':
                            ao = bpy.context.active_object
                            if ao != None and not ao.is_library_indirect:
                                target_object = bpy.context.active_object.name
                                target_slot = bpy.context.active_object.active_material_index
                            else:
                                target_object = ''
                                target_slot = ''
                    # FIRST START SEARCH
    
                    if asset_search_index == -3:
                        return {'RUNNING_MODAL'}
                    if asset_search_index > -3:
                        if ui_props.asset_type == 'MATERIAL':
                            if target_object != '':
                                # position is for downloader:
                                loc = ui_props.snapped_location
                                rotation = (0, 0, 0)
    
                                asset_data = sr[asset_search_index]
                                utils.automap(target_object, target_slot=target_slot,
                                              tex_size=asset_data.get('texture_size_meters', 1.0))
                                bpy.ops.scene.blenderkit_download(True,
                                                                  asset_type=ui_props.asset_type,
                                                                  asset_index=asset_search_index,
                                                                  model_location=loc,
                                                                  model_rotation=rotation,
                                                                  target_object=target_object,
                                                                  material_target_slot=target_slot)
    
    
                        elif ui_props.asset_type == 'MODEL':
                            if ui_props.has_hit and ui_props.dragging:
                                loc = ui_props.snapped_location
                                rotation = ui_props.snapped_rotation
                            else:
                                loc = s.cursor.location
                                rotation = s.cursor.rotation_euler
    
                            bpy.ops.scene.blenderkit_download(True,
                                                              asset_type=ui_props.asset_type,
                                                              asset_index=asset_search_index,
                                                              model_location=loc,
                                                              model_rotation=rotation,
                                                              target_object=target_object)
    
                        else:
                            bpy.ops.scene.blenderkit_download(asset_type=ui_props.asset_type,
                                                              asset_index=asset_search_index)
    
                        ui_props.dragging = False
                        return {'RUNNING_MODAL'}
                else:
                    return {'RUNNING_MODAL'}
    
    
    Vilem Duha's avatar
    Vilem Duha committed
            if event.type == 'A' and ui_props.active_index != -3:
                sr = bpy.context.scene['search results']
                asset_data = sr[ui_props.active_index]
                a = bpy.context.window_manager['bkit authors'].get(asset_data['author_id'])
                if a is not None:
                    if a.get('aboutMeUrl') is not None:
                        bpy.ops.wm.url_open(url=a['aboutMeUrl'])
    
    Vilem Duha's avatar
    Vilem Duha committed
            if event.type == 'X' and ui_props.active_index != -3:
                sr = bpy.context.scene['search results']
                asset_data = sr[ui_props.active_index]
                print(asset_data['name'])
                print('delete')
                paths.delete_asset_debug(asset_data)
                asset_data['downloaded'] = 0
                return {'RUNNING_MODAL'}
            return {'PASS_THROUGH'}
    
        def invoke(self, context, event):
            # FIRST START SEARCH
            ui_props = context.scene.blenderkitUI
    
            if self.do_search:
                # we erase search keywords for cateogry search now, since these combinations usually return nothing now.
                # when the db gets bigger, this can be deleted.
                if self.category != '':
                    sprops = utils.get_search_props()
                    sprops.search_keywords = ''
                search.search(category=self.category, free_only=self.free_only)
    
            if ui_props.assetbar_on:
                # we don't want to run the assetbar many times, that's why it has a switch on/off behaviour,
                # unless being called with 'keep_running' prop.
                if not self.keep_running:
                    # this sends message to the originally running operator, so it quits, and then it ends this one too.
                    # If it initiated a search, the search will finish in a thread. The switch off procedure is run
                    # by the 'original' operator, since if we get here, it means
                    # same operator is allready running.
                    ui_props.turn_off = True
                    # if there was an error, we need to turn off these props so we can restart after 2 clicks
                    ui_props.assetbar_on = False
    
                else:
                    pass
                return {'FINISHED'}
    
    
            ui_props.dragging = False  # only for cases where assetbar ended with an error.
    
    Vilem Duha's avatar
    Vilem Duha committed
            ui_props.assetbar_on = True
            ui_props.turn_off = False
    
            sr = bpy.context.scene.get('search results')
            if sr is None:
                bpy.context.scene['search results'] = []
    
            if context.area.type == 'VIEW_3D':
                # the arguments we pass the the callback
                args = (self, context)
                self.area = context.area
                self.scene = bpy.context.scene
    
                for r in self.area.regions:
                    if r.type == 'WINDOW':
                        self.region = r
    
                update_ui_size(self.area, self.region)
    
                self._handle_2d = bpy.types.SpaceView3D.draw_handler_add(draw_callback_2d, args, 'WINDOW', 'POST_PIXEL')
                self._handle_3d = bpy.types.SpaceView3D.draw_handler_add(draw_callback_3d, args, 'WINDOW', 'POST_VIEW')
    
                context.window_manager.modal_handler_add(self)
                ui_props.assetbar_on = True
                return {'RUNNING_MODAL'}
            else:
    
                self.report({'WARNING'}, "View3D not found, cannot run operator")
                return {'CANCELLED'}
    
        def execute(self, context):
            return {'RUNNING_MODAL'}
    
    
    classess = (
        AssetBarOperator,
    
    )
    
    # store keymaps here to access after registration
    addon_keymaps = []
    
    
    def register_ui():
    
    Vilem Duha's avatar
    Vilem Duha committed
        for c in classess:
            bpy.utils.register_class(c)
    
        args = (None, bpy.context)
    
    
        handler_2d = bpy.types.SpaceView3D.draw_handler_add(draw_callback_2d_progress, args, 'WINDOW', 'POST_PIXEL')
        handler_3d = bpy.types.SpaceView3D.draw_handler_add(draw_callback_3d_progress, args, 'WINDOW', 'POST_VIEW')
    
    Vilem Duha's avatar
    Vilem Duha committed
    
        wm = bpy.context.window_manager
    
        # spaces solved by registering shortcut to Window. Couldn't register object mode before somehow.
        if not wm.keyconfigs.addon:
            return
        km = wm.keyconfigs.addon.keymaps.new(name="Window", space_type='EMPTY')
        kmi = km.keymap_items.new(AssetBarOperator.bl_idname, 'SEMI_COLON', 'PRESS', ctrl=False, shift=False)
        kmi.properties.keep_running = False
        kmi.properties.do_search = False
    
        addon_keymaps.append(km)
    
    
    def unregister_ui():
    
        bpy.types.SpaceView3D.draw_handler_remove(handler_2d, 'WINDOW')
        bpy.types.SpaceView3D.draw_handler_remove(handler_3d, 'WINDOW')
    
    
    Vilem Duha's avatar
    Vilem Duha committed
        for c in classess:
            bpy.utils.unregister_class(c)
    
        args = (None, bpy.context)
    
        wm = bpy.context.window_manager
        if not wm.keyconfigs.addon:
            return
    
        for km in addon_keymaps:
            wm.keyconfigs.addon.keymaps.remove(km)
        del addon_keymaps[:]