f84fb12f5d
This commit adds functionality for operations that require pixel translation or resizing on "Full Frame" mode, allowing to adjust their canvas. It fixes most cropping issues in translate, scale, rotate and transform nodes by adjusting their canvas to the result, instead of the input canvas. Operations output buffer is still always on (0,0) position for easier image algorithm implementation, even when the canvas is not. Current limitations (will be addressed on bcon2): - Displayed translation in Viewer node is limited to 6000px. - When scaling up the canvas size is limited to the scene resolution size x 1.5 . From that point it crops. If none of these limitations are hit, the Viewer node displays the full input with any translation. Differential Revision: https://developer.blender.org/D12466
113 lines
3.1 KiB
C++
113 lines
3.1 KiB
C++
/*
|
|
* 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.
|
|
*
|
|
* Copyright 2011, Blender Foundation.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "COM_MultiThreadedOperation.h"
|
|
#include "DNA_node_types.h"
|
|
|
|
struct RNG;
|
|
|
|
namespace blender::compositor {
|
|
|
|
class ScreenLensDistortionOperation : public MultiThreadedOperation {
|
|
private:
|
|
/**
|
|
* Cached reference to the inputProgram
|
|
*/
|
|
SocketReader *m_inputProgram;
|
|
struct RNG *m_rng;
|
|
|
|
bool m_fit;
|
|
bool m_jitter;
|
|
|
|
float m_dispersion;
|
|
float m_distortion;
|
|
bool m_dispersion_const;
|
|
bool m_distortion_const;
|
|
bool m_variables_ready;
|
|
float m_k[3];
|
|
float m_k4[3];
|
|
float m_dk4[3];
|
|
float m_maxk;
|
|
float m_sc, m_cx, m_cy;
|
|
|
|
public:
|
|
ScreenLensDistortionOperation();
|
|
|
|
void init_data() override;
|
|
|
|
/**
|
|
* The inner loop of this operation.
|
|
*/
|
|
void executePixel(float output[4], int x, int y, void *data) override;
|
|
|
|
/**
|
|
* Initialize the execution
|
|
*/
|
|
void initExecution() override;
|
|
|
|
void *initializeTileData(rcti *rect) override;
|
|
/**
|
|
* Deinitialize the execution
|
|
*/
|
|
void deinitExecution() override;
|
|
|
|
void setFit(bool fit)
|
|
{
|
|
m_fit = fit;
|
|
}
|
|
void setJitter(bool jitter)
|
|
{
|
|
m_jitter = jitter;
|
|
}
|
|
|
|
/** Set constant distortion value */
|
|
void setDistortion(float distortion);
|
|
/** Set constant dispersion value */
|
|
void setDispersion(float dispersion);
|
|
|
|
bool determineDependingAreaOfInterest(rcti *input,
|
|
ReadBufferOperation *readOperation,
|
|
rcti *output) override;
|
|
|
|
void determine_canvas(const rcti &preferred_area, rcti &r_area) override;
|
|
void get_area_of_interest(int input_idx, const rcti &output_area, rcti &r_input_area) override;
|
|
void update_memory_buffer_partial(MemoryBuffer *output,
|
|
const rcti &area,
|
|
Span<MemoryBuffer *> inputs) override;
|
|
|
|
private:
|
|
void determineUV(float result[6], float x, float y) const;
|
|
void updateVariables(float distortion, float dispersion);
|
|
|
|
void get_uv(const float xy[2], float uv[2]) const;
|
|
void distort_uv(const float uv[2], float t, float xy[2]) const;
|
|
bool get_delta(float r_sq, float k4, const float uv[2], float delta[2]) const;
|
|
void accumulate(const MemoryBuffer *buffer,
|
|
int a,
|
|
int b,
|
|
float r_sq,
|
|
const float uv[2],
|
|
const float delta[3][2],
|
|
float sum[4],
|
|
int count[3]) const;
|
|
};
|
|
|
|
} // namespace blender::compositor
|