def get_bands(qpoints, frequencies, labels, path_connections, label=None):
qpoints_list = list(qpoints[0])
frequencies_list = list(frequencies[0])
labels_list = [
(0, labels[0]),
]
label_index = 1
for pc, qs, fs in zip(path_connections[:-1], qpoints[1:], frequencies[1:]):
if labels[label_index] == 'GAMMA' and pc:
labels_list.append((len(qpoints_list) - 1, labels[label_index]))
if label_index < len(labels):
labels_list.append((len(qpoints_list), labels[label_index]))
label_index += 1
qpoints_list += list(qs)
frequencies_list += list(fs)
elif pc:
labels_list.append((len(qpoints_list) - 1, labels[label_index]))
label_index += 1
qpoints_list += list(qs[1:])
frequencies_list += list(fs[1:])
else:
labels_list.append((len(qpoints_list) - 1, labels[label_index]))
label_index += 1
if label_index < len(labels):
labels_list.append((len(qpoints_list), labels[label_index]))
label_index += 1
qpoints_list += list(qs)
frequencies_list += list(fs)
labels_list.append((len(qpoints_list) - 1, labels[-1]))
bs = DataFactory('array.bands')()
bs.set_kpoints(np.array(qpoints_list))
bs.set_bands(np.array(frequencies_list), units='THz')
bs.labels = labels_list
if label is not None:
bs.label = label
return bs
