123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286 |
- #ifndef OPENCV_CORE_CVSTDINL_HPP
- #define OPENCV_CORE_CVSTDINL_HPP
- #include <complex>
- #include <ostream>
- #ifdef _MSC_VER
- #pragma warning( push )
- #pragma warning( disable: 4127 )
- #endif
- namespace cv
- {
- template<typename _Tp> class DataType< std::complex<_Tp> >
- {
- public:
- typedef std::complex<_Tp> value_type;
- typedef value_type work_type;
- typedef _Tp channel_type;
- enum { generic_type = 0,
- depth = DataType<channel_type>::depth,
- channels = 2,
- fmt = DataType<channel_type>::fmt + ((channels - 1) << 8),
- type = CV_MAKETYPE(depth, channels) };
- typedef Vec<channel_type, channels> vec_type;
- };
- inline
- String::String(const std::string& str)
- : cstr_(0), len_(0)
- {
- if (!str.empty())
- {
- size_t len = str.size();
- if (len) memcpy(allocate(len), str.c_str(), len);
- }
- }
- inline
- String::String(const std::string& str, size_t pos, size_t len)
- : cstr_(0), len_(0)
- {
- size_t strlen = str.size();
- pos = min(pos, strlen);
- len = min(strlen - pos, len);
- if (!len) return;
- memcpy(allocate(len), str.c_str() + pos, len);
- }
- inline
- String& String::operator = (const std::string& str)
- {
- deallocate();
- if (!str.empty())
- {
- size_t len = str.size();
- if (len) memcpy(allocate(len), str.c_str(), len);
- }
- return *this;
- }
- inline
- String& String::operator += (const std::string& str)
- {
- *this = *this + str;
- return *this;
- }
- inline
- String::operator std::string() const
- {
- return std::string(cstr_, len_);
- }
- inline
- String operator + (const String& lhs, const std::string& rhs)
- {
- String s;
- size_t rhslen = rhs.size();
- s.allocate(lhs.len_ + rhslen);
- if (lhs.len_) memcpy(s.cstr_, lhs.cstr_, lhs.len_);
- if (rhslen) memcpy(s.cstr_ + lhs.len_, rhs.c_str(), rhslen);
- return s;
- }
- inline
- String operator + (const std::string& lhs, const String& rhs)
- {
- String s;
- size_t lhslen = lhs.size();
- s.allocate(lhslen + rhs.len_);
- if (lhslen) memcpy(s.cstr_, lhs.c_str(), lhslen);
- if (rhs.len_) memcpy(s.cstr_ + lhslen, rhs.cstr_, rhs.len_);
- return s;
- }
- inline
- FileNode::operator std::string() const
- {
- String value;
- read(*this, value, value);
- return value;
- }
- template<> inline
- void operator >> (const FileNode& n, std::string& value)
- {
- read(n, value, std::string());
- }
- template<> inline
- FileStorage& operator << (FileStorage& fs, const std::string& value)
- {
- return fs << cv::String(value);
- }
- static inline
- std::ostream& operator << (std::ostream& os, const String& str)
- {
- return os << str.c_str();
- }
- static inline
- std::ostream& operator << (std::ostream& out, Ptr<Formatted> fmtd)
- {
- fmtd->reset();
- for(const char* str = fmtd->next(); str; str = fmtd->next())
- out << str;
- return out;
- }
- static inline
- std::ostream& operator << (std::ostream& out, const Mat& mtx)
- {
- return out << Formatter::get()->format(mtx);
- }
- static inline
- std::ostream& operator << (std::ostream& out, const UMat& m)
- {
- return out << m.getMat(ACCESS_READ);
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const Complex<_Tp>& c)
- {
- return out << "(" << c.re << "," << c.im << ")";
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const std::vector<Point_<_Tp> >& vec)
- {
- return out << Formatter::get()->format(Mat(vec));
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const std::vector<Point3_<_Tp> >& vec)
- {
- return out << Formatter::get()->format(Mat(vec));
- }
- template<typename _Tp, int m, int n> static inline
- std::ostream& operator << (std::ostream& out, const Matx<_Tp, m, n>& matx)
- {
- return out << Formatter::get()->format(Mat(matx));
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const Point_<_Tp>& p)
- {
- out << "[" << p.x << ", " << p.y << "]";
- return out;
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const Point3_<_Tp>& p)
- {
- out << "[" << p.x << ", " << p.y << ", " << p.z << "]";
- return out;
- }
- template<typename _Tp, int n> static inline
- std::ostream& operator << (std::ostream& out, const Vec<_Tp, n>& vec)
- {
- out << "[";
- if (cv::traits::Depth<_Tp>::value <= CV_32S)
- {
- for (int i = 0; i < n - 1; ++i) {
- out << (int)vec[i] << ", ";
- }
- out << (int)vec[n-1] << "]";
- }
- else
- {
- for (int i = 0; i < n - 1; ++i) {
- out << vec[i] << ", ";
- }
- out << vec[n-1] << "]";
- }
- return out;
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const Size_<_Tp>& size)
- {
- return out << "[" << size.width << " x " << size.height << "]";
- }
- template<typename _Tp> static inline
- std::ostream& operator << (std::ostream& out, const Rect_<_Tp>& rect)
- {
- return out << "[" << rect.width << " x " << rect.height << " from (" << rect.x << ", " << rect.y << ")]";
- }
- static inline std::ostream& operator << (std::ostream& out, const MatSize& msize)
- {
- int i, dims = msize.p[-1];
- for( i = 0; i < dims; i++ )
- {
- out << msize.p[i];
- if( i < dims-1 )
- out << " x ";
- }
- return out;
- }
- }
- #ifdef _MSC_VER
- #pragma warning( pop )
- #endif
- #endif
|