SofaPython3/stlib/__init__.py at 6721f3f1a37a3f9936c356115578d9eda62197e7 · sofa-framework/SofaPython3 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
__all__ = ["core","entities","prefabs","shapes"]

import Sofa.Core

from stlib.core.baseParameters import BaseParameters
from stlib.core.basePrefab import BasePrefab


def __genericAdd(self : Sofa.Core.Node, typeName, *param, **kwargs):


    def findName(cname, names):
        """Compute a working unique name in the node"""
        rname = cname
        for i in range(0, len(names)):
            if rname not in names:
                return rname
            rname = cname + str(i+1)
        return rname

    def checkName(context : Sofa.Core.Node, name):
        # Check if the name already exists, if this happens, create a new one.
        if name in context.children or name in context.objects:
            names = {node.name.value for node in context.children}
            names = names.union({object.name.value for object in context.objects})
            name = findName(name, names)
        return name


    if len(param) == 1 and isinstance(typeName, type) and not issubclass(typeName, BasePrefab):
        raise RuntimeError("Invalid argument : only prefabs take positionnal argument which should be a parameter")
    elif len(param) > 1:
        raise RuntimeError("Invalid argument : only one positionnal argument accepted and only when used with a prefab")
    elif len(param) == 0 and isinstance(typeName, type) and issubclass(typeName, BasePrefab):
        raise RuntimeError("Invalid argument : one positionnal argument is required when calling add with prefab type")
    elif len(param) == 1 and isinstance(typeName, type) and issubclass(typeName, BasePrefab) and not isinstance(param[0], BaseParameters):
        raise RuntimeError("Invalid argument : when calling add with prefab type the positionnal argument is expected to be a type derived from stlib.core.BaseParameter")


    if len(param) == 1 and isinstance(typeName, type) and issubclass(typeName, BasePrefab):
        param[0].name = checkName(self, param[0].name)
        if(len(kwargs)):
            param[0].kwargs = kwargs.copy()

        newEntity = self.addChild(typeName(param[0]))
        newEntity.init()
        return newEntity

    ## If we ever get here, it means we are not adding a prefab by giving its type name and its parameter set

    # Check if a name is provided, if not, use the one of the class
    params = kwargs.copy()
    isNode = False
    if "name" not in params:
        if isinstance(typeName, str):
            params["name"] = typeName
            isNode=True
        elif isinstance(typeName, type) and issubclass(typeName, Sofa.Core.Node):
            params["name"] = "Node"
            isNode=True
        elif isinstance(typeName, Sofa.Core.Node):
            params["name"] = "Node"
            isNode=True
        elif isinstance(typeName, type) and issubclass(typeName, Sofa.Core.Object):
            params["name"] = typeName.name.value
        elif isinstance(typeName, type) and issubclass(typeName, Sofa.Core.ObjectDeclaration):
            params["name"] = typeName.__name__
        else:
            raise RuntimeError("Invalid argument ", typeName)

    params["name"] = checkName(self, params["name"])

    # Dispatch the creation to either addObject or addChild
    if isinstance(typeName, type) and issubclass(typeName, Sofa.Core.Node):
        pref = self.addChild(typeName(params["name"]))
    elif isinstance(typeName, Sofa.Core.Node):
        pref = self.addChild(typeName)
    elif isinstance(typeName, type) and issubclass(typeName, Sofa.Core.Object):
        pref = self.addObject(typeName(**params))
    elif isinstance(typeName, type) and issubclass(typeName, Sofa.Core.ObjectDeclaration):
        pref = self.addObject(typeName.__name__, **params)
    elif isinstance(typeName, str):
        pref = self.addObject(typeName, **params)
    else:
        raise RuntimeError("Invalid argument", typeName)
    return pref

# Inject the method so it become available as if it was part of Sofa.Core.Node
Sofa.Core.Node.add = __genericAdd