diff options
author | Aleksandr <ognelisar@gmail.com> | 2017-10-17 12:59:45 +0700 |
---|---|---|
committer | Aleksandr <ognelisar@gmail.com> | 2017-10-17 12:59:45 +0700 |
commit | 55756e6206822ca1784f91e3266c48adc084905f (patch) | |
tree | 8095ef7157edbbb7e0997840eb654eb23278abb6 /src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala | |
parent | cf68e1eb3b352d8c5045dea703d4ae3fe9c46543 (diff) | |
parent | c5d36593cc14bfdfb7b8f06d680ad6fc2eb6dde0 (diff) | |
download | rest-query-55756e6206822ca1784f91e3266c48adc084905f.tar.gz rest-query-55756e6206822ca1784f91e3266c48adc084905f.tar.bz2 rest-query-55756e6206822ca1784f91e3266c48adc084905f.zip |
Merge branch 'master' into PDSUI-rep-spray
Diffstat (limited to 'src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala')
-rw-r--r-- | src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala | 101 |
1 files changed, 0 insertions, 101 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala deleted file mode 100644 index 70ddd10..0000000 --- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala +++ /dev/null @@ -1,101 +0,0 @@ -package xyz.driver.pdsuidomain.fakes.entities.rep - -import xyz.driver.core.generators -import xyz.driver.core.generators._ -import xyz.driver.entities.labels.{Label, LabelCategory, LabelValue} -import xyz.driver.pdsuicommon.domain.{LongId, TextJson} -import xyz.driver.pdsuidomain.entities._ -import xyz.driver.pdsuidomain.entities.ExtractedData.Meta -import xyz.driver.pdsuidomain.fakes.entities.common._ -import xyz.driver.pdsuidomain.services.ExtractedDataService.RichExtractedData - -object ExtractedDataGen { - private val maxItemsInCollectionNumber = 50 - - private val maxPageNumber = 100 - private val maxIndexNumber = 100 - private val maxOffsetNumber = 10 - - implicit private class TextLayerPositionOrdering(textLayerPosition: ExtractedData.Meta.TextLayerPosition) - extends Ordered[ExtractedData.Meta.TextLayerPosition] { - override def compare(that: Meta.TextLayerPosition): Int = { - if (this.textLayerPosition.page < that.page) -1 - else if (this.textLayerPosition.page > that.page) 1 - else if (this.textLayerPosition.index < that.index) -1 - else if (this.textLayerPosition.index > that.index) 1 - else if (this.textLayerPosition.offset < that.offset) -1 - else if (this.textLayerPosition.offset > that.offset) 1 - else 0 - } - } - - def nextExtractedDataMetaKeyword(): Meta.Keyword = { - ExtractedData.Meta.Keyword( - page = nextInt(maxPageNumber, minValue = 0), - pageRatio = nextOption(nextDouble()), - index = nextInt(maxIndexNumber, minValue = 0), - sortIndex = nextString() - ) - } - - def nextExtractedDataMetaTextLayerPosition(): Meta.TextLayerPosition = { - ExtractedData.Meta.TextLayerPosition( - page = nextInt(maxPageNumber, minValue = 0), - index = nextInt(maxIndexNumber, minValue = 0), - offset = nextInt(maxOffsetNumber, minValue = 0) - ) - } - - def nextExtractedDataMetaEvidence(): Meta.Evidence = { - val layersPosition = - genBoundedRange[ExtractedData.Meta.TextLayerPosition]( - nextExtractedDataMetaTextLayerPosition(), - nextExtractedDataMetaTextLayerPosition() - ) - - ExtractedData.Meta.Evidence( - pageRatio = nextDouble(), - start = layersPosition._1, - end = layersPosition._2 - ) - } - - def nextExtractedDataMeta(): Meta = { - ExtractedData.Meta( - nextExtractedDataMetaKeyword(), - nextExtractedDataMetaEvidence() - ) - } - - def nextExtractedDataMetaJson(): TextJson[Meta] = - TextJson(ExtractedData.Meta(nextExtractedDataMetaKeyword(), nextExtractedDataMetaEvidence())) - - def nextExtractedData(documentId: LongId[Document]): ExtractedData = { - ExtractedData( - id = nextLongId[ExtractedData], - documentId = documentId, - keywordId = nextOption(nextLongId[xyz.driver.pdsuidomain.entities.Keyword]), - evidenceText = nextOption(nextString()), - meta = nextOption(nextExtractedDataMetaJson()) - ) - } - - def nextExtractedDataLabel(): ExtractedDataLabel = { - ExtractedDataLabel( - id = nextLongId[ExtractedDataLabel], - dataId = nextLongId[ExtractedData], - labelId = nextOption(nextLongId[Label]), - categoryId = nextOption(nextLongId[LabelCategory]), - value = nextOption(generators.oneOf[LabelValue](LabelValue.Yes, LabelValue.No, LabelValue.Maybe)) - ) - } - - def nextRichExtractedData(documentId: LongId[Document] = nextLongId): RichExtractedData = { - RichExtractedData( - extractedData = nextExtractedData(documentId), - labels = List.fill( - nextInt(maxItemsInCollectionNumber, minValue = 0) - )(nextExtractedDataLabel()) - ) - } -} |