Да, можно и так. Сделать виртуальную функцию для отрисовки (вызываемую в базовом классе), и передавать ей контекст. То есть, то, как ты до этого сделал, в принципе верно, но "мясо" надо было в виртуальную функцию перенести, а не передавать параметр, который говорит, что рисовать надо. Тогда и принципы ООП не пострадают