¿Hay alguna forma de hacer que la fila cambie el fondo haciendo clic en el botón? Tengo el siguiente código:

 rowClassRules: { "a-grade": params => params.api.getValue("Grade", params.node) == 'A', "b-grade": params => params.api.getValue("Grade", params.node) == 'B', "c-grade": params => params.api.getValue("Grade", params.node) == 'C', },

Cambia el color de la fila según el valor de la columna Grado. Pero quiero hacer lo mismo solo después de presionar un botón, y también otro botón debería apagarlo (como Activar/Desactivar resaltado). No tengo idea de cómo convertir este código en una función que pueda usar al hacer clic en el botón.