added implementation of CvHaarFeatureParams::read() in traincascade application ...
authorMaria Dimashova <no@email>
Fri, 10 Jun 2011 13:07:31 +0000 (13:07 +0000)
committerMaria Dimashova <no@email>
Fri, 10 Jun 2011 13:07:31 +0000 (13:07 +0000)
modules/traincascade/haarfeatures.cpp
modules/traincascade/haarfeatures.h

index 3b22edf..6344af5 100644 (file)
@@ -29,6 +29,22 @@ void CvHaarFeatureParams::write( FileStorage &fs ) const
     fs << CC_MODE << modeStr;
 }
 
+bool CvHaarFeatureParams::read( const FileNode &node )
+{
+    if( !CvFeatureParams::read( node ) )
+        return false;
+
+    FileNode rnode = node[CC_MODE];
+    if( !rnode.isString() )
+        return false;
+    String modeStr;
+    rnode >> modeStr;
+    mode = !modeStr.compare( CC_MODE_BASIC ) ? BASIC :
+           !modeStr.compare( CC_MODE_CORE ) ? CORE :
+           !modeStr.compare( CC_MODE_ALL ) ? ALL : -1;
+    return (mode >= 0);
+}
+
 void CvHaarFeatureParams::printDefaults() const
 {
     CvFeatureParams::printDefaults();
index da0067d..a871c54 100644 (file)
@@ -19,6 +19,7 @@ public:
     
     virtual void init( const CvFeatureParams& fp );
     virtual void write( FileStorage &fs ) const;
+    virtual bool read( const FileNode &node );
 
     virtual void printDefaults() const;  
     virtual void printAttrs() const;