It could be, but it needs significant effort to modify the codes. Please explore more.
However, please think it over if 3 or 4 values are practical, since ultimately, you will need to convert to real materials. You may consider dispersive material if needed.
The online example is for binary, either has the waveguide material, or no, with fixed refractive index.
You are navigating away from the AIS Discovery experience