Sure! It is in matlab (I know...) so I assume you may want to ask an llm to translate it to python or sth but it should be fairly simple

https://pastebin.com/n1mvcxP1