Yes, you can use profile of a surface to control the groove and the ball hex, with reference to datum features A and B. For defining the true profile the tolerance zones will be based on, you will need to either define the geometry of the ball hex and the groove with basic dimensions, or include the CAD model in your product definition data set, and refer to it from a note on the drawing.