static int
action_colorconvert (int argc, const char *argv[])
{
ASSERT (argc == 3);
if (ot.postpone_callback (1, action_colorconvert, argc, argv))
return 0;
std::string fromspace = argv[1];
std::string tospace = argv[2];
ot.read ();
bool need_transform = false;
ImageRecRef A = ot.curimg;
ot.read (A);
for (int s = 0, send = A->subimages(); s < send; ++s) {
for (int m = 0, mend = A->miplevels(s); m < mend; ++m) {
const ImageSpec *spec = A->spec(s,m);
need_transform |=
spec->get_string_attribute("oiio:ColorSpace") != tospace;
}
}
if (! need_transform)
return 1; // no need to do anything
ot.pop ();
ot.push (new ImageRec (*A, ot.allsubimages ? -1 : 0,
ot.allsubimages ? -1 : 0, true, false));
if (fromspace == "current")
fromspace = A->spec(0,0)->get_string_attribute ("oiio:Colorspace", "Linear");
ColorProcessor *processor =
ot.colorconfig.createColorProcessor (fromspace.c_str(), tospace.c_str());
if (! processor)
return 1;
for (int s = 0, send = A->subimages(); s < send; ++s) {
for (int m = 0, mend = A->miplevels(s); m < mend; ++m) {
ImageBufAlgo::colorconvert ((*ot.curimg)(s,m), (*A)(s,m), processor, false);
ot.curimg->spec(s,m)->attribute ("oiio::Colorspace", tospace);
}
}
ot.colorconfig.deleteColorProcessor (processor);
return 1;
}
static int
action_flip (int argc, const char *argv[])
{
if (ot.postpone_callback (1, action_flip, argc, argv))
return 0;
ot.read ();
ImageRecRef A = ot.pop();
ot.push (new ImageRec (*A, ot.allsubimages ? -1 : 0,
ot.allsubimages ? -1 : 0, true, false));
int subimages = ot.curimg->subimages();
for (int s = 0; s < subimages; ++s) {
int miplevels = ot.curimg->miplevels(s);
for (int m = 0; m < miplevels; ++m) {
const ImageBuf &Aib ((*A)(s,m));
ImageBuf &Rib ((*ot.curimg)(s,m));
ImageBuf::ConstIterator<float> a (Aib);
ImageBuf::Iterator<float> r (Rib);
int nchans = Rib.nchannels();
int firstscanline = Rib.ymin();
int lastscanline = Rib.ymax();
for ( ; ! r.done(); ++r) {
a.pos (r.x(), lastscanline - (r.y() - firstscanline));
for (int c = 0; c < nchans; ++c)
r[c] = a[c];
}
}
}
return 0;
}
static int
action_add (int argc, const char *argv[])
{
if (ot.postpone_callback (2, action_add, argc, argv))
return 0;
ImageRecRef B (ot.pop());
ImageRecRef A (ot.pop());
ot.read (A);
ot.read (B);
ot.push (new ImageRec (*A, ot.allsubimages ? -1 : 0,
ot.allsubimages ? -1 : 0, true, false));
int subimages = ot.curimg->subimages();
for (int s = 0; s < subimages; ++s) {
int miplevels = ot.curimg->miplevels(s);
for (int m = 0; m < miplevels; ++m) {
const ImageBuf &Aib ((*A)(s,m));
const ImageBuf &Bib ((*B)(s,m));
if (! same_size (Aib, Bib)) {
// FIXME: some day, there should be options of combining
// differing images somehow.
std::cerr << "oiiotool: " << argv[0] << " could not combine images of differing sizes\n";
continue;
}
ImageBuf &Rib ((*ot.curimg)(s,m));
ImageBufAlgo::add (Rib, Aib, Bib);
}
}
return 0;
}
static int
action_select_subimage (int argc, const char *argv[])
{
if (ot.postpone_callback (1, action_select_subimage, argc, argv))
return 0;
ot.read ();
if (ot.curimg->subimages() == 1)
return 0; // --subimage on a single-image file is a no-op
int subimage = std::min (atoi(argv[1]), ot.curimg->subimages());
ImageRecRef A = ot.pop();
ot.push (new ImageRec (*A, subimage));
return 0;
}
static int
set_full_to_pixels (int argc, const char *argv[])
{
if (ot.postpone_callback (1, set_full_to_pixels, argc, argv))
return 0;
ot.read ();
ImageRecRef A = ot.curimg;
ImageSpec &spec (*A->spec(0,0));
spec.full_x = spec.x;
spec.full_y = spec.y;
spec.full_width = spec.width;
spec.full_height = spec.height;
A->metadata_modified (true);
return 0;
}
bool
OiioTool::set_attribute (ImageRecRef img, const std::string &attribname,
TypeDesc type, const std::string &value)
{
ot.read (img);
img->metadata_modified (true);
if (! value.length()) {
// If the value is the empty string, clear the attribute
return apply_spec_mod (*img, do_erase_attribute,
attribname, ot.allsubimages);
}
// Does it seem to be an int, or did the caller explicitly request
// that it be set as an int?
char *p = NULL;
int i = strtol (value.c_str(), &p, 10);
while (*p && isspace(*p))
++p;
if ((! *p && type == TypeDesc::UNKNOWN) || type == TypeDesc::INT) {
// int conversion succeeded and accounted for the whole string --
// so set an int attribute.
return apply_spec_mod (*img, do_set_any_attribute<int>,
std::pair<std::string,int>(attribname,i),
ot.allsubimages);
}
// Does it seem to be a float, or did the caller explicitly request
// that it be set as a float?
p = NULL;
float f = (float)strtod (value.c_str(), &p);
while (*p && isspace(*p))
++p;
if ((! *p && type == TypeDesc::UNKNOWN) || type == TypeDesc::FLOAT) {
// float conversion succeeded and accounted for the whole string --
// so set a float attribute.
return apply_spec_mod (*img, do_set_any_attribute<float>,
std::pair<std::string,float>(attribname,f),
ot.allsubimages);
}
// Otherwise, set it as a string attribute
return apply_spec_mod (*img, do_set_any_attribute<std::string>,
std::pair<std::string,std::string>(attribname,value),
ot.allsubimages);
}
static int
action_selectmip (int argc, const char *argv[])
{
if (ot.postpone_callback (1, action_unmip, argc, argv))
return 0;
ot.read ();
bool mipmapped = false;
for (int s = 0, send = ot.curimg->subimages(); s < send; ++s)
mipmapped |= (ot.curimg->miplevels(s) > 1);
if (! mipmapped) {
return 0; // --selectmip on an unmipped image is a no-op
}
ImageRecRef newimg (new ImageRec (*ot.curimg, -1, atoi(argv[1]), true, true));
ot.curimg = newimg;
return 0;
}
static int
action_sub (int argc, const char *argv[])
{
if (ot.postpone_callback (2, action_sub, argc, argv))
return 0;
ImageRecRef B (ot.pop());
ImageRecRef A (ot.pop());
ot.read (A);
ot.read (B);
ot.push (new ImageRec (*A, ot.allsubimages ? -1 : 0,
ot.allsubimages ? -1 : 0, true, false));
int subimages = ot.curimg->subimages();
for (int s = 0; s < subimages; ++s) {
int miplevels = ot.curimg->miplevels(s);
for (int m = 0; m < miplevels; ++m) {
const ImageBuf &Aib ((*A)(s,m));
const ImageBuf &Bib ((*B)(s,m));
if (! same_size (Aib, Bib)) {
// FIXME: some day, there should be options of combining
// differing images somehow.
std::cerr << "oiiotool: " << argv[0] << " could not combine images of differing sizes\n";
continue;
}
ImageBuf &Rib ((*ot.curimg)(s,m));
ImageBuf::ConstIterator<float> a (Aib);
ImageBuf::ConstIterator<float> b (Bib);
ImageBuf::Iterator<float> r (Rib);
int nchans = Rib.nchannels();
for ( ; ! r.done(); ++r) {
a.pos (r.x(), r.y());
b.pos (r.x(), r.y());
for (int c = 0; c < nchans; ++c)
r[c] = a[c] - b[c];
}
}
}
return 0;
}
static int
set_origin (int argc, const char *argv[])
{
if (ot.postpone_callback (1, set_origin, argc, argv))
return 0;
ot.read ();
ImageRecRef A = ot.curimg;
ImageSpec &spec (*A->spec(0,0));
int x = spec.x, y = spec.y;
int w = spec.width, h = spec.height;
adjust_geometry (w, h, x, y, argv[1]);
if (spec.width != w || spec.height != h)
std::cerr << argv[0] << " can't be used to change the size, only the origin\n";
if (spec.x != x || spec.y != y) {
spec.x = x;
spec.y = y;
A->metadata_modified (true);
}
return 0;
}
static int
set_fullsize (int argc, const char *argv[])
{
if (ot.postpone_callback (1, set_fullsize, argc, argv))
return 0;
ot.read ();
ImageRecRef A = ot.curimg;
ImageSpec &spec (*A->spec(0,0));
int x = spec.full_x, y = spec.full_y;
int w = spec.full_width, h = spec.full_height;
adjust_geometry (w, h, x, y, argv[1]);
if (spec.full_x != x || spec.full_y != y ||
spec.full_width != w || spec.full_height != h) {
spec.full_x = x;
spec.full_y = y;
spec.full_width = w;
spec.full_height = h;
A->metadata_modified (true);
}
return 0;
}
static int
output_file (int argc, const char *argv[])
{
ASSERT (argc == 2 && !strcmp(argv[0],"-o"));
std::string filename = argv[1];
if (! ot.curimg.get()) {
std::cerr << "oiiotool ERROR: -o " << filename << " did not have any current image to output.\n";
return 0;
}
if (ot.noclobber && Filesystem::exists(filename)) {
std::cerr << "oiiotool ERROR: Output file \"" << filename
<< "\" already exists, not overwriting.\n";
return 0;
}
if (ot.verbose)
std::cout << "Writing " << argv[1] << "\n";
ImageOutput *out = ImageOutput::create (filename.c_str());
if (! out) {
std::cerr << "oiiotool ERROR: " << geterror() << "\n";
return 0;
}
bool supports_displaywindow = out->supports ("displaywindow");
ot.read ();
ImageRecRef saveimg = ot.curimg;
ImageRecRef ir (ot.curimg);
if (! supports_displaywindow && ot.output_autocrop &&
(ir->spec()->x != ir->spec()->full_x ||
ir->spec()->y != ir->spec()->full_y ||
ir->spec()->width != ir->spec()->full_width ||
ir->spec()->height != ir->spec()->full_height)) {
const char *argv[] = { "croptofull" };
int action_croptofull (int argc, const char *argv[]); // forward decl
action_croptofull (1, argv);
ir = ot.curimg;
}
ImageOutput::OpenMode mode = ImageOutput::Create; // initial open
for (int s = 0, send = ir->subimages(); s < send; ++s) {
for (int m = 0, mend = ir->miplevels(s); m < mend; ++m) {
ImageSpec spec = *ir->spec(s,m);
adjust_output_options (spec, ot);
if (! out->open (filename, spec, mode)) {
std::cerr << "oiiotool ERROR: " << out->geterror() << "\n";
return 0;
}
if (! (*ir)(s,m).write (out)) {
std::cerr << "oiiotool ERROR: " << (*ir)(s,m).geterror() << "\n";
return 0;
}
if (mend > 1) {
if (out->supports("mipmap")) {
mode = ImageOutput::AppendMIPLevel; // for next level
} else if (out->supports("multiimage")) {
mode = ImageOutput::AppendSubimage;
} else {
std::cout << "oiiotool WARNING: " << out->format_name()
<< " does not support MIP-maps for "
<< filename << "\n";
break;
}
}
}
mode = ImageOutput::AppendSubimage; // for next subimage
if (send > 1 && ! out->supports("multiimage")) {
std::cout << "oiiotool WARNING: " << out->format_name()
<< " does not support multiple subimages for "
<< filename << "\n";
break;
}
}
out->close ();
delete out;
if (ot.output_adjust_time) {
std::string metadatatime = ir->spec(0,0)->get_string_attribute ("DateTime");
std::time_t in_time = ir->time();
if (! metadatatime.empty())
DateTime_to_time_t (metadatatime.c_str(), in_time);
boost::filesystem::last_write_time (filename, in_time);
}
ot.curimg = saveimg;
return 0;
}