It’s not unlikely that he based it on some (maybe even commonly know at the time) ancient myth. Plenty of civilizations had flood related ones (and there is some historical basis for it). Of course all the government/society descriptions were invented by him