1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744 |
- //
- // ImgprocTest.swift
- //
- // Created by Giles Payne on 2020/02/08.
- //
- import XCTest
- import OpenCV
- class ImgprocTest: OpenCVTestCase {
- let anchorPoint = Point(x: 2, y: 2)
- let imgprocSz: Int32 = 2
- let size = Size(width: 3, height: 3)
- func testAccumulateMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulate(src: src, dst: dst)
- Imgproc.accumulate(src: src, dst: dst2)
- try assertMatEqual(src, dst, OpenCVTestCase.EPS)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [4]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateMatMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let mask = makeMask(getMat(CvType.CV_8U, vals: [1]))
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulate(src: src, dst: dst, mask: mask)
- Imgproc.accumulate(src: src, dst: dst2, mask: mask)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [2])), dst, OpenCVTestCase.EPS)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [4]), vals: [2]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateProductMatMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulateProduct(src1: src, src2: src, dst: dst)
- Imgproc.accumulateProduct(src1: src, src2: dst, dst: dst2)
- try assertMatEqual(getMat(CvType.CV_64F, vals:[4]), dst, OpenCVTestCase.EPS)
- try assertMatEqual(getMat(CvType.CV_64F, vals:[10]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateProductMatMatMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let mask = makeMask(getMat(CvType.CV_8U, vals: [1]))
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulateProduct(src1: src, src2: src, dst: dst, mask: mask)
- Imgproc.accumulateProduct(src1: src, src2: dst, dst: dst2, mask: mask)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [4])), dst, OpenCVTestCase.EPS)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [10]), vals:[2]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateSquareMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulateSquare(src: src, dst: dst)
- Imgproc.accumulateSquare(src: src, dst: dst2)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [4]), dst, OpenCVTestCase.EPS)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [6]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateSquareMatMatMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let mask = makeMask(getMat(CvType.CV_8U, vals: [1]))
- let dst = getMat(CvType.CV_64F, vals: [0])
- let dst2 = src.clone()
- Imgproc.accumulateSquare(src: src, dst: dst, mask: mask)
- Imgproc.accumulateSquare(src: src, dst: dst2, mask: mask)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [4])), dst, OpenCVTestCase.EPS)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [6]), vals: [2]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateWeightedMatMatDouble() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let dst = getMat(CvType.CV_64F, vals: [4])
- let dst2 = src.clone()
- Imgproc.accumulateWeighted(src: src, dst: dst, alpha: 0.5)
- Imgproc.accumulateWeighted(src: src, dst: dst2, alpha: 2)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [3]), dst, OpenCVTestCase.EPS)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [2]), dst2, OpenCVTestCase.EPS)
- }
- func testAccumulateWeightedMatMatDoubleMat() throws {
- let src = getMat(CvType.CV_64F, vals: [2])
- let mask = makeMask(getMat(CvType.CV_8U, vals: [1]))
- let dst = getMat(CvType.CV_64F, vals: [4])
- let dst2 = src.clone()
- Imgproc.accumulateWeighted(src: src, dst: dst, alpha: 0.5, mask: mask)
- Imgproc.accumulateWeighted(src: src, dst: dst2, alpha: 2, mask: mask)
- try assertMatEqual(makeMask(getMat(CvType.CV_64F, vals: [3]), vals: [4]), dst, OpenCVTestCase.EPS)
- try assertMatEqual(getMat(CvType.CV_64F, vals: [2]), dst2, OpenCVTestCase.EPS)
- }
- func testAdaptiveThreshold() {
- let src = makeMask(getMat(CvType.CV_8U, vals: [50]), vals:[20])
- let dst = Mat()
- Imgproc.adaptiveThreshold(src: src, dst: dst, maxValue: 1, adaptiveMethod: .ADAPTIVE_THRESH_MEAN_C, thresholdType: .THRESH_BINARY, blockSize: 3, C: 0)
- XCTAssertEqual(src.rows(), Core.countNonZero(src: dst))
- }
- func testApproxPolyDP() {
- let curve = [Point2f(x: 1, y: 3), Point2f(x: 2, y: 4), Point2f(x: 3, y: 5), Point2f(x: 4, y: 4), Point2f(x: 5, y: 3)]
- var approxCurve = [Point2f]()
- Imgproc.approxPolyDP(curve: curve, approxCurve: &approxCurve, epsilon: OpenCVTestCase.EPS, closed: true)
- let approxCurveGold = [Point2f(x: 1, y: 3), Point2f(x: 3, y: 5), Point2f(x: 5, y: 3)]
- XCTAssert(approxCurve == approxCurveGold)
- }
- func testArcLength() {
- let curve = [Point2f(x: 1, y: 3), Point2f(x: 2, y: 4), Point2f(x: 3, y: 5), Point2f(x: 4, y: 4), Point2f(x: 5, y: 3)]
- let arcLength = Imgproc.arcLength(curve: curve, closed: false)
- XCTAssertEqual(5.656854249, arcLength, accuracy:0.000001)
- }
- func testBilateralFilterMatMatIntDoubleDouble() throws {
- Imgproc.bilateralFilter(src: gray255, dst: dst, d: 5, sigmaColor: 10, sigmaSpace: 5)
- try assertMatEqual(gray255, dst)
- }
- func testBilateralFilterMatMatIntDoubleDoubleInt() throws {
- Imgproc.bilateralFilter(src: gray255, dst: dst, d: 5, sigmaColor: 10, sigmaSpace: 5, borderType: .BORDER_REFLECT)
- try assertMatEqual(gray255, dst)
- }
- func testBlurMatMatSize() throws {
- Imgproc.blur(src: gray0, dst: dst, ksize: size)
- try assertMatEqual(gray0, dst)
- Imgproc.blur(src: gray255, dst: dst, ksize: size)
- try assertMatEqual(gray255, dst)
- }
- func testBlurMatMatSizePoint() throws {
- Imgproc.blur(src: gray0, dst: dst, ksize: size, anchor: anchorPoint)
- try assertMatEqual(gray0, dst)
- }
- func testBlurMatMatSizePointInt() throws {
- Imgproc.blur(src: gray0, dst: dst, ksize: size, anchor: anchorPoint, borderType: .BORDER_REFLECT)
- try assertMatEqual(gray0, dst)
- }
- func testBoundingRect() {
- let points = [Point(x: 0, y: 0), Point(x: 0, y: 4), Point(x: 4, y: 0), Point(x: 4, y: 4)]
- let p1 = Point(x: 1, y: 1)
- let p2 = Point(x: -5, y: -2)
- let bbox = Imgproc.boundingRect(array: MatOfPoint(array: points))
- XCTAssert(bbox.contains(p1))
- XCTAssertFalse(bbox.contains(p2))
- }
- func testBoxFilterMatMatIntSize() throws {
- let size = Size(width: 3, height: 3)
- Imgproc.boxFilter(src: gray0, dst: dst, ddepth: 8, ksize: size)
- try assertMatEqual(gray0, dst)
- }
- func testBoxFilterMatMatIntSizePointBoolean() throws {
- Imgproc.boxFilter(src: gray255, dst: dst, ddepth: 8, ksize: size, anchor: anchorPoint, normalize: false)
- try assertMatEqual(gray255, dst)
- }
- func testBoxFilterMatMatIntSizePointBooleanInt() throws {
- Imgproc.boxFilter(src: gray255, dst: dst, ddepth: 8, ksize: size, anchor: anchorPoint, normalize: false, borderType: .BORDER_REFLECT)
- try assertMatEqual(gray255, dst)
- }
- func testCalcBackProject() {
- let images = [grayChess]
- let channels:[Int32] = [0]
- let histSize:[Int32] = [10]
- let ranges:[Float] = [0, 256]
- let hist = Mat()
- Imgproc.calcHist(images: images, channels: channels, mask: Mat(), hist: hist, histSize: histSize, ranges: ranges)
- Core.normalize(src: hist, dst: hist)
- Imgproc.calcBackProject(images: images, channels: channels, hist: hist, dst: dst, ranges: ranges, scale: 255)
- XCTAssertEqual(grayChess.size(), dst.size())
- XCTAssertEqual(grayChess.depth(), dst.depth())
- XCTAssertFalse(0 == Core.countNonZero(src: dst))
- }
- func testCalcHistListOfMatListOfIntegerMatMatListOfIntegerListOfFloat() throws {
- let images = [gray128]
- let channels:[Int32] = [0]
- let histSize:[Int32] = [10]
- let ranges:[Float] = [0, 256]
- let hist = Mat()
- Imgproc.calcHist(images: images, channels: channels, mask: Mat(), hist: hist, histSize: histSize, ranges: ranges)
- truth = Mat(rows: 10, cols: 1, type: CvType.CV_32F, scalar: Scalar.all(0))
- try truth!.put(row: 5, col: 0, data: [100] as [Float])
- try assertMatEqual(truth!, hist, OpenCVTestCase.EPS)
- }
- func testCalcHistListOfMatListOfIntegerMatMatListOfIntegerListOfFloat2D() throws {
- let images = [gray255, gray128]
- let channels:[Int32] = [0, 1]
- let histSize:[Int32] = [10, 10]
- let ranges:[Float] = [0, 256, 0, 256]
- let hist = Mat()
- Imgproc.calcHist(images: images, channels: channels, mask: Mat(), hist: hist, histSize: histSize, ranges: ranges)
- truth = Mat(rows: 10, cols: 10, type: CvType.CV_32F, scalar: Scalar.all(0))
- try truth!.put(row: 9, col: 5, data: [100] as [Float])
- try assertMatEqual(truth!, hist, OpenCVTestCase.EPS)
- }
- func testCalcHistListOfMatListOfIntegerMatMatListOfIntegerListOfFloat3D() throws {
- let images = [rgbLena]
- let hist3D = Mat()
- let histList = [Mat(), Mat(), Mat()]
- let histSize: [Int32] = [10]
- let ranges: [Float] = [0, 256]
- for i:Int in 0..<Int(rgbLena.channels()) {
- Imgproc.calcHist(images: images, channels: [Int32(i)], mask: Mat(), hist: histList[i], histSize: histSize, ranges: ranges)
- XCTAssertEqual(10, histList[i].checkVector(elemChannels: 1))
- }
- Core.merge(mv: histList, dst: hist3D)
- XCTAssertEqual(CvType.CV_32FC3, hist3D.type())
- XCTAssertEqual(10, hist3D.checkVector(elemChannels: 3))
- let truth = Mat(rows: 10, cols: 1, type: CvType.CV_32FC3)
- try truth.put(row: 0, col: 0,
- data: [0, 24870, 0,
- 1863, 31926, 1,
- 56682, 37677, 2260,
- 77278, 44751, 32436,
- 69397, 41343, 18526,
- 27180, 40407, 18658,
- 21101, 15993, 32042,
- 8343, 18585, 47786,
- 300, 6567, 80988,
- 0, 25, 29447] as [Float])
- try assertMatEqual(truth, hist3D, OpenCVTestCase.EPS)
- }
- func testCalcHistListOfMatListOfIntegerMatMatListOfIntegerListOfFloatBoolean() throws {
- let images = [gray255, gray128]
- let channels:[Int32] = [0, 1]
- let histSize:[Int32] = [10, 10]
- let ranges:[Float] = [0, 256, 0, 256]
- let hist = Mat()
- Imgproc.calcHist(images: images, channels: channels, mask: Mat(), hist: hist, histSize: histSize, ranges: ranges, accumulate: true)
- truth = Mat(rows: 10, cols: 10, type: CvType.CV_32F, scalar: Scalar.all(0))
- try truth!.put(row: 9, col: 5, data: [100] as [Float])
- try assertMatEqual(truth!, hist, OpenCVTestCase.EPS)
- }
- func testCannyMatMatDoubleDouble() throws {
- Imgproc.Canny(image: gray255, edges: dst, threshold1: 5, threshold2: 10)
- try assertMatEqual(gray0, dst)
- }
- func testCannyMatMatDoubleDoubleIntBoolean() throws {
- Imgproc.Canny(image: gray0, edges: dst, threshold1: 5, threshold2: 10, apertureSize: 5, L2gradient: true)
- try assertMatEqual(gray0, dst)
- }
- func testCompareHist() throws {
- let H1 = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- let H2 = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- try H1.put(row: 0, col: 0, data: [1, 2, 3] as [Float])
- try H2.put(row: 0, col: 0, data: [4, 5, 6] as [Float])
- let distance = Imgproc.compareHist(H1: H1, H2: H2, method: .HISTCMP_CORREL)
- XCTAssertEqual(1.0, distance, accuracy: OpenCVTestCase.EPS)
- }
- func testContourAreaMat() throws {
- let contour = Mat(rows: 1, cols: 4, type: CvType.CV_32FC2)
- try contour.put(row: 0, col: 0, data: [0, 0, 10, 0, 10, 10, 5, 4] as [Float])
- let area = Imgproc.contourArea(contour: contour)
- XCTAssertEqual(45.0, area, accuracy: OpenCVTestCase.EPS)
- }
- func testContourAreaMatBoolean() throws {
- let contour = Mat(rows: 1, cols: 4, type: CvType.CV_32FC2)
- try contour.put(row: 0, col: 0, data: [0, 0, 10, 0, 10, 10, 5, 4] as [Float])
- let area = Imgproc.contourArea(contour: contour, oriented: true)
- XCTAssertEqual(45.0, area, accuracy: OpenCVTestCase.EPS)
- }
- func testConvertMapsMatMatMatMatInt() throws {
- let map1 = Mat(rows: 1, cols: 4, type: CvType.CV_32FC1, scalar: Scalar(1))
- let map2 = Mat(rows: 1, cols: 4, type: CvType.CV_32FC1, scalar: Scalar(2))
- let dstmap1 = Mat(rows: 1, cols: 4, type: CvType.CV_16SC2)
- let dstmap2 = Mat(rows: 1, cols: 4, type: CvType.CV_16UC1)
- Imgproc.convertMaps(map1: map1, map2: map2, dstmap1: dstmap1, dstmap2: dstmap2, dstmap1type: CvType.CV_16SC2)
- let truthMap1 = Mat(rows: 1, cols: 4, type: CvType.CV_16SC2)
- try truthMap1.put(row: 0, col: 0, data: [1, 2, 1, 2, 1, 2, 1, 2] as [Int16])
- try assertMatEqual(truthMap1, dstmap1)
- let truthMap2 = Mat(rows: 1, cols: 4, type: CvType.CV_16UC1, scalar: Scalar(0))
- try assertMatEqual(truthMap2, dstmap2)
- }
- func testConvertMapsMatMatMatMatIntBoolean() throws {
- let map1 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1, scalar: Scalar(2))
- let map2 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1, scalar: Scalar(4))
- let dstmap1 = Mat(rows: 1, cols: 3, type: CvType.CV_16SC2)
- let dstmap2 = Mat(rows: 1, cols: 3, type: CvType.CV_16UC1)
- Imgproc.convertMaps(map1: map1, map2: map2, dstmap1: dstmap1, dstmap2: dstmap2, dstmap1type: CvType.CV_16SC2, nninterpolation: false)
- // TODO_: write better test (last param == true)
- let truthMap1 = Mat(rows: 1, cols: 3, type: CvType.CV_16SC2)
- try truthMap1.put(row: 0, col: 0, data: [2, 4, 2, 4, 2, 4] as [Int16])
- try assertMatEqual(truthMap1, dstmap1)
- let truthMap2 = Mat(rows: 1, cols: 3, type: CvType.CV_16UC1, scalar: Scalar(0))
- try assertMatEqual(truthMap2, dstmap2)
- }
- func testConvexHullMatMat() {
- let points = [Point(x: 20, y: 0),
- Point(x: 40, y: 0),
- Point(x: 30, y: 20),
- Point(x: 0, y: 20),
- Point(x: 20, y: 10),
- Point(x: 30, y: 10)]
- var hull = [Int32]()
- Imgproc.convexHull(points: points, hull: &hull)
- XCTAssert([0, 1, 2, 3] == hull)
- }
- func testConvexHullMatMatBooleanBoolean() {
- let points = [Point(x: 2, y: 0),
- Point(x: 4, y: 0),
- Point(x: 3, y: 2),
- Point(x: 0, y: 2),
- Point(x: 2, y: 1),
- Point(x: 3, y: 1)]
- var hull = [Int32]()
- Imgproc.convexHull(points: points, hull: &hull, clockwise: true)
- XCTAssert([3, 2, 1, 0] == hull)
- }
- func testConvexityDefects() throws {
- let points = [Point(x: 20, y: 0),
- Point(x: 40, y: 0),
- Point(x: 30, y: 20),
- Point(x: 0, y: 20),
- Point(x: 20, y: 10),
- Point(x: 30, y: 10)]
- var hull = [Int32]()
- Imgproc.convexHull(points: points, hull: &hull)
- var convexityDefects = [Int4]()
- Imgproc.convexityDefects(contour: points, convexhull: hull, convexityDefects: &convexityDefects)
- XCTAssertTrue(Int4(v0: 3, v1: 0, v2: 5, v3: 3620) == convexityDefects[0])
- }
- func testCornerEigenValsAndVecsMatMatIntInt() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32FC1)
- try src.put(row: 0, col: 0, data: [1, 2] as [Float])
- try src.put(row: 1, col: 0, data: [4, 2] as [Float])
- let blockSize:Int32 = 3
- let ksize:Int32 = 5
- // TODO: eigen vals and vectors returned = 0 for most src matrices
- Imgproc.cornerEigenValsAndVecs(src: src, dst: dst, blockSize: blockSize, ksize: ksize)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32FC(6), scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerEigenValsAndVecsMatMatIntIntInt() throws {
- let src = Mat(rows: 4, cols: 4, type: CvType.CV_32FC1, scalar: Scalar(128))
- let blockSize:Int32 = 3
- let ksize:Int32 = 5
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32FC(6), scalar: Scalar(0))
- Imgproc.cornerEigenValsAndVecs(src: src, dst: dst, blockSize: blockSize, ksize: ksize, borderType: .BORDER_REFLECT)
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerHarrisMatMatIntIntDouble() throws {
- truth = Mat(rows: OpenCVTestCase.matSize, cols: OpenCVTestCase.matSize, type: CvType.CV_32FC1, scalar: Scalar(0))
- let blockSize:Int32 = 5
- let ksize:Int32 = 7
- let k = 0.1
- Imgproc.cornerHarris(src: gray128, dst: dst, blockSize: blockSize, ksize: ksize, k: k)
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerHarrisMatMatIntIntDoubleInt() throws {
- truth = Mat(rows: OpenCVTestCase.matSize, cols: OpenCVTestCase.matSize, type: CvType.CV_32FC1, scalar: Scalar(0))
- let blockSize:Int32 = 5
- let ksize:Int32 = 7
- let k = 0.1
- Imgproc.cornerHarris(src: gray255, dst: dst, blockSize: blockSize, ksize: ksize, k: k, borderType: .BORDER_REFLECT)
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerMinEigenValMatMatInt() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32FC1)
- try src.put(row: 0, col: 0, data: [1, 2] as [Float])
- try src.put(row: 1, col: 0, data: [2, 1] as [Float])
- let blockSize:Int32 = 5
- Imgproc.cornerMinEigenVal(src: src, dst: dst, blockSize: blockSize)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32FC1, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- Imgproc.cornerMinEigenVal(src: gray255, dst: dst, blockSize: blockSize)
- truth = Mat(rows: OpenCVTestCase.matSize, cols: OpenCVTestCase.matSize, type: CvType.CV_32FC1, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerMinEigenValMatMatIntInt() throws {
- let src = Mat.eye(rows: 3, cols: 3, type: CvType.CV_32FC1)
- let blockSize:Int32 = 3
- let ksize:Int32 = 5
- Imgproc.cornerMinEigenVal(src: src, dst: dst, blockSize: blockSize, ksize: ksize)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32FC1)
- try truth!.put(row: 0, col: 0, data: [1.0 / 18, 1.0 / 36, 1.0 / 18] as [Float])
- try truth!.put(row: 1, col: 0, data: [1.0 / 36, 1.0 / 18, 1.0 / 36] as [Float])
- try truth!.put(row: 2, col: 0, data: [1.0 / 18, 1.0 / 36, 1.0 / 18] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerMinEigenValMatMatIntIntInt() throws {
- let src = Mat.eye(rows: 3, cols: 3, type: CvType.CV_32FC1)
- let blockSize:Int32 = 3
- let ksize:Int32 = 5
- Imgproc.cornerMinEigenVal(src: src, dst: dst, blockSize: blockSize, ksize: ksize, borderType: .BORDER_REFLECT)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32FC1)
- try truth!.put(row: 0, col: 0, data: [0.68055558, 0.92708349, 0.5868057])
- try truth!.put(row: 1, col: 0, data: [0.92708343, 0.92708343, 0.92708343])
- try truth!.put(row: 2, col: 0, data: [0.58680564, 0.92708343, 0.68055564])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testCornerSubPix() {
- let img = Mat(rows: OpenCVTestCase.matSize, cols: OpenCVTestCase.matSize, type: CvType.CV_8U, scalar: Scalar(128))
- let truthPosition = Point(x: img.cols() / 2, y: img.rows() / 2)
- let r = Rect(point: Point(x: 0, y: 0), point: truthPosition)
- Imgproc.rectangle(img: img, pt1: r.tl(), pt2: r.br(), color: Scalar(0), thickness: Core.FILLED)
- let corners = MatOfPoint2f(array: [Point2f(x: Float(truthPosition.x + 1), y: Float(truthPosition.y + 1))])
- let winSize = Size(width: 2, height: 2)
- let zeroZone = Size(width: -1, height: -1)
- let criteria = TermCriteria(type: TermCriteria.eps, maxCount: 0, epsilon: 0.01)
- Imgproc.cornerSubPix(image: img, corners: corners, winSize: winSize, zeroZone: zeroZone, criteria: criteria)
- assertPoint2fEquals(Point2f(x: Float(truthPosition.x), y: Float(truthPosition.y)), corners.toArray()[0], OpenCVTestCase.weakFEPS)
- }
- func testDilateMatMatMat() throws {
- let kernel = Mat()
- Imgproc.dilate(src: gray255, dst: dst, kernel: kernel)
- try assertMatEqual(gray255, dst)
- Imgproc.dilate(src: gray1, dst: dst, kernel: kernel)
- try assertMatEqual(gray1, dst)
- }
- func testDistanceTransformWithLabels() throws {
- let dstLables = getMat(CvType.CV_32SC1, vals: [0])
- let labels = Mat()
- Imgproc.distanceTransform(src: gray128, dst: dst, labels: labels, distanceType: .DIST_L2, maskSize: .DIST_MASK_3)
- try assertMatEqual(dstLables, labels)
- try assertMatEqual(getMat(CvType.CV_32FC1, vals: [8192]), dst, OpenCVTestCase.EPS)
- }
- func testDrawContoursMatListOfMatIntScalar() {
- let gray0clone = gray0.clone()
- Imgproc.rectangle(img: gray0clone, pt1: Point(x: 1, y: 2), pt2: Point(x: 7, y: 8), color: Scalar(100))
- var contours = [[Point]]()
- Imgproc.findContours(image: gray0clone, contours: &contours, hierarchy: Mat(), mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE)
- Imgproc.drawContours(image: gray0clone, contours: contours, contourIdx: -1, color: Scalar(0))
- XCTAssertEqual(0, Core.countNonZero(src: gray0clone))
- }
- func testDrawContoursMatListOfMatIntScalarInt() {
- let gray0clone = gray0.clone()
- Imgproc.rectangle(img: gray0clone, pt1: Point(x: 1, y: 2), pt2: Point(x: 7, y: 8), color: Scalar(100))
- var contours = [[Point]]()
- Imgproc.findContours(image: gray0clone, contours: &contours, hierarchy: Mat(), mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE)
- Imgproc.drawContours(image: gray0clone, contours: contours, contourIdx: -1, color: Scalar(0), thickness: Core.FILLED)
- XCTAssertEqual(0, Core.countNonZero(src: gray0clone))
- }
- func testEqualizeHist() throws {
- Imgproc.equalizeHist(src: gray0, dst: dst)
- try assertMatEqual(gray0, dst)
- Imgproc.equalizeHist(src: gray255, dst: dst)
- try assertMatEqual(gray255, dst)
- }
- func testErodeMatMatMat() throws {
- let kernel = Mat()
- Imgproc.erode(src: gray128, dst: dst, kernel: kernel)
- try assertMatEqual(gray128, dst)
- }
- func testErodeMatMatMatPointInt() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_8U)
- try src.put(row: 0, col: 0, data: [15, 9, 10] as [Int8])
- try src.put(row: 1, col: 0, data: [10, 8, 12] as [Int8])
- try src.put(row: 2, col: 0, data: [12, 20, 25] as [Int8])
- let kernel = Mat()
- Imgproc.erode(src: src, dst: dst, kernel: kernel, anchor: anchorPoint, iterations: 10)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_8U, scalar: Scalar(8))
- try assertMatEqual(truth!, dst)
- }
- func testErodeMatMatMatPointIntIntScalar() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_8U)
- try src.put(row: 0, col: 0, data: [15, 9, 10] as [Int8])
- try src.put(row: 1, col: 0, data: [10, 8, 12] as [Int8])
- try src.put(row: 2, col: 0, data: [12, 20, 25] as [Int8])
- let kernel = Mat()
- let sc = Scalar(3, 3)
- Imgproc.erode(src: src, dst: dst, kernel: kernel, anchor: anchorPoint, iterations: 10, borderType: .BORDER_REFLECT, borderValue: sc)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_8U, scalar: Scalar(8))
- try assertMatEqual(truth!, dst)
- }
- func testFilter2DMatMatIntMat() throws {
- let src = Mat.eye(rows: 4, cols: 4, type: CvType.CV_32F)
- let kernel = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(1))
- Imgproc.filter2D(src: src, dst: dst, ddepth: -1, kernel: kernel)
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [2, 2, 1, 0] as [Float])
- try truth!.put(row: 1, col: 0, data: [2, 2, 1, 0] as [Float])
- try truth!.put(row: 2, col: 0, data: [1, 1, 2, 1] as [Float])
- try truth!.put(row: 3, col: 0, data: [0, 0, 1, 2] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testFilter2DMatMatIntMatPointDoubleInt() throws {
- let kernel = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(0))
- let point = Point(x: 0, y: 0)
- Imgproc.filter2D(src: gray128, dst: dst, ddepth: -1, kernel: kernel, anchor: point, delta: 2, borderType: .BORDER_CONSTANT)
- try assertMatEqual(gray2, dst)
- }
- func testFindContoursMatListOfMatMatIntInt() {
- let img = Mat(rows: 50, cols: 50, type: CvType.CV_8UC1, scalar: Scalar(0))
- var contours = [[Point]]()
- let hierarchy = Mat()
- Imgproc.findContours(image: img, contours: &contours, hierarchy: hierarchy, mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE)
- // no contours on empty image
- XCTAssertEqual(contours.count, 0)
- XCTAssertEqual(contours.count, hierarchy.total())
- Imgproc.rectangle(img: img, pt1: Point(x: 10, y: 20), pt2: Point(x: 20, y: 30), color: Scalar(100), thickness: 3, lineType: .LINE_AA, shift: 0)
- Imgproc.rectangle(img: img, pt1: Point(x: 30, y: 35), pt2: Point(x: 40, y: 45), color: Scalar(200))
- Imgproc.findContours(image: img, contours: &contours, hierarchy: hierarchy, mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE)
- // two contours of two rectangles
- XCTAssertEqual(contours.count, 2)
- XCTAssertEqual(contours.count, hierarchy.total())
- }
- func testFindContoursMatListOfMatMatIntIntPoint() throws {
- let img = Mat(rows: 50, cols: 50, type: CvType.CV_8UC1, scalar: Scalar(0))
- let img2 = img.submat(rowStart: 5, rowEnd: 50, colStart: 3, colEnd: 50)
- var contours = [[Point]]()
- var contours2 = [[Point]]()
- let hierarchy = Mat()
- Imgproc.rectangle(img: img, pt1: Point(x: 10, y: 20), pt2: Point(x: 20, y: 30), color: Scalar(100), thickness: 3, lineType: .LINE_AA, shift: 0)
- Imgproc.rectangle(img: img, pt1: Point(x: 30, y: 35), pt2: Point(x: 40, y: 45), color: Scalar(200))
- Imgproc.findContours(image: img, contours: &contours, hierarchy: hierarchy, mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE)
- Imgproc.findContours(image: img2, contours: &contours2, hierarchy: hierarchy, mode: .RETR_EXTERNAL, method: .CHAIN_APPROX_SIMPLE, offset: Point(x: 3, y: 5))
- XCTAssertEqual(contours.count, contours2.count)
- XCTAssert(contours[0] == contours2[0])
- }
- func testFitEllipse() {
- let points = [Point2f(x: 0, y: 0), Point2f(x: -1, y: 1), Point2f(x: 1, y: 1), Point2f(x: 1, y: -1), Point2f(x: -1, y: -1)]
- let rrect = Imgproc.fitEllipse(points: points)
- let FIT_ELLIPSE_CENTER_EPS:Float = 0.01
- let FIT_ELLIPSE_SIZE_EPS:Float = 0.4
- assertPoint2fEquals(Point2f(x: 0, y: 0), rrect.center, FIT_ELLIPSE_CENTER_EPS)
- XCTAssertEqual(Float(2.828), rrect.size.width, accuracy: FIT_ELLIPSE_SIZE_EPS)
- XCTAssertEqual(Float(2.828), rrect.size.height, accuracy: FIT_ELLIPSE_SIZE_EPS)
- }
- func testFitLine() throws {
- let points = Mat(rows: 1, cols: 4, type: CvType.CV_32FC2)
- try points.put(row: 0, col: 0, data: [0, 0, 2, 3, 3, 4, 5, 8] as [Float])
- let linePoints = Mat(rows: 4, cols: 1, type: CvType.CV_32FC1)
- try linePoints.put(row: 0, col: 0, data: [0.53198653, 0.84675282, 2.5, 3.75] as [Float])
- Imgproc.fitLine(points: points, line: dst, distType: .DIST_L12, param: 0, reps: 0.01, aeps: 0.01)
- try assertMatEqual(linePoints, dst, OpenCVTestCase.EPS)
- }
- func testFloodFillMatMatPointScalar() throws {
- let mask = Mat(rows: OpenCVTestCase.matSize + 2, cols: OpenCVTestCase.matSize + 2, type: CvType.CV_8U, scalar: Scalar(0))
- let img = gray0
- Imgproc.circle(img: mask, center: Point(x: OpenCVTestCase.matSize / 2 + 1, y: OpenCVTestCase.matSize / 2 + 1), radius: 3, color: Scalar(2))
- let retval = Imgproc.floodFill(image: img, mask: mask, seedPoint: Point(x: OpenCVTestCase.matSize / 2, y: OpenCVTestCase.matSize / 2), newVal: Scalar(1))
- XCTAssertEqual(Core.countNonZero(src: img), retval)
- Imgproc.circle(img: mask, center: Point(x: OpenCVTestCase.matSize / 2 + 1, y: OpenCVTestCase.matSize / 2 + 1), radius: 3, color: Scalar(0))
- XCTAssertEqual(retval + 4 * (OpenCVTestCase.matSize + 1), Core.countNonZero(src: mask))
- try assertMatEqual(mask.submat(rowStart: 1, rowEnd: OpenCVTestCase.matSize + 1, colStart: 1, colEnd: OpenCVTestCase.matSize + 1), img)
- }
- func testFloodFillMatMatPointScalar_WithoutMask() {
- let img = gray0
- Imgproc.circle(img: img, center: Point(x: OpenCVTestCase.matSize / 2, y: OpenCVTestCase.matSize / 2), radius: 3, color: Scalar(2))
- // TODO: ideally we should pass null instead of "new Mat()"
- let retval = Imgproc.floodFill(image: img, mask: Mat(), seedPoint: Point(x: OpenCVTestCase.matSize / 2, y: OpenCVTestCase.matSize / 2), newVal: Scalar(1))
- Imgproc.circle(img: img, center: Point(x: OpenCVTestCase.matSize / 2, y: OpenCVTestCase.matSize / 2), radius: 3, color: Scalar(0))
- XCTAssertEqual(Core.countNonZero(src: img), retval)
- }
- func testGaussianBlurMatMatSizeDouble() throws {
- Imgproc.GaussianBlur(src: gray0, dst: dst, ksize: size, sigmaX: 1)
- try assertMatEqual(gray0, dst)
- Imgproc.GaussianBlur(src: gray2, dst: dst, ksize: size, sigmaX: 1)
- try assertMatEqual(gray2, dst)
- }
- func testGaussianBlurMatMatSizeDoubleDouble() throws {
- Imgproc.GaussianBlur(src: gray2, dst: dst, ksize: size, sigmaX: 0, sigmaY: 0)
- try assertMatEqual(gray2, dst)
- }
- func testGaussianBlurMatMatSizeDoubleDoubleInt() throws {
- Imgproc.GaussianBlur(src: gray2, dst: dst, ksize: size, sigmaX: 1, sigmaY: 3, borderType: .BORDER_REFLECT)
- try assertMatEqual(gray2, dst)
- }
- func testGetAffineTransform() throws {
- let src = [Point2f(x: 2, y: 3), Point2f(x: 3, y: 1), Point2f(x: 1, y: 4)]
- let dst = [Point2f(x: 3, y: 3), Point2f(x: 7, y: 4), Point2f(x: 5, y: 6)]
- let transform = Imgproc.getAffineTransform(src: src, dst: dst)
- let truth = Mat(rows: 2, cols: 3, type: CvType.CV_64FC1)
- try truth.put(row: 0, col: 0, data: [-8.0, -6.0, 37.0])
- try truth.put(row: 1, col: 0, data: [-7.0, -4.0, 29.0])
- try assertMatEqual(truth, transform, OpenCVTestCase.EPS)
- }
- func testGetDerivKernelsMatMatIntIntInt() throws {
- let kx = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- let ky = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- let expKx = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- let expKy = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- try kx.put(row: 0, col: 0, data: [1, 1] as [Float])
- try kx.put(row: 1, col: 0, data: [1, 1] as [Float])
- try ky.put(row: 0, col: 0, data: [2, 2] as [Float])
- try ky.put(row: 1, col: 0, data: [2, 2] as [Float])
- try expKx.put(row: 0, col: 0, data: [1, -2, 1] as [Float])
- try expKy.put(row: 0, col: 0, data: [1, -2, 1] as [Float])
- Imgproc.getDerivKernels(kx: kx, ky: ky, dx: 2, dy: 2, ksize: 3)
- try assertMatEqual(expKx, kx, OpenCVTestCase.EPS)
- try assertMatEqual(expKy, ky, OpenCVTestCase.EPS)
- }
- func testGetDerivKernelsMatMatIntIntIntBooleanInt() throws {
- let kx = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- let ky = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- let expKx = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- let expKy = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- try kx.put(row: 0, col: 0, data: [1, 1] as [Float])
- try kx.put(row: 1, col: 0, data: [1, 1] as [Float])
- try ky.put(row: 0, col: 0, data: [2, 2] as [Float])
- try ky.put(row: 1, col: 0, data: [2, 2] as [Float])
- try expKx.put(row: 0, col: 0, data: [1, -2, 1] as [Float])
- try expKy.put(row: 0, col: 0, data: [1, -2, 1] as [Float])
- Imgproc.getDerivKernels(kx: kx, ky: ky, dx: 2, dy: 2, ksize: 3, normalize: true, ktype: CvType.CV_32F)
- try assertMatEqual(expKx, kx, OpenCVTestCase.EPS)
- try assertMatEqual(expKy, ky, OpenCVTestCase.EPS)
- // TODO_: write better test
- }
- func testGetGaussianKernelIntDouble() throws {
- dst = Imgproc.getGaussianKernel(ksize: 1, sigma: 0.5)
- truth = Mat(rows: 1, cols: 1, type: CvType.CV_64FC1, scalar: Scalar(1))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testGetGaussianKernelIntDoubleInt() throws {
- dst = Imgproc.getGaussianKernel(ksize: 3, sigma: 0.8, ktype: CvType.CV_32F)
- truth = Mat(rows: 3, cols: 1, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [0.23899426, 0.52201146, 0.23899426] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testGetRectSubPixMatSizePointMat() throws {
- let size = Size(width: 3, height: 3)
- let center = Point2f(x: Float(gray255.cols() / 2), y: Float(gray255.rows() / 2))
- Imgproc.getRectSubPix(image: gray255, patchSize: size, center: center, patch: dst)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_8U, scalar: Scalar(255))
- try assertMatEqual(truth!, dst)
- }
- func testGetRectSubPixMatSizePointMatInt() throws {
- let src = Mat(rows: 10, cols: 10, type: CvType.CV_32F, scalar: Scalar(2))
- let patchSize = Size(width: 5, height: 5)
- let center = Point2f(x: Float(src.cols() / 2), y: Float(src.rows() / 2))
- Imgproc.getRectSubPix(image: src, patchSize: patchSize, center: center, patch: dst)
- truth = Mat(rows: 5, cols: 5, type: CvType.CV_32F, scalar: Scalar(2))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testGetRotationMatrix2D() throws {
- let center = Point2f(x: 0, y: 0)
- dst = Imgproc.getRotationMatrix2D(center: center, angle: 0, scale: 1)
- truth = Mat(rows: 2, cols: 3, type: CvType.CV_64F)
- try truth!.put(row: 0, col: 0, data: [1.0, 0.0, 0.0])
- try truth!.put(row: 1, col: 0, data: [0.0, 1.0, 0.0])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testGetStructuringElementIntSize() throws {
- dst = Imgproc.getStructuringElement(shape: .MORPH_RECT, ksize: size)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_8UC1, scalar: Scalar(1))
- try assertMatEqual(truth!, dst)
- }
- func testGetStructuringElementIntSizePoint() throws {
- dst = Imgproc.getStructuringElement(shape: .MORPH_CROSS, ksize: size, anchor: anchorPoint)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_8UC1)
- try truth!.put(row: 0, col: 0, data: [0, 0, 1] as [Int8])
- try truth!.put(row: 1, col: 0, data: [0, 0, 1] as [Int8])
- try truth!.put(row: 2, col: 0, data: [1, 1, 1] as [Int8])
- try assertMatEqual(truth!, dst)
- }
- func testGoodFeaturesToTrackMatListOfPointIntDoubleDouble() {
- let src = gray0
- Imgproc.rectangle(img: src, pt1: Point(x: 2, y: 2), pt2: Point(x: 8, y: 8), color: Scalar(100), thickness: -1)
- var lp = [Point]()
- Imgproc.goodFeaturesToTrack(image: src, corners: &lp, maxCorners: 100, qualityLevel: 0.01, minDistance: 3)
- XCTAssertEqual(4, lp.count)
- }
- func testGoodFeaturesToTrackMatListOfPointIntDoubleDoubleMatIntBooleanDouble() {
- let src = gray0
- Imgproc.rectangle(img: src, pt1: Point(x: 2, y: 2), pt2: Point(x: 8, y: 8), color: Scalar(100), thickness: -1)
- var lp = [Point]()
- Imgproc.goodFeaturesToTrack(image: src, corners: &lp, maxCorners: 100, qualityLevel: 0.01, minDistance: 3, mask: gray1, blockSize: 4, gradientSize: 3, useHarrisDetector: true, k: 0)
- XCTAssertEqual(4, lp.count)
- }
- func testHoughCirclesMatMatIntDoubleDouble() {
- let sz:Int32 = 512
- let img = Mat(rows: sz, cols: sz, type: CvType.CV_8U, scalar: Scalar(128))
- let circles = Mat()
- Imgproc.HoughCircles(image: img, circles: circles, method: .HOUGH_GRADIENT, dp: 2.0, minDist: Double(img.rows() / 4))
- XCTAssertEqual(0, circles.cols())
- }
- func testHoughCirclesMatMatIntDoubleDouble1() {
- let sz: Int32 = 512
- let img = Mat(rows: sz, cols: sz, type: CvType.CV_8U, scalar: Scalar(128))
- let circles = Mat()
- let center = Point(x: img.cols() / 2, y: img.rows() / 2)
- let radius = min(img.cols() / 4, img.rows() / 4)
- Imgproc.circle(img: img, center: center, radius: radius, color: colorBlack, thickness: 3)
- Imgproc.HoughCircles(image: img, circles: circles, method: .HOUGH_GRADIENT, dp: 2.0, minDist: Double(img.rows() / 4))
- XCTAssertEqual(1, circles.cols())
- }
- func testHoughLinesMatMatDoubleDoubleInt() {
- let sz:Int32 = 512
- let img = Mat(rows: sz, cols: sz, type: CvType.CV_8U, scalar: Scalar(0))
- let point1 = Point(x: 50, y: 50)
- let point2 = Point(x: img.cols() / 2, y: img.rows() / 2)
- Imgproc.line(img: img, pt1: point1, pt2: point2, color: colorWhite, thickness: 1)
- let lines = Mat()
- Imgproc.HoughLines(image: img, lines: lines, rho: 1, theta: 3.1415926/180, threshold: 100)
- XCTAssertEqual(1, lines.cols())
- }
- func testHoughLinesPMatMatDoubleDoubleInt() {
- let sz:Int32 = 512
- let img = Mat(rows: sz, cols: sz, type: CvType.CV_8U, scalar: Scalar(0))
- let point1 = Point(x: 0, y: 0)
- let point2 = Point(x: sz, y: sz)
- let point3 = Point(x: sz, y: 0)
- let point4 = Point(x: 2*sz/3, y: sz/3)
- Imgproc.line(img: img, pt1: point1, pt2: point2, color: Scalar.all(255), thickness: 1)
- Imgproc.line(img: img, pt1: point3, pt2: point4, color: Scalar.all(255), thickness: 1)
- let lines = Mat()
- Imgproc.HoughLinesP(image: img, lines: lines, rho: 1, theta: 3.1415926/180, threshold: 100)
- XCTAssertEqual(2, lines.rows())
- }
- func testIntegral2MatMatMat() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F, scalar: Scalar(3))
- let expSum = Mat(rows: 4, cols: 4, type: CvType.CV_64F)
- let expSqsum = Mat(rows: 4, cols: 4, type: CvType.CV_64F)
- let sum = Mat()
- let sqsum = Mat()
- try expSum.put(row: 0, col: 0, data: [0.0, 0.0, 0.0, 0.0])
- try expSum.put(row: 1, col: 0, data: [0.0, 3.0, 6.0, 9.0])
- try expSum.put(row: 2, col: 0, data: [0.0, 6.0, 12.0, 18.0])
- try expSum.put(row: 3, col: 0, data: [0.0, 9.0, 18.0, 27.0])
- try expSqsum.put(row: 0, col: 0, data: [0.0, 0.0, 0.0, 0.0])
- try expSqsum.put(row: 1, col: 0, data: [0.0, 9.0, 18.0, 27.0])
- try expSqsum.put(row: 2, col: 0, data: [0.0, 18.0, 36.0, 54.0])
- try expSqsum.put(row: 3, col: 0, data: [0.0, 27.0, 54.0, 81.0])
- Imgproc.integral(src: src, sum: sum, sqsum: sqsum)
- try assertMatEqual(expSum, sum, OpenCVTestCase.EPS)
- try assertMatEqual(expSqsum, sqsum, OpenCVTestCase.EPS)
- }
- func testIntegral2MatMatMatInt() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F, scalar: Scalar(3))
- let expSum = Mat(rows: 4, cols: 4, type: CvType.CV_64F)
- let expSqsum = Mat(rows: 4, cols: 4, type: CvType.CV_64F)
- let sum = Mat()
- let sqsum = Mat()
- try expSum.put(row: 0, col: 0, data: [0.0, 0.0, 0.0, 0.0])
- try expSum.put(row: 1, col: 0, data: [0.0, 3.0, 6.0, 9.0])
- try expSum.put(row: 2, col: 0, data: [0.0, 6.0, 12.0, 18.0])
- try expSum.put(row: 3, col: 0, data: [0.0, 9.0, 18.0, 27.0])
- try expSqsum.put(row: 0, col: 0, data: [0.0, 0.0, 0.0, 0.0])
- try expSqsum.put(row: 1, col: 0, data: [0.0, 9.0, 18.0, 27.0])
- try expSqsum.put(row: 2, col: 0, data: [0.0, 18.0, 36.0, 54.0])
- try expSqsum.put(row: 3, col: 0, data: [0.0, 27.0, 54.0, 81.0])
- Imgproc.integral(src: src, sum: sum, sqsum: sqsum, sdepth: CvType.CV_64F, sqdepth: CvType.CV_64F)
- try assertMatEqual(expSum, sum, OpenCVTestCase.EPS)
- try assertMatEqual(expSqsum, sqsum, OpenCVTestCase.EPS)
- }
- func testIntegral3MatMatMatMat() throws {
- let src = Mat(rows: 1, cols: 1, type: CvType.CV_32F, scalar: Scalar(1))
- let expSum = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let expSqsum = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let expTilted = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let sum = Mat()
- let sqsum = Mat()
- let tilted = Mat()
- try expSum.put(row: 0, col: 0, data: [0.0, 0.0])
- try expSum.put(row: 1, col: 0, data: [0.0, 1.0])
- try expSqsum.put(row: 0, col: 0, data: [0.0, 0.0])
- try expSqsum.put(row: 1, col: 0, data: [0.0, 1.0])
- try expTilted.put(row: 0, col: 0, data: [0.0, 0.0])
- try expTilted.put(row: 1, col: 0, data: [0.0, 1.0])
- Imgproc.integral(src: src, sum: sum, sqsum: sqsum, tilted: tilted)
- try assertMatEqual(expSum, sum, OpenCVTestCase.EPS)
- try assertMatEqual(expSqsum, sqsum, OpenCVTestCase.EPS)
- try assertMatEqual(expTilted, tilted, OpenCVTestCase.EPS)
- }
- func testIntegral3MatMatMatMatInt() throws {
- let src = Mat(rows: 1, cols: 1, type: CvType.CV_32F, scalar: Scalar(1))
- let expSum = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let expSqsum = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let expTilted = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_64F)
- let sum = Mat()
- let sqsum = Mat()
- let tilted = Mat()
- try expSum.put(row: 0, col: 0, data: [0.0, 0.0])
- try expSum.put(row: 1, col: 0, data: [0.0, 1.0])
- try expSqsum.put(row: 0, col: 0, data: [0.0, 0.0])
- try expSqsum.put(row: 1, col: 0, data: [0.0, 1.0])
- try expTilted.put(row: 0, col: 0, data: [0.0, 0.0])
- try expTilted.put(row: 1, col: 0, data: [0.0, 1.0])
- Imgproc.integral(src: src, sum: sum, sqsum: sqsum, tilted: tilted, sdepth: CvType.CV_64F, sqdepth: CvType.CV_64F)
- try assertMatEqual(expSum, sum, OpenCVTestCase.EPS)
- try assertMatEqual(expSqsum, sqsum, OpenCVTestCase.EPS)
- try assertMatEqual(expTilted, tilted, OpenCVTestCase.EPS)
- }
- func testIntegralMatMat() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(2))
- Imgproc.integral(src: src, sum: dst)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_64F)
- try truth!.put(row: 0, col: 0, data: [0.0, 0.0, 0.0])
- try truth!.put(row: 1, col: 0, data: [0.0, 2.0, 4.0])
- try truth!.put(row: 2, col: 0, data: [0.0, 4.0, 8.0])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testIntegralMatMatInt() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(2))
- Imgproc.integral(src: src, sum: dst, sdepth: CvType.CV_64F)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_64F)
- try truth!.put(row: 0, col: 0, data: [0.0, 0.0, 0.0])
- try truth!.put(row: 1, col: 0, data: [0.0, 2.0, 4.0])
- try truth!.put(row: 2, col: 0, data: [0.0, 4.0, 8.0])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testInvertAffineTransform() throws {
- let src = Mat(rows: 2, cols: 3, type: CvType.CV_64F, scalar: Scalar(1))
- Imgproc.invertAffineTransform(M: src, iM: dst)
- truth = Mat(rows: 2, cols: 3, type: CvType.CV_64F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testIsContourConvex() {
- let contour1 = [Point(x: 0, y: 0), Point(x: 10, y: 0), Point(x: 10, y: 10), Point(x: 5, y: 4)]
- XCTAssertFalse(Imgproc.isContourConvex(contour: contour1))
- let contour2 = [Point(x: 0, y: 0), Point(x: 10, y: 0), Point(x: 10, y: 10), Point(x: 5, y: 6)]
- XCTAssert(Imgproc.isContourConvex(contour: contour2))
- }
- func testLaplacianMatMatInt() throws {
- Imgproc.Laplacian(src: gray0, dst: dst, ddepth: CvType.CV_8U)
- try assertMatEqual(gray0, dst)
- }
- func testLaplacianMatMatIntIntDoubleDouble() throws {
- let src = Mat.eye(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- Imgproc.Laplacian(src: src, dst: dst, ddepth: CvType.CV_32F, ksize: 1, scale: 2, delta: OpenCVTestCase.EPS)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [-7.9990001, 8.0009995])
- try truth!.put(row: 1, col: 0, data: [8.0009995, -7.9990001])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testLaplacianMatMatIntIntDoubleDoubleInt() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F, scalar: Scalar(2))
- Imgproc.Laplacian(src: src, dst: dst, ddepth: CvType.CV_32F, ksize: 1, scale: 2, delta: OpenCVTestCase.EPS, borderType: .BORDER_REFLECT)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32F, scalar: Scalar(0.00099945068))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testMatchShapes() throws {
- let contour1 = Mat(rows: 1, cols: 4, type: CvType.CV_32FC2)
- let contour2 = Mat(rows: 1, cols: 4, type: CvType.CV_32FC2)
- try contour1.put(row: 0, col: 0, data: [1, 1, 5, 1, 4, 3, 6, 2] as [Float])
- try contour2.put(row: 0, col: 0, data: [1, 1, 6, 1, 4, 1, 2, 5] as [Float])
- let distance = Imgproc.matchShapes(contour1: contour1, contour2: contour2, method: .CONTOURS_MATCH_I1, parameter: 1)
- XCTAssertEqual(2.81109697365334, distance, accuracy:OpenCVTestCase.EPS)
- }
- func testMatchTemplate() throws {
- let image = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- let templ = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- try image.put(row: 0, col: 0, data: [1, 2, 3, 4] as [Int8])
- try templ.put(row: 0, col: 0, data: [5, 6, 7, 8] as [Int8])
- Imgproc.matchTemplate(image: image, templ: templ, result: dst, method: .TM_CCORR)
- truth = Mat(rows: 1, cols: 1, type: CvType.CV_32F, scalar: Scalar(70))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- Imgproc.matchTemplate(image: gray255, templ: gray0, result: dst, method: .TM_CCORR)
- truth = Mat(rows: 1, cols: 1, type: CvType.CV_32F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testMedianBlur() throws {
- Imgproc.medianBlur(src: gray255, dst: dst, ksize: 5)
- try assertMatEqual(gray255, dst)
- Imgproc.medianBlur(src: gray2, dst: dst, ksize: 3)
- try assertMatEqual(gray2, dst)
- // TODO_: write better test
- }
- func testMinAreaRect() {
- let points = [Point2f(x: 1, y: 1), Point2f(x: 5, y: 1), Point2f(x: 4, y: 3), Point2f(x: 6, y: 2)]
- let rrect = Imgproc.minAreaRect(points: points)
- XCTAssertEqual(Size2f(width: 5, height: 2), rrect.size)
- XCTAssertEqual(0.0, rrect.angle)
- XCTAssertEqual(Point2f(x: 3.5, y: 2), rrect.center)
- }
- func testMinEnclosingCircle() {
- let points = [Point2f(x: 0, y: 0), Point2f(x: -100, y: 0), Point2f(x: 0, y: -100), Point2f(x: 100, y: 0), Point2f(x: 0, y: 100)]
- let actualCenter = Point2f()
- var radius:Float = 0
- Imgproc.minEnclosingCircle(points: points, center: actualCenter, radius: &radius)
- XCTAssertEqual(Point2f(x: 0, y: 0), actualCenter)
- XCTAssertEqual(100.0, radius, accuracy: 1.0)
- }
- func testMorphologyExMatMatIntMat() throws {
- Imgproc.morphologyEx(src: gray255, dst: dst, op: MorphTypes.MORPH_GRADIENT, kernel: gray0)
- try assertMatEqual(gray0, dst)
- }
- func testMorphologyExMatMatIntMatPointInt() throws {
- let src = Mat.eye(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- let kernel = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U, scalar: Scalar(0))
- let point = Point(x: 0, y: 0)
- Imgproc.morphologyEx(src: src, dst: dst, op: MorphTypes.MORPH_CLOSE, kernel: kernel, anchor: point, iterations: 10)
- truth = Mat.eye(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- try assertMatEqual(truth!, dst)
- }
- func testMorphologyExMatMatIntMatPointIntIntScalar() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- try src.put(row: 0, col: 0, data: [2, 1] as [Int8])
- try src.put(row: 1, col: 0, data: [2, 1] as [Int8])
- let kernel = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U, scalar: Scalar(1))
- let point = Point(x: 1, y: 1)
- let sc = Scalar(3, 3)
- Imgproc.morphologyEx(src: src, dst: dst, op: MorphTypes.MORPH_TOPHAT, kernel: kernel, anchor: point, iterations: 10, borderType: .BORDER_REFLECT, borderValue: sc)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8U)
- try truth!.put(row: 0, col: 0, data: [1, 0] as [Int8])
- try truth!.put(row: 1, col: 0, data: [1, 0] as [Int8])
- try assertMatEqual(truth!, dst)
- }
- func testPointPolygonTest() {
- let contour = [Point2f(x: 0, y: 0), Point2f(x: 1, y: 3), Point2f(x: 3, y: 4), Point2f(x: 4, y: 3), Point2f(x: 2, y: 1)]
- let sign1 = Imgproc.pointPolygonTest(contour: contour, pt: Point2f(x: 2, y: 2), measureDist: false)
- XCTAssertEqual(1.0, sign1)
- let sign2 = Imgproc.pointPolygonTest(contour: contour, pt: Point2f(x: 4, y: 4), measureDist: true)
- XCTAssertEqual(-sqrt(0.5), sign2)
- }
- func testPreCornerDetectMatMatInt() throws {
- let src = Mat(rows: 4, cols: 4, type: CvType.CV_32F, scalar: Scalar(1))
- let ksize:Int32 = 3
- Imgproc.preCornerDetect(src: src, dst: dst, ksize: ksize)
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testPreCornerDetectMatMatIntInt() throws {
- let src = Mat(rows: 4, cols: 4, type: CvType.CV_32F, scalar: Scalar(1))
- let ksize:Int32 = 3
- Imgproc.preCornerDetect(src: src, dst: dst, ksize: ksize, borderType: .BORDER_REFLECT)
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testPyrDownMatMat() throws {
- let src = Mat(rows: 4, cols: 4, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 1, 4, 2] as [Float])
- try src.put(row: 1, col: 0, data: [3, 2, 6, 8] as [Float])
- try src.put(row: 2, col: 0, data: [4, 6, 8, 10] as [Float])
- try src.put(row: 3, col: 0, data: [12, 32, 6, 18] as [Float])
- Imgproc.pyrDown(src: src, dst: dst)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [2.78125, 4.609375])
- try truth!.put(row: 1, col: 0, data: [8.546875, 8.8515625])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testPyrDownMatMatSize() throws {
- let src = Mat(rows: 4, cols: 4, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 1, 4, 2] as [Float])
- try src.put(row: 1, col: 0, data: [3, 2, 6, 8] as [Float])
- try src.put(row: 2, col: 0, data: [4, 6, 8, 10] as [Float])
- try src.put(row: 3, col: 0, data: [12, 32, 6, 18] as [Float])
- let dstSize = Size(width: 2, height: 2)
- Imgproc.pyrDown(src: src, dst: dst, dstsize: dstSize)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [2.78125, 4.609375])
- try truth!.put(row: 1, col: 0, data: [8.546875, 8.8515625])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testPyrMeanShiftFilteringMatMatDoubleDouble() throws {
- let src = Mat(rows: OpenCVTestCase.matSize, cols: OpenCVTestCase.matSize, type: CvType.CV_8UC3, scalar: Scalar(0))
- Imgproc.pyrMeanShiftFiltering(src: src, dst: dst, sp: 10, sr: 50)
- try assertMatEqual(src, dst)
- }
- func testPyrUpMatMat() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 1] as [Float])
- try src.put(row: 1, col: 0, data: [3, 2] as [Float])
- Imgproc.pyrUp(src: src, dst: dst)
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [2, 1.75, 1.375, 1.25])
- try truth!.put(row: 1, col: 0, data: [2.25, 2, 1.625, 1.5])
- try truth!.put(row: 2, col: 0, data: [2.625, 2.375, 2, 1.875])
- try truth!.put(row: 3, col: 0, data: [2.75, 2.5, 2.125, 2])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testRemapMatMatMatMatInt() throws {
- // this test does something weird
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(2))
- let map1 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- let map2 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try map1.put(row: 0, col: 0, data: [3, 6, 5] as [Float])
- try map2.put(row: 0, col: 0, data: [4, 8, 12] as [Float])
- Imgproc.remap(src: src, dst: dst, map1: map1, map2: map2, interpolation: InterpolationFlags.INTER_LINEAR.rawValue)
- truth = Mat(rows: 1, cols: 3, type: CvType.CV_32F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testRemapMatMatMatMatIntIntScalar() throws {
- // this test does something weird
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(2))
- let map1 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- let map2 = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- let sc = Scalar(0)
- try map1.put(row: 0, col: 0, data: [3, 6, 5, 0] as [Float])
- try map2.put(row: 0, col: 0, data: [4, 8, 12] as [Float])
- truth = Mat(rows: 1, cols: 3, type: CvType.CV_32F, scalar: Scalar(2))
- Imgproc.remap(src: src, dst: dst, map1: map1, map2: map2, interpolation: InterpolationFlags.INTER_LINEAR.rawValue, borderMode: .BORDER_REFLECT, borderValue: sc)
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testResizeMatMatSize() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_8UC1, scalar: Scalar(1))
- let dsize = Size(width: 1, height: 1)
- Imgproc.resize(src: src, dst: dst, dsize: dsize, fx: 0, fy: 0, interpolation: InterpolationFlags.INTER_LINEAR_EXACT.rawValue)
- truth = Mat(rows: 1, cols: 1, type: CvType.CV_8UC1, scalar: Scalar(1))
- try assertMatEqual(truth!, dst)
- }
- func testResizeMatMatSizeDoubleDoubleInt() throws {
- Imgproc.resize(src: gray255, dst: dst, dsize: Size(width: 2, height: 2), fx: 0, fy: 0, interpolation: InterpolationFlags.INTER_AREA.rawValue)
- truth = Mat(rows: 2, cols: 2, type: CvType.CV_8UC1, scalar: Scalar(255))
- try assertMatEqual(truth!, dst)
- }
- func testScharrMatMatIntIntInt() throws {
- let src = Mat.eye(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- Imgproc.Scharr(src: src, dst: dst, ddepth: CvType.CV_32F, dx: 1, dy: 0)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(0))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testScharrMatMatIntIntIntDoubleDouble() throws {
- let src = Mat.eye(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F)
- Imgproc.Scharr(src: src, dst: dst, ddepth: CvType.CV_32F, dx: 1, dy: 0, scale: 1.5, delta: 0.001)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(0.001))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testScharrMatMatIntIntIntDoubleDoubleInt() throws {
- let src = Mat.eye(rows: 3, cols: 3, type: CvType.CV_32F)
- Imgproc.Scharr(src: src, dst: dst, ddepth: CvType.CV_32F, dx: 1, dy: 0, scale: 1.5, delta: 0, borderType: .BORDER_REFLECT)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [-15, -19.5, -4.5])
- try truth!.put(row: 1, col: 0, data: [10.5, 0, -10.5])
- try truth!.put(row: 2, col: 0, data: [4.5, 19.5, 15])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testSepFilter2DMatMatIntMatMat() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(2))
- let kernelX = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- let kernelY = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try kernelX.put(row: 0, col: 0, data: [4, 3, 7] as [Float])
- try kernelY.put(row: 0, col: 0, data: [9, 4, 2] as [Float])
- Imgproc.sepFilter2D(src: src, dst: dst, ddepth: CvType.CV_32F, kernelX: kernelX, kernelY: kernelY)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(420))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testSepFilter2DMatMatIntMatMatPointDouble() throws {
- let src = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32FC1, scalar: Scalar(2))
- let kernelX = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try kernelX.put(row: 0, col: 0, data: [2, 2, 2] as [Float])
- let kernelY = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try kernelY.put(row: 0, col: 0, data: [1, 1, 1] as [Float])
- Imgproc.sepFilter2D(src: src, dst: dst, ddepth: CvType.CV_32F, kernelX: kernelX, kernelY: kernelY, anchor: anchorPoint, delta: OpenCVTestCase.weakEPS)
- truth = Mat(rows: imgprocSz, cols: imgprocSz, type: CvType.CV_32F, scalar: Scalar(36 + OpenCVTestCase.weakEPS))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testSepFilter2DMatMatIntMatMatPointDoubleInt() throws {
- let kernelX = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try kernelX.put(row: 0, col: 0, data: [2, 2, 2] as [Float])
- let kernelY = Mat(rows: 1, cols: 3, type: CvType.CV_32FC1)
- try kernelY.put(row: 0, col: 0, data: [1, 1, 1] as [Float])
- Imgproc.sepFilter2D(src: gray0, dst: dst, ddepth: CvType.CV_32F, kernelX: kernelX, kernelY: kernelY, anchor: anchorPoint, delta: OpenCVTestCase.weakEPS, borderType: .BORDER_REFLECT)
- truth = Mat(rows: 10, cols: 10, type: CvType.CV_32F, scalar: Scalar(OpenCVTestCase.weakEPS))
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testSobelMatMatIntIntInt() throws {
- Imgproc.Sobel(src: gray255, dst: dst, ddepth: CvType.CV_8U, dx: 1, dy: 0)
- try assertMatEqual(gray0, dst)
- }
- func testSobelMatMatIntIntIntIntDoubleDouble() throws {
- Imgproc.Sobel(src: gray255, dst: dst, ddepth: CvType.CV_8U, dx: 1, dy: 0, ksize: 3, scale: 2, delta: 0.001)
- try assertMatEqual(gray0, dst)
- }
- func testSobelMatMatIntIntIntIntDoubleDoubleInt() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 0, 1] as [Float])
- try src.put(row: 1, col: 0, data: [6, 4, 3] as [Float])
- try src.put(row: 2, col: 0, data: [1, 0, 2] as [Float])
- Imgproc.Sobel(src: src, dst: dst, ddepth: CvType.CV_32F, dx: 1, dy: 0, ksize: 3, scale: 2, delta: 0, borderType: .BORDER_REPLICATE)
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [-16, -12, 4] as [Float])
- try truth!.put(row: 1, col: 0, data: [-14, -12, 2] as [Float])
- try truth!.put(row: 2, col: 0, data: [-10, 0, 10] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testThreshold() throws {
- Imgproc.threshold(src: makeMask(gray0.clone(), vals: [10]), dst: dst, thresh: 5, maxval: 255, type: .THRESH_TRUNC)
- try assertMatEqual(makeMask(gray0.clone(), vals: [5]), dst)
- Imgproc.threshold(src: makeMask(gray2.clone(), vals: [10]), dst: dst, thresh: 1, maxval: 255, type: .THRESH_BINARY)
- try assertMatEqual(gray255, dst)
- Imgproc.threshold(src: makeMask(gray2.clone(), vals: [10]), dst: dst, thresh: 3, maxval: 255, type: .THRESH_BINARY_INV)
- try assertMatEqual(makeMask(gray255.clone(), vals: [0]), dst)
- }
- func testWarpAffineMatMatMatSize() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 0, 1] as [Float])
- try src.put(row: 1, col: 0, data: [6, 4, 3] as [Float])
- try src.put(row: 2, col: 0, data: [1, 0, 2] as [Float])
- let M = Mat(rows: 2, cols: 3, type: CvType.CV_32F)
- try M.put(row: 0, col: 0, data: [1, 0, 1] as [Float])
- try M.put(row: 1, col: 0, data: [0, 1, 1] as [Float])
- Imgproc.warpAffine(src: src, dst: dst, M: M, dsize: Size(width: 3, height: 3))
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [0, 0, 0] as [Float])
- try truth!.put(row: 1, col: 0, data: [0, 2, 0] as [Float])
- try truth!.put(row: 2, col: 0, data: [0, 6, 4] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testWarpAffineMatMatMatSizeInt() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 4, 1] as [Float])
- try src.put(row: 1, col: 0, data: [6, 4, 3] as [Float])
- try src.put(row: 2, col: 0, data: [0, 2, 2] as [Float])
- let M = Mat(rows: 2, cols: 3, type: CvType.CV_32F)
- try M.put(row: 0, col: 0, data: [1, 0, 0] as [Float])
- try M.put(row: 1, col: 0, data: [0, 0, 1] as [Float])
- Imgproc.warpAffine(src: src, dst: dst, M: M, dsize: Size(width: 2, height: 2), flags: InterpolationFlags.WARP_INVERSE_MAP.rawValue)
- truth = Mat(rows: 2, cols: 2, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [6, 4] as [Float])
- try truth!.put(row: 1, col: 0, data: [6, 4] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testWarpPerspectiveMatMatMatSize() throws {
- let src = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try src.put(row: 0, col: 0, data: [2, 4, 1] as [Float])
- try src.put(row: 1, col: 0, data: [0, 4, 5] as [Float])
- try src.put(row: 2, col: 0, data: [1, 2, 2] as [Float])
- let M = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try M.put(row: 0, col: 0, data: [1, 0, 1] as [Float])
- try M.put(row: 1, col: 0, data: [0, 1, 1] as [Float])
- try M.put(row: 2, col: 0, data: [0, 0, 1] as [Float])
- Imgproc.warpPerspective(src: src, dst: dst, M: M, dsize: Size(width: 3, height: 3))
- truth = Mat(rows: 3, cols: 3, type: CvType.CV_32F)
- try truth!.put(row: 0, col: 0, data: [0, 0, 0] as [Float])
- try truth!.put(row: 1, col: 0, data: [0, 2, 4] as [Float])
- try truth!.put(row: 2, col: 0, data: [0, 0, 4] as [Float])
- try assertMatEqual(truth!, dst, OpenCVTestCase.EPS)
- }
- func testWatershed() throws {
- let image = Mat.eye(rows: 4, cols: 4, type: CvType.CV_8UC(3))
- let markers = Mat(rows: 4, cols: 4, type: CvType.CV_32SC1, scalar: Scalar(0))
- Imgproc.watershed(image: image, markers: markers)
- truth = Mat(rows: 4, cols: 4, type: CvType.CV_32SC1)
- try truth!.put(row: 0, col: 0, data: [-1, -1, -1, -1] as [Int32])
- try truth!.put(row: 1, col: 0, data: [-1, 0, 0, -1] as [Int32])
- try truth!.put(row: 2, col: 0, data: [-1, 0, 0, -1] as [Int32])
- try truth!.put(row: 3, col: 0, data: [-1, -1, -1, -1] as [Int32])
- try assertMatEqual(truth!, markers)
- }
- func testGetTextSize() {
- let text = "iOS all the way"
- let fontScale:Int32 = 2
- let thickness:Int32 = 3
- var baseLine:Int32 = 0
- Imgproc.getTextSize(text: text, fontFace: .FONT_HERSHEY_SCRIPT_SIMPLEX, fontScale: Double(fontScale), thickness: thickness, baseLine: &baseLine)
- let res = Imgproc.getTextSize(text: text, fontFace: .FONT_HERSHEY_SCRIPT_SIMPLEX, fontScale: Double(fontScale), thickness: thickness, baseLine: &baseLine)
- XCTAssertEqual(431, res.width)
- XCTAssertEqual(44, res.height)
- XCTAssertEqual(20, baseLine)
- }
- func testCircleMatPointIntScalar() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let radius = min(gray0clone.cols() / 4, gray0clone.rows() / 4)
- let color = Scalar(128)
- Imgproc.circle(img: gray0clone, center: center, radius: radius, color: color)
- XCTAssert(0 != Core.countNonZero(src: gray0clone))
- }
- func testCircleMatPointIntScalarInt() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let radius = min(gray0clone.cols() / 4, gray0clone.rows() / 4)
- let color = Scalar(128)
- Imgproc.circle(img: gray0clone, center: center, radius: radius, color: color, thickness: Core.FILLED)
- XCTAssert(0 != Core.countNonZero(src: gray0clone))
- }
- func testCircleMatPointIntScalarIntIntInt() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let center2 = Point(x: gray0clone.cols(), y: gray0clone.rows())
- let radius = min(gray0clone.cols() / 4, gray0clone.rows() / 4)
- let color128 = Scalar(128)
- let color0 = Scalar(0)
- Imgproc.circle(img: gray0clone, center: center2, radius: radius * 2, color: color128, thickness: 2, lineType: .LINE_4, shift: 1)
- XCTAssertFalse(0 == Core.countNonZero(src: gray0clone))
- Imgproc.circle(img: gray0clone, center: center, radius: radius, color: color0, thickness: 2, lineType: .LINE_4, shift: 0)
- XCTAssert(0 == Core.countNonZero(src: gray0clone))
- }
- func testClipLine() {
- let r = Rect(x: 10, y: 10, width: 10, height: 10)
- var pt1 = Point(x: 5, y: 15)
- var pt2 = Point(x: 25, y: 15)
- XCTAssert(Imgproc.clipLine(imgRect: r, pt1: pt1, pt2: pt2))
- var pt1Clipped = Point(x: 10, y: 15)
- var pt2Clipped = Point(x: 19, y: 15)
- XCTAssertEqual(pt1Clipped, pt1)
- XCTAssertEqual(pt2Clipped, pt2)
- pt1 = Point(x: 5, y: 5)
- pt2 = Point(x: 25, y: 5)
- pt1Clipped = Point(x: 5, y: 5)
- pt2Clipped = Point(x: 25, y: 5)
- XCTAssertFalse(Imgproc.clipLine(imgRect: r, pt1: pt1, pt2: pt2))
- XCTAssertEqual(pt1Clipped, pt1)
- XCTAssertEqual(pt2Clipped, pt2)
- }
- func testEllipse2Poly() {
- let center = Point(x: 4, y: 4)
- let axes = Size(width: 2, height: 2)
- let angle:Int32 = 30
- let arcStart:Int32 = 30
- let arcEnd:Int32 = 60
- let delta:Int32 = 2
- var pts = [Point]()
- Imgproc.ellipse2Poly(center: center, axes: axes, angle: angle, arcStart: arcStart, arcEnd: arcEnd, delta: delta, pts: &pts)
- let truth = [Point(x: 5, y: 6), Point(x: 4, y: 6)]
- XCTAssert(truth == pts)
- }
- func testEllipseMatPointSizeDoubleDoubleDoubleScalar() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let axes = Size(width: 2, height: 2)
- let angle = 30.0, startAngle = 60.0, endAngle = 90.0
- Imgproc.ellipse(img: gray0clone, center: center, axes: axes, angle: angle, startAngle: startAngle, endAngle: endAngle, color: colorWhite)
- XCTAssert(0 != Core.countNonZero(src: gray0clone))
- }
- func testEllipseMatPointSizeDoubleDoubleDoubleScalarInt() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let axes = Size(width: 2, height: 2)
- let angle = 30.0, startAngle = 60.0, endAngle = 90.0
- Imgproc.ellipse(img: gray0clone, center: center, axes: axes, angle: angle, startAngle: startAngle, endAngle: endAngle, color: colorWhite, thickness: Core.FILLED)
- XCTAssert(0 != Core.countNonZero(src: gray0clone))
- }
- func testEllipseMatPointSizeDoubleDoubleDoubleScalarIntIntInt() {
- let gray0clone = gray0.clone()
- let center = Point(x: gray0clone.cols() / 2, y: gray0clone.rows() / 2)
- let axes = Size(width: 2, height: 2)
- let center2 = Point(x: gray0clone.cols(), y: gray0clone.rows())
- let axes2 = Size(width: 4, height: 4)
- let angle = 30.0, startAngle = 0.0, endAngle = 30.0
- Imgproc.ellipse(img: gray0clone, center: center, axes: axes, angle: angle, startAngle: startAngle, endAngle: endAngle, color: colorWhite, thickness: Core.FILLED, lineType: .LINE_4, shift: 0)
- XCTAssert(0 != Core.countNonZero(src: gray0clone))
- Imgproc.ellipse(img: gray0clone, center: center2, axes: axes2, angle: angle, startAngle: startAngle, endAngle: endAngle, color: colorBlack, thickness: Core.FILLED, lineType: .LINE_4, shift: 1)
- XCTAssertEqual(0, Core.countNonZero(src: gray0clone))
- }
- func testEllipseMatRotatedRectScalar() throws {
- let matSize:Int32 = 10
- let gray0 = Mat.zeros(matSize, cols: matSize, type: CvType.CV_8U)
- let center = Point2f(x: Float(matSize / 2), y: Float(matSize / 2))
- let size = Size2f(width: Float(matSize / 4), height: Float(matSize / 2))
- let box = RotatedRect(center: center, size: size, angle: 45)
- Imgproc.ellipse(img: gray0, box: box, color: Scalar(1))
- let truth = Mat(rows: matSize, cols: matSize, type: CvType.CV_8U)
- try truth.put(row: 0, col: 0, data: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
- 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
- 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
- 0, 0, 0, 0, 0, 0, 1, 1, 0, 0,
- 0, 0, 0, 0, 1, 1, 0, 1, 0, 0,
- 0, 0, 0, 0, 1, 0, 1, 0, 0, 0,
- 0, 0, 0, 1, 0, 1, 1, 0, 0, 0,
- 0, 0, 0, 1, 1, 0, 0, 0, 0, 0,
- 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
- 0, 0, 0, 0, 0, 0, 0, 0, 0, 0] as [Int8])
- try assertMatEqual(truth, gray0)
- }
- func testEllipseMatRotatedRectScalarInt() {
- let gray0clone = gray0.clone()
- let center = Point2f(x: Float(OpenCVTestCase.matSize / 2), y: Float(OpenCVTestCase.matSize / 2))
- let size = Size2f(width: Float(OpenCVTestCase.matSize / 4), height: Float(OpenCVTestCase.matSize / 2))
- let box = RotatedRect(center: center, size: size, angle: 45)
- Imgproc.ellipse(img: gray0clone, box: box, color: Scalar(1), thickness: Core.FILLED)
- Imgproc.ellipse(img: gray0clone, box: box, color: Scalar(0))
- XCTAssert(0 < Core.countNonZero(src: gray0clone))
- }
- func testEllipseMatRotatedRectScalarIntInt() {
- let gray0clone = gray0.clone()
- let center = Point2f(x: Float(OpenCVTestCase.matSize / 2), y: Float(OpenCVTestCase.matSize / 2))
- let size = Size2f(width: 2, height: Float(OpenCVTestCase.matSize * 2 / 3))
- let box = RotatedRect(center: center, size: size, angle: 20)
- Imgproc.ellipse(img: gray0clone, box: box, color: Scalar(9), thickness: 1, lineType: .LINE_AA)
- Imgproc.ellipse(img: gray0clone, box: box, color: Scalar(0), thickness: 1, lineType: .LINE_4)
- XCTAssert(0 < Core.countNonZero(src: gray0clone))
- }
- func testPolylinesMatListOfListOfPointBooleanScalar() {
- let img = gray0.clone()
- let polyline = [[Point(x: 1, y: 1), Point(x: 7, y: 1), Point(x: 7, y: 6), Point(x: 1, y: 6)]]
- Imgproc.polylines(img: img, pts: polyline, isClosed: true, color: Scalar(100))
- XCTAssertEqual(22, Core.countNonZero(src: img))
- Imgproc.polylines(img: img, pts: polyline, isClosed: false, color: Scalar(0))
- XCTAssertEqual(4, Core.countNonZero(src: img))
- }
- func testPolylinesMatListOfListOfPointBooleanScalarInt() {
- let img = gray0.clone()
- let polyline = [[Point(x: 1, y: 1), Point(x: 7, y: 1), Point(x: 7, y: 6), Point(x: 1, y: 6)]]
- Imgproc.polylines(img: img, pts: polyline, isClosed: true, color: Scalar(100), thickness: 2)
- XCTAssertEqual(62, Core.countNonZero(src: img))
- }
- func testPolylinesMatListOfListOfPointBooleanScalarIntIntInt() {
- let img = gray0.clone()
- let polyline1 = [[Point(x: 1, y: 1), Point(x: 7, y: 1), Point(x: 7, y: 6), Point(x: 1, y: 6)]]
- let polyline2 = [[Point(x: 2, y: 2), Point(x: 14, y: 2), Point(x: 14, y: 12), Point(x: 2, y: 12)]]
- Imgproc.polylines(img: img, pts: polyline1, isClosed: true, color: Scalar(100), thickness: 2, lineType: .LINE_8, shift: 0)
- XCTAssert(Core.countNonZero(src: img) > 0)
- Imgproc.polylines(img: img, pts: polyline2, isClosed: true, color: Scalar(0), thickness: 2, lineType: .LINE_8, shift: 1)
- XCTAssertEqual(0, Core.countNonZero(src: img))
- }
- func testPutTextMatStringPointIntDoubleScalar() {
- let text = "Hello World"
- let labelSize = Size(width: 175, height: 22)
- let img = Mat(rows: 20 + labelSize.height, cols: 20 + labelSize.width, type: CvType.CV_8U, scalar: colorBlack)
- let origin = Point(x: 10, y: labelSize.height + 10)
- Imgproc.putText(img: img, text: text, org: origin, fontFace: .FONT_HERSHEY_SIMPLEX, fontScale: 1.0, color: colorWhite)
- XCTAssert(Core.countNonZero(src: img) > 0)
- // check that border is not corrupted
- Imgproc.rectangle(img: img, pt1: Point(x: 11, y: 11), pt2: Point(x: labelSize.width + 10, y: labelSize.height + 10), color: colorBlack, thickness: Core.FILLED)
- XCTAssertEqual(0, Core.countNonZero(src: img))
- }
- func testPutTextMatStringPointIntDoubleScalarInt() {
- let text = "Hello World"
- let labelSize = Size(width: 176, height: 22)
- let img = Mat(rows: 20 + labelSize.height, cols: 20 + labelSize.width, type: CvType.CV_8U, scalar: colorBlack)
- let origin = Point(x: 10, y: labelSize.height + 10)
- Imgproc.putText(img: img, text: text, org: origin, fontFace: .FONT_HERSHEY_SIMPLEX, fontScale: 1.0, color: colorWhite, thickness: 2)
- XCTAssert(Core.countNonZero(src: img) > 0)
- // check that border is not corrupted
- Imgproc.rectangle(img: img, pt1: Point(x: 10, y: 10), pt2: Point(x: labelSize.width + 10 + 1, y: labelSize.height + 10 + 1), color: colorBlack, thickness: Core.FILLED)
- XCTAssertEqual(0, Core.countNonZero(src: img))
- }
- func testPutTextMatStringPointIntDoubleScalarIntIntBoolean() {
- let text = "Hello World"
- let labelSize = Size(width: 175, height: 22)
- let img = Mat(rows: 20 + labelSize.height, cols: 20 + labelSize.width, type: CvType.CV_8U, scalar: colorBlack)
- let origin = Point(x: 10, y: 10)
- Imgproc.putText(img: img, text: text, org: origin, fontFace: .FONT_HERSHEY_SIMPLEX, fontScale: 1.0, color: colorWhite, thickness: 1, lineType: .LINE_8, bottomLeftOrigin: true)
- XCTAssert(Core.countNonZero(src: img) > 0)
- // check that border is not corrupted
- Imgproc.rectangle(img: img, pt1: Point(x: 10, y: 10), pt2: Point(x: labelSize.width + 9, y: labelSize.height + 9), color: colorBlack, thickness: Core.FILLED)
- XCTAssertEqual(0, Core.countNonZero(src: img))
- }
- }
|