gtests/TestFixture.cpp

   1 //
   2 // Copyright (C) 2016 Google, Inc.
   3 //
   4 // All rights reserved.
   5 //
   6 // Redistribution and use in source and binary forms, with or without
   7 // modification, are permitted provided that the following conditions
   8 // are met:
   9 //
  10 //    Redistributions of source code must retain the above copyright
  11 //    notice, this list of conditions and the following disclaimer.
  12 //
  13 //    Redistributions in binary form must reproduce the above
  14 //    copyright notice, this list of conditions and the following
  15 //    disclaimer in the documentation and/or other materials provided
  16 //    with the distribution.
  17 //
  18 //    Neither the name of Google Inc. nor the names of its
  19 //    contributors may be used to endorse or promote products derived
  20 //    from this software without specific prior written permission.
  21 //
  22 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  23 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  24 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  25 // FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  26 // COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  27 // INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  28 // BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  29 // LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  30 // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31 // LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  32 // ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  33 // POSSIBILITY OF SUCH DAMAGE.
  34
  35 #include "TestFixture.h"
  36
  37 namespace glslangtest {
  38
  39 std::string FileNameAsCustomTestSuffix(
  40     const ::testing::TestParamInfo<std::string>& info)
  41 {
  42     std::string name = info.param;
  43     // A valid test case suffix cannot have '.' and '-' inside.
  44     std::replace(name.begin(), name.end(), '.', '_');
  45     std::replace(name.begin(), name.end(), '-', '_');
  46     return name;
  47 }
  48
  49 EShLanguage GetShaderStage(const std::string& stage)
  50 {
  51     if (stage == "vert") {
  52         return EShLangVertex;
  53     } else if (stage == "tesc") {
  54         return EShLangTessControl;
  55     } else if (stage == "tese") {
  56         return EShLangTessEvaluation;
  57     } else if (stage == "geom") {
  58         return EShLangGeometry;
  59     } else if (stage == "frag") {
  60         return EShLangFragment;
  61     } else if (stage == "comp") {
  62         return EShLangCompute;
  63     } else if (stage == "rgen") {
  64         return EShLangRayGenNV;
  65     } else if (stage == "rint") {
  66         return EShLangIntersectNV;
  67     } else if (stage == "rahit") {
  68         return EShLangAnyHitNV;
  69     } else if (stage == "rchit") {
  70         return EShLangClosestHitNV;
  71     } else if (stage == "rmiss") {
  72         return EShLangMissNV;
  73     } else if (stage == "rcall") {
  74         return EShLangCallableNV;
  75     } else if (stage == "task") {
  76         return EShLangTaskNV;
  77     } else if (stage == "mesh") {
  78         return EShLangMeshNV;
  79     } else {
  80         assert(0 && "Unknown shader stage");
  81         return EShLangCount;
  82     }
  83 }
  84
  85 EShMessages DeriveOptions(Source source, Semantics semantics, Target target)
  86 {
  87     EShMessages result = EShMsgCascadingErrors;
  88
  89     switch (source) {
  90         case Source::GLSL:
  91             break;
  92         case Source::HLSL:
  93             result = static_cast<EShMessages>(result | EShMsgReadHlsl);
  94             break;
  95     }
  96
  97     switch (target) {
  98         case Target::AST:
  99             result = static_cast<EShMessages>(result | EShMsgAST);
 100             break;
 101         case Target::Spv:
 102             result = static_cast<EShMessages>(result | EShMsgSpvRules);
 103             result = static_cast<EShMessages>(result | EShMsgKeepUncalled);
 104             break;
 105         case Target::BothASTAndSpv:
 106             result = static_cast<EShMessages>(result | EShMsgSpvRules | EShMsgAST);
 107             result = static_cast<EShMessages>(result | EShMsgKeepUncalled);
 108             break;
 109     };
 110
 111     switch (semantics) {
 112         case Semantics::OpenGL:
 113             break;
 114         case Semantics::Vulkan:
 115             result = static_cast<EShMessages>(result | EShMsgVulkanRules | EShMsgSpvRules);
 116             break;
 117     }
 118
 119     result = static_cast<EShMessages>(result | EShMsgHlslLegalization);
 120
 121     return result;
 122 }
 123
 124 std::pair<bool, std::string> ReadFile(const std::string& path)
 125 {
 126     std::ifstream fstream(path, std::ios::in);
 127     if (fstream) {
 128         std::string contents;
 129         fstream.seekg(0, std::ios::end);
 130         contents.reserve((std::string::size_type)fstream.tellg());
 131         fstream.seekg(0, std::ios::beg);
 132         contents.assign((std::istreambuf_iterator<char>(fstream)),
 133                         std::istreambuf_iterator<char>());
 134         return std::make_pair(true, contents);
 135     }
 136     return std::make_pair(false, "");
 137 }
 138
 139 std::pair<bool, std::vector<std::uint32_t> > ReadSpvBinaryFile(const std::string& path)
 140 {
 141     std::ifstream fstream(path, std::fstream::in | std::fstream::binary);
 142
 143     if (!fstream)
 144         return std::make_pair(false, std::vector<std::uint32_t>());
 145
 146     std::vector<std::uint32_t> contents;
 147
 148     // Reserve space (for efficiency, not for correctness)
 149     fstream.seekg(0, fstream.end);
 150     contents.reserve(size_t(fstream.tellg()) / sizeof(std::uint32_t));
 151     fstream.seekg(0, fstream.beg);
 152
 153     // There is no istream iterator traversing by uint32_t, so we must loop.
 154     while (!fstream.eof()) {
 155         std::uint32_t inWord;
 156         fstream.read((char *)&inWord, sizeof(inWord));
 157
 158         if (!fstream.eof())
 159             contents.push_back(inWord);
 160     }
 161
 162     return std::make_pair(true, contents); // hopefully, c++11 move semantics optimizes the copy away.
 163 }
 164
 165 bool WriteFile(const std::string& path, const std::string& contents)
 166 {
 167     std::ofstream fstream(path, std::ios::out);
 168     if (!fstream) return false;
 169     fstream << contents;
 170     fstream.flush();
 171     return true;
 172 }
 173
 174 std::string GetSuffix(const std::string& name)
 175 {
 176     const size_t pos = name.rfind('.');
 177     return (pos == std::string::npos) ? "" : name.substr(name.rfind('.') + 1);
 178 }
 179
 180 }  // namespace glslangtest