From 6949be161d94665345250427505a11e0a8a32e67 Mon Sep 17 00:00:00 2001
From: antoniolrm <am@antoniomanesco.org>
Date: Wed, 25 Oct 2023 00:14:10 +0200
Subject: [PATCH] add metadate for kwant-based systems
---
codes/utils.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/codes/utils.py b/codes/utils.py
index 65c27ae..b513320 100644
--- a/codes/utils.py
+++ b/codes/utils.py
@@ -68,6 +68,7 @@ def builder2tb_model(builder):
tb_model = {}
sites_list = [*builder.sites()]
norbs_list = [site[0].norbs for site in builder.sites()]
+ positions_list = [site[0].pos for site in builder.sites()]
norbs_tot = sum(norbs_list)
for hop, val in builder.hopping_value_pairs():
a, b = hop
@@ -89,6 +90,9 @@ def builder2tb_model(builder):
tb_model[tuple(b_dom)] = coo_array(
(data, (row, col)), shape=(norbs_tot, norbs_tot)
).toarray()
+ tb_model['norbs'] = norbs_list
+ tb_model['positions'] = positions_list
+
return tb_model
--
GitLab