com.applitools.eyes.selenium.universal.mapper.OCRExtractSettingsDtoMapper Maven / Gradle / Ivy

Go to download

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

Applitools Eyes SDK for Selenium Java WebDriver

There is a newer version: 5.73.0

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

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

import com.applitools.eyes.locators.BaseOcrRegion;
import com.applitools.eyes.locators.OcrRegion;
import com.applitools.eyes.universal.dto.OCRExtractSettingsDto;
import com.applitools.eyes.universal.mapper.RectangleRegionMapper;

/**
 * OCRExtractSettingsDtoMapper
 */
public class OCRExtractSettingsDtoMapper {

  public static OCRExtractSettingsDto toOCRExtractSettingsDto(BaseOcrRegion baseOcrRegion) {
    if (!(baseOcrRegion instanceof OcrRegion)) {
      return null;
    }

    OcrRegion ocrRegion = (OcrRegion) baseOcrRegion;


    OCRExtractSettingsDto ocrExtractSettingsDto = new OCRExtractSettingsDto();
    if (ocrRegion.getSelector() != null) {
      ocrExtractSettingsDto.setRegion(SelectorRegionMapper.toSelectorRegionDto(ocrRegion.getSelector()));
    } else if (ocrRegion.getElement() != null) {
      ocrExtractSettingsDto.setRegion(ElementRegionMapper.toElementRegionDto(ocrRegion.getElement()));
    } 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) {
    if (baseOcrRegionList == null || baseOcrRegionList.isEmpty()) {
      return null;
    }

    return baseOcrRegionList.stream().map(OCRExtractSettingsDtoMapper::toOCRExtractSettingsDto).collect(Collectors.toList());
  }
}