141 строка
5.1 KiB
C++
141 строка
5.1 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// //
|
|
// DxilShaderModel.cpp //
|
|
// Copyright (C) Microsoft Corporation. All rights reserved. //
|
|
// This file is distributed under the University of Illinois Open Source //
|
|
// License. See LICENSE.TXT for details. //
|
|
// //
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "dxc/HLSL/DxilShaderModel.h"
|
|
#include "dxc/HLSL/DxilSemantic.h"
|
|
#include "dxc/Support/Global.h"
|
|
|
|
|
|
namespace hlsl {
|
|
|
|
ShaderModel::ShaderModel(Kind Kind, unsigned Major, unsigned Minor, const char *pszName,
|
|
unsigned NumInputRegs, unsigned NumOutputRegs,
|
|
bool bUAVs, bool bTypedUavs, unsigned NumUAVRegs)
|
|
: m_Kind(Kind)
|
|
, m_Major(Major)
|
|
, m_Minor(Minor)
|
|
, m_pszName(pszName)
|
|
, m_NumInputRegs(NumInputRegs)
|
|
, m_NumOutputRegs(NumOutputRegs)
|
|
, m_bUAVs(bUAVs)
|
|
, m_bTypedUavs(bTypedUavs)
|
|
, m_NumUAVRegs(NumUAVRegs) {
|
|
}
|
|
|
|
bool ShaderModel::IsValid() const {
|
|
DXASSERT(IsPS() || IsVS() || IsGS() || IsHS() || IsDS() || IsCS() || m_Kind == Kind::Invalid, "invalid shader model");
|
|
return m_Kind != Kind::Invalid;
|
|
}
|
|
|
|
const ShaderModel *ShaderModel::Get(unsigned Idx) {
|
|
DXASSERT_NOMSG(Idx < kNumShaderModels - 1);
|
|
if (Idx < kNumShaderModels - 1)
|
|
return &ms_ShaderModels[Idx];
|
|
else
|
|
return GetInvalid();
|
|
}
|
|
|
|
const ShaderModel *ShaderModel::Get(Kind Kind, unsigned Major, unsigned Minor) {
|
|
// Linear search. Replaced by binary search if necessary.
|
|
for (unsigned i = 0; i < kNumShaderModels; i++) {
|
|
const ShaderModel *pSM = &ms_ShaderModels[i];
|
|
if (pSM->m_Kind == Kind && pSM->m_Major == Major && pSM->m_Minor == Minor)
|
|
return pSM;
|
|
}
|
|
|
|
return GetInvalid();
|
|
}
|
|
|
|
const ShaderModel *ShaderModel::GetByName(const char *pszName) {
|
|
// [ps|vs|gs|hs|ds|cs]_[major]_[minor]
|
|
Kind Kind;
|
|
switch (pszName[0]) {
|
|
case 'p': Kind = Kind::Pixel; break;
|
|
case 'v': Kind = Kind::Vertex; break;
|
|
case 'g': Kind = Kind::Geometry; break;
|
|
case 'h': Kind = Kind::Hull; break;
|
|
case 'd': Kind = Kind::Domain; break;
|
|
case 'c': Kind = Kind::Compute; break;
|
|
default: return GetInvalid();
|
|
}
|
|
unsigned Idx = 3;
|
|
if (pszName[1] != 's' || pszName[2] != '_')
|
|
return GetInvalid();
|
|
|
|
unsigned Major;
|
|
switch (pszName[Idx++]) {
|
|
case '4': Major = 4; break;
|
|
case '5': Major = 5; break;
|
|
case '6': Major = 6; break;
|
|
default: return GetInvalid();
|
|
}
|
|
if (pszName[Idx++] != '_')
|
|
return GetInvalid();
|
|
|
|
unsigned Minor;
|
|
switch (pszName[Idx++]) {
|
|
case '0': Minor = 0; break;
|
|
case '1': Minor = 1; break;
|
|
default: return GetInvalid();
|
|
}
|
|
if (pszName[Idx++] != 0)
|
|
return GetInvalid();
|
|
|
|
return Get(Kind, Major, Minor);
|
|
}
|
|
|
|
std::string ShaderModel::GetKindName() const {
|
|
return std::string(m_pszName).substr(0, 2);
|
|
}
|
|
|
|
const ShaderModel *ShaderModel::GetInvalid() {
|
|
return &ms_ShaderModels[kNumShaderModels - 1];
|
|
}
|
|
|
|
typedef ShaderModel SM;
|
|
typedef Semantic SE;
|
|
const ShaderModel ShaderModel::ms_ShaderModels[kNumShaderModels] = {
|
|
// IR OR UAV? TyUAV? UAV base
|
|
SM(Kind::Compute, 4, 0, "cs_4_0", 0, 0, true, false, 1),
|
|
SM(Kind::Compute, 4, 1, "cs_4_1", 0, 0, true, false, 1),
|
|
SM(Kind::Compute, 5, 0, "cs_5_0", 0, 0, true, true, 64),
|
|
SM(Kind::Compute, 5, 1, "cs_5_1", 0, 0, true, true, UINT_MAX),
|
|
SM(Kind::Compute, 6, 0, "cs_6_0", 0, 0, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Domain, 5, 0, "ds_5_0", 32, 32, true, true, 64),
|
|
SM(Kind::Domain, 5, 1, "ds_5_1", 32, 32, true, true, UINT_MAX),
|
|
SM(Kind::Domain, 6, 0, "ds_6_0", 32, 32, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Geometry, 4, 0, "gs_4_0", 16, 32, false, false, 0),
|
|
SM(Kind::Geometry, 4, 1, "gs_4_1", 32, 32, false, false, 0),
|
|
SM(Kind::Geometry, 5, 0, "gs_5_0", 32, 32, true, true, 64),
|
|
SM(Kind::Geometry, 5, 1, "gs_5_1", 32, 32, true, true, UINT_MAX),
|
|
SM(Kind::Geometry, 6, 0, "gs_6_0", 32, 32, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Hull, 5, 0, "hs_5_0", 32, 32, true, true, 64),
|
|
SM(Kind::Hull, 5, 1, "hs_5_1", 32, 32, true, true, UINT_MAX),
|
|
SM(Kind::Hull, 6, 0, "hs_6_0", 32, 32, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Pixel, 4, 0, "ps_4_0", 32, 8, false, false, 0),
|
|
SM(Kind::Pixel, 4, 1, "ps_4_1", 32, 8, false, false, 0),
|
|
SM(Kind::Pixel, 5, 0, "ps_5_0", 32, 8, true, true, 64),
|
|
SM(Kind::Pixel, 5, 1, "ps_5_1", 32, 8, true, true, UINT_MAX),
|
|
SM(Kind::Pixel, 6, 0, "ps_6_0", 32, 8, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Vertex, 4, 0, "vs_4_0", 16, 16, false, false, 0),
|
|
SM(Kind::Vertex, 4, 1, "vs_4_1", 32, 32, false, false, 0),
|
|
SM(Kind::Vertex, 5, 0, "vs_5_0", 32, 32, true, true, 64),
|
|
SM(Kind::Vertex, 5, 1, "vs_5_1", 32, 32, true, true, UINT_MAX),
|
|
SM(Kind::Vertex, 6, 0, "vs_6_0", 32, 32, true, true, UINT_MAX),
|
|
|
|
SM(Kind::Invalid, 0, 0, "invalid", 0, 0, false, false, 0),
|
|
};
|
|
|
|
} // namespace hlsl
|