Commit 423b721d authored by Maria Dimashova's avatar Maria Dimashova

fixed AlgorithmInfo::read for read only parameters (thanks to Vadim)

parent 78ee81c4
...@@ -4312,10 +4312,10 @@ public: ...@@ -4312,10 +4312,10 @@ public:
class CV_EXPORTS AlgorithmInfo class CV_EXPORTS AlgorithmInfo
{ {
public: public:
friend class Algorithm;
AlgorithmInfo(const string& name, Algorithm::Constructor create); AlgorithmInfo(const string& name, Algorithm::Constructor create);
~AlgorithmInfo(); ~AlgorithmInfo();
void get(const Algorithm* algo, const char* name, int argType, void* value) const; void get(const Algorithm* algo, const char* name, int argType, void* value) const;
void set(Algorithm* algo, const char* name, int argType, const void* value) const;
void addParam_(Algorithm& algo, const char* name, int argType, void addParam_(Algorithm& algo, const char* name, int argType,
void* value, bool readOnly, void* value, bool readOnly,
Algorithm::Getter getter, Algorithm::Setter setter, Algorithm::Getter getter, Algorithm::Setter setter,
...@@ -4365,6 +4365,8 @@ public: ...@@ -4365,6 +4365,8 @@ public:
const string& help=string()); const string& help=string());
protected: protected:
AlgorithmInfoData* data; AlgorithmInfoData* data;
void set(Algorithm* algo, const char* name, int argType,
const void* value, bool force=false) const;
}; };
......
...@@ -296,9 +296,9 @@ AlgorithmInfo::~AlgorithmInfo() ...@@ -296,9 +296,9 @@ AlgorithmInfo::~AlgorithmInfo()
void AlgorithmInfo::write(const Algorithm* algo, FileStorage& fs) const void AlgorithmInfo::write(const Algorithm* algo, FileStorage& fs) const
{ {
size_t i = 0, n = data->params.vec.size(); size_t i = 0, nparams = data->params.vec.size();
cv::write(fs, "name", algo->name()); cv::write(fs, "name", algo->name());
for( i = 0; i < n; i++ ) for( i = 0; i < nparams; i++ )
{ {
const Param& p = data->params.vec[i].second; const Param& p = data->params.vec[i].second;
const string& pname = data->params.vec[i].first; const string& pname = data->params.vec[i].first;
...@@ -327,9 +327,10 @@ void AlgorithmInfo::write(const Algorithm* algo, FileStorage& fs) const ...@@ -327,9 +327,10 @@ void AlgorithmInfo::write(const Algorithm* algo, FileStorage& fs) const
void AlgorithmInfo::read(Algorithm* algo, const FileNode& fn) const void AlgorithmInfo::read(Algorithm* algo, const FileNode& fn) const
{ {
size_t i = 0, n = data->params.vec.size(); size_t i = 0, nparams = data->params.vec.size();
AlgorithmInfo* info = algo->info();
for( i = 0; i < n; i++ ) for( i = 0; i < nparams; i++ )
{ {
const Param& p = data->params.vec[i].second; const Param& p = data->params.vec[i].second;
const string& pname = data->params.vec[i].first; const string& pname = data->params.vec[i].first;
...@@ -337,31 +338,43 @@ void AlgorithmInfo::read(Algorithm* algo, const FileNode& fn) const ...@@ -337,31 +338,43 @@ void AlgorithmInfo::read(Algorithm* algo, const FileNode& fn) const
if( n.empty() ) if( n.empty() )
continue; continue;
if( p.type == Param::INT ) if( p.type == Param::INT )
algo->set(pname, (int)n); {
int val = (int)n;
info->set(algo, pname.c_str(), p.type, &val, true);
}
else if( p.type == Param::BOOLEAN ) else if( p.type == Param::BOOLEAN )
algo->set(pname, (int)n != 0); {
bool val = (int)n != 0;
info->set(algo, pname.c_str(), p.type, &val, true);
}
else if( p.type == Param::REAL ) else if( p.type == Param::REAL )
algo->set(pname, (double)n); {
double val = (double)n;
info->set(algo, pname.c_str(), p.type, &val, true);
}
else if( p.type == Param::STRING ) else if( p.type == Param::STRING )
algo->set(pname, (string)n); {
string val = (string)n;
info->set(algo, pname.c_str(), p.type, &val, true);
}
else if( p.type == Param::MAT ) else if( p.type == Param::MAT )
{ {
Mat m; Mat m;
cv::read(n, m); cv::read(n, m);
algo->set(pname, m); info->set(algo, pname.c_str(), p.type, &m, true);
} }
else if( p.type == Param::MAT_VECTOR ) else if( p.type == Param::MAT_VECTOR )
{ {
vector<Mat> mv; vector<Mat> mv;
cv::read(n, mv); cv::read(n, mv);
algo->set(pname, mv); info->set(algo, pname.c_str(), p.type, &mv, true);
} }
else if( p.type == Param::ALGORITHM ) else if( p.type == Param::ALGORITHM )
{ {
Ptr<Algorithm> nestedAlgo = Algorithm::_create((string)n["name"]); Ptr<Algorithm> nestedAlgo = Algorithm::_create((string)n["name"]);
CV_Assert( !nestedAlgo.empty() ); CV_Assert( !nestedAlgo.empty() );
nestedAlgo->read(n); nestedAlgo->read(n);
algo->set(pname, nestedAlgo); info->set(algo, pname.c_str(), p.type, &nestedAlgo, true);
} }
else else
CV_Error( CV_StsUnsupportedFormat, "unknown/unsupported parameter type"); CV_Error( CV_StsUnsupportedFormat, "unknown/unsupported parameter type");
...@@ -392,14 +405,14 @@ union GetSetParam ...@@ -392,14 +405,14 @@ union GetSetParam
void (Algorithm::*set_algo)(const Ptr<Algorithm>&); void (Algorithm::*set_algo)(const Ptr<Algorithm>&);
}; };
void AlgorithmInfo::set(Algorithm* algo, const char* name, int argType, const void* value) const void AlgorithmInfo::set(Algorithm* algo, const char* name, int argType, const void* value, bool force) const
{ {
const Param* p = findstr(data->params, name); const Param* p = findstr(data->params, name);
if( !p ) if( !p )
CV_Error_( CV_StsBadArg, ("No parameter '%s' is found", name ? name : "<NULL>") ); CV_Error_( CV_StsBadArg, ("No parameter '%s' is found", name ? name : "<NULL>") );
if( p->readonly ) if( !force && p->readonly )
CV_Error_( CV_StsError, ("Parameter '%s' is readonly", name)); CV_Error_( CV_StsError, ("Parameter '%s' is readonly", name));
GetSetParam f; GetSetParam f;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment