godot/thirdparty/embree/kernels/subdiv/gregory_patch_dense.h
2021-01-14 18:02:07 +01:00

114 lines
4.2 KiB
C++

// Copyright 2009-2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include "gregory_patch.h"
namespace embree
{
class __aligned(64) DenseGregoryPatch3fa
{
typedef Vec3fa Vec3fa_4x4[4][4];
public:
__forceinline DenseGregoryPatch3fa (const GregoryPatch3fa& patch)
{
for (size_t y=0; y<4; y++)
for (size_t x=0; x<4; x++)
matrix[y][x] = Vec3ff(patch.v[y][x], 0.0f);
matrix[0][0].w = patch.f[0][0].x;
matrix[0][1].w = patch.f[0][0].y;
matrix[0][2].w = patch.f[0][0].z;
matrix[0][3].w = 0.0f;
matrix[1][0].w = patch.f[0][1].x;
matrix[1][1].w = patch.f[0][1].y;
matrix[1][2].w = patch.f[0][1].z;
matrix[1][3].w = 0.0f;
matrix[2][0].w = patch.f[1][1].x;
matrix[2][1].w = patch.f[1][1].y;
matrix[2][2].w = patch.f[1][1].z;
matrix[2][3].w = 0.0f;
matrix[3][0].w = patch.f[1][0].x;
matrix[3][1].w = patch.f[1][0].y;
matrix[3][2].w = patch.f[1][0].z;
matrix[3][3].w = 0.0f;
}
__forceinline void extract_f_m(Vec3fa f_m[2][2]) const
{
f_m[0][0] = Vec3fa( matrix[0][0].w, matrix[0][1].w, matrix[0][2].w );
f_m[0][1] = Vec3fa( matrix[1][0].w, matrix[1][1].w, matrix[1][2].w );
f_m[1][1] = Vec3fa( matrix[2][0].w, matrix[2][1].w, matrix[2][2].w );
f_m[1][0] = Vec3fa( matrix[3][0].w, matrix[3][1].w, matrix[3][2].w );
}
__forceinline Vec3fa eval(const float uu, const float vv) const
{
__aligned(64) Vec3fa f_m[2][2]; extract_f_m(f_m);
return GregoryPatch3fa::eval(*(Vec3fa_4x4*)&matrix,f_m,uu,vv);
}
__forceinline Vec3fa normal(const float uu, const float vv) const
{
__aligned(64) Vec3fa f_m[2][2]; extract_f_m(f_m);
return GregoryPatch3fa::normal(*(Vec3fa_4x4*)&matrix,f_m,uu,vv);
}
template<class T>
__forceinline Vec3<T> eval(const T &uu, const T &vv) const
{
Vec3<T> f_m[2][2];
f_m[0][0] = Vec3<T>( matrix[0][0].w, matrix[0][1].w, matrix[0][2].w );
f_m[0][1] = Vec3<T>( matrix[1][0].w, matrix[1][1].w, matrix[1][2].w );
f_m[1][1] = Vec3<T>( matrix[2][0].w, matrix[2][1].w, matrix[2][2].w );
f_m[1][0] = Vec3<T>( matrix[3][0].w, matrix[3][1].w, matrix[3][2].w );
return GregoryPatch3fa::eval_t(*(Vec3fa_4x4*)&matrix,f_m,uu,vv);
}
template<class T>
__forceinline Vec3<T> normal(const T &uu, const T &vv) const
{
Vec3<T> f_m[2][2];
f_m[0][0] = Vec3<T>( matrix[0][0].w, matrix[0][1].w, matrix[0][2].w );
f_m[0][1] = Vec3<T>( matrix[1][0].w, matrix[1][1].w, matrix[1][2].w );
f_m[1][1] = Vec3<T>( matrix[2][0].w, matrix[2][1].w, matrix[2][2].w );
f_m[1][0] = Vec3<T>( matrix[3][0].w, matrix[3][1].w, matrix[3][2].w );
return GregoryPatch3fa::normal_t(*(Vec3fa_4x4*)&matrix,f_m,uu,vv);
}
__forceinline void eval(const float u, const float v,
Vec3fa* P, Vec3fa* dPdu, Vec3fa* dPdv, Vec3fa* ddPdudu, Vec3fa* ddPdvdv, Vec3fa* ddPdudv,
const float dscale = 1.0f) const
{
__aligned(64) Vec3fa f_m[2][2]; extract_f_m(f_m);
if (P) {
*P = GregoryPatch3fa::eval(*(Vec3fa_4x4*)&matrix,f_m,u,v);
}
if (dPdu) {
assert(dPdu); *dPdu = GregoryPatch3fa::eval_du(*(Vec3fa_4x4*)&matrix,f_m,u,v)*dscale;
assert(dPdv); *dPdv = GregoryPatch3fa::eval_dv(*(Vec3fa_4x4*)&matrix,f_m,u,v)*dscale;
}
if (ddPdudu) {
assert(ddPdudu); *ddPdudu = GregoryPatch3fa::eval_dudu(*(Vec3fa_4x4*)&matrix,f_m,u,v)*sqr(dscale);
assert(ddPdvdv); *ddPdvdv = GregoryPatch3fa::eval_dvdv(*(Vec3fa_4x4*)&matrix,f_m,u,v)*sqr(dscale);
assert(ddPdudv); *ddPdudv = GregoryPatch3fa::eval_dudv(*(Vec3fa_4x4*)&matrix,f_m,u,v)*sqr(dscale);
}
}
template<typename vbool, typename vfloat>
__forceinline void eval(const vbool& valid, const vfloat& uu, const vfloat& vv, float* P, float* dPdu, float* dPdv, const float dscale, const size_t dstride, const size_t N) const
{
__aligned(64) Vec3fa f_m[2][2]; extract_f_m(f_m);
GregoryPatch3fa::eval(matrix,f_m,valid,uu,vv,P,dPdu,dPdv,dscale,dstride,N);
}
private:
Vec3ff matrix[4][4]; // f_p/m points are stored in 4th component
};
}