/src/core/ParseUtils.cpp
http://github.com/imageworks/OpenColorIO · C++ · 510 lines · 382 code · 90 blank · 38 comment · 230 complexity · 375bac55a1b3a0a108d30e042d2766b3 MD5 · raw file
- /*
- Copyright (c) 2003-2010 Sony Pictures Imageworks Inc., et al.
- 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 Sony Pictures Imageworks 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
- OWNER 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 <iostream>
- #include <set>
- #include <sstream>
- #include <OpenColorIO/OpenColorIO.h>
- #include "ParseUtils.h"
- #include "pystring/pystring.h"
- OCIO_NAMESPACE_ENTER
- {
- const char * BoolToString(bool val)
- {
- if(val) return "true";
- return "false";
- }
-
- bool BoolFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if((str == "true") || (str=="yes")) return true;
- return false;
- }
-
- const char * LoggingLevelToString(LoggingLevel level)
- {
- if(level == LOGGING_LEVEL_NONE) return "none";
- else if(level == LOGGING_LEVEL_WARNING) return "warning";
- else if(level == LOGGING_LEVEL_INFO) return "info";
- else if(level == LOGGING_LEVEL_DEBUG) return "debug";
- return "unknown";
- }
-
- LoggingLevel LoggingLevelFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "0" || str == "none") return LOGGING_LEVEL_NONE;
- else if(str == "1" || str == "warning") return LOGGING_LEVEL_WARNING;
- else if(str == "2" || str == "info") return LOGGING_LEVEL_INFO;
- else if(str == "3" || str == "debug") return LOGGING_LEVEL_DEBUG;
- return LOGGING_LEVEL_UNKNOWN;
- }
-
- const char * TransformDirectionToString(TransformDirection dir)
- {
- if(dir == TRANSFORM_DIR_FORWARD) return "forward";
- else if(dir == TRANSFORM_DIR_INVERSE) return "inverse";
- return "unknown";
- }
-
- TransformDirection TransformDirectionFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "forward") return TRANSFORM_DIR_FORWARD;
- else if(str == "inverse") return TRANSFORM_DIR_INVERSE;
- return TRANSFORM_DIR_UNKNOWN;
- }
-
- TransformDirection CombineTransformDirections(TransformDirection d1,
- TransformDirection d2)
- {
- // Any unknowns always combine to be unknown.
- if(d1 == TRANSFORM_DIR_UNKNOWN || d2 == TRANSFORM_DIR_UNKNOWN)
- return TRANSFORM_DIR_UNKNOWN;
-
- if(d1 == TRANSFORM_DIR_FORWARD && d2 == TRANSFORM_DIR_FORWARD)
- return TRANSFORM_DIR_FORWARD;
-
- if(d1 == TRANSFORM_DIR_INVERSE && d2 == TRANSFORM_DIR_INVERSE)
- return TRANSFORM_DIR_FORWARD;
-
- return TRANSFORM_DIR_INVERSE;
- }
-
- TransformDirection GetInverseTransformDirection(TransformDirection dir)
- {
- if(dir == TRANSFORM_DIR_FORWARD) return TRANSFORM_DIR_INVERSE;
- else if(dir == TRANSFORM_DIR_INVERSE) return TRANSFORM_DIR_FORWARD;
- return TRANSFORM_DIR_UNKNOWN;
- }
-
- const char * ColorSpaceDirectionToString(ColorSpaceDirection dir)
- {
- if(dir == COLORSPACE_DIR_TO_REFERENCE) return "to_reference";
- else if(dir == COLORSPACE_DIR_FROM_REFERENCE) return "from_reference";
- return "unknown";
- }
-
- ColorSpaceDirection ColorSpaceDirectionFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "to_reference") return COLORSPACE_DIR_TO_REFERENCE;
- else if(str == "from_reference") return COLORSPACE_DIR_FROM_REFERENCE;
- return COLORSPACE_DIR_UNKNOWN;
- }
-
- const char * BitDepthToString(BitDepth bitDepth)
- {
- if(bitDepth == BIT_DEPTH_UINT8) return "8ui";
- else if(bitDepth == BIT_DEPTH_UINT10) return "10ui";
- else if(bitDepth == BIT_DEPTH_UINT12) return "12ui";
- else if(bitDepth == BIT_DEPTH_UINT14) return "14ui";
- else if(bitDepth == BIT_DEPTH_UINT16) return "16ui";
- else if(bitDepth == BIT_DEPTH_UINT32) return "32ui";
- else if(bitDepth == BIT_DEPTH_F16) return "16f";
- else if(bitDepth == BIT_DEPTH_F32) return "32f";
- return "unknown";
- }
-
- BitDepth BitDepthFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "8ui") return BIT_DEPTH_UINT8;
- else if(str == "10ui") return BIT_DEPTH_UINT10;
- else if(str == "12ui") return BIT_DEPTH_UINT12;
- else if(str == "14ui") return BIT_DEPTH_UINT14;
- else if(str == "16ui") return BIT_DEPTH_UINT16;
- else if(str == "32ui") return BIT_DEPTH_UINT32;
- else if(str == "16f") return BIT_DEPTH_F16;
- else if(str == "32f") return BIT_DEPTH_F32;
- return BIT_DEPTH_UNKNOWN;
- }
-
- bool BitDepthIsFloat(BitDepth bitDepth)
- {
- if(bitDepth == BIT_DEPTH_F16) return true;
- else if(bitDepth == BIT_DEPTH_F32) return true;
- return false;
- }
-
- int BitDepthToInt(BitDepth bitDepth)
- {
- if(bitDepth == BIT_DEPTH_UINT8) return 8;
- else if(bitDepth == BIT_DEPTH_UINT10) return 10;
- else if(bitDepth == BIT_DEPTH_UINT12) return 12;
- else if(bitDepth == BIT_DEPTH_UINT14) return 14;
- else if(bitDepth == BIT_DEPTH_UINT16) return 16;
- else if(bitDepth == BIT_DEPTH_UINT32) return 32;
-
- return 0;
- }
-
- const char * AllocationToString(Allocation alloc)
- {
- if(alloc == ALLOCATION_UNIFORM) return "uniform";
- else if(alloc == ALLOCATION_LG2) return "lg2";
- return "unknown";
- }
-
- Allocation AllocationFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "uniform") return ALLOCATION_UNIFORM;
- else if(str == "lg2") return ALLOCATION_LG2;
- return ALLOCATION_UNKNOWN;
- }
-
- const char * InterpolationToString(Interpolation interp)
- {
- if(interp == INTERP_NEAREST) return "nearest";
- else if(interp == INTERP_LINEAR) return "linear";
- else if(interp == INTERP_TETRAHEDRAL) return "tetrahedral";
- else if(interp == INTERP_BEST) return "best";
- return "unknown";
- }
-
- Interpolation InterpolationFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "nearest") return INTERP_NEAREST;
- else if(str == "linear") return INTERP_LINEAR;
- else if(str == "tetrahedral") return INTERP_TETRAHEDRAL;
- else if(str == "best") return INTERP_BEST;
- return INTERP_UNKNOWN;
- }
-
- const char * GpuLanguageToString(GpuLanguage language)
- {
- if(language == GPU_LANGUAGE_CG) return "cg";
- else if(language == GPU_LANGUAGE_GLSL_1_0) return "glsl_1.0";
- else if(language == GPU_LANGUAGE_GLSL_1_3) return "glsl_1.3";
- return "unknown";
- }
-
- GpuLanguage GpuLanguageFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "cg") return GPU_LANGUAGE_CG;
- else if(str == "glsl_1.0") return GPU_LANGUAGE_GLSL_1_0;
- else if(str == "glsl_1.3") return GPU_LANGUAGE_GLSL_1_3;
- return GPU_LANGUAGE_UNKNOWN;
- }
-
-
- const char * EnvironmentModeToString(EnvironmentMode mode)
- {
- if(mode == ENV_ENVIRONMENT_LOAD_PREDEFINED) return "loadpredefined";
- else if(mode == ENV_ENVIRONMENT_LOAD_ALL) return "loadall";
- return "unknown";
- }
-
- EnvironmentMode EnvironmentModeFromString(const char * s)
- {
- std::string str = pystring::lower(s);
- if(str == "loadpredefined") return ENV_ENVIRONMENT_LOAD_PREDEFINED;
- else if(str == "loadall") return ENV_ENVIRONMENT_LOAD_ALL;
- return ENV_ENVIRONMENT_UNKNOWN;
- }
-
- const char * ROLE_DEFAULT = "default";
- const char * ROLE_REFERENCE = "reference";
- const char * ROLE_DATA = "data";
- const char * ROLE_COLOR_PICKING = "color_picking";
- const char * ROLE_SCENE_LINEAR = "scene_linear";
- const char * ROLE_COMPOSITING_LOG = "compositing_log";
- const char * ROLE_COLOR_TIMING = "color_timing";
- const char * ROLE_TEXTURE_PAINT = "texture_paint";
- const char * ROLE_MATTE_PAINT = "matte_paint";
-
- namespace
- {
- const int FLOAT_DECIMALS = 7;
- const int DOUBLE_DECIMALS = 16;
- }
-
- std::string FloatToString(float value)
- {
- std::ostringstream pretty;
- pretty.precision(FLOAT_DECIMALS);
- pretty << value;
- return pretty.str();
- }
-
- std::string FloatVecToString(const float * fval, unsigned int size)
- {
- if(size<=0) return "";
-
- std::ostringstream pretty;
- pretty.precision(FLOAT_DECIMALS);
- for(unsigned int i=0; i<size; ++i)
- {
- if(i!=0) pretty << " ";
- pretty << fval[i];
- }
-
- return pretty.str();
- }
-
- bool StringToFloat(float * fval, const char * str)
- {
- if(!str) return false;
-
- std::istringstream inputStringstream(str);
- float x;
- if(!(inputStringstream >> x))
- {
- return false;
- }
-
- if(fval) *fval = x;
- return true;
- }
-
- bool StringToInt(int * ival, const char * str, bool failIfLeftoverChars)
- {
- if(!str) return false;
- if(!ival) return false;
-
- std::istringstream i(str);
- char c=0;
- if (!(i >> *ival) || (failIfLeftoverChars && i.get(c))) return false;
- return true;
- }
-
-
- std::string DoubleToString(double value)
- {
- std::ostringstream pretty;
- pretty.precision(DOUBLE_DECIMALS);
- pretty << value;
- return pretty.str();
- }
-
-
- bool StringVecToFloatVec(std::vector<float> &floatArray,
- const std::vector<std::string> &lineParts)
- {
- floatArray.resize(lineParts.size());
-
- for(unsigned int i=0; i<lineParts.size(); i++)
- {
- std::istringstream inputStringstream(lineParts[i]);
- float x;
- if(!(inputStringstream >> x))
- {
- return false;
- }
- floatArray[i] = x;
- }
-
- return true;
- }
-
-
- bool StringVecToIntVec(std::vector<int> &intArray,
- const std::vector<std::string> &lineParts)
- {
- intArray.resize(lineParts.size());
-
- for(unsigned int i=0; i<lineParts.size(); i++)
- {
- std::istringstream inputStringstream(lineParts[i]);
- int x;
- if(!(inputStringstream >> x))
- {
- return false;
- }
- intArray[i] = x;
- }
-
- return true;
- }
-
- ////////////////////////////////////////////////////////////////////////////
-
- // read the next non empty line, and store it in 'line'
- // return 'true' on success
-
- bool nextline(std::istream &istream, std::string &line)
- {
- while ( istream.good() )
- {
- std::getline(istream, line);
- if(line.size() > 0 && line[line.size() - 1] == '\r')
- {
- line.resize(line.size() - 1);
- }
- if(!pystring::strip(line).empty())
- {
- return true;
- }
- }
-
- line = "";
- return false;
- }
-
-
- bool StrEqualsCaseIgnore(const std::string & a, const std::string & b)
- {
- return (pystring::lower(a) == pystring::lower(b));
- }
-
- // If a ',' is in the string, split on it
- // If a ':' is in the string, split on it
- // Otherwise, assume a single string.
- // Also, strip whitespace from all parts.
-
- void SplitStringEnvStyle(std::vector<std::string> & outputvec, const char * str)
- {
- if(!str) return;
-
- std::string s = pystring::strip(str);
- if(pystring::find(s, ",") > -1)
- {
- pystring::split(s, outputvec, ",");
- }
- else if(pystring::find(s, ":") > -1)
- {
- pystring::split(s, outputvec, ":");
- }
- else
- {
- outputvec.push_back(s);
- }
-
- for(unsigned int i=0; i<outputvec.size(); ++i)
- {
- outputvec[i] = pystring::strip(outputvec[i]);
- }
- }
-
- std::string JoinStringEnvStyle(const std::vector<std::string> & outputvec)
- {
- return pystring::join(", ", outputvec);
- }
-
- // Ordering and capitalization from vec1 is preserved
- std::vector<std::string> IntersectStringVecsCaseIgnore(const std::vector<std::string> & vec1,
- const std::vector<std::string> & vec2)
- {
- std::vector<std::string> newvec;
- std::set<std::string> allvalues;
-
- // Seed the set with all values from vec2
- for(unsigned int i=0; i<vec2.size(); ++i)
- {
- allvalues.insert(pystring::lower(vec2[i]));
- }
-
- for(unsigned int i=0; i<vec1.size(); ++i)
- {
- std::string key = pystring::lower(vec1[i]);
- if(allvalues.find(key) != allvalues.end())
- {
- newvec.push_back(vec1[i]);
- }
- }
-
- return newvec;
- }
-
-
- int FindInStringVecCaseIgnore(const std::vector<std::string> & vec, const std::string & str)
- {
- std::string teststr = pystring::lower(str);
- for(unsigned int i=0; i<vec.size(); ++i)
- {
- if(pystring::lower(vec[i]) == teststr) return static_cast<int>(i);
- }
-
- return -1;
- }
- }
- OCIO_NAMESPACE_EXIT
- ///////////////////////////////////////////////////////////////////////////////
- #ifdef OCIO_UNIT_TEST
- OCIO_NAMESPACE_USING
- #include "UnitTest.h"
- OIIO_ADD_TEST(ParseUtils, StringToInt)
- {
- int ival = 0;
- bool success = false;
-
- success = StringToInt(&ival, "", false);
- OIIO_CHECK_EQUAL(success, false);
-
- success = StringToInt(&ival, "9", false);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, 9);
-
- success = StringToInt(&ival, " 10 ", false);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, 10);
-
- success = StringToInt(&ival, " 101", true);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, 101);
-
- success = StringToInt(&ival, " 11x ", false);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, 11);
-
- success = StringToInt(&ival, " 12x ", true);
- OIIO_CHECK_EQUAL(success, false);
-
- success = StringToInt(&ival, "13", true);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, 13);
-
- success = StringToInt(&ival, "-14", true);
- OIIO_CHECK_EQUAL(success, true);
- OIIO_CHECK_EQUAL(ival, -14);
-
- success = StringToInt(&ival, "x-15", false);
- OIIO_CHECK_EQUAL(success, false);
-
- success = StringToInt(&ival, "x-16", false);
- OIIO_CHECK_EQUAL(success, false);
-
-
- }
- #endif // OCIO_UNIT_TEST