How to custom the animation editor

I want to custom the animation editor like drawing some lines in the preview scene or add some display text showing on the screen.What is the best way?Do I need to inherit and register some class or just edit the source code of FAnimationEditor and other class of AnimationEditorModule?