I want to use the BCPD algorithm to implement the rigid transformation, how to modify bcpd_nonrigid.py to achieve the rigid transformation