mirror of
https://github.com/KhronosGroup/glslang
synced 2024-11-13 05:30:06 +00:00
1868b14435
This PR adds handling of the numthreads attribute for compute shaders, as well as a general infrastructure for returning attribute values from acceptAttributes, which may be needed in other cases, e.g, unroll(x), or merely to know if some attribute without params was given. A map of enum values from TAttributeType to TIntermAggregate nodes is built and returned. It can be queried with operator[] on the map. In the future there may be a need to also handle strings (e.g, for patchconstantfunc), and those can be easily added into the class if needed. New test is in hlsl.numthreads.comp.
109 lines
4.0 KiB
C++
109 lines
4.0 KiB
C++
//
|
|
//Copyright (C) 2016 LunarG, Inc.
|
|
//
|
|
//All rights reserved.
|
|
//
|
|
//Redistribution and use in source and binary forms, with or without
|
|
//modification, are permitted provided that the following conditions
|
|
//are met:
|
|
//
|
|
// Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
//
|
|
// Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following
|
|
// disclaimer in the documentation and/or other materials provided
|
|
// with the distribution.
|
|
//
|
|
// Neither the name of Google, Inc., nor the names of its
|
|
// contributors may be used to endorse or promote products derived
|
|
// from this software without specific prior written permission.
|
|
//
|
|
//THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
//"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
//LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
//FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
//COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
//INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
//BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
//LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
//CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
//LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
|
//ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
//POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
|
|
#include "hlslAttributes.h"
|
|
#include <cstdlib>
|
|
#include <cctype>
|
|
|
|
namespace glslang {
|
|
// Map the given string to an attribute enum from TAttributeType,
|
|
// or EatNone if invalid.
|
|
TAttributeType TAttributeMap::attributeFromName(const TString& name)
|
|
{
|
|
// These are case insensitive.
|
|
TString lowername(name);
|
|
std::transform(lowername.begin(), lowername.end(), lowername.begin(), ::tolower);
|
|
|
|
if (lowername == "allow_uav_condition")
|
|
return EatAllow_uav_condition;
|
|
else if (lowername == "branch")
|
|
return EatBranch;
|
|
else if (lowername == "call")
|
|
return EatCall;
|
|
else if (lowername == "domain")
|
|
return EatDomain;
|
|
else if (lowername == "earlydepthstencil")
|
|
return EatEarlydepthstencil;
|
|
else if (lowername == "fastopt")
|
|
return EatFastopt;
|
|
else if (lowername == "flatten")
|
|
return EatFlatten;
|
|
else if (lowername == "forcecase")
|
|
return EatForcecase;
|
|
else if (lowername == "instance")
|
|
return EatInstance;
|
|
else if (lowername == "maxtessfactor")
|
|
return EatMaxtessfactor;
|
|
else if (lowername == "numthreads")
|
|
return EatNumthreads;
|
|
else if (lowername == "outputcontrolpoints")
|
|
return EatOutputcontrolpoints;
|
|
else if (lowername == "outputtopology")
|
|
return EatOutputtopology;
|
|
else if (lowername == "partitioning")
|
|
return EatPartitioning;
|
|
else if (lowername == "patchconstantfunc")
|
|
return EatPatchconstantfunc;
|
|
else if (lowername == "unroll")
|
|
return EatUnroll;
|
|
else
|
|
return EatNone;
|
|
}
|
|
|
|
// Look up entry, inserting if it's not there, and if name is a valid attribute name
|
|
// as known by attributeFromName.
|
|
TAttributeType TAttributeMap::setAttribute(const TString* name, TIntermAggregate* value)
|
|
{
|
|
if (name == nullptr)
|
|
return EatNone;
|
|
|
|
const TAttributeType attr = attributeFromName(*name);
|
|
|
|
if (attr != EatNone)
|
|
attributes[attr] = value;
|
|
|
|
return attr;
|
|
}
|
|
|
|
// Look up entry (const version), and return aggregate node. This cannot change the map.
|
|
const TIntermAggregate* TAttributeMap::operator[](TAttributeType attr) const
|
|
{
|
|
const auto entry = attributes.find(attr);
|
|
|
|
return (entry == attributes.end()) ? nullptr : entry->second;
|
|
}
|
|
|
|
} // end namespace glslang
|