com.applitools.eyes.playwright.universal.mapper.PlaywrightOCRExtractSettingsDtoMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of eyes-playwright-java5 Show documentation

Applitools Eyes SDK for Playwright Java

There is a newer version: 5.70.6

Show newest version

package com.applitools.eyes.playwright.universal.mapper;

import com.applitools.eyes.locators.BaseOcrRegion;
import com.applitools.eyes.playwright.locators.OcrRegion;
import com.applitools.eyes.playwright.universal.Refer;
import com.applitools.eyes.playwright.universal.dto.Element;
import com.applitools.eyes.playwright.universal.dto.Selector;
import com.applitools.eyes.universal.Reference;
import com.applitools.eyes.universal.dto.OCRExtractSettingsDto;
import com.applitools.eyes.universal.mapper.RectangleRegionMapper;

import java.util.List;
import java.util.stream.Collectors;

public class PlaywrightOCRExtractSettingsDtoMapper {

    public static OCRExtractSettingsDto toOCRExtractSettingsDto(BaseOcrRegion baseOcrRegion, Refer refer, Reference root) {
        if (!(baseOcrRegion instanceof OcrRegion)) {
            return null;
        }

        OcrRegion ocrRegion = (OcrRegion) baseOcrRegion;


        OCRExtractSettingsDto ocrExtractSettingsDto = new OCRExtractSettingsDto();
        if (ocrRegion.getSelector() != null) {
            Selector selector = new Selector(ocrRegion.getSelector());
            ocrExtractSettingsDto.setRegion(selector);
        } else if (ocrRegion.getLocator() != null) {
            Selector selector = new Selector(ocrRegion.getLocator());
            selector.setApplitoolsRefId(refer.ref(ocrRegion.getLocator(), root));
            ocrExtractSettingsDto.setRegion(selector);
        } else if (ocrRegion.getElement() != null) {
            Element element = new Element(ocrRegion.getElement());
            element.setApplitoolsRefId(refer.ref(ocrRegion.getElement(), root));
            ocrExtractSettingsDto.setRegion(element);
        } else if (ocrRegion.getRegion() != null) {
            ocrExtractSettingsDto.setRegion(RectangleRegionMapper.toRectangleRegionDto(ocrRegion.getRegion()));
        }

        ocrExtractSettingsDto.setHint(ocrRegion.getHint());
        ocrExtractSettingsDto.setMinMatch(ocrRegion.getMinMatch());
        ocrExtractSettingsDto.setLanguage(ocrRegion.getLanguage());
        return ocrExtractSettingsDto;
    }


    public static List toOCRExtractSettingsDtoList(List baseOcrRegionList, Refer refer, Reference root) {
        if (baseOcrRegionList == null || baseOcrRegionList.isEmpty()) {
            return null;
        }

        return baseOcrRegionList.stream()
                .map(baseOcrRegion -> toOCRExtractSettingsDto(baseOcrRegion, refer, root))
                .collect(Collectors.toList());
    }
}