BB2_MLP_PANEL.py

# Blender modules
# 2020-03-28
import bpy
from bpy import *
import bpy.path
from bpy.path import abspath
from mathutils import *

# Python standard modules
from urllib.parse import urlencode
from urllib.request import *
from html.parser import *
from smtplib import *
from email.mime.text import MIMEText
import time
import platform
import os
import codecs
import base64
from math import *
import pickle
import shutil
import subprocess
import sys
import traceback
import copy

from .BioBlender2 import *
from . import BB2_GUI_PDB_IMPORT as ImportPDB
from . import BB2_PANEL_VIEW as panel
from .BB2_PHYSICS_SIM_PANEL import *
from .BB2_PDB_OUTPUT_PANEL import *
from .BB2_OUTPUT_PANEL import *
from .BB2_NMA_PANEL import *
from .BB2_EP_PANEL import *

class BB2_MLP_PANEL(types.Panel):
    bl_label = "BioBlender2 MLP Visualization"
    bl_idname = "BB2_MLP_PANEL"
    bl_space_type = "PROPERTIES"
    bl_region_type = "WINDOW"
    bl_context = "scene"
    bl_options = {'DEFAULT_CLOSED'}
    bpy.types.Scene.BBAtomic = bpy.props.EnumProperty(attr="BBAtomic", name="BBAtomic",
                                                      description="Atomic or Surface MLP",
                                                      items=(("0", "Atomic", ""), ("1", "Surface", "")), default="0")
    bpy.types.Scene.BBMLPFormula = bpy.props.EnumProperty(attr="BBMLPFormula", name="Formula",
                                                          description="Select a formula for MLP calculation", items=(
            ("0", "Dubost", ""), ("1", "Testa", ""), ("2", "Fauchere", ""), ("3", "Brasseur", ""),
            ("4", "Buckingham", "")),
                                                          default="1")
    bpy.types.Scene.BBMLPGridSpacing = bpy.props.FloatProperty(attr="BBMLPGridSpacing", name="Grid Spacing",
                                                               description="MLP Calculation step size (Smaller is better, but slower)",
                                                               default=1, min=0.01, max=20, soft_min=1.4, soft_max=10)
    bpy.types.Scene.BBAtomicMLP = bpy.props.BoolProperty(attr="BBAtomicMLP", name="Atomic MLP", default=False)

    def draw(self, context):
        scene = context.scene
        layout = self.layout
        r = layout.row()
        r.prop(scene, "BBAtomic", expand=True)
        r = layout.row()
        if bpy.context.scene.BBAtomic == "0":
            r.label("Calculate Atomic MLP")
            r = layout.row()
            r.scale_y = 2
            r.operator("ops.bb2_operator_atomic_mlp")
        else:
            split = layout.split()
            c = split.column()
            c.prop(scene, "BBMLPFormula")
            c.prop(scene, "BBMLPGridSpacing")
            r = split.row()
            r.scale_y = 2
            r.operator("ops.bb2_operator_mlp")
            split = layout.split()
            r = split.column(align=True)
            r = split.column()
            r.scale_y = 2
            r.operator("ops.bb2_operator_mlp_render")


class bb2_operator_atomic_mlp(types.Operator):
    bl_idname = "ops.bb2_operator_atomic_mlp"
    bl_label = "Atomic MLP"
    bl_description = "Atomic MLP"

    def invoke(self, context, event):
        try:
            selectedPDBidS = []
            for b in bpy.context.scene.objects:
                if b.select:
                    try:
                        if b.bb2_pdbID not in selectedPDBidS:
                            t = copy.copy(b.bb2_pdbID)
                            selectedPDBidS.append(t)
                    except Exception as E:
                        str1 = str(E)  # Do not print...
                        print("AtomicMLP ERROR 1" + str1)
            context.user_preferences.edit.use_global_undo = False
            for id in selectedPDBidS:
                bpy.ops.object.select_all(action="DESELECT")
                for o in bpy.data.objects:
                    o.select = False
                for obj in bpy.context.scene.objects:
                    try:
                        if obj.bb2_pdbID == id:
                            obj.select = True
                    except Exception as E:
                        str2 = str(E)  # Do not print...
                        print("AtomicMLP ERROR 2" + str2)
                tID = copy.copy(id)
                atomicMLP(tID)
            context.user_preferences.edit.use_global_undo = True
        except Exception as E:
            s = "Generate Atomic MLP visualization Failed: " + str(E)
            print(s)
            return {'CANCELLED'}
        else:
            return {'FINISHED'}


bpy.utils.register_class(bb2_operator_atomic_mlp)


def atomicMLP(tID):
    for obj in bpy.data.objects:
        try:
            if (obj.bb2_pdbID == tID) and (obj.bb2_objectType == "ATOM"):
                aminoName = ImportPDB.PDBString(obj.BBInfo).get("aminoName")
                name = ImportPDB.PDBString(obj.BBInfo).get("name")
                material_this = ImportPDB.retrieve_fi_materials(am_name=aminoName, at_name=name)
                obj.material_slots[0].material = bpy.data.materials[material_this]
                bpy.context.scene.BBAtomicMLP = True
        except Exception as E:
            print(str(E))
    print("Atomic MLP Color set")


class bb2_operator_mlp(types.Operator):
    bl_idname = "ops.bb2_operator_mlp"
    bl_label = "Show MLP on Surface"
    bl_description = "Calculate Molecular Lipophilicity Potential on surface"

    def invoke(self, context, event):
        try:
            bpy.context.user_preferences.edit.use_global_undo = False
            selectedPDBidS = []
            for b in bpy.context.scene.objects:
                if b.select == True:
                    try:
                        if b.bb2_pdbID not in selectedPDBidS:
                            t = copy.copy(b.bb2_pdbID)
                            selectedPDBidS.append(t)
                    except Exception as E:
                        str1 = str(E)
                        print("Error operator_mlp 1" + str(E))
            context.user_preferences.edit.use_global_undo = False
            for id in selectedPDBidS:
                bpy.ops.object.select_all(action="DESELECT")
                for o in bpy.data.objects:
                    o.select = False
                for obj in bpy.context.scene.objects:
                    try:
                        if obj.bb2_pdbID == id:
                            obj.select = True
                    except Exception as E:
                        str2 = str(E)
                        print("Error operator_mlp 2" + str(E))
                tID = copy.copy(id)
                mlp(tID, force=True)
                panel.todoAndviewpoints()
            bpy.context.scene.BBViewFilter = "4"
            bpy.context.user_preferences.edit.use_global_undo = True
        except Exception as E:
            s = "Generate MLP visualization Failed 1: " + str(E)
            print(s)
            return {'CANCELLED'}
        else:
            return {'FINISHED'}


bpy.utils.register_class(bb2_operator_mlp)


class bb2_operator_mlp_render(types.Operator):
    bl_idname = "ops.bb2_operator_mlp_render"
    bl_label = "Render MLP to Surface"
    bl_description = "Visualize Molecular Lipophilicity Potential on surface"

    def invoke(self, context, event):
        try:
            context.user_preferences.edit.use_global_undo = False
            selectedPDBidS = []
            for b in bpy.context.scene.objects:
                if b.select:
                    try:
                        if (b.bb2_pdbID not in selectedPDBidS) and (b.bb2_objectType == "SURFACE"):
                            t = copy.copy(b.bb2_pdbID)
                            selectedPDBidS.append(t)
                    except Exception as E:
                        str1 = str(E)  # Do not print...
            context.user_preferences.edit.use_global_undo = False
            for id in selectedPDBidS:
                tID = copy.copy(id)
                mlpRender(tID)
                panel.todoAndviewpoints()
            context.scene.BBViewFilter = "4"
            bpy.context.scene.world.light_settings.use_environment_light = False
            context.user_preferences.edit.use_global_undo = True
        except Exception as E:
            s = "Generate MLP visualization Failed 2: " + str(E)
            print(s)
            return {'CANCELLED'}
        else:
            return {'FINISHED'}


bpy.utils.register_class(bb2_operator_mlp_render)


# do MLP visualization
def mlp(tID, force):
    global dxCacheM
    global method
    global dxData
    global dimension
    global origin
    global delta

    scene = bpy.context.scene
    formula = bpy.context.scene.BBMLPFormula
    spacing = bpy.context.scene.BBMLPGridSpacing
    namemlp = "MLP_Surface_" + NamePDBMLP(tID) + "_" + getNumFrameMLP()
    if ExistMLP(namemlp) == False:
        scene.render.engine = 'BLENDER_RENDER'

        def getVar(rawID):
            try:
                val = dxCache[rawID]
            except:
                v = ob.data.vertices[rawID].co
                dimx = dimension[0]
                dimy = dimension[1]
                dimz = dimension[2]
                deltax = delta[0]
                deltay = delta[1]
                deltaz = delta[2]
                cellx = int((v[0] - origin[0]) / deltax)
                celly = int((v[1] - origin[1]) / deltay)
                cellz = int((v[2] - origin[2]) / deltaz)
                mmm = dxData[cellz + ((celly) * dimz) + ((cellx) * dimz * dimy)]
                pmm = dxData[cellz + ((celly) * dimz) + ((cellx + 1) * dimz * dimy)]
                mpm = dxData[cellz + ((celly + 1) * dimz) + ((cellx) * dimz * dimy)]
                mmp = dxData[cellz + 1 + ((celly) * dimz) + ((cellx) * dimz * dimy)]
                ppm = dxData[cellz + ((celly + 1) * dimz) + ((cellx +1) * dimz * dimy)]
                mpp = dxData[cellz + 1 + ((celly + 1) * dimz) + ((cellx) * dimz * dimy)]
                pmp = dxData[cellz + 1 + ((celly) * dimz) + ((cellx + 1) * dimz * dimy)]
                ppp = dxData[cellz + 1 + ((celly + 1) * dimz) + ((cellx + 1) * dimz * dimy)]
                wxp = 1.0 - (fabs(v[0] - (origin[0] + (deltax * (cellx + 0.8))))) / deltax
                wxm = 1.0 - (fabs(v[0] - (origin[0] + (deltax * (cellx))))) / deltax
                wyp = 1.0 - (fabs(v[1] - (origin[1] + (deltay * (celly + 0.8))))) / deltay
                wym = 1.0 - (fabs(v[1] - (origin[1] + (deltay * (celly))))) / deltay
                wzp = 1.0 - (fabs(v[2] - (origin[2] + (deltaz * (cellz + 0.8))))) / deltaz
                wzm = 1.0 - (fabs(v[2] - (origin[2] + (deltaz * (cellz))))) / deltaz
                onz_xmym = (wzp * mmp) + (wzm * mmm)
                onz_xpym = (wzp * pmp) + (wzm * pmm)
                onz_xmyp = (wzp * mpp) + (wzm * mpm)
                onz_xpyp = (wzp * ppp) + (wzm * ppm)
                onx_yp = (wxp * onz_xpyp) + (wxm * onz_xmyp)
                onx_ym = (wxp * onz_xpym) + (wxm * onz_xmym)
                val = (wyp * onx_yp) + (wym * onx_ym)
                dxCache[rawID] = val

            # map values
            if (val >= 0.0):
                val = (val + 1.0) / 2.0
            else:
                val = (val + 3.0) / 6.0
            return [val, val, val]

        if force:
            panel.setup(setupPDBid=tID)
            # select formula for PyMLP script
            if formula == "0":
                method = "dubost"
            elif formula == "1":
                method = "testa"
            elif formula == "2":
                method = "fauchere"
            elif formula == "3":
                method = "brasseur"
            elif formula == "4":
                method = "buckingham"

            # Launch this in a separate process
            if opSystem == "linux":
                command = "chmod 755 %s" % (
                    panel.quotedPath(homePath + "bin" + os.sep + "pyMLP-1.0" + os.sep + "pyMLP.py"))
                command = panel.quotedPath(command)
                panel.launch(exeName=command)
            elif opSystem == "darwin":
                command = "chmod 755 %s" % (
                    panel.quotedPath(homePath + "bin" + os.sep + "pyMLP-1.0" + os.sep + "pyMLP.py"))
                command = panel.quotedPath(command)
                panel.launch(exeName=command)
            print("Running PyMLP")
            global pyPath
            if not pyPath:
                pyPath = "python"
            command = "%s %s -i %s -m %s -s %f -o %s -v" % (panel.quotedPath(pyPath), panel.quotedPath(homePath + "bin" + os.sep + "pyMLP-1.0" + os.sep + "pyMLP.py"), panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "tmp.pdb"), method, spacing, panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "tmp.dx"))

            p = panel.launch(exeName=command, async=True)

            print("PyMLP command succeded")
            panel.surface(sPid=tID, optName=namemlp)

            wait(p)

            # purge the all old data
            dxCache = {}
            dxData = []  # list[n] of Potential data
            dimension = []  # list[3] of dx grid store.dimension
            origin = []  # list[3] of dx grid store.origin
            delta = []  # list[3] of dx grid store.increment

            print("Loading MLP values into Blender")

            try:
                tmpPathO = homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "tmp.dx"
                with open(tmpPathO) as dx:
                    for line in dx:
                        # skip comments starting with #
                        if line[0] == "#": continue
                        if not dimension:
                            # get the store.dimension and convert to integer
                            dim = line.split()[-3:]
                            dimension = [int(d) for d in dim]
                            size = dimension[0] * dimension[1] * dimension[2]
                            continue

                        if not origin:
                            # get the store.origin
                            org = line.split()[-3:]
                            origin = [float(o) for o in org]
                            continue

                        if not delta:
                            # get the increment delta
                            x = float(line.split()[-3:][0])
                            line = dx.readline()
                            y = float(line.split()[-3:][1])
                            line = dx.readline()
                            z = float(line.split()[-3:][2])
                            delta = [x, y, z]

                            # ignore more garbage lines
                            dx.readline()
                            dx.readline()
                            continue

                        # load as much data as we should, ignoring the rest of the file
                        if (len(dxData) >= size):
                            break

                        # Load the data
                        # Convert dx data from str to float, then save to list
                        [dxData.append(float(coord)) for coord in line.split()]
            except Exception as E:
                print("An error occured in MLP while loading values into Blender; be careful; " + str(E))

            # quick and dirty update starts here
        if dxData:
            ob = bpy.data.objects[namemlp]
            ob.bb2_pdbID = copy.copy(tID)
            ob.select = True
            bpy.context.scene.objects.active = ob

            if not bpy.context.vertex_paint_object:
                bpy.ops.paint.vertex_paint_toggle()
            try:
                bpy.ops.object.editmode_toggle()
                bpy.ops.mesh.remove_doubles(threshold=0.0001, use_unselected=False)
                bpy.ops.object.editmode_toggle()
                bpy.ops.object.shade_smooth()
            except Exception as E:
                print("Error in MLP: remove doubles and shade smooth failed; " + str(E))

            try:
                ob.data.update(calc_tessface=True)
            except Exception as E:
                print("Error in MLP: ob.data.update tessface failed; " + str(E))


            try:
                vColor0 = []
                vColor1 = []
                vColor2 = []
                for f in ob.data.tessfaces:
                    vColor0.extend(getVar(f.vertices_raw[0]))
                    vColor1.extend(getVar(f.vertices_raw[1]))
                    vColor2.extend(getVar(f.vertices_raw[2]))
                for i in range(len(ob.data.vertex_colors[0].data)):
                    tmp = ((0.21 * vColor0[i]) + (0.71 * vColor1[i]) + (0.07 * vColor2[i]))
                    ob.data.vertex_colors[0].data[i].color = (tmp, tmp, tmp)
            except Exception as E:
                print("Error in MLP: tessfaces vColor extend failed; " + str(E))

            try:
                me = ob.data
            except Exception as E:
                print("Error in MLP: me = ob.data failed; " + str(E))

            try:
                bpy.ops.paint.vertex_paint_toggle()
                me.use_paint_mask = False
                bpy.ops.paint.vertex_color_smooth()
                bpy.ops.paint.vertex_paint_toggle()
            except Exception as E:
                print("Error in MLP: vertex color smooth failed; " + str(E))

            try:
                # needed to make sure VBO is up to date
                ob.data.update()
            except Exception as E:
                print("Error in MLP: VBO ob.data.update failed; " + str(E))

            try:
                for area in bpy.context.screen.areas:
                    if area.type == 'VIEW_3D':
                        area.spaces[0].viewport_shade = "TEXTURED"
            except Exception as E:
                print("Error in MLP: view3D viewport shade textured failed; " + str(E))

        try:
            for obj in bpy.context.scene.objects:
                if obj.BBInfo:
                    obj.hide = True
                    obj.hide_render = True
        except Exception as E:
            print("Error in MLP: obj.BBInfo")
        panel.ClearLigth(1)
        DeleteSurface()
        print("MLP function completed")
    else:
        bpy.ops.object.select_all(action="DESELECT")
        for o in bpy.data.objects:
            o.select = False


def mlpRender(tID):
    print("MLP RENDER Start")
    scene = bpy.context.scene
    # Stop if no surface is found

    scene.render.engine = 'BLENDER_RENDER'

    for obj in bpy.data.objects:
        try:
            if (obj.bb2_pdbID == tID) and (obj.bb2_objectType == "SURFACE") and (obj.name.split("_")[0] != "SURFACE"):
                surfaceName = str(copy.copy(obj.name))

        except Exception as E:
            print("mlpRender(tID)" + str(E))
    ob = bpy.data.objects[surfaceName]

    bpy.ops.object.select_all(action="DESELECT")
    for o in bpy.data.objects:
        o.select = False
    bpy.context.scene.objects.active = None
    bpy.data.objects[surfaceName].select = True
    bpy.context.scene.objects.active = bpy.data.objects[surfaceName]

    if not ob:
        raise Exception("No MLP Surface Found, select surface view first")

    # Stop if no dx data is loaded
    if not dxData:
        raise Exception("No MLP data is loaded.  Run MLP calculation first")

    # create image data block
    try:
        print("MLP Render first time: False")
        firstTime = False
        image = bpy.data.images["MLPBaked_" + NamePDBMLP(tID)]
    except:
        print("MLP Render first time: True")
        firstTime = True
        image = bpy.data.images.new(name="MLPBaked_" + NamePDBMLP(tID), width=2048, height=2048)

    # set material
    if firstTime:
        mat = bpy.data.materials.new("matMLP" + NamePDBMLP(tID))
        mat.use_shadeless = True
        mat.use_vertex_color_paint = True
        ob.data.materials.append(mat)
    else:
        mat = bpy.data.materials["matMLP" + NamePDBMLP(tID)]
        mat.use_shadeless = True
        mat.use_vertex_color_paint = True
        if not ob.data.materials:
            ob.data.materials.append(mat)

    print("Baking MLP textures")
    # save and bake
    image.source = "GENERATED"
    image.generated_height = 2048
    image.generated_width = 2048

    if not ob.data.uv_textures:
        bpy.context.active_object.data.uv_textures.new()
    if bpy.context.mode != "EDIT":
        bpy.ops.object.editmode_toggle()
    # ====
    for uv in ob.data.uv_textures[0].data:
        uv.image = image

    bpy.data.screens['UV Editing'].areas[1].spaces[0].image = bpy.data.images['MLPBaked_' + NamePDBMLP(tID)]

    bpy.ops.uv.smart_project(angle_limit=66, island_margin=0, user_area_weight=0)
    bpy.context.scene.render.bake_type = 'TEXTURE'

    bpy.context.scene.render.use_raytrace = False
    print("===== BAKING... =====")
    bpy.ops.object.bake_image()
    print("=====          ... BAKED! =====")
    bpy.context.scene.render.use_raytrace = True

    if opSystem == "linux":
        os.chdir(panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep))
    elif opSystem == "darwin":
        os.chdir(panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep))
    else:
        os.chdir(r"\\?\\" + homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep)

    print("Image Save Render")
    image.save_render(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "MLPBaked.png")
    # copy the needed files
    print("Copy the needed files")
    uriSource = homePath + "data" + os.sep + "noise.png"
    uriDest = homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "noise.png"

    if opSystem == "linux":
        shutil.copy(uriSource, uriDest)
    elif opSystem == "darwin":
        shutil.copy(uriSource, uriDest)
    else:
        shutil.copy(r"\\?\\" + uriSource, r"\\?\\" + uriDest)

    uriSource = homePath + "data" + os.sep + "composite.blend"
    uriDest = homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "composite.blend"

    if opSystem == "linux":
        shutil.copy(uriSource, uriDest)
    elif opSystem == "darwin":
        shutil.copy(uriSource, uriDest)
    else:
        shutil.copy(r"\\?\\" + uriSource, r"\\?\\" + uriDest)


    # render out composite texture
    if blenderPath == "":
        bP = panel.quotedPath(str(os.environ['PWD']) + os.sep + "blender")
        command = "%s -b %s -f 1" % (panel.quotedPath(bP), panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "composite.blend"))
    else:
        print(str(panel.quotedPath(blenderPath)))
        command = "%s -b %s -f 1" % (panel.quotedPath(blenderPath), panel.quotedPath(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "composite.blend"))

    panel.launch(exeName=command)

    # set materials
    mat.specular_shader = ("TOON")
    mat.specular_toon_size = 0.2
    mat.specular_toon_smooth = 0.0
    mat.specular_intensity = 0.0
    mat.use_shadeless = False
    mat.use_vertex_color_paint = False
    mat.use_shadows = False

    # setup textures
    if firstTime:
        img_bump = bpy.data.images.load(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "0001.png")
        img_bump.name = "0001_" + NamePDBMLP(tID) + ".png"
        tex_bump = bpy.data.textures.new('bump', type="IMAGE")
        tex_bump.image = img_bump
        mtex = mat.texture_slots.add()
        mtex.texture = tex_bump
        mtex.texture_coords = 'UV'
        mtex.use_map_normal = True
        mtex.use_map_color_diffuse = False
        bpy.data.textures["bump"]
        mat.texture_slots[0].normal_factor = 1
        img_baked = bpy.data.images.load(homePath + "tmp" + os.sep + NamePDBMLP(tID) + os.sep + "MLPBaked.png")
        img_baked.name = "MLPBaked_" + NamePDBMLP(tID) + ".png"
        tex_spec = bpy.data.textures.new('specular', type="IMAGE")
        tex_spec.image = img_baked
        tex_spec.contrast = 4.0
        mtex = mat.texture_slots.add()
        mtex.texture = tex_spec
        mtex.texture_coords = 'UV'
        mtex.use_map_color_diffuse = False
        mtex.use_map_specular = True
        mat.texture_slots[1].use_rgb_to_intensity = True
        mat.texture_slots[1].default_value = 1

    # refresh all images
    for img in bpy.data.images:
        img.reload()

    for area in bpy.context.screen.areas:
        if area.type == 'VIEW_3D':
            area.spaces[0].viewport_shade = "SOLID"
    bpy.ops.object.editmode_toggle()

    ob.data.materials[0] = mat

    ob.data.materials[0].specular_toon_smooth = 0.3
    ob.data.materials[0].texture_slots[0].normal_factor = 1
    ob.data.materials[0].texture_slots[1].default_value = 1
    ob.data.materials[0].texture_slots[1].specular_factor = 0.1

    bpy.ops.paint.vertex_paint_toggle()
    meshData = ob.data
    vColLayer0 = meshData.vertex_colors[0]
    for vCol in vColLayer0.data:
        vCol.color = Color((((vCol.color[0] - 0.5) * 0.6) + 0.5, ((vCol.color[1] - 0.5) * 0.6) + 0.5,
                            ((vCol.color[2] - 0.5) * 0.6) + 0.5))
    meshData.update()
    bpy.ops.paint.vertex_paint_toggle()

    ob.data.materials[0].use_vertex_color_paint = True

    for obj in bpy.context.scene.objects:
        if obj.BBInfo:
            obj.hide = True
            obj.hide_render = True
    panel.ClearLigth(1)


# Wait until process finishes
def wait(process):
    while process.poll() is None:
        time.sleep(0.1)


def DeleteSurface():
    for surface in bpy.data.objects:
        surface.select = False
        if surface.name.split("_")[0] == "SURFACE":
            surface.select = True
    bpy.ops.object.delete()


def getNumFrameMLP():
    global frame
    cifras = len(str(bpy.data.scenes["Scene"].frame_current))
    if cifras == 1:
        frame = "000" + str(bpy.data.scenes["Scene"].frame_current)
    elif cifras == 2:
        frame = "00" + str(bpy.data.scenes["Scene"].frame_current)
    elif cifras == 3:
        frame = "0" + str(bpy.data.scenes["Scene"].frame_current)
    else:
        frame = str(bpy.data.scenes["Scene"].frame_current)
    return frame


def ExistMLP(nm):
    if nm in bpy.data.objects.keys():
        return True
    return False


def NamePDBMLP(tID):
    PDB = ""
    for o1 in bpy.context.scene.objects:
        try:
            if o1.bb2_pdbID == tID:
                if o1.bb2_objectType == "PDBEMPTY":
                    pE = copy.copy(o1.name)
                    PDB = str(pE)
        except Exception as E:
            str3 = str(E)  # Do not print...
            print("Error Name PDB " + str3)
    return PDB.split(".")[0]


if __name__ == "__main__":
    print("MLP module created")