Function main

code/dask/save_KCMb.py:36–55

()

from run_fr import global_matrices, read_tri

def main():
    for i in range(1,8):
        model = f'satellite/satellite.{i}'
        node_xy = read_tri(f'{model}.ele', f'{model}.node')[1]
        K, C, M, b, u_to_c = global_matrices(model)
        keep_nodes = np.argwhere((np.abs(23.5 - node_xy[:,0]) < 1.5) *
                                 (np.abs(10.5 - node_xy[:,1]) < 1.5)).ravel()
        keep_dof_full = sorted(np.hstack([keep_nodes*2, keep_nodes*2+1]))
        keep_dof = sorted([u_to_c[_] for _ in keep_dof_full]) # constrained DOF

#       u_to_c2 = [[_,u_to_c[_]] for _ in u_to_c]
        d = {
                'K'      : K       ,
                'C'      : C       ,
                'M'      : M       ,
                'b'      : b       ,
                'keep_dof' : keep_dof , }
        outF = f'KCMb_{i}.mat'
        scipy.io.savemat(outF, d)
        print(f'wrote {outF}')

if __name__ == "__main__": main()