You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.6 KiB
C++
74 lines
2.6 KiB
C++
/*
|
|
Copyright 2007 nVidia, Inc.
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License.
|
|
|
|
You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
See the License for the specific language governing permissions and limitations under the License.
|
|
*/
|
|
|
|
// utility class holding common routines
|
|
#pragma once
|
|
#ifndef _ZOH_UTILS_H
|
|
#define _ZOH_UTILS_H
|
|
|
|
#include "nvmath/Vector.h"
|
|
|
|
namespace ZOH {
|
|
|
|
inline int SIGN_EXTEND(int x, int nb) { return ((((signed(x))&(1<<((nb)-1)))?((~0)<<(nb)):0)|(signed(x))); }
|
|
|
|
enum Field {
|
|
FIELD_M = 1, // mode
|
|
FIELD_D = 2, // distribution/shape
|
|
FIELD_RW = 10+0, FIELD_RX = 10+1, FIELD_RY = 10+2, FIELD_RZ = 10+3, // red channel endpoints or deltas
|
|
FIELD_GW = 20+0, FIELD_GX = 20+1, FIELD_GY = 20+2, FIELD_GZ = 20+3, // green channel endpoints or deltas
|
|
FIELD_BW = 30+0, FIELD_BX = 30+1, FIELD_BY = 30+2, FIELD_BZ = 30+3, // blue channel endpoints or deltas
|
|
};
|
|
|
|
// some constants
|
|
static const int F16S_MASK = 0x8000; // f16 sign mask
|
|
static const int F16EM_MASK = 0x7fff; // f16 exp & mantissa mask
|
|
static const int U16MAX = 0xffff;
|
|
static const int S16MIN = -0x8000;
|
|
static const int S16MAX = 0x7fff;
|
|
static const int INT16_MASK = 0xffff;
|
|
static const int F16MAX = 0x7bff; // MAXFLT bit pattern for halfs
|
|
|
|
enum Format { UNSIGNED_F16, SIGNED_F16 };
|
|
|
|
class Utils
|
|
{
|
|
public:
|
|
static Format FORMAT; // this is a global -- we're either handling unsigned or unsigned half values
|
|
|
|
// error metrics
|
|
static float norm(const nv::Vector3 &a, const nv::Vector3 &b);
|
|
static float mpsnr_norm(const nv::Vector3 &a, int exposure, const nv::Vector3 &b);
|
|
|
|
// conversion & clamp
|
|
static int ushort_to_format(unsigned short input);
|
|
static unsigned short format_to_ushort(int input);
|
|
|
|
// clamp to format
|
|
static void clamp(nv::Vector3 &v);
|
|
|
|
// quantization and unquantization
|
|
static int finish_unquantize(int q, int prec);
|
|
static int unquantize(int q, int prec);
|
|
static int quantize(float value, int prec);
|
|
|
|
static void parse(const char *encoding, int &ptr, Field & field, int &endbit, int &len);
|
|
|
|
// lerping
|
|
static int lerp(int a, int b, int i, int denom);
|
|
static nv::Vector3 lerp(const nv::Vector3 & a, const nv::Vector3 & b, int i, int denom);
|
|
};
|
|
|
|
}
|
|
|
|
#endif // _ZOH_UTILS_H
|