123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412 |
- /*M///////////////////////////////////////////////////////////////////////////////////////
- //
- // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
- //
- // By downloading, copying, installing or using the software you agree to this license.
- // If you do not agree to this license, do not download, install,
- // copy or use the software.
- //
- //
- // License Agreement
- // For Open Source Computer Vision Library
- //
- // Copyright (C) 2013, OpenCV Foundation, all rights reserved.
- // Third party copyrights are property of their respective owners.
- //
- // Redistribution and use in source and binary forms, with or without modification,
- // are permitted provided that the following conditions are met:
- //
- // * Redistribution's of source code must retain the above copyright notice,
- // this list of conditions and the following disclaimer.
- //
- // * Redistribution's 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.
- //
- // * The name of the copyright holders may not 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 Intel Corporation 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.
- //
- //M*/
- #ifndef OPENCV_DNN_DNN_INL_HPP
- #define OPENCV_DNN_DNN_INL_HPP
- #include <opencv2/dnn.hpp>
- namespace cv {
- namespace dnn {
- CV__DNN_INLINE_NS_BEGIN
- template<typename TypeIter>
- DictValue DictValue::arrayInt(TypeIter begin, int size)
- {
- DictValue res(Param::INT, new AutoBuffer<int64, 1>(size));
- for (int j = 0; j < size; begin++, j++)
- (*res.pi)[j] = *begin;
- return res;
- }
- template<typename TypeIter>
- DictValue DictValue::arrayReal(TypeIter begin, int size)
- {
- DictValue res(Param::REAL, new AutoBuffer<double, 1>(size));
- for (int j = 0; j < size; begin++, j++)
- (*res.pd)[j] = *begin;
- return res;
- }
- template<typename TypeIter>
- DictValue DictValue::arrayString(TypeIter begin, int size)
- {
- DictValue res(Param::STRING, new AutoBuffer<String, 1>(size));
- for (int j = 0; j < size; begin++, j++)
- (*res.ps)[j] = *begin;
- return res;
- }
- template<>
- inline DictValue DictValue::get<DictValue>(int idx) const
- {
- CV_Assert(idx == -1);
- return *this;
- }
- template<>
- inline int64 DictValue::get<int64>(int idx) const
- {
- CV_Assert((idx == -1 && size() == 1) || (idx >= 0 && idx < size()));
- idx = (idx == -1) ? 0 : idx;
- if (type == Param::INT)
- {
- return (*pi)[idx];
- }
- else if (type == Param::REAL)
- {
- double doubleValue = (*pd)[idx];
- double fracpart, intpart;
- fracpart = std::modf(doubleValue, &intpart);
- CV_Assert(fracpart == 0.0);
- return (int64)doubleValue;
- }
- else if (type == Param::STRING)
- {
- return std::atoi((*ps)[idx].c_str());
- }
- else
- {
- CV_Assert(isInt() || isReal() || isString());
- return 0;
- }
- }
- template<>
- inline int DictValue::get<int>(int idx) const
- {
- return (int)get<int64>(idx);
- }
- inline int DictValue::getIntValue(int idx) const
- {
- return (int)get<int64>(idx);
- }
- template<>
- inline unsigned DictValue::get<unsigned>(int idx) const
- {
- return (unsigned)get<int64>(idx);
- }
- template<>
- inline bool DictValue::get<bool>(int idx) const
- {
- return (get<int64>(idx) != 0);
- }
- template<>
- inline double DictValue::get<double>(int idx) const
- {
- CV_Assert((idx == -1 && size() == 1) || (idx >= 0 && idx < size()));
- idx = (idx == -1) ? 0 : idx;
- if (type == Param::REAL)
- {
- return (*pd)[idx];
- }
- else if (type == Param::INT)
- {
- return (double)(*pi)[idx];
- }
- else if (type == Param::STRING)
- {
- return std::atof((*ps)[idx].c_str());
- }
- else
- {
- CV_Assert(isReal() || isInt() || isString());
- return 0;
- }
- }
- inline double DictValue::getRealValue(int idx) const
- {
- return get<double>(idx);
- }
- template<>
- inline float DictValue::get<float>(int idx) const
- {
- return (float)get<double>(idx);
- }
- template<>
- inline String DictValue::get<String>(int idx) const
- {
- CV_Assert(isString());
- CV_Assert((idx == -1 && ps->size() == 1) || (idx >= 0 && idx < (int)ps->size()));
- return (*ps)[(idx == -1) ? 0 : idx];
- }
- inline String DictValue::getStringValue(int idx) const
- {
- return get<String>(idx);
- }
- inline void DictValue::release()
- {
- switch (type)
- {
- case Param::INT:
- delete pi;
- break;
- case Param::STRING:
- delete ps;
- break;
- case Param::REAL:
- delete pd;
- break;
- case Param::BOOLEAN:
- case Param::MAT:
- case Param::MAT_VECTOR:
- case Param::ALGORITHM:
- case Param::FLOAT:
- case Param::UNSIGNED_INT:
- case Param::UINT64:
- case Param::UCHAR:
- case Param::SCALAR:
- break; // unhandled
- }
- }
- inline DictValue::~DictValue()
- {
- release();
- }
- inline DictValue & DictValue::operator=(const DictValue &r)
- {
- if (&r == this)
- return *this;
- if (r.type == Param::INT)
- {
- AutoBuffer<int64, 1> *tmp = new AutoBuffer<int64, 1>(*r.pi);
- release();
- pi = tmp;
- }
- else if (r.type == Param::STRING)
- {
- AutoBuffer<String, 1> *tmp = new AutoBuffer<String, 1>(*r.ps);
- release();
- ps = tmp;
- }
- else if (r.type == Param::REAL)
- {
- AutoBuffer<double, 1> *tmp = new AutoBuffer<double, 1>(*r.pd);
- release();
- pd = tmp;
- }
- type = r.type;
- return *this;
- }
- inline DictValue::DictValue(const DictValue &r)
- : pv(NULL)
- {
- type = r.type;
- if (r.type == Param::INT)
- pi = new AutoBuffer<int64, 1>(*r.pi);
- else if (r.type == Param::STRING)
- ps = new AutoBuffer<String, 1>(*r.ps);
- else if (r.type == Param::REAL)
- pd = new AutoBuffer<double, 1>(*r.pd);
- }
- inline bool DictValue::isString() const
- {
- return (type == Param::STRING);
- }
- inline bool DictValue::isInt() const
- {
- return (type == Param::INT);
- }
- inline bool DictValue::isReal() const
- {
- return (type == Param::REAL || type == Param::INT);
- }
- inline int DictValue::size() const
- {
- switch (type)
- {
- case Param::INT:
- return (int)pi->size();
- case Param::STRING:
- return (int)ps->size();
- case Param::REAL:
- return (int)pd->size();
- case Param::BOOLEAN:
- case Param::MAT:
- case Param::MAT_VECTOR:
- case Param::ALGORITHM:
- case Param::FLOAT:
- case Param::UNSIGNED_INT:
- case Param::UINT64:
- case Param::UCHAR:
- case Param::SCALAR:
- break; // unhandled
- }
- CV_Error_(Error::StsInternal, ("Unhandled type (%d)", static_cast<int>(type)));
- }
- inline std::ostream &operator<<(std::ostream &stream, const DictValue &dictv)
- {
- int i;
- if (dictv.isInt())
- {
- for (i = 0; i < dictv.size() - 1; i++)
- stream << dictv.get<int64>(i) << ", ";
- stream << dictv.get<int64>(i);
- }
- else if (dictv.isReal())
- {
- for (i = 0; i < dictv.size() - 1; i++)
- stream << dictv.get<double>(i) << ", ";
- stream << dictv.get<double>(i);
- }
- else if (dictv.isString())
- {
- for (i = 0; i < dictv.size() - 1; i++)
- stream << "\"" << dictv.get<String>(i) << "\", ";
- stream << dictv.get<String>(i);
- }
- return stream;
- }
- /////////////////////////////////////////////////////////////////
- inline bool Dict::has(const String &key) const
- {
- return dict.count(key) != 0;
- }
- inline DictValue *Dict::ptr(const String &key)
- {
- _Dict::iterator i = dict.find(key);
- return (i == dict.end()) ? NULL : &i->second;
- }
- inline const DictValue *Dict::ptr(const String &key) const
- {
- _Dict::const_iterator i = dict.find(key);
- return (i == dict.end()) ? NULL : &i->second;
- }
- inline const DictValue &Dict::get(const String &key) const
- {
- _Dict::const_iterator i = dict.find(key);
- if (i == dict.end())
- CV_Error(Error::StsObjectNotFound, "Required argument \"" + key + "\" not found into dictionary");
- return i->second;
- }
- template <typename T>
- inline T Dict::get(const String &key) const
- {
- return this->get(key).get<T>();
- }
- template <typename T>
- inline T Dict::get(const String &key, const T &defaultValue) const
- {
- _Dict::const_iterator i = dict.find(key);
- if (i != dict.end())
- return i->second.get<T>();
- else
- return defaultValue;
- }
- template<typename T>
- inline const T &Dict::set(const String &key, const T &value)
- {
- _Dict::iterator i = dict.find(key);
- if (i != dict.end())
- i->second = DictValue(value);
- else
- dict.insert(std::make_pair(key, DictValue(value)));
- return value;
- }
- inline void Dict::erase(const String &key)
- {
- dict.erase(key);
- }
- inline std::ostream &operator<<(std::ostream &stream, const Dict &dict)
- {
- Dict::_Dict::const_iterator it;
- for (it = dict.dict.begin(); it != dict.dict.end(); it++)
- stream << it->first << " : " << it->second << "\n";
- return stream;
- }
- inline std::map<String, DictValue>::const_iterator Dict::begin() const
- {
- return dict.begin();
- }
- inline std::map<String, DictValue>::const_iterator Dict::end() const
- {
- return dict.end();
- }
- CV__DNN_INLINE_NS_END
- }
- }
- #endif
|