apps/sft/config.cpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                           License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2008-2012, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #include <sft/config.hpp>
  44 #include <iomanip>
  45
  46 sft::Config::Config(): seed(0) {}
  47
  48 void sft::Config::write(cv::FileStorage& fs) const
  49 {
  50     fs << "{"
  51        << "trainPath"    << trainPath
  52        << "testPath"     << testPath
  53
  54        << "modelWinSize" << modelWinSize
  55        << "offset"       << offset
  56        << "octaves"      << octaves
  57
  58        << "positives"    << positives
  59        << "negatives"    << negatives
  60        << "btpNegatives" << btpNegatives
  61
  62        << "shrinkage"    << shrinkage
  63
  64        << "treeDepth"    << treeDepth
  65        << "weaks"        << weaks
  66        << "poolSize"     << poolSize
  67
  68        << "cascadeName"  << cascadeName
  69        << "outXmlPath"   << outXmlPath
  70
  71        << "seed"         << seed
  72        << "featureType"  << featureType
  73        << "}";
  74 }
  75
  76 void sft::Config::read(const cv::FileNode& node)
  77 {
  78     trainPath = (string)node["trainPath"];
  79     testPath  = (string)node["testPath"];
  80
  81     cv::FileNodeIterator  nIt = node["modelWinSize"].end();
  82     modelWinSize = cv::Size((int)*(--nIt), (int)*(--nIt));
  83
  84     nIt = node["offset"].end();
  85     offset = cv::Point2i((int)*(--nIt), (int)*(--nIt));
  86
  87     node["octaves"] >> octaves;
  88
  89     positives =    (int)node["positives"];
  90     negatives =    (int)node["negatives"];
  91     btpNegatives = (int)node["btpNegatives"];
  92
  93     shrinkage = (int)node["shrinkage"];
  94
  95     treeDepth = (int)node["treeDepth"];
  96     weaks =     (int)node["weaks"];
  97     poolSize =  (int)node["poolSize"];
  98
  99     cascadeName = (std::string)node["cascadeName"];
 100     outXmlPath =  (std::string)node["outXmlPath"];
 101
 102     seed = (int)node["seed"];
 103     featureType = (std::string)node["featureType"];
 104 }
 105
 106 void sft::write(cv::FileStorage& fs, const string&, const Config& x)
 107 {
 108     x.write(fs);
 109 }
 110
 111 void sft::read(const cv::FileNode& node, Config& x, const Config& default_value)
 112 {
 113     x = default_value;
 114
 115     if(!node.empty())
 116         x.read(node);
 117 }
 118
 119 namespace {
 120
 121 struct Out
 122 {
 123     Out(std::ostream& _out): out(_out) {}
 124     template<typename T>
 125     void operator ()(const T a) const {out << a << " ";}
 126
 127     std::ostream& out;
 128 private:
 129     Out& operator=(Out const& other);
 130 };
 131 }
 132
 133 std::ostream& sft::operator<<(std::ostream& out, const Config& m)
 134 {
 135     out << std::setw(14) << std::left << "trainPath"    << m.trainPath     << std::endl
 136         << std::setw(14) << std::left << "testPath"     << m.testPath      << std::endl
 137
 138         << std::setw(14) << std::left << "modelWinSize" << m.modelWinSize  << std::endl
 139         << std::setw(14) << std::left << "offset"       << m.offset        << std::endl
 140         << std::setw(14) << std::left << "octaves";
 141
 142     Out o(out);
 143     for_each(m.octaves.begin(), m.octaves.end(), o);
 144
 145     out << std::endl
 146         << std::setw(14) << std::left  << "positives"    << m.positives    << std::endl
 147         << std::setw(14) << std::left  << "negatives"    << m.negatives    << std::endl
 148         << std::setw(14) << std::left  << "btpNegatives" << m.btpNegatives << std::endl
 149
 150         << std::setw(14) << std::left  << "shrinkage"    << m.shrinkage    << std::endl
 151
 152         << std::setw(14) << std::left  << "treeDepth"    << m.treeDepth    << std::endl
 153         << std::setw(14) << std::left  << "weaks"        << m.weaks        << std::endl
 154         << std::setw(14) << std::left  << "poolSize"     << m.poolSize     << std::endl
 155
 156         << std::setw(14) << std::left  << "cascadeName"  << m.cascadeName  << std::endl
 157         << std::setw(14) << std::left  << "outXmlPath"   << m.outXmlPath   << std::endl
 158         << std::setw(14) << std::left  << "seed"         << m.seed         << std::endl
 159         << std::setw(14) << std::left  << "featureType"  << m.featureType  << std::endl;
 160
 161     return out;
 162 }