I tried with below code for Galaxy S4 portrait mode it works. screenshot will save in root folder of project directory
WebDriver driver1 = new Augmenter().augment(driver);
File file = ((TakesScreenshot)driver1).getScreenshotAs(OutputType.FILE);
FileUtils.copyFile(file, new File(“Screenshot.jpg”));