ru.yandex.qatools.ashot.screentaker.HeadCuttingShootingStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ashot Show documentation

There is a newer version: 1.5.4

package ru.yandex.qatools.ashot.screentaker;

import org.openqa.selenium.WebDriver;

import java.awt.image.BufferedImage;

/**
 * @author Pavel Zorin
 */

public class HeadCuttingShootingStrategy extends ShootingStrategy {

    protected int headerToCut;

    public HeadCuttingShootingStrategy(int headerToCut) {
        this.headerToCut = headerToCut;
    }

    @Override
    public BufferedImage getScreenshot(WebDriver wd) {
        BufferedImage baseImage = simple().getScreenshot(wd);
        int h = baseImage.getHeight();
        int w = baseImage.getWidth();
        return baseImage.getSubimage(0, headerToCut, w, h - headerToCut);
    }
}