framework/common/tcuStringTemplate.cpp

   1 /*-------------------------------------------------------------------------
   2  * drawElements Quality Program Tester Core
   3  * ----------------------------------------
   4  *
   5  * Copyright 2014 The Android Open Source Project
   6  *
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  *
  19  *//*!
  20  * \file
  21  * \brief String template class.
  22  *//*--------------------------------------------------------------------*/
  23
  24 #include "tcuStringTemplate.hpp"
  25 #include "tcuDefs.hpp"
  26
  27 #include <sstream>
  28
  29 using std::string;
  30 using std::map;
  31 using std::ostringstream;
  32
  33 namespace tcu
  34 {
  35
  36 StringTemplate::StringTemplate (void)
  37 {
  38 }
  39
  40 StringTemplate::StringTemplate (const std::string& str)
  41 {
  42         setString(str);
  43 }
  44
  45 StringTemplate::~StringTemplate (void)
  46 {
  47 }
  48
  49 void StringTemplate::setString (const std::string& str)
  50 {
  51         m_template = str;
  52 }
  53
  54 string StringTemplate::specialize (const map<string, string>& params) const
  55 {
  56         ostringstream res;
  57
  58         size_t curNdx = 0;
  59         for (;;)
  60         {
  61                 size_t paramNdx = m_template.find("${", curNdx);
  62                 if (paramNdx != string::npos)
  63                 {
  64                         // Append in-between stuff.
  65                         res << m_template.substr(curNdx, paramNdx - curNdx);
  66
  67                         // Find end-of-param.
  68                         size_t paramEndNdx = m_template.find("}", paramNdx);
  69                         if (paramEndNdx == string::npos)
  70                                 TCU_THROW(InternalError, "No '}' found in template parameter");
  71
  72                         // Parse parameter contents.
  73                         string  paramStr                = m_template.substr(paramNdx+2, paramEndNdx-2-paramNdx);
  74                         bool    paramSingleLine = false;
  75                         bool    paramOptional   = false;
  76                         string  paramName;
  77                         size_t colonNdx = paramStr.find(":");
  78                         if (colonNdx != string::npos)
  79                         {
  80                                 paramName = paramStr.substr(0, colonNdx);
  81                                 string flagsStr = paramStr.substr(colonNdx+1);
  82                                 if (flagsStr == "single-line")
  83                                 {
  84                                         paramSingleLine = true;
  85                                 }
  86                                 else if (flagsStr == "opt")
  87                                 {
  88                                         paramOptional = true;
  89                                 }
  90                                 else
  91                                 {
  92                                         TCU_THROW(InternalError, (string("Unrecognized flag") + paramStr).c_str());
  93                                 }
  94                         }
  95                         else
  96                                 paramName = paramStr;
  97
  98                         // Fill in parameter value.
  99                         if (params.find(paramName) != params.end())
 100                         {
 101                                 const string& val = (*params.find(paramName)).second;
 102                                 if (paramSingleLine)
 103                                 {
 104                                         string tmp = val;
 105                                         for (size_t ndx = tmp.find("\n"); ndx != string::npos; ndx = tmp.find("\n"))
 106                                                 tmp = tmp.replace(ndx, 1, " ");
 107                                         res << tmp;
 108                                 }
 109                                 else
 110                                         res << val;
 111                         }
 112                         else if (!paramOptional)
 113                                 TCU_THROW(InternalError, (string("Value for parameter '") + paramName + "' not found in map").c_str());
 114
 115                         // Skip over template.
 116                         curNdx = paramEndNdx + 1;
 117                 }
 118                 else
 119                 {
 120                         if (curNdx < m_template.length())
 121                                 res << &m_template[curNdx];
 122
 123                         break;
 124                 }
 125         }
 126
 127         return res.str();
 128 }
 129
 130 } // tcu