summaryrefslogtreecommitdiff
path: root/src/apps/ociobakelut/main.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/apps/ociobakelut/main.cpp')
-rw-r--r--src/apps/ociobakelut/main.cpp537
1 files changed, 537 insertions, 0 deletions
diff --git a/src/apps/ociobakelut/main.cpp b/src/apps/ociobakelut/main.cpp
new file mode 100644
index 0000000..265d464
--- /dev/null
+++ b/src/apps/ociobakelut/main.cpp
@@ -0,0 +1,537 @@
+/*
+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 <cmath>
+#include <cstdlib>
+#include <iostream>
+#include <sstream>
+#include <fstream>
+#include <vector>
+
+#include <OpenColorIO/OpenColorIO.h>
+namespace OCIO = OCIO_NAMESPACE;
+
+#include "argparse.h"
+#include "ocioicc.h"
+
+static std::string outputfile;
+
+static int
+parse_end_args(int argc, const char *argv[])
+{
+ if(argc>0)
+ {
+ outputfile = argv[0];
+ }
+
+ return 0;
+}
+
+OCIO::GroupTransformRcPtr
+parse_luts(int argc, const char *argv[]);
+
+int main (int argc, const char* argv[])
+{
+
+ bool help = false;
+ int cubesize = -1;
+ int shapersize = -1; // cubsize^2
+ std::string format;
+ std::string inputconfig;
+ std::string inputspace;
+ std::string shaperspace;
+ std::string looks;
+ std::string outputspace;
+ bool usestdout = false;
+ bool verbose = false;
+
+ int whitepointtemp = 6505;
+ std::string displayicc;
+ std::string description;
+ std::string copyright = "OpenColorIO (Sony Imageworks)";
+
+ // What are the allowed baker output formats?
+ std::ostringstream formats;
+ formats << "the lut format to bake: ";
+ for(int i=0; i<OCIO::Baker::getNumFormats(); ++i)
+ {
+ if(i!=0) formats << ", ";
+ formats << OCIO::Baker::getFormatNameByIndex(i);
+ formats << " (." << OCIO::Baker::getFormatExtensionByIndex(i) << ")";
+ }
+ formats << ", icc (.icc)";
+
+ std::string formatstr = formats.str();
+
+ std::string dummystr;
+ float dummyf1, dummyf2, dummyf3;
+
+ ArgParse ap;
+ ap.options("ociobakelut -- create a new LUT or icc profile from an OCIO config or lut file(s)\n\n"
+ "usage: ociobakelut [options] <OUTPUTFILE.LUT>\n\n"
+ "example: ociobakelut --inputspace lg10 --outputspace srgb8 --format flame lg_to_srgb.3dl\n"
+ "example: ociobakelut --lut filmlut.3dl --lut calibration.3dl --format flame display.3dl\n"
+ "example: ociobakelut --lut look.3dl --offset 0.01 -0.02 0.03 --lut display.3dl --format flame display_with_look.3dl\n"
+ "example: ociobakelut --inputspace lg10 --outputspace srgb8 --format icc ~/Library/ColorSync/Profiles/test.icc\n"
+ "example: ociobakelut --lut filmlut.3dl --lut calibration.3dl --format icc ~/Library/ColorSync/Profiles/test.icc\n\n",
+ "%*", parse_end_args, "",
+ "<SEPARATOR>", "Using Existing OCIO Configurations",
+ "--inputspace %s", &inputspace, "Input OCIO ColorSpace (or Role)",
+ "--outputspace %s", &outputspace, "Output OCIO ColorSpace (or Role)",
+ "--shaperspace %s", &shaperspace, "the OCIO ColorSpace or Role, for the shaper",
+ "--looks %s", &looks, "the OCIO looks to apply",
+ "--iconfig %s", &inputconfig, "Input .ocio configuration file (default: $OCIO)\n",
+ "<SEPARATOR>", "Config-Free LUT Baking",
+ "<SEPARATOR>", " (all options can be specified multiple times, each is applied in order)",
+ "--lut %s", &dummystr, "Specify a LUT (forward direction)",
+ "--invlut %s", &dummystr, "Specify a LUT (inverse direction)",
+ "--slope %f %f %f", &dummyf1, &dummyf2, &dummyf3, "slope",
+ "--offset %f %f %f", &dummyf1, &dummyf2, &dummyf3, "offset (float)",
+ "--offset10 %f %f %f", &dummyf1, &dummyf2, &dummyf3, "offset (10-bit)",
+ "--power %f %f %f", &dummyf1, &dummyf2, &dummyf3, "power",
+ "--sat %f", &dummyf1, "saturation (ASC-CDL luma coefficients)\n",
+ "<SEPARATOR>", "Baking Options",
+ "--format %s", &format, formatstr.c_str(),
+ "--shapersize %d", &shapersize, "size of the shaper (default: format specific)",
+ "--cubesize %d", &cubesize, "size of the cube (default: format specific)",
+ "--stdout", &usestdout, "Write to stdout (rather than file)",
+ "--v", &verbose, "Verbose",
+ "--help", &help, "Print help message\n",
+ "<SEPARATOR>", "ICC Options",
+ //"--cubesize %d", &cubesize, "size of the icc CLUT cube (default: 32)",
+ "--whitepoint %d", &whitepointtemp, "whitepoint for the profile (default: 6505)",
+ "--displayicc %s", &displayicc , "an icc profile which matches the OCIO profiles target display",
+ "--description %s", &description , "a meaningful description, this will show up in UI like photoshop",
+ "--copyright %s", &copyright , "a copyright field\n",
+ // TODO: add --metadata option
+ NULL);
+
+ if (ap.parse(argc, argv) < 0)
+ {
+ std::cout << ap.geterror() << std::endl;
+ ap.usage();
+ std::cout << "\n";
+ return 1;
+ }
+
+ if (help || (argc == 1 ))
+ {
+ ap.usage();
+ std::cout << "\n";
+ return 1;
+ }
+
+ // If we're printing to stdout, disable verbose printouts
+ if(usestdout)
+ {
+ verbose = false;
+ }
+
+ // Create the OCIO processor for the specified transform.
+ OCIO::ConstConfigRcPtr config;
+
+
+ OCIO::GroupTransformRcPtr groupTransform;
+
+ try
+ {
+ groupTransform = parse_luts(argc, argv);
+ }
+ catch(const OCIO::Exception & e)
+ {
+ std::cerr << "\nERROR: " << e.what() << std::endl;
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ catch(...)
+ {
+ std::cerr << "\nERROR: An unknown error occurred in parse_luts" << std::endl;
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(!groupTransform)
+ {
+ std::cerr << "\nERROR: parse_luts returned null transform" << std::endl;
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ // If --luts have been specified, synthesize a new (temporary) configuration
+ // with the transformation embedded in a colorspace.
+ if(!groupTransform->empty())
+ {
+ if(!inputspace.empty())
+ {
+ std::cerr << "\nERROR: --inputspace is not allowed when using --lut\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ if(!outputspace.empty())
+ {
+ std::cerr << "\nERROR: --outputspace is not allowed when using --lut\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ if(!looks.empty())
+ {
+ std::cerr << "\nERROR: --looks is not allowed when using --lut\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ if(!shaperspace.empty())
+ {
+ std::cerr << "\nERROR: --shaperspace is not allowed when using --lut\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ OCIO::ConfigRcPtr editableConfig = OCIO::Config::Create();
+
+ OCIO::ColorSpaceRcPtr inputColorSpace = OCIO::ColorSpace::Create();
+ inputspace = "RawInput";
+ inputColorSpace->setName(inputspace.c_str());
+ editableConfig->addColorSpace(inputColorSpace);
+
+ OCIO::ColorSpaceRcPtr outputColorSpace = OCIO::ColorSpace::Create();
+ outputspace = "ProcessedOutput";
+ outputColorSpace->setName(outputspace.c_str());
+
+ outputColorSpace->setTransform(groupTransform,
+ OCIO::COLORSPACE_DIR_FROM_REFERENCE);
+
+ if(verbose)
+ {
+ std::cout << "[OpenColorIO DEBUG]: Specified Transform:";
+ std::cout << *groupTransform;
+ std::cout << "\n";
+ }
+
+ editableConfig->addColorSpace(outputColorSpace);
+ config = editableConfig;
+ }
+ else
+ {
+
+ if(inputspace.empty())
+ {
+ std::cerr << "\nERROR: You must specify the --inputspace.\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(outputspace.empty())
+ {
+ std::cerr << "\nERROR: You must specify the --outputspace.\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(format.empty())
+ {
+ std::cerr << "\nERROR: You must specify the lut format using --format.\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(!inputconfig.empty())
+ {
+ if(!usestdout && verbose)
+ std::cout << "[OpenColorIO INFO]: Loading " << inputconfig << std::endl;
+ config = OCIO::Config::CreateFromFile(inputconfig.c_str());
+ }
+ else if(getenv("OCIO"))
+ {
+ if(!usestdout && verbose)
+ std::cout << "[OpenColorIO INFO]: Loading $OCIO " << getenv("OCIO") << std::endl;
+ config = OCIO::Config::CreateFromEnv();
+ }
+ else
+ {
+ std::cerr << "ERROR: You must specify an input ocio configuration ";
+ std::cerr << "(either with --iconfig or $OCIO).\n\n";
+ ap.usage ();
+ return 1;
+ }
+ }
+
+ if(outputfile.empty() && !usestdout)
+ {
+ std::cerr << "\nERROR: You must specify the outputfile or --stdout.\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ try
+ {
+ if(format == "icc")
+ {
+ if(usestdout)
+ {
+ std::cerr << "\nERROR: --stdout not supported when writing icc profiles.\n\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(outputfile.empty())
+ {
+ std::cerr << "ERROR: you need to specify a output icc path\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(copyright.empty())
+ {
+ std::cerr << "ERROR: need to specify a --copyright to embed in the icc profile\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ if(cubesize<2) cubesize = 32; // default
+
+ OCIO::ConstProcessorRcPtr processor;
+ if (!looks.empty())
+ {
+ OCIO::LookTransformRcPtr transform =
+ OCIO::LookTransform::Create();
+ transform->setLooks(looks.c_str());
+ transform->setSrc(inputspace.c_str());
+ transform->setDst(outputspace.c_str());
+ processor = config->getProcessor(transform,
+ OCIO::TRANSFORM_DIR_FORWARD);
+ }
+ else
+ {
+ processor = config->getProcessor(inputspace.c_str(),
+ outputspace.c_str());
+ }
+
+ SaveICCProfileToFile(outputfile,
+ processor,
+ cubesize,
+ whitepointtemp,
+ displayicc,
+ description,
+ copyright,
+ verbose);
+ }
+ else
+ {
+
+ OCIO::BakerRcPtr baker = OCIO::Baker::Create();
+
+ // setup the baker for our lut type
+ baker->setConfig(config);
+ baker->setFormat(format.c_str());
+ baker->setInputSpace(inputspace.c_str());
+ baker->setShaperSpace(shaperspace.c_str());
+ baker->setLooks(looks.c_str());
+ baker->setTargetSpace(outputspace.c_str());
+ if(shapersize!=-1) baker->setShaperSize(shapersize);
+ if(cubesize!=-1) baker->setCubeSize(cubesize);
+
+ // output lut
+ std::ostringstream output;
+
+ if(!usestdout && verbose)
+ std::cout << "[OpenColorIO INFO]: Baking '" << format << "' lut" << std::endl;
+
+ if(usestdout)
+ {
+ baker->bake(std::cout);
+ }
+ else
+ {
+ std::ofstream f(outputfile.c_str());
+ baker->bake(f);
+ if(verbose)
+ std::cout << "[OpenColorIO INFO]: Wrote '" << outputfile << "'" << std::endl;
+ }
+ }
+ }
+ catch(OCIO::Exception & exception)
+ {
+ std::cerr << "OCIO Error: " << exception.what() << std::endl;
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ catch (std::exception& exception)
+ {
+ std::cerr << "Error: " << exception.what() << "\n";
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+ catch(...)
+ {
+ std::cerr << "Unknown OCIO error encountered." << std::endl;
+ std::cerr << "See --help for more info." << std::endl;
+ return 1;
+ }
+
+ return 0;
+}
+
+
+// TODO: Replace this dirty argument parsing code with a clean version
+// that leverages the same codepath for the standard arguments. If
+// the OIIO derived argparse does not suffice, options we may want to consider
+// include simpleopt, tclap, ultraopt
+
+// TODO: Use better input validation, instead of atof.
+// If too few arguments are provides for scale (let's say only two) and
+// the following argument is the start of another flag (let's say "--invlut")
+// then atof() will likely try to convert "--invlut" to its double equivalent,
+// resulting in an invalid (or at least undesired) scale value.
+
+OCIO::GroupTransformRcPtr
+parse_luts(int argc, const char *argv[])
+{
+ OCIO::GroupTransformRcPtr groupTransform = OCIO::GroupTransform::Create();
+
+ for(int i=0; i<argc; ++i)
+ {
+ std::string arg(argv[i]);
+
+ if(arg == "--lut" || arg == "-lut")
+ {
+ if(i+1>=argc)
+ {
+ throw OCIO::Exception("Error parsing --lut. Invalid num args");
+ }
+
+ OCIO::FileTransformRcPtr t = OCIO::FileTransform::Create();
+ t->setSrc(argv[i+1]);
+ t->setInterpolation(OCIO::INTERP_BEST);
+ groupTransform->push_back(t);
+
+ i += 1;
+ }
+ else if(arg == "--invlut" || arg == "-invlut")
+ {
+ if(i+1>=argc)
+ {
+ throw OCIO::Exception("Error parsing --invlut. Invalid num args");
+ }
+
+ OCIO::FileTransformRcPtr t = OCIO::FileTransform::Create();
+ t->setSrc(argv[i+1]);
+ t->setInterpolation(OCIO::INTERP_BEST);
+ t->setDirection(OCIO::TRANSFORM_DIR_INVERSE);
+ groupTransform->push_back(t);
+
+ i += 1;
+ }
+ else if(arg == "--slope" || arg == "-slope")
+ {
+ if(i+3>=argc)
+ {
+ throw OCIO::Exception("Error parsing --slope. Invalid num args");
+ }
+
+ OCIO::CDLTransformRcPtr t = OCIO::CDLTransform::Create();
+
+ float scale[3];
+ scale[0] = (float) atof(argv[i+1]);
+ scale[1] = (float) atof(argv[i+2]);
+ scale[2] = (float) atof(argv[i+3]);
+ t->setSlope(scale);
+ groupTransform->push_back(t);
+
+ i += 3;
+ }
+ else if(arg == "--offset" || arg == "-offset")
+ {
+ if(i+3>=argc)
+ {
+ throw OCIO::Exception("Error parsing --offset. Invalid num args");
+ }
+
+ OCIO::CDLTransformRcPtr t = OCIO::CDLTransform::Create();
+
+ float offset[3];
+ offset[0] = (float) atof(argv[i+1]);
+ offset[1] = (float) atof(argv[i+2]);
+ offset[2] = (float) atof(argv[i+3]);
+ t->setOffset(offset);
+ groupTransform->push_back(t);
+
+ i += 3;
+ }
+ else if(arg == "--offset10" || arg == "-offset10")
+ {
+ if(i+3>=argc)
+ {
+ throw OCIO::Exception("Error parsing --offset10. Invalid num args");
+ }
+
+ OCIO::CDLTransformRcPtr t = OCIO::CDLTransform::Create();
+
+ float offset[3];
+ offset[0] = (float) atof(argv[i+1]) / 1023.0f;
+ offset[1] = (float) atof(argv[i+2]) / 1023.0f;
+ offset[2] = (float) atof(argv[i+3]) / 1023.0f;
+ t->setOffset(offset);
+ groupTransform->push_back(t);
+ i += 3;
+ }
+ else if(arg == "--power" || arg == "-power")
+ {
+ if(i+3>=argc)
+ {
+ throw OCIO::Exception("Error parsing --power. Invalid num args");
+ }
+
+ OCIO::CDLTransformRcPtr t = OCIO::CDLTransform::Create();
+
+ float power[3];
+ power[0] = (float) atof(argv[i+1]);
+ power[1] = (float) atof(argv[i+2]);
+ power[2] = (float) atof(argv[i+3]);
+ t->setPower(power);
+ groupTransform->push_back(t);
+
+ i += 3;
+ }
+ else if(arg == "--sat" || arg == "-sat")
+ {
+ if(i+1>=argc)
+ {
+ throw OCIO::Exception("Error parsing --sat. Invalid num args");
+ }
+
+ OCIO::CDLTransformRcPtr t = OCIO::CDLTransform::Create();
+ t->setSat((float) atof(argv[i+1]));
+ groupTransform->push_back(t);
+
+ i += 1;
+ }
+ }
+
+ return groupTransform;
+}
+
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-15 02:35:25 +0000