Affine transform

base/CMakeLists.txt

@@ -293,6 +293,7 @@ IF(ENABLE_ARM64)
 		src/NvTransform.cpp
 		src/ApraEGLDisplay.cpp
 		src/DMAFDToHostCopy.cpp
+		src/AffineTransform.cpp
 	)
 ELSE()
 	SET(CUDA_IP_FILES ${CUDA_IP_FILES}		
@@ -316,7 +317,7 @@ SET(CUDA_IP_FILES_H
 IF(ENABLE_ARM64)
 	SET(CUDA_IP_FILES_H ${CUDA_IP_FILES_H}
 		include/JPEGDecoderL4TMHelper.h
-		include/JPEGDecoderL4TM.h	
+		include/JPEGDecoderL4TM.h	 
 		include/JPEGEncoderL4TMHelper.h
 		include/JPEGEncoderL4TM.h
 		include/AV4L2Buffer.h
@@ -338,6 +339,7 @@ IF(ENABLE_ARM64)
 		include/ApraEGLDisplay.h
 		include/DMAFrameUtils.h
 		include/DMAFDToHostCopy.h
+		include/AffineTransform.h
 	)
 ELSE()
 	SET(CUDA_IP_FILES_H ${CUDA_IP_FILES_H}		

base/include/AffineTransform.h

@@ -0,0 +1,64 @@
+#pragma once
+
+#include "Module.h"
+#include "CudaCommon.h"
+
+class AffineTransformProps : public ModuleProps
+{
+public:
+	AffineTransformProps(cudastream_sp &_stream, double _angle, int _x=0, int _y=0, float _scale=1.0f)
+	{
+		stream = _stream;
+		angle = _angle;
+		x = _x;
+		y = _y;
+		scale = _scale;
+	}
+
+	int x=0;
+	int y = 0;
+	float scale = 1.0f;
+	double angle;
+	cudastream_sp stream;
+	size_t getSerializeSize()
+	{
+		return ModuleProps::getSerializeSize() + sizeof(int) * 2 + sizeof(double) + sizeof(float) ;
+	}
+
+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 ;
+	}
+};
+
+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);
+	void setProps(AffineTransform);
+	bool handlePropsChange(frame_sp &frame);
+
+private:
+	class Detail;
+	boost::shared_ptr<Detail> mDetail;
+};