removeModelFace.cpp

Go to the documentation of this file.

// Usage: removeModelFace <model_dir> <face_idx>

#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
#include <string>
#include <opencv/cv.h>
#include <opencv/highgui.h>

#include "../ObjectModel.h"
#include "../MetaFile.h"

#include "ros/ros.h"
#include "DUtils.h"
#include "DUtilsCV.h"

#include <stdio.h>

using namespace std;

// ---------------------------------------------------------------------------

int main(int argc, char *argv[])
{
  ros::init(argc, argv, "removeModelFace");
  
  if(argc < 2)
  {
    cout << "Usage: " << argv[0] << " <model dir> <face idx>" << endl;
    return 0;
  }

  string model_dir = argv[1];
  int face_idx = atoi(argv[2]);
  
  // check model
  if(!ObjectModel::checkDirectory(model_dir))
  {
    ROS_ERROR("Directory %s does not seem to contain a valid object model",
      model_dir.c_str());
    return 1;
  }
  
  // check index
  ObjectModel model(model_dir, false);
  
  if(model.Faces.empty())
  {
    ROS_WARN("Object %s does not contain any face", model.Name.c_str());
    return 0;
  }
  
  if(face_idx < 0 || face_idx >= (int)model.Faces.size())
  {
    ROS_WARN("Face index out of bounds. Object %s has %d faces (0 - %d)",
      model.Name.c_str(), model.Faces.size(), model.Faces.size() - 1);
    return 2;
  }
  
  vector<string> extensions;
  extensions.push_back(".key.gz");
  extensions.push_back(".ply");
  extensions.push_back(".png");
  extensions.push_back(".txt");
  
  for(int idx = face_idx; idx < (int)model.Faces.size()-1; ++idx)
  {
    // mv face_%03d .key.gz / .ply / .png / .txt one index backwards
    stringstream source, target;
    source << model_dir << "/face_" << setw(3) << setfill('0') << idx+1;
    target << model_dir << "/face_" << setw(3) << setfill('0') << idx;
    
    for(unsigned int j = 0; j < extensions.size(); ++j)
    {
      rename((source.str() + extensions[j]).c_str(), 
        (target.str() + extensions[j]).c_str() );
    }
  }
  
  // delete face_%03d with index |model.Faces|-1
  stringstream source;
  source << model_dir << "/face_" << setw(3) << setfill('0') 
    << model.Faces.size() - 1;

  for(unsigned int j = 0; j < extensions.size(); ++j)
  {
    remove((source.str() + extensions[j]).c_str());
  }
  
  // update meta
  ROS_INFO("Updating meta data...");
  
  MetaFile::MetaData data;
  string meta_file = model_dir + "/meta.xml";
  MetaFile::readFile(meta_file, data);
  
  data.NFaces--;
  if(data.Type == "planar")
  {
    data.Dimensions.Planar.Faces.erase(
      data.Dimensions.Planar.Faces.begin() + face_idx);
  }
  
  MetaFile::saveFile(meta_file, data);

  ROS_INFO("Done. Face %d of object %s has been deleted", face_idx, 
    model.Name.c_str());
}

// ---------------------------------------------------------------------------