Wpf style datacontext?

How would I go about having the focused button be determined by the datacontext? For both styles split the ControlTemplate in 2, one with an expander and one with out. What i want to achieve is something like this: