I know this is an old thread, but I have written a library for this screen. It provides an 'Adafruit_GFX' style graphics interface, and touchscreen support that's reasonably easy to use. You can find it here:
I've run it on an Uno, a Mega, and a Wemos D1 R1 (but that required re-wiring). It is substantially faster than the one the Waveshare website points people to, and has more functionality.