python: store group velocities and fields at every k-point

[mahar]

Hello,

I am trying to calculate and store in arrays the fields and group velocities at every k-point.
Probably I don't understand exactly the way the band functions are working.

Currently, I am using the following code (and add more k-points manually

efieldsG = []
groupVelocG = []
efieldsX = []
groupVelocX = []

def get_efieldsG(ms,band):
    efieldsG.append(ms.get_efield(band,bloch_phase=True))
    groupVelocG.append(ms.compute_one_group_velocity(band))

def get_efieldsX(ms,band):
    efieldsX.append(ms.get_efield(band,bloch_phase=True))
    groupVelocX.append(ms.compute_one_group_velocity(band))

ms.run_tm(mpb.output_at_kpoint(G,mpb.fix_efield_phase,get_efieldsG),
          mpb.output_at_kpoint(X,mpb.fix_efield_phase,get_efieldsX))

Thank you