Source code for dpdata.plugins.list
from dpdata.format import Format
[docs]
@Format.register("list")
class ListFormat(Format):
[docs]
def to_system(self, data, **kwargs):
"""Convert system to list, usefull for data collection."""
from dpdata import LabeledSystem, System
if "forces" in data:
system = LabeledSystem(data=data)
else:
system = System(data=data)
if len(system) == 0:
return []
if len(system) == 1:
return [system]
else:
systems = []
for ii in range(len(system)):
systems.append(system.sub_system([ii]))
return systems