framework/cg/blender/scripts/addons/Robossembler/__init__.py

#  ***** 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 3
#  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, see <http://www.gnu.org/licenses/>
#  and write to the Free Software Foundation, Inc., 51 Franklin Street,
#  Fifth Floor, Boston, MA  02110-1301, USA..
#
#  The Original Code is Copyright (C) 2023 by Kurochkin Ilia  ###
#  All rights reserved.
#
#  Contact:      brothermechanic@yandex.com    ###
#  Information:  https://gitlab.com/robossembler   ###
#
#  The Original Code is: all of this file.
#
#  ***** END GPL LICENSE BLOCK *****
#
# -*- coding: utf-8 -*-

import os
import bpy

from bpy.types import (
    Panel,
    Operator,
    PropertyGroup)

from bpy.props import (
    StringProperty,
    EnumProperty,
    PointerProperty)

from .io_scene_json import export_json
from .io_anim_ros2bag import set_animation_data

bl_info = {
    'name': 'Robossembler Tools',
    'author': 'brothermechanic@gmail.com',
    'version': (0, 2),
    'blender': (4, 2, 0),
    'location': '3D View > Toolbox',
    'description': 'Robossembler pipeline tools',
    'warning': '',
    'wiki_url': '',
    'tracker_url': 'https://gitlab.com/robossembler',
    'category': 'Robossembler',
}


class addon_Properties(PropertyGroup):

    engine: EnumProperty(
        name='Physics Engine',
        description='Selest Target Engine',
        items=[('BULLET', 'Bullet', ''),
               ('ODE', 'O D E', ''),
               ('SIMBODY', 'Simbody', ''),
               ('OPENSIM', 'OpenSim', '')
               ]
    )

    conf_file_path: StringProperty(
        name='Config File Path',
        description='Input/output config file path',
        default='',
        maxlen=1023,
        subtype='FILE_PATH'
    )


class RobossemblerPanel1(Panel):
    ''' Robossembler UI'''
    bl_idname = 'ROBOSSEMBLER_PT_EXPORT_JSON'
    bl_label = 'Export Scene as Json'
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = 'Robossembler'

    def draw(self, context):
        prop = context.scene.robossembler_properties

        layout = self.layout
        layout.prop(prop, 'conf_file_path')
        layout.prop(prop, 'engine')

        col = layout.column()
        col.alert = True
        col.scale_y = 2.0
        col.operator('export.scene_config',
                     icon='WORLD_DATA',
                     text='Export Scene')


class RobossemblerPanel2(Panel):
    '''Doc'''
    bl_idname = 'ROBOSSEMBLER_PT_IMPORT_ANIMATION'
    bl_label = 'Import Ros2Bag Animation'
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = 'Robossembler'

    def draw(self, context):
        prop = context.scene.robossembler_properties

        layout = self.layout
        layout.prop(prop, 'conf_file_path')

        col = layout.column()
        col.alert = True
        col.scale_y = 2.0
        col.operator('export.scene_config',
                     icon='ACTION',
                     text='Import Animation')


class RobossemblerOperator1(Operator):
    '''Tooltip'''
    bl_idname = 'export.scene_config'
    bl_label = ''
    bl_description = 'Export scene liks to json config.'
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):
        prop = context.scene.robossembler_properties

        file_path = os.path.realpath(bpy.path.abspath((prop.conf_file_path)))
        physics_engine = prop.engine
        export_json(context, file_path, physics_engine)

        return {'FINISHED'}


class RobossemblerOperator2(Operator):
    '''Tooltip'''
    bl_idname = 'export.scene_config'
    bl_label = ''
    bl_description = 'Export scene liks to json config.'
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):
        prop = context.scene.robossembler_properties

        file_path = os.path.realpath(bpy.path.abspath((prop.conf_file_path)))
        set_animation_data(context, file_path)

        return {'FINISHED'}


classes = (
    RobossemblerPanel1,
    RobossemblerPanel2,
    RobossemblerOperator1,
    RobossemblerOperator2,
    addon_Properties)


def register():
    for cls in classes:
        bpy.utils.register_class(cls)
    bpy.types.Scene.robossembler_properties = PointerProperty(type=addon_Properties)


def unregister():
    for cls in classes:
        bpy.utils.unregister_class(cls)
    del bpy.types.Scene.robossembler_properties


if __name__ == '__main__':
    register()