Split the single utils.py file into a package of modules.

Author: angavrilov

utils.py

@@ -0,0 +1,33 @@
+# These forwarding imports are for backwards compatibility with legacy code
+# that expects a single utils.py file. New code should import directly from
+# the modules that contain the utilities. Also, don't add more imports here.
+
+from .errors import MetarigError
+
+from .misc import angle_on_plane, linsrgb_to_srgb, gamma_correct, copy_attributes
+
+from .naming import ORG_PREFIX, MCH_PREFIX, DEF_PREFIX, ROOT_NAME
+from .naming import strip_trailing_number, unique_name, org_name, strip_org, strip_mch, strip_def
+from .naming import org, make_original_name, mch, make_mechanism_name, deformer, make_deformer_name
+from .naming import insert_before_lr, random_id
+
+from .bones import new_bone, copy_bone_simple, copy_bone, flip_bone, put_bone, make_nonscaling_child
+from .bones import align_bone_roll, align_bone_x_axis, align_bone_z_axis, align_bone_y_axis
+
+from .widgets import WGT_PREFIX,  WGT_LAYERS, obj_to_bone, create_widget, write_widget, create_circle_polygon
+
+from .widgets_basic import create_line_widget, create_circle_widget, create_cube_widget, create_chain_widget
+from .widgets_basic import create_sphere_widget, create_limb_widget, create_bone_widget
+
+from .widgets_special import create_compass_widget, create_root_widget
+from .widgets_special import create_neck_bend_widget, create_neck_tweak_widget
+
+from .animation import get_keyed_frames, bones_in_frame, overwrite_prop_animation
+
+from .glue import make_constraints_from_string
+
+from .rig import RIG_DIR, METARIG_DIR, MODULE_NAME, outdated_types, upgradeMetarigTypes
+from .rig import get_rig_type, get_metarig_module, write_metarig
+from .rig import connected_children_names, has_connected_children
+
+from .layers import get_layers

utils/__init__.py

@@ -0,0 +1,84 @@
+#====================== 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 2
+#  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, write to the Free Software Foundation,
+#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+#
+#======================= END GPL LICENSE BLOCK ========================
+
+# <pep8 compliant>
+
+
+#=============================================
+# Keyframing functions
+#=============================================
+
+
+def get_keyed_frames(rig):
+    frames = []
+    if rig.animation_data:
+        if rig.animation_data.action:
+            fcus = rig.animation_data.action.fcurves
+            for fc in fcus:
+                for kp in fc.keyframe_points:
+                    if kp.co[0] not in frames:
+                        frames.append(kp.co[0])
+
+    frames.sort()
+
+    return frames
+
+
+def bones_in_frame(f, rig, *args):
+    """
+    True if one of the bones listed in args is animated at frame f
+    :param f: the frame
+    :param rig: the rig
+    :param args: bone names
+    :return:
+    """
+
+    if rig.animation_data and rig.animation_data.action:
+        fcus = rig.animation_data.action.fcurves
+    else:
+        return False
+
+    for fc in fcus:
+        animated_frames = [kp.co[0] for kp in fc.keyframe_points]
+        for bone in args:
+            if bone in fc.data_path.split('"') and f in animated_frames:
+                return True
+
+    return False
+
+
+def overwrite_prop_animation(rig, bone, prop_name, value, frames):
+    act = rig.animation_data.action
+    if not act:
+        return
+
+    bone_name = bone.name
+    curve = None
+
+    for fcu in act.fcurves:
+        words = fcu.data_path.split('"')
+        if words[0] == "pose.bones[" and words[1] == bone_name and words[-2] == prop_name:
+            curve = fcu
+            break
+
+    if not curve:
+        return
+
+    for kp in curve.keyframe_points:
+        if kp.co[0] in frames:
+            kp.co[1] = value