At a guess, you can reuse the entropy part, but you'd need to plug in a new probability distribution.