git.maemo.org Git - opencv/blob - 3rdparty/flann/util/saving.cpp

   1 /***********************************************************************
   2  * Software License Agreement (BSD License)
   3  *
   4  * Copyright 2008-2009  Marius Muja (mariusm@cs.ubc.ca). All rights reserved.
   5  * Copyright 2008-2009  David G. Lowe (lowe@cs.ubc.ca). All rights reserved.
   6  *
   7  * THE BSD LICENSE
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  *
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  *************************************************************************/
  30
  31 #include "common.h"
  32 #include "saving.h"
  33 #include "nn_index.h"
  34 #include <cstdio>
  35
  36 namespace flann
  37 {
  38
  39 const char FLANN_SIGNATURE[] = "FLANN_INDEX";
  40
  41 void save_header(FILE* stream, const NNIndex& index)
  42 {
  43         IndexHeader header;
  44         memset(header.signature, 0 , sizeof(header.signature));
  45         strcpy(header.signature, FLANN_SIGNATURE);
  46         header.flann_version = (int)FLANN_VERSION;
  47         header.index_type = index.getType();
  48         header.rows = index.size();
  49         header.cols = index.veclen();
  50
  51         std::fwrite(&header, sizeof(header),1,stream);
  52 }
  53
  54
  55
  56 IndexHeader load_header(FILE* stream)
  57 {
  58         IndexHeader header;
  59         int read_size = fread(&header,sizeof(header),1,stream);
  60
  61         if (read_size!=1) {
  62                 throw FLANNException("Invalid index file, cannot read");
  63         }
  64
  65         if (strcmp(header.signature,FLANN_SIGNATURE)!=0) {
  66                 throw FLANNException("Invalid index file, wrong signature");
  67         }
  68
  69         return header;
  70
  71 }
  72
  73 }