nwb-linkml/nwb_linkml/lang_elements.py

77 lines
2 KiB
Python
Raw Normal View History

"""
Language elements in nwb schema language that have a fixed, alternative representation
in LinkML. These are exported as an nwb.language.yml file along with every generated namespace
"""
from nwb_schema_language.datamodel.nwb_schema_pydantic import FlatDtype as FlatDtype_source
from linkml_runtime.linkml_model import \
ClassDefinition, \
EnumDefinition, \
SchemaDefinition, \
SlotDefinition, \
TypeDefinition,\
Prefix,\
PermissibleValue
from nwb_linkml.maps.dtype import flat_to_linkml
FlatDType = EnumDefinition(
name="FlatDType",
permissible_values=[PermissibleValue(p) for p in FlatDtype_source.__members__.keys()],
)
DimNameSlot = SlotDefinition(
name="dim_name",
range="string",
description="The name of a dimension"
)
DimShapeSlot = SlotDefinition(
name="dim_shape",
range="integer",
required=False
)
DimClass = ClassDefinition(
name="Dimension",
slots=[DimNameSlot.name, DimShapeSlot.name],
description="A single dimension within a shape"
)
DimSlot = SlotDefinition(
name="dim",
range=DimClass.name,
multivalued=True,
description="Slot representing the dimensions that a Shape can have"
)
# ShapeClass = ClassDefinition(
# name="Shape",
# description="A possible shape for an array-like dataset",
# slots=[DimSlot.name]
# )
DTypeTypes = []
for nwbtype, linkmltype in flat_to_linkml.items():
amin = None
if nwbtype.startswith('uint'):
amin = 0
atype = TypeDefinition(
name=nwbtype,
minimum_value=amin,
typeof=linkmltype
)
DTypeTypes.append(atype)
NwbLangSchema = SchemaDefinition(
name="nwb.language",
id='nwb.language',
description="Adapter objects to mimic the behavior of elements in the nwb-schema-language",
enums=[FlatDType],
slots=[DimNameSlot, DimShapeSlot, DimSlot],
classes=[DimClass],
types=DTypeTypes,
imports=['linkml:types'],
prefixes={'linkml': Prefix('linkml','https://w3id.org/linkml')}
)