import bpy
from . import bake_utilities

from .. Functions import constants
from .. Functions import visibility_functions


def bake_texture(self, selected_objects, bake_settings):
    parent_operator = self
    # ----------------------- CREATE INSTANCE --------------------#
    lightmap_utilities = bake_utilities.BakeUtilities(parent_operator, selected_objects, bake_settings)  

    if not lightmap_utilities.checkPBR():
        return
    
    # -----------------------SET LIGHTMAP UV--------------------#
    lightmap_utilities.set_active_uv_to_lightmap()
    
    # -----------------------SETUP UV'S--------------------#
    lightmap_utilities.unwrap_selected()

    # -----------------------SETUP ENGINE--------------------#
    lightmap_utilities.setup_engine()

    # -----------------------SWITCH BACK TO SHOW ORG MATERIAL --------------------#
    visibility_functions.preview_bake_texture(self,context=bpy.context)
    
    # ----------------------- CREATE NEW MATERIAL FOR BAKING --------------------#
    lightmap_utilities.create_bake_material("_AO")
     
    # -----------------------SETUP NODES--------------------#
    lightmap_utilities.add_node_setup()
        
    # ----------------------- BAKING --------------------#
    if bake_settings.lightmap_bake:
        lightmap_utilities.save_metal_value()
        lightmap_utilities.bake(constants.Bake_Passes.lightmap)
        lightmap_utilities.load_metal_value()
        lightmap_utilities.add_lightmap_flag()
    
    if bake_settings.ao_bake:
        lightmap_utilities.bake(constants.Bake_Passes.ao)

    lightmap_utilities.cleanup()
    del lightmap_utilities
    return


def bake_on_plane(self,selected_objects,bake_settings):

    parent_operator = self
    
    # ----------------------- CREATE INSTANCE --------------------#
    pbr_utilities = bake_utilities.PbrBakeUtilities(parent_operator,selected_objects,bake_settings)

    # -----------------------SETUP ENGINE--------------------#

    pbr_utilities.setup_engine()

    # ----------------------- BAKE --------------------#

    pbr_utilities.bake_materials_on_object()
    del pbr_utilities

    return