mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
396 lines
8.7 KiB
396 lines
8.7 KiB
/////////////////////////////////////////////////////////////////////////// |
|
// |
|
// Copyright (c) 2007, Weta Digital Ltd |
|
// |
|
// 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 Weta Digital 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. |
|
// |
|
/////////////////////////////////////////////////////////////////////////// |
|
|
|
//----------------------------------------------------------------------------- |
|
// |
|
// Functions related to accessing channels |
|
// and views in multi-view OpenEXR files. |
|
// |
|
//----------------------------------------------------------------------------- |
|
|
|
#include <ImfMultiView.h> |
|
|
|
using namespace std; |
|
|
|
namespace Imf { |
|
namespace { |
|
|
|
StringVector |
|
parseString (string name, char c = '.') |
|
{ |
|
// |
|
// Turn name into a list of strings, separating |
|
// on char 'c' with whitespace stripped. |
|
// |
|
|
|
StringVector r; |
|
|
|
while (name.size() > 0) |
|
{ |
|
size_t s = name.find (c); |
|
string sec = name.substr (0, s); |
|
|
|
// |
|
// Strip spaces from beginning |
|
// |
|
|
|
while (sec.size() > 0 && sec[0] == ' ') |
|
sec.erase (0, 1); |
|
|
|
// |
|
// Strip spaces from end |
|
// |
|
|
|
while (sec.size() > 0 && sec[sec.size() - 1] == ' ') |
|
sec.erase (sec.size() - 1); |
|
|
|
r.push_back (sec); |
|
|
|
// |
|
// Strip off name including ending 'c' |
|
// |
|
|
|
if (s == name.npos) |
|
name = ""; |
|
else |
|
name = name.substr (s + 1); |
|
} |
|
|
|
return r; |
|
} |
|
|
|
|
|
int |
|
viewNum (const string &view, const StringVector &multiView) |
|
{ |
|
// |
|
// returns which view number is called 'view' |
|
// returns -1 if no member of multiView is 'view' |
|
// (i.e. if viewNum() returns -1, 'view' isn't a view name |
|
// if viewNum() returns 0, 'view' is the default view |
|
// otherwise, it's some other (valid) view |
|
// |
|
|
|
for (int i = 0; i < multiView.size(); ++i) |
|
{ |
|
if (multiView[i] == view) |
|
return i; |
|
} |
|
|
|
return -1; |
|
} |
|
|
|
} // namespace |
|
|
|
|
|
string |
|
defaultViewName (const StringVector &multiView) |
|
{ |
|
if (multiView.size() > 0) |
|
return multiView[0]; |
|
else |
|
return ""; |
|
} |
|
|
|
|
|
string |
|
viewFromChannelName (const string &channel, |
|
const StringVector &multiView) |
|
{ |
|
// |
|
// Given the name of a channel, return the name of the view to |
|
// which it belongs. |
|
// |
|
|
|
// |
|
// View name is penultimate section of name separated by periods ('.'s) |
|
// |
|
|
|
StringVector s = parseString (channel, '.'); |
|
|
|
if (s.size() == 0) |
|
return ""; // nothing in, nothing out |
|
|
|
if (s.size() == 1) |
|
{ |
|
// |
|
// Return default view name. |
|
// The rules say ALL channels with no periods |
|
// in the name belong to the default view. |
|
// |
|
|
|
return multiView[0]; |
|
} |
|
else |
|
{ |
|
// |
|
// size >= 2 - the last part is the channel name, |
|
// the next-to-last part is the view name. |
|
// Check if that part of the name really is |
|
// a valid view and, if it is, return it. |
|
// |
|
|
|
const string &viewName = s[s.size() - 2]; |
|
|
|
if (viewNum (viewName, multiView) >= 0) |
|
return viewName; |
|
else |
|
return ""; // not associated with any particular view |
|
} |
|
} |
|
|
|
|
|
ChannelList |
|
channelsInView (const string & viewName, |
|
const ChannelList & channelList, |
|
const StringVector & multiView) |
|
{ |
|
// |
|
// Return a list of all channels belonging to view viewName. |
|
// |
|
|
|
ChannelList q; |
|
|
|
for (ChannelList::ConstIterator i = channelList.begin(); |
|
i != channelList.end(); |
|
++i) |
|
{ |
|
// |
|
// Get view name for this channel |
|
// |
|
|
|
string view = viewFromChannelName (i.name(), multiView); |
|
|
|
|
|
// |
|
// Insert channel into q if it's a member of view viewName |
|
// |
|
|
|
if (view == viewName) |
|
q.insert (i.name(), i.channel()); |
|
} |
|
|
|
return q; |
|
} |
|
|
|
|
|
ChannelList |
|
channelsInNoView (const ChannelList &channelList, |
|
const StringVector &multiView) |
|
{ |
|
// |
|
// Return a list of channels not associated with any named view. |
|
// |
|
|
|
return channelsInView ("", channelList, multiView); |
|
} |
|
|
|
|
|
|
|
bool |
|
areCounterparts (const string &channel1, |
|
const string &channel2, |
|
const StringVector &multiView) |
|
{ |
|
// |
|
// Given two channels, return true if they are the same |
|
// channel in two different views. |
|
// |
|
|
|
StringVector chan1 = parseString (channel1); |
|
unsigned int size1 = chan1.size(); // number of SECTIONS in string |
|
// name (not string length) |
|
|
|
StringVector chan2 = parseString (channel2); |
|
unsigned int size2 = chan2.size(); |
|
|
|
if (size1 == 0 || size2 == 0) |
|
return false; |
|
|
|
// |
|
// channel1 and channel2 can't be counterparts |
|
// if either channel is in no view. |
|
// |
|
|
|
if (size1 > 1 && viewNum (chan1[size1 - 2], multiView) == -1) |
|
return false; |
|
|
|
if (size2 > 1 && viewNum (chan2[size2 - 2], multiView) == -1) |
|
return false; |
|
|
|
if (viewFromChannelName (channel1, multiView) == |
|
viewFromChannelName (channel2, multiView)) |
|
{ |
|
// |
|
// channel1 and channel2 are not counterparts |
|
// if they are in the same view. |
|
// |
|
|
|
return false; |
|
} |
|
|
|
if (size1 == 1) |
|
{ |
|
// |
|
// channel1 is a default channel - the channels will only be |
|
// counterparts if channel2 is of the form <view>.<channel1> |
|
// |
|
|
|
return size2 == 2 && chan1[0] == chan2[1]; |
|
} |
|
|
|
if (size2 == 1) |
|
{ |
|
// |
|
// channel2 is a default channel - the channels will only be |
|
// counterparts if channel1 is of the form <view>.<channel2> |
|
// |
|
|
|
return size1 == 2 && chan2[0] == chan1[1]; |
|
} |
|
|
|
// |
|
// Neither channel is a default channel. To be counterparts both |
|
// channel names must have the same number of components, and |
|
// all components except the penultimate one must be the same. |
|
// |
|
|
|
if (size1 != size2) |
|
return false; |
|
|
|
for(int i = 0; i < size1; ++i) |
|
{ |
|
if (i != size1 - 2 && chan1[i] != chan2[i]) |
|
return false; |
|
} |
|
|
|
return true; |
|
} |
|
|
|
|
|
ChannelList |
|
channelInAllViews (const string &channelName, |
|
const ChannelList &channelList, |
|
const StringVector &multiView) |
|
{ |
|
// |
|
// Given the name of a channel, return a |
|
// list of the same channel in all views. |
|
// |
|
|
|
ChannelList q; |
|
|
|
for (ChannelList::ConstIterator i=channelList.begin(); |
|
i != channelList.end(); |
|
++i) |
|
{ |
|
if (i.name() == channelName || |
|
areCounterparts (i.name(), channelName, multiView)) |
|
{ |
|
q.insert (i.name(), i.channel()); |
|
} |
|
} |
|
|
|
return q; |
|
} |
|
|
|
|
|
string |
|
channelInOtherView (const string &channelName, |
|
const ChannelList &channelList, |
|
const StringVector &multiView, |
|
const string &otherViewName) |
|
{ |
|
// |
|
// Given the name of a channel in one view, return the |
|
// corresponding channel name for view otherViewName. |
|
// |
|
|
|
for (ChannelList::ConstIterator i=channelList.begin(); |
|
i != channelList.end(); |
|
++i) |
|
{ |
|
if (viewFromChannelName (i.name(), multiView) == otherViewName && |
|
areCounterparts (i.name(), channelName, multiView)) |
|
{ |
|
return i.name(); |
|
} |
|
} |
|
|
|
return ""; |
|
} |
|
|
|
|
|
string |
|
insertViewName (const string &channel, |
|
const StringVector &multiView, |
|
int i) |
|
{ |
|
// |
|
// Insert multiView[i] into the channel name if appropriate. |
|
// |
|
|
|
StringVector s = parseString (channel, '.'); |
|
|
|
if (s.size() == 0) |
|
return ""; // nothing in, nothing out |
|
|
|
if (s.size() == 1 && i == 0) |
|
{ |
|
// |
|
// Channel in the default view, with no periods in its name. |
|
// Do not insert view name. |
|
// |
|
|
|
return channel; |
|
} |
|
|
|
// |
|
// View name becomes penultimate section of new channel name. |
|
// |
|
|
|
string newName; |
|
|
|
for (int j = 0; j < s.size(); ++j) |
|
{ |
|
if (j < s.size() - 1) |
|
newName += s[j] + "."; |
|
else |
|
newName += multiView[i] + "." + s[j]; |
|
} |
|
|
|
return newName; |
|
} |
|
|
|
|
|
} // namespace Imf
|
|
|