itsmattkc
a0b6bb842d
OCIO config was set on a per-project basis, but we were using a singleton for the ColorManager that would break if more than one project was ever open. Now the ColorManager belongs to the Project and is always accessed through the Project.
156 lines
3.4 KiB
C++
156 lines
3.4 KiB
C++
#include "colormanager.h"
|
|
|
|
#include <QFloat16>
|
|
|
|
#include "common/define.h"
|
|
#include "config/config.h"
|
|
|
|
ColorManager::ColorManager()
|
|
{
|
|
// Ensures config is set to something
|
|
config_ = OCIO::GetCurrentConfig();
|
|
}
|
|
|
|
OCIO::ConstConfigRcPtr ColorManager::GetConfig() const
|
|
{
|
|
return config_;
|
|
}
|
|
|
|
void ColorManager::SetConfig(const QString &filename)
|
|
{
|
|
SetConfig(OCIO::Config::CreateFromFile(filename.toUtf8()));
|
|
}
|
|
|
|
void ColorManager::SetConfig(OCIO::ConstConfigRcPtr config)
|
|
{
|
|
config_ = config;
|
|
|
|
emit ConfigChanged();
|
|
}
|
|
|
|
void ColorManager::DisassociateAlpha(FramePtr f)
|
|
{
|
|
AssociateAlphaPixFmtFilter(kDisassociate, f);
|
|
}
|
|
|
|
void ColorManager::AssociateAlpha(FramePtr f)
|
|
{
|
|
AssociateAlphaPixFmtFilter(kAssociate, f);
|
|
}
|
|
|
|
void ColorManager::ReassociateAlpha(FramePtr f)
|
|
{
|
|
AssociateAlphaPixFmtFilter(kReassociate, f);
|
|
}
|
|
|
|
QStringList ColorManager::ListAvailableDisplays()
|
|
{
|
|
QStringList displays;
|
|
|
|
int number_of_displays = config_->getNumDisplays();
|
|
|
|
for (int i=0;i<number_of_displays;i++) {
|
|
displays.append(config_->getDisplay(i));
|
|
}
|
|
|
|
return displays;
|
|
}
|
|
|
|
QString ColorManager::GetDefaultDisplay()
|
|
{
|
|
return config_->getDefaultDisplay();
|
|
}
|
|
|
|
QStringList ColorManager::ListAvailableViews(QString display)
|
|
{
|
|
QStringList views;
|
|
|
|
int number_of_views = config_->getNumViews(display.toUtf8());
|
|
|
|
for (int i=0;i<number_of_views;i++) {
|
|
views.append(config_->getView(display.toUtf8(), i));
|
|
}
|
|
|
|
return views;
|
|
}
|
|
|
|
QString ColorManager::GetDefaultView(const QString &display)
|
|
{
|
|
return config_->getDefaultView(display.toUtf8());
|
|
}
|
|
|
|
QStringList ColorManager::ListAvailableLooks()
|
|
{
|
|
QStringList looks;
|
|
|
|
int number_of_looks = config_->getNumLooks();
|
|
|
|
for (int i=0;i<number_of_looks;i++) {
|
|
looks.append(config_->getLookNameByIndex(i));
|
|
}
|
|
|
|
return looks;
|
|
}
|
|
|
|
QStringList ColorManager::ListAvailableInputColorspaces()
|
|
{
|
|
return ListAvailableInputColorspaces(config_);
|
|
}
|
|
|
|
QStringList ColorManager::ListAvailableInputColorspaces(OCIO::ConstConfigRcPtr config)
|
|
{
|
|
QStringList spaces;
|
|
|
|
int number_of_colorspaces = config->getNumColorSpaces();
|
|
|
|
for (int i=0;i<number_of_colorspaces;i++) {
|
|
spaces.append(config->getColorSpaceNameByIndex(i));
|
|
}
|
|
|
|
return spaces;
|
|
}
|
|
|
|
void ColorManager::AssociateAlphaPixFmtFilter(ColorManager::AlphaAction action, FramePtr f)
|
|
{
|
|
int pixel_count = f->width() * f->height() * kRGBAChannels;
|
|
|
|
switch (static_cast<olive::PixelFormat>(f->format())) {
|
|
case olive::PIX_FMT_INVALID:
|
|
case olive::PIX_FMT_COUNT:
|
|
qWarning() << "Alpha association functions received an invalid pixel format";
|
|
break;
|
|
case olive::PIX_FMT_RGBA8:
|
|
case olive::PIX_FMT_RGBA16U:
|
|
qWarning() << "Alpha association functions only works on float-based pixel formats at this time";
|
|
break;
|
|
case olive::PIX_FMT_RGBA16F:
|
|
{
|
|
AssociateAlphaInternal<qfloat16>(action, reinterpret_cast<qfloat16*>(f->data()), pixel_count);
|
|
break;
|
|
}
|
|
case olive::PIX_FMT_RGBA32F:
|
|
{
|
|
AssociateAlphaInternal<float>(action, reinterpret_cast<float*>(f->data()), pixel_count);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void ColorManager::AssociateAlphaInternal(ColorManager::AlphaAction action, T *data, int pix_count)
|
|
{
|
|
for (int i=0;i<pix_count;i+=kRGBAChannels) {
|
|
T alpha = data[i+kRGBChannels];
|
|
|
|
if (action == kAssociate || alpha > 0) {
|
|
for (int j=0;j<kRGBChannels;j++) {
|
|
if (action == kDisassociate) {
|
|
data[i+j] /= alpha;
|
|
} else {
|
|
data[i+j] *= alpha;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|