blender-addons/mesh_tissue/config.py

74 lines
2.4 KiB
Python

# SPDX-FileCopyrightText: 2022-2023 Blender Foundation
#
# SPDX-License-Identifier: GPL-2.0-or-later
import bpy
from bpy.props import (
IntProperty,
BoolProperty
)
evaluatedDepsgraph = None
class tissuePreferences(bpy.types.AddonPreferences):
bl_idname = __package__
print_stats : IntProperty(
name="Print Stats",
description="Print in the console all details about the computing time.",
default=1,
min=0,
max=4
)
use_numba_tess : BoolProperty(
name="Numba Tessellate",
description="Boost the Tessellation using Numba module. It will be slower during the first execution",
default=True
)
def draw(self, context):
from .utils_pip import Pip
Pip._ensure_user_site_package()
layout = self.layout
layout.prop(self, "print_stats")
import importlib
numba_spec = importlib.util.find_spec('numba')
found = numba_spec is not None
if found:
try:
import numba
layout.label(text='Numba module installed correctly!', icon='INFO')
layout.prop(self, "use_numba_tess")
except:
found = False
if not found:
layout.label(text='Numba module not installed!', icon='ERROR')
layout.label(text='Installing Numba will make Tissue faster', icon='INFO')
row = layout.row()
row.operator('scene.tissue_install_numba')
layout.label(text='Internet connection required. It may take few minutes', icon='URL')
class tissue_install_numba(bpy.types.Operator):
bl_idname = "scene.tissue_install_numba"
bl_label = "Install Numba"
bl_description = ("Install Numba python module")
bl_options = {'REGISTER'}
def execute(self, context):
try:
from .utils_pip import Pip
#Pip.upgrade_pip()
Pip.uninstall('llvmlite')
Pip.uninstall('numba')
Pip.install('llvmlite')
Pip.install('numba')
from numba import jit, njit, guvectorize, float64, int32, prange
bool_numba = True
print('Tissue: Numba successfully installed!')
self.report({'INFO'}, 'Tissue: Numba successfully installed!')
except:
print('Tissue: Numba not loaded correctly. Try restarting Blender')
return {'FINISHED'}