105 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			105 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| 
 | |
| /*
 | |
|  * Copyright 2013, Blender Foundation.
 | |
|  *
 | |
|  * This program is free software; you can redistribute it and/or
 | |
|  * modify it under the terms of the GNU General Public License
 | |
|  * as published by the Free Software Foundation; either version 2
 | |
|  * of the License, or (at your option) any later version.
 | |
|  *
 | |
|  * This program is distributed in the hope that it will be useful,
 | |
|  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|  * GNU General Public License for more details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU General Public License
 | |
|  * along with this program; if not, write to the Free Software Foundation,
 | |
|  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 | |
|  *
 | |
|  * Contributor:
 | |
|  *		Sergey Sharybin
 | |
|  */
 | |
| 
 | |
| #ifndef _COM_PlaneTrackCommonOperation_h
 | |
| #define _COM_PlaneTrackCommonOperation_h
 | |
| 
 | |
| #include <string.h>
 | |
| 
 | |
| #include "COM_NodeOperation.h"
 | |
| 
 | |
| #include "DNA_movieclip_types.h"
 | |
| #include "DNA_tracking_types.h"
 | |
| 
 | |
| #include "BLI_listbase.h"
 | |
| #include "BLI_string.h"
 | |
| 
 | |
| #define PLANE_DISTORT_MAX_SAMPLES 64
 | |
| 
 | |
| class PlaneDistortWarpImageOperation : public NodeOperation {
 | |
| protected:
 | |
| 	struct MotionSample {
 | |
| 		float frameSpaceCorners[4][2];  /* Corners coordinates in pixel space. */
 | |
| 		float perspectiveMatrix[3][3];
 | |
| 	};
 | |
| 	SocketReader *m_pixelReader;
 | |
| 	MotionSample m_samples[PLANE_DISTORT_MAX_SAMPLES];
 | |
| 	int m_motion_blur_samples;
 | |
| 	float m_motion_blur_shutter;
 | |
| 
 | |
| public:
 | |
| 	PlaneDistortWarpImageOperation();
 | |
| 
 | |
| 	void calculateCorners(const float corners[4][2],
 | |
| 	                      bool normalized,
 | |
| 	                      int sample);
 | |
| 
 | |
| 	void initExecution();
 | |
| 	void deinitExecution();
 | |
| 
 | |
| 	void executePixelSampled(float output[4], float x, float y, PixelSampler sampler);
 | |
| 
 | |
| 	bool determineDependingAreaOfInterest(rcti *input, ReadBufferOperation *readOperation, rcti *output);
 | |
| 
 | |
| 	void setMotionBlurSamples(int samples) {
 | |
| 		BLI_assert(samples <= PLANE_DISTORT_MAX_SAMPLES);
 | |
| 		this->m_motion_blur_samples = samples;
 | |
| 	}
 | |
| 	void setMotionBlurShutter(float shutter) {
 | |
| 		this->m_motion_blur_shutter = shutter;
 | |
| 	}
 | |
| };
 | |
| 
 | |
| 
 | |
| class PlaneDistortMaskOperation : public NodeOperation {
 | |
| protected:
 | |
| 	struct MotionSample {
 | |
| 		float frameSpaceCorners[4][2];  /* Corners coordinates in pixel space. */
 | |
| 	};
 | |
| 	int m_osa;
 | |
| 	MotionSample m_samples[PLANE_DISTORT_MAX_SAMPLES];
 | |
| 	float m_jitter[32][2];
 | |
| 	int m_motion_blur_samples;
 | |
| 	float m_motion_blur_shutter;
 | |
| 
 | |
| public:
 | |
| 	PlaneDistortMaskOperation();
 | |
| 
 | |
| 	void calculateCorners(const float corners[4][2],
 | |
| 	                      bool normalized,
 | |
| 	                      int sample);
 | |
| 
 | |
| 	void initExecution();
 | |
| 
 | |
| 	void executePixelSampled(float output[4], float x, float y, PixelSampler sampler);
 | |
| 
 | |
| 	void setMotionBlurSamples(int samples) {
 | |
| 		BLI_assert(samples <= PLANE_DISTORT_MAX_SAMPLES);
 | |
| 		this->m_motion_blur_samples = samples;
 | |
| 	}
 | |
| 	void setMotionBlurShutter(float shutter) {
 | |
| 		this->m_motion_blur_shutter = shutter;
 | |
| 	}
 | |
| };
 | |
| 
 | |
| #endif
 |