Affine transform module - support DMA/CUDA via NPPI and HOST via open cv

base/CMakeLists.txt

@@ -263,6 +263,7 @@ SET(IP_FILES
 	src/FacialLandmarksCV.cpp
 	src/ImageEncoderCV.cpp
 	src/RotateCV.cpp
+	src/AffineTransform.cpp
 	src/BrightnessContrastControlXform.cpp
 	src/VirtualPTZ.cpp
 	src/WebCamSource.cpp
@@ -293,6 +294,7 @@ SET(IP_FILES_H
 	include/FacialLandmarksCV.h
 	include/ImageEncoderCV.h
 	include/RotateCV.h
+	include/AffineTransform.h
 	include/BrightnessContrastControlXform.h
 	include/VirtualPTZ.h
 	include/WebCamSource.h
@@ -526,6 +528,7 @@ SET(UT_FILES
 	test/faciallandmarkscv_tests.cpp
 	test/ImageEncodeCV_tests.cpp
 	test/rotatecv_tests.cpp
+	test/affinetransform_tests.cpp
 	test/brightness_contrast_tests.cpp
 	test/virtualptz_tests.cpp
 	test/webcam_source_tests.cpp

base/include/AffineTransform.h

@@ -0,0 +1,115 @@
+#pragma once
+
+#include "Module.h"
+#ifdef APRA_CUDA_ENABLED
+#include "CudaCommon.h"
+#endif
+
+class DetailMemoryAbstract;
+class DeatilCUDA;
+class DetailDMA;
+class DetailHost;
+class DetailGPU;
+
+class AffineTransformProps : public ModuleProps
+{
+public:
+	enum Interpolation {
+		NN = 0,     
+		LINEAR, 
+		CUBIC,
+		UNDEFINED,
+		CUBIC2P_BSPLINE,
+		CUBIC2P_CATMULLROM,
+		CUBIC2P_B05C03,
+		SUPER,
+		LANCZOS,
+		LANCZOS3_ADVANCED
+	};
+
+	enum TransformType
+	{
+		USING_OPENCV = 0,
+		USING_NPPI
+	};
+
+	AffineTransformProps(TransformType _type, double _angle, int _x = 0, int _y = 0, double _scale = 1.0)
+	{
+		if (_type != TransformType::USING_OPENCV)
+		{
+			throw AIPException(AIP_FATAL, "This constructor only supports Opencv");
+		}
+		angle = _angle;
+		x = _x;
+		y = _y;
+		scale = _scale;
+		type = _type;
+	}
+#ifdef APRA_CUDA_ENABLED
+	AffineTransformProps(TransformType _type, Interpolation _interpolation, cudastream_sp &_stream, double _angle, int _x=0, int _y=0, double _scale = 1.0)
+	{
+		if (_type != TransformType::USING_NPPI)
+		{
+			throw AIPException(AIP_FATAL, "This constructor only supports using NPPI");
+		}
+		stream = _stream;
+		angle = _angle;
+		x = _x;
+		y = _y;
+		scale = _scale;
+		interpolation = _interpolation;
+		type = _type;
+	}
+#endif
+
+	int x = 0;
+	int y = 0;
+	double scale = 1.0;
+	double angle;
+#ifdef APRA_CUDA_ENABLED
+	cudastream_sp stream;
+#endif
+	Interpolation interpolation = AffineTransformProps::NN;
+	TransformType type;
+
+	size_t getSerializeSize()
+	{
+		return ModuleProps::getSerializeSize() + sizeof(int) * 2 + sizeof(double) + sizeof(float) + sizeof(interpolation) + sizeof(type);
+	}
+
+private:
+	friend class boost::serialization::access;
+
+	template <class Archive>
+	void serialize(Archive &ar, const unsigned int version)
+	{
+		ar &boost::serialization::base_object<ModuleProps>(*this);
+		ar &angle &x &y &scale ;
+		ar& interpolation;
+		ar& type;
+	}
+};
+
+class AffineTransform : public Module
+{
+
+public:
+	AffineTransform(AffineTransformProps props);
+	virtual ~AffineTransform();
+	bool init();
+	bool term();
+	void setProps(AffineTransformProps &props);
+	AffineTransformProps getProps();
+
+protected:
+	bool process(frame_container &frames);
+	bool processSOS(frame_sp &frame);
+	bool validateInputPins();
+	bool validateOutputPins();
+	void addInputPin(framemetadata_sp &metadata, string &pinId); // throws exception if validation fails
+	bool shouldTriggerSOS();
+	bool processEOS(string &pinId);
+	AffineTransformProps mProp;
+	bool handlePropsChange(frame_sp& frame);
+	boost::shared_ptr<DetailMemoryAbstract> mDetail;
+};