Heads being ok could be a big assumption, I'd be looking at a long block.
Pretty much needs to be a ej20x quad avcs motor, Ideally get the same to ensure cam/compression is OK for the factory tune (might not even be an issue with these ecus), but you could just get the car reflashed with whatever motor you put in to know its safe.