How to write class-dependant transformations?

vision
2

This code snippet might be what you are looking for.