2019-11-04 02:23:13 -05:00
|
|
|
package com.chameleonvision.classabstraction.pipeline;
|
|
|
|
|
|
2019-11-21 05:32:19 -05:00
|
|
|
import com.chameleonvision.classabstraction.pipeline.pipes.Draw2dContoursPipe;
|
|
|
|
|
import org.apache.commons.lang3.tuple.Pair;
|
2019-11-04 02:23:13 -05:00
|
|
|
import org.opencv.core.Mat;
|
|
|
|
|
|
2019-11-10 11:47:56 -05:00
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.function.Supplier;
|
|
|
|
|
|
2019-11-21 05:32:19 -05:00
|
|
|
import static com.chameleonvision.classabstraction.pipeline.DriverVisionPipeline.DriverPipelineResult;
|
2019-11-10 11:47:56 -05:00
|
|
|
|
|
|
|
|
public class DriverVisionPipeline extends CVPipeline<DriverPipelineResult, CVPipelineSettings> {
|
|
|
|
|
|
2019-11-23 04:05:37 -05:00
|
|
|
public DriverVisionPipeline(CVPipelineSettings settings) {
|
|
|
|
|
super(settings);
|
2019-11-04 02:23:13 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
2019-11-10 11:47:56 -05:00
|
|
|
public DriverPipelineResult runPipeline(Mat inputMat) {
|
|
|
|
|
|
2019-11-21 05:32:19 -05:00
|
|
|
outputMat = inputMat;
|
|
|
|
|
|
|
|
|
|
var camProps = cameraProcess.getProperties().staticProperties;
|
|
|
|
|
|
|
|
|
|
Draw2dContoursPipe.Draw2dContoursSettings draw2dContoursSettings = new Draw2dContoursPipe.Draw2dContoursSettings();
|
|
|
|
|
draw2dContoursSettings.showCrosshair = true;
|
|
|
|
|
Draw2dContoursPipe draw2dContoursPipe = new Draw2dContoursPipe(draw2dContoursSettings, camProps);
|
|
|
|
|
|
|
|
|
|
outputMat = draw2dContoursPipe.run(Pair.of(outputMat, null)).getLeft();
|
2019-11-10 11:47:56 -05:00
|
|
|
|
2019-11-16 14:30:44 -05:00
|
|
|
return new DriverPipelineResult(null, inputMat, 0);
|
2019-11-04 02:23:13 -05:00
|
|
|
}
|
|
|
|
|
|
2019-11-07 11:15:54 -05:00
|
|
|
public static class DriverPipelineResult extends CVPipelineResult<Void> {
|
2019-11-16 14:30:44 -05:00
|
|
|
public DriverPipelineResult(List<Void> targets, Mat outputMat, long processTime) {
|
|
|
|
|
super(targets, outputMat, processTime);
|
2019-11-07 11:15:54 -05:00
|
|
|
}
|
2019-11-04 02:23:13 -05:00
|
|
|
}
|
|
|
|
|
}
|