I like this idea, but as others have pointed out it would get really tricky to maintain that as you adjusted the other controls…
But I’m imagining a “unity” switch that would engage the following behavior: the drive block monitors the signal level at its input (peak or RMS, idk which makes more sense), and compares that to the same measurement at the output. Then it applies the necessary positive/negative dB gain to eliminate the difference. Kinda the way “auto make up” works in the compressor block.
Could also be useful in the amp block…