streamline logic for setting waveform colors