The treasurer says Australia's foreign investment settings won't discriminate against certain countries, even as his reforms respond to geopolitical threats, the energy transition and shifts in the global economy.