removed cv prefix
This commit is contained in:
@@ -0,0 +1,306 @@
|
||||
/*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) 2000-2008, Intel Corporation, all rights reserved.
|
||||
// Copyright (C) 2009-2010, Willow Garage Inc., 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*/
|
||||
|
||||
#include "precomp.hpp"
|
||||
#include <iterator>
|
||||
|
||||
namespace cv
|
||||
{
|
||||
|
||||
static inline char getCloseBrace(char c)
|
||||
{
|
||||
return c == '[' ? ']' : c == '(' ? ')' : c == '{' ? '}' : '\0';
|
||||
}
|
||||
|
||||
|
||||
template<typename _Tp> static void writeElems(std::ostream& out, const _Tp* data,
|
||||
int nelems, int cn, char obrace, char cbrace)
|
||||
{
|
||||
typedef typename DataType<_Tp>::work_type _WTp;
|
||||
nelems *= cn;
|
||||
for(int i = 0; i < nelems; i += cn)
|
||||
{
|
||||
if(cn == 1)
|
||||
{
|
||||
out << (_WTp)data[i] << (i+1 < nelems ? ", " : "");
|
||||
continue;
|
||||
}
|
||||
out << obrace;
|
||||
for(int j = 0; j < cn; j++)
|
||||
out << (_WTp)data[i + j] << (j+1 < cn ? ", " : "");
|
||||
out << cbrace << (i+cn < nelems ? ", " : "");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static void writeElems(std::ostream& out, const void* data, int nelems, int type, char brace)
|
||||
{
|
||||
int depth = CV_MAT_DEPTH(type), cn = CV_MAT_CN(type);
|
||||
char cbrace = ' ';
|
||||
if(!brace || isspace(brace))
|
||||
{
|
||||
nelems *= cn;
|
||||
cn = 1;
|
||||
}
|
||||
else
|
||||
cbrace = getCloseBrace(brace);
|
||||
if(depth == CV_8U)
|
||||
writeElems(out, (const uchar*)data, nelems, cn, brace, cbrace);
|
||||
else if(depth == CV_8S)
|
||||
writeElems(out, (const schar*)data, nelems, cn, brace, cbrace);
|
||||
else if(depth == CV_16U)
|
||||
writeElems(out, (const ushort*)data, nelems, cn, brace, cbrace);
|
||||
else if(depth == CV_16S)
|
||||
writeElems(out, (const short*)data, nelems, cn, brace, cbrace);
|
||||
else if(depth == CV_32S)
|
||||
writeElems(out, (const int*)data, nelems, cn, brace, cbrace);
|
||||
else if(depth == CV_32F)
|
||||
{
|
||||
std::streamsize pp = out.precision();
|
||||
out.precision(8);
|
||||
writeElems(out, (const float*)data, nelems, cn, brace, cbrace);
|
||||
out.precision(pp);
|
||||
}
|
||||
else if(depth == CV_64F)
|
||||
{
|
||||
std::streamsize pp = out.precision();
|
||||
out.precision(16);
|
||||
writeElems(out, (const double*)data, nelems, cn, brace, cbrace);
|
||||
out.precision(pp);
|
||||
}
|
||||
else
|
||||
CV_Error(CV_StsUnsupportedFormat, "");
|
||||
}
|
||||
|
||||
|
||||
static void writeMat(std::ostream& out, const Mat& m, char rowsep, char elembrace, bool singleLine)
|
||||
{
|
||||
CV_Assert(m.dims <= 2);
|
||||
int type = m.type();
|
||||
|
||||
char crowbrace = getCloseBrace(rowsep);
|
||||
char orowbrace = crowbrace ? rowsep : '\0';
|
||||
|
||||
if( orowbrace || isspace(rowsep) )
|
||||
rowsep = '\0';
|
||||
|
||||
for( int i = 0; i < m.rows; i++ )
|
||||
{
|
||||
if(orowbrace)
|
||||
out << orowbrace;
|
||||
writeElems(out, m.ptr(i), m.cols, type, elembrace);
|
||||
if(orowbrace)
|
||||
out << crowbrace << (i+1 < m.rows ? ", " : "");
|
||||
if(i+1 < m.rows)
|
||||
{
|
||||
if(rowsep)
|
||||
out << rowsep << (singleLine ? " " : "");
|
||||
if(!singleLine)
|
||||
out << "\n ";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class MatlabFormatter : public Formatter
|
||||
{
|
||||
public:
|
||||
virtual ~MatlabFormatter() {}
|
||||
void write(std::ostream& out, const Mat& m, const int*, int) const
|
||||
{
|
||||
out << "[";
|
||||
writeMat(out, m, ';', ' ', m.cols == 1);
|
||||
out << "]";
|
||||
}
|
||||
|
||||
void write(std::ostream& out, const void* data, int nelems, int type, const int*, int) const
|
||||
{
|
||||
writeElems(out, data, nelems, type, ' ');
|
||||
}
|
||||
};
|
||||
|
||||
class PythonFormatter : public Formatter
|
||||
{
|
||||
public:
|
||||
virtual ~PythonFormatter() {}
|
||||
void write(std::ostream& out, const Mat& m, const int*, int) const
|
||||
{
|
||||
out << "[";
|
||||
writeMat(out, m, m.cols > 1 ? '[' : ' ', '[', m.cols*m.channels() == 1);
|
||||
out << "]";
|
||||
}
|
||||
|
||||
void write(std::ostream& out, const void* data, int nelems, int type, const int*, int) const
|
||||
{
|
||||
writeElems(out, data, nelems, type, '[');
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
class NumpyFormatter : public Formatter
|
||||
{
|
||||
public:
|
||||
virtual ~NumpyFormatter() {}
|
||||
void write(std::ostream& out, const Mat& m, const int*, int) const
|
||||
{
|
||||
static const char* numpyTypes[] =
|
||||
{
|
||||
"uint8", "int8", "uint16", "int16", "int32", "float32", "float64", "uint64"
|
||||
};
|
||||
out << "array([";
|
||||
writeMat(out, m, m.cols > 1 ? '[' : ' ', '[', m.cols*m.channels() == 1);
|
||||
out << "], type='" << numpyTypes[m.depth()] << "')";
|
||||
}
|
||||
|
||||
void write(std::ostream& out, const void* data, int nelems, int type, const int*, int) const
|
||||
{
|
||||
writeElems(out, data, nelems, type, '[');
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
class CSVFormatter : public Formatter
|
||||
{
|
||||
public:
|
||||
virtual ~CSVFormatter() {}
|
||||
void write(std::ostream& out, const Mat& m, const int*, int) const
|
||||
{
|
||||
writeMat(out, m, ' ', ' ', m.cols*m.channels() == 1);
|
||||
if(m.rows > 1)
|
||||
out << "\n";
|
||||
}
|
||||
|
||||
void write(std::ostream& out, const void* data, int nelems, int type, const int*, int) const
|
||||
{
|
||||
writeElems(out, data, nelems, type, ' ');
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
class CFormatter : public Formatter
|
||||
{
|
||||
public:
|
||||
virtual ~CFormatter() {}
|
||||
void write(std::ostream& out, const Mat& m, const int*, int) const
|
||||
{
|
||||
out << "{";
|
||||
writeMat(out, m, ',', ' ', m.cols==1);
|
||||
out << "}";
|
||||
}
|
||||
|
||||
void write(std::ostream& out, const void* data, int nelems, int type, const int*, int) const
|
||||
{
|
||||
writeElems(out, data, nelems, type, ' ');
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
static MatlabFormatter matlabFormatter;
|
||||
static PythonFormatter pythonFormatter;
|
||||
static NumpyFormatter numpyFormatter;
|
||||
static CSVFormatter csvFormatter;
|
||||
static CFormatter cFormatter;
|
||||
|
||||
static const Formatter* g_defaultFormatter0 = &matlabFormatter;
|
||||
static const Formatter* g_defaultFormatter = &matlabFormatter;
|
||||
|
||||
bool my_streq(const char* a, const char* b)
|
||||
{
|
||||
size_t i, alen = strlen(a), blen = strlen(b);
|
||||
if( alen != blen )
|
||||
return false;
|
||||
for( i = 0; i < alen; i++ )
|
||||
if( a[i] != b[i] && a[i] - 32 != b[i] )
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
const Formatter* Formatter::get(const char* fmt)
|
||||
{
|
||||
if(!fmt || my_streq(fmt, ""))
|
||||
return g_defaultFormatter;
|
||||
if( my_streq(fmt, "MATLAB"))
|
||||
return &matlabFormatter;
|
||||
if( my_streq(fmt, "CSV"))
|
||||
return &csvFormatter;
|
||||
if( my_streq(fmt, "PYTHON"))
|
||||
return &pythonFormatter;
|
||||
if( my_streq(fmt, "NUMPY"))
|
||||
return &numpyFormatter;
|
||||
if( my_streq(fmt, "C"))
|
||||
return &cFormatter;
|
||||
CV_Error(CV_StsBadArg, "Unknown formatter");
|
||||
return g_defaultFormatter;
|
||||
}
|
||||
|
||||
const Formatter* Formatter::setDefault(const Formatter* fmt)
|
||||
{
|
||||
const Formatter* prevFmt = g_defaultFormatter;
|
||||
if(!fmt)
|
||||
fmt = g_defaultFormatter0;
|
||||
g_defaultFormatter = fmt;
|
||||
return prevFmt;
|
||||
}
|
||||
|
||||
Formatted::Formatted(const Mat& _m, const Formatter* _fmt,
|
||||
const vector<int>& _params)
|
||||
{
|
||||
mtx = _m;
|
||||
fmt = _fmt ? _fmt : Formatter::get();
|
||||
std::copy(_params.begin(), _params.end(), back_inserter(params));
|
||||
}
|
||||
|
||||
Formatted::Formatted(const Mat& _m, const Formatter* _fmt, const int* _params)
|
||||
{
|
||||
mtx = _m;
|
||||
fmt = _fmt ? _fmt : Formatter::get();
|
||||
|
||||
if( _params )
|
||||
{
|
||||
int i, maxParams = 100;
|
||||
for(i = 0; i < maxParams && _params[i] != 0; i+=2)
|
||||
;
|
||||
std::copy(_params, _params + i, back_inserter(params));
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user