crop_vol_test.cpp File Reference

#include "moab/Core.hpp"
#include <iostream>
#include <fstream>
#include <cstring>
#include "moab/FBEngine.hpp"
#include "moab/GeomTopoTool.hpp"
#include "TestUtil.hpp"

Include dependency graph for crop_vol_test.cpp:

Go to the source code of this file.

Functions
ErrorCode	volume_test (FBEngine *pFacet)
void	handle_error_code (ErrorCode rv, int &number_failed, int &number_successful)
int	main (int argc, char *argv[])

Variables
std::string	filename_top
std::string	filename_bot
std::string	polygon_file_name
double	min_dot = 0.8
std::string	vol_file
int	number_tests_successful = 0
int	number_tests_failed = 0

Function Documentation

handle_error_code()

void handle_error_code	(	ErrorCode	rv,
		int &	number_failed,
		int &	number_successful
	)

Definition at line 36 of file crop_vol_test.cpp.

{
 if( rv == MB_SUCCESS )
 {
 std::cout << "Success";
 number_successful++;
 }
 else
 {
 std::cout << "Failure";
 number_failed++;
 }
}

References MB_SUCCESS.

Referenced by main().

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 50 of file crop_vol_test.cpp.

{
 filename_bot = TestDir + "unittest/BedCrop2.h5m";
 filename_top = TestDir + "unittest/SECrop2.h5m"; //"/polyPB.txt";
 polygon_file_name = TestDir + "unittest/poly14.txt";
 vol_file = "volIce.h5m"; // output
 
 number_tests_successful = 0;
 bool remove_output = true;
 
 if( argc == 1 )
 {
 std::cout << "Using default input files: " << filename_bot << " " << filename_top << " " << polygon_file_name
 << " " << min_dot << " " << vol_file << std::endl;
 std::cout << " default output file: " << vol_file << " will be deleted \n";
 }
 else if( argc == 6 )
 {
 remove_output = false;
 filename_bot = argv[1];
 filename_top = argv[2];
 polygon_file_name = argv[3];
 min_dot = atof( argv[4] );
 vol_file = argv[5];
 }
 else
 {
 std::cout << " wrong number of arguments\n";
 return 1; // fail
 }
 Core mbcore;
 Interface* mb = &mbcore;
 
 ErrorCode rval = mb->load_file( filename_bot.c_str() );MB_CHK_SET_ERR( rval, "failed to load bed file" );
 
 rval = mb->load_file( filename_top.c_str() ); // load the second face (we know we have one face in each file)
 MB_CHK_SET_ERR( rval, "failed to load top file" );
 
 FBEngine* pFacet = new FBEngine( mb, NULL, true ); // smooth facetting, no OBB tree passed
 
 if( !pFacet ) return 1; // error
 
 // should the init be part of constructor or not?
 // this is where the obb tree is constructed, and smooth faceting initialized, too.
 rval = pFacet->Init();MB_CHK_SET_ERR( rval, "failed to initialize smoothing" );
 
 std::cout << "volume creation test: ";
 rval = volume_test( pFacet );
 handle_error_code( rval, number_tests_failed, number_tests_successful );
 std::cout << "\n";
 
 if( remove_output )
 {
 remove( vol_file.c_str() );
 }
 return number_tests_failed;
}

References ErrorCode, filename_bot, filename_top, handle_error_code(), moab::FBEngine::Init(), moab::Core::load_file(), mb, MB_CHK_SET_ERR, mbcore, min_dot, number_tests_failed, number_tests_successful, polygon_file_name, vol_file, and volume_test().

volume_test()

ErrorCode volume_test

(

FBEngine *

pFacet

)

Definition at line 108 of file crop_vol_test.cpp.

{
 // there should be 2 faces in the model
 std::ifstream datafile( polygon_file_name.c_str(), std::ifstream::in );
 if( !datafile )
 {
 std::cout << "can't read file\n";
 return MB_FAILURE;
 }
 //
 char temp[100];
 double direction[3]; // normalized
 double gridSize;
 datafile.getline( temp, 100 ); // first line
 
 // get direction and mesh size along polygon segments, from file
 sscanf( temp, " %lf %lf %lf %lf ", direction, direction + 1, direction + 2, &gridSize );
 // NORMALIZE(direction);// just to be sure
 
 std::vector< double > xyz;
 while( !datafile.eof() )
 {
 datafile.getline( temp, 100 );
 // int id = 0;
 double x, y, z;
 int nr = sscanf( temp, "%lf %lf %lf", &x, &y, &z );
 if( nr == 3 )
 {
 xyz.push_back( x );
 xyz.push_back( y );
 xyz.push_back( z );
 }
 }
 int sizePolygon = (int)xyz.size() / 3;
 if( sizePolygon < 3 )
 {
 std::cerr << " Not enough points in the polygon" << std::endl;
 return MB_FAILURE;
 }
 EntityHandle root_set;
 ErrorCode rval = pFacet->getRootSet( &root_set );MB_CHK_SET_ERR( rval, "ERROR : getRootSet failed!" );
 
 int top = 2; // iBase_FACE;
 
 Range faces;
 rval = pFacet->getEntities( root_set, top, faces );MB_CHK_SET_ERR( rval, "Failed to get faces in volume_test." );
 
 if( 2 != faces.size() )
 {
 std::cerr << "expected 2 faces, top and bottom\n";
 return MB_FAILURE;
 }
 EntityHandle newFace1; // first test is with closed surface
 rval = pFacet->split_surface_with_direction( faces[0], xyz, direction, /*closed*/ 1, min_dot, newFace1 );MB_CHK_SET_ERR( rval, "Failed to crop first face." );
 
 EntityHandle newFace2; // first test is with closed surface
 rval = pFacet->split_surface_with_direction( faces[1], xyz, direction, /*closed*/ 1, min_dot, newFace2 );MB_CHK_SET_ERR( rval, "Failed to crop second face." );
 
 // here we make the assumption that the edges, vertices are matching fine...
 EntityHandle volume;
 rval = pFacet->create_volume_with_direction( newFace1, newFace2, direction, volume );MB_CHK_SET_ERR( rval, "Failed to create volume." );
 
 Interface* mb = pFacet->moab_instance();
 pFacet->delete_smooth_tags();
 // duplicate -- extract a new geom topo tool
 GeomTopoTool* gtt = pFacet->get_gtt();
 GeomTopoTool* duplicate = NULL;
 std::vector< EntityHandle > gents;
 gents.push_back( volume );
 rval = gtt->duplicate_model( duplicate, &gents );MB_CHK_SET_ERR( rval, "Failed to extract volume." );
 EntityHandle newRootSet = duplicate->get_root_model_set();
 delete pFacet;
 pFacet = NULL; // try not to write the obb tree
 delete duplicate;
 rval = mb->write_file( vol_file.c_str(), NULL, NULL, &newRootSet, 1 );
 
 return rval;
}

References moab::FBEngine::create_volume_with_direction(), moab::FBEngine::delete_smooth_tags(), moab::GeomTopoTool::duplicate_model(), ErrorCode, moab::FBEngine::get_gtt(), moab::GeomTopoTool::get_root_model_set(), moab::FBEngine::getEntities(), moab::FBEngine::getRootSet(), mb, MB_CHK_SET_ERR, min_dot, moab::FBEngine::moab_instance(), nr, polygon_file_name, root_set, moab::Range::size(), moab::FBEngine::split_surface_with_direction(), vol_file, and moab::Core::write_file().

Referenced by main().

Variable Documentation

filename_bot

std::string filename_bot

Definition at line 24 of file crop_vol_test.cpp.

Referenced by main().

filename_top

std::string filename_top

This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version.

Definition at line 23 of file crop_vol_test.cpp.

Referenced by main().

min_dot

double min_dot = 0.8

Definition at line 26 of file crop_vol_test.cpp.

Referenced by moab::FBEngine::chain_able_edge(), moab::FBEngine::chain_edges(), moab::SmoothFace::compute_control_points_on_edges(), moab::FBEngine::initializeSmoothing(), main(), moab::FBEngine::split_surface_with_direction(), and volume_test().

number_tests_failed

int number_tests_failed = 0

Definition at line 30 of file crop_vol_test.cpp.

Referenced by main().

number_tests_successful

int number_tests_successful = 0

Definition at line 29 of file crop_vol_test.cpp.

Referenced by main().

polygon_file_name

std::string polygon_file_name

Definition at line 25 of file crop_vol_test.cpp.

Referenced by main(), and volume_test().

vol_file

std::string vol_file

Definition at line 27 of file crop_vol_test.cpp.

Referenced by main(), and volume_test().