make_watertight.cpp

Go to the documentation of this file.

// ********************************************************************
// Brandon Smith
// August, 2009

/* _curve_to_be_tested_for_watertightness_
      vert1 X X vert1
            | |
      vert2 X |
  surf1     | |    surf2
            | |
      vert3 X X vert2
            | |
      vert4 X X vert3                   */

// input:  h5m filename, tolerance
// output: watertight h5m

// make CXXFLAGS=-g for debug
// make CXXFLAGS=-pg for profiling

// modified by Andrew Davis 2012
// Updated deprecated MOAB calls

#include <iostream>
#include <sstream>
#include <iomanip> // for setprecision
#include <limits> // for min/max values
#include <assert.h>
#include <math.h>
#include <time.h>
#include <vector>

#include "moab/Core.hpp"
#include "MBTagConventions.hpp"
#include "moab/Range.hpp"
#include "moab/Skinner.hpp"
#include "moab/GeomTopoTool.hpp"

#include "meshkit/mw_func.hpp"
#include "meshkit/gen.hpp"
#include "meshkit/arc.hpp"
#include "meshkit/zip.hpp"
#include "meshkit/cleanup.hpp"


using namespace moab;

ErrorCode write_sealed_file( std::string root_filename, double facet_tol, bool is_acis);



int main(int argc, char **argv) 
  {

// ******************************************************************
// Load the h5m file 
// ******************************************************************

    clock_t start_time = clock();


    // check input args
    if( 2 > argc || 3 < argc ) 
      {
        std::cout << "To zip a faceted h5m file:" << std::endl;
        std::cout << "$ ./make_watertight <input_file.h5m>" << std::endl;
        std::cout << "To facet and zip an ACIS file using the default facet tolerance:" << std::endl;
        std::cout << "$ ./make_watertight <input_file.sat>" << std::endl;
        std::cout << "To facet and zip an ACIS file using a specified facet tolerance:" << std::endl;
        std::cout << "$ ./make_watertight <input_file.sat> <facet_tolerance>" << std::endl;
        return MB_FAILURE;
      }

    // The root name does not have an extension
    std::string input_name = argv[1];
    std::string root_name = argv[1];
    int len = root_name.length();
    root_name.erase(len - 4);
    bool is_acis;

    // load the input file
    ErrorCode result, rval;
    EntityHandle input_set;

    rval = MBI()->create_meshset( MESHSET_SET, input_set );

    if(gen::error(MB_SUCCESS!=rval,"failed to create_meshset"))
      {
        return rval;
      }

    std::cout << "Loading input file..." << std::endl;

    // If reading an h5m file, the facet tolerance has already been determined.
    // Read the facet_tol from the file_set. There should only be one input
    // argument.

    if(std::string::npos!=input_name.find("h5m") && (2==argc)) 
      {
        rval = MBI()->load_file( input_name.c_str(), &input_set );
        if(gen::error(MB_SUCCESS!=rval,"failed to load_file 0"))
          {
            return rval;      
          }
        
        is_acis = false;

       }
    //loading completed at this point
    clock_t load_time = clock();  
    //seal the input mesh set
    double facet_tol;
    result= mw_func::make_mesh_watertight(input_set, facet_tol);
    if(gen::error(MB_SUCCESS!=result, "could not make model watertight")) return result;

  
    //write file
    clock_t zip_time = clock();  
    std::cout << "Writing zipped file..." << std::endl;
    write_sealed_file( root_name, facet_tol, is_acis);
    if(gen::error(MB_SUCCESS!=result, "could not write the sealed mesh to a new file"))
    return result; 

    clock_t write_time = clock();
    std::cout << "Timing(seconds): loading=" 
              << (double) (load_time -start_time)/CLOCKS_PER_SEC << ", sealing=" 
              << (double) (zip_time  -load_time )/CLOCKS_PER_SEC << ", writing=" 
              << (double) (write_time-zip_time  )/CLOCKS_PER_SEC << std::endl;

  return 0;  
  }


ErrorCode write_sealed_file( std::string root_filename, double facet_tol, bool is_acis){

    ErrorCode result;
    std::string output_filename;
    if(is_acis) {  
      std::stringstream facet_tol_ss;
      facet_tol_ss << facet_tol; 
      output_filename = root_filename + "_" + facet_tol_ss.str() + "_zip.h5m";
    } else {
      output_filename = root_filename + "_zip.h5m";
    }
    // PROBLEM: If I write the input meshset the writer returns MB_FAILURE.
    // This happens only if I delete vertices when merging.
    // result = MBI()->write_mesh( filename_new.c_str(), &input_meshset, 1);
    result = MBI()->write_mesh( output_filename.c_str() );
    if (MB_SUCCESS != result) std::cout << "result= " << result << std::endl;
    assert(MB_SUCCESS == result);

  return result; 
}