How to take screen shot when running device tests on Grid?

How to take screen shot when running device tests on Grid?

I have four different Android drivers, how do I pass driver uniquely to onTestFailure(ITestResult result) for failures on different devices.