detectfeas.C File Reference

#include "roccom.h"
#include <cstdio>
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cmath>
#include <cassert>
#include <sstream>
#include "roccom_assertion.h"
#include "../Rocsurf/test/IM_Reader.h"

Include dependency graph for detectfeas.C:

Go to the source code of this file.

Classes
struct	Control_parameter

Functions
	COM_EXTERN_MODULE (Rocprop)
	COM_EXTERN_MODULE (Rocout)
void	load_modules ()
void	print_usage (int argc, char *argv[])
void	read_control_file (const char *fname, Control_parameter &cp)
void	init_parameters (const Control_parameter &cntr_param)
std::string	read_in_mesh (const char *fname)
void	init_attributes (const string &wname, const Control_parameter &cntr_param)
void	output_solution (const string &wname, const char *timelevel, const string &format)
int	main (int argc, char *argv[])

Variables
static int	rank = 0

Function Documentation

COM_EXTERN_MODULE

(

Rocprop

)

COM_EXTERN_MODULE

(

Rocout

)

void init_attributes	(	const string &	wname,
		const Control_parameter &	cntr_param
	)

Definition at line 137 of file detectfeas.C.

References COM_DOUBLE, COM_INT, COM_new_attribute(), COM_resize_array(), COM_set_size(), and COM_window_init_done().

                                                           {
  COM_new_attribute((wname+".disps").c_str(), 'n', COM_DOUBLE, 3, "m");

  // Ridges of each pane.
  COM_new_attribute((wname+".ridges").c_str(), 'p', COM_INT, 2, "");
  COM_set_size((wname+".ridges").c_str(), 0, 0);

  COM_new_attribute((wname+".tangranks").c_str(), 'n', COM_INT, 1, "");
  // COM_new_attribute((wname+".weaks").c_str(), 'n', COM_INT, 1, "");

  COM_resize_array( (wname+".atts").c_str());
  COM_window_init_done( wname.c_str());
}

Here is the call graph for this function:

void init_parameters

(

const Control_parameter &

cntr_param

)

Definition at line 98 of file detectfeas.C.

References COM_call_function(), COM_get_function_handle(), Control_parameter::fangle, rank, and Control_parameter::sangle.

                                                           {
  int PROP_set_option = COM_get_function_handle( "PROP.set_option");

  if ( !cntr_param.fangle.empty()) {
    COM_call_function( PROP_set_option, "fangle", cntr_param.fangle.c_str());
    if ( rank==0) std::cout << "Set weak-feature angle to " << cntr_param.fangle << std::endl;
  }
  else
    COM_call_function( PROP_set_option, "fangle", "15");

  if ( !cntr_param.sangle.empty()) {
    COM_call_function( PROP_set_option, "sangle", cntr_param.sangle.c_str());
    if ( rank==0) std::cout << "Set strong-feature angle to " << cntr_param.sangle << std::endl;
  }
  else
    COM_call_function( PROP_set_option, "sangle", "60");

  COM_call_function( PROP_set_option, "rediter", "0");
  COM_call_function( PROP_set_option, "weight", "3"); // Angle-based
}

Here is the call graph for this function:

void load_modules

(

)

Definition at line 46 of file detectfeas.C.

References COM_LOAD_MODULE_STATIC_DYNAMIC.

                    {
  COM_LOAD_MODULE_STATIC_DYNAMIC(Rocprop, "PROP");
  COM_LOAD_MODULE_STATIC_DYNAMIC(Rocout, "OUT");
}

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

Definition at line 185 of file detectfeas.C.

References COM_call_function(), COM_finalize(), COM_get_attribute_handle(), COM_get_function_handle(), COM_init(), COMMPI_Comm_rank(), COMMPI_Initialized(), Control_parameter::format, init_attributes(), init_parameters(), load_modules(), MPI_COMM_WORLD, output_solution(), print_usage(), rank, read_control_file(), and read_in_mesh().

                                 {
  COM_init( &argc, &argv);
  load_modules();
  print_usage( argc, argv);

  if ( COMMPI_Initialized()) rank = COMMPI_Comm_rank( MPI_COMM_WORLD);

  // Read in mesh file.
  string wname = read_in_mesh( argv[1]);

  // Read in control parameters
  Control_parameter cntr_param;
  if ( argc>2) read_control_file( argv[2], cntr_param);

  // Initialize the attributes.
  init_attributes( wname, cntr_param);
  
  // Attributes
  int pmesh = COM_get_attribute_handle( (wname+".pmesh").c_str());
  int disps = COM_get_attribute_handle( (wname+".disps").c_str());

  // Funcitions
  int PROP_init = COM_get_function_handle( "PROP.initialize");
  int PROP_propagate = COM_get_function_handle( "PROP.propagate");
  init_parameters( cntr_param);

  if ( rank==0) cout << "Initializing data structures..." << endl;
  // Initialize control parameters
  COM_call_function( PROP_init, &pmesh);

  if ( rank==0) cout << "Detecting features..." << endl;
  double zero=0.;
  // Perform one step of smoothing to detect features.
  COM_call_function( PROP_propagate, &pmesh, &disps, &zero, &disps, &zero);

  if ( rank==0) cout << "Writing out solutions..." << endl;
  output_solution( wname, "00000", cntr_param.format);

  COM_finalize();
}

Here is the call graph for this function:

void output_solution	(	const string &	wname,
		const char *	timelevel,
		const string &	format
	)

Definition at line 152 of file detectfeas.C.

References COM_call_function(), COM_get_attribute_handle(), COM_get_function_handle(), and COMMPI_Initialized().

                                            {
  static int OUT_write = 0, hdl;

  if ( OUT_write==0) {
    OUT_write = COM_get_function_handle( "OUT.write_attribute");
    hdl = COM_get_attribute_handle( (wname+".all").c_str());

    int OUT_set = COM_get_function_handle( "OUT.set_option");
#if USE_CGNS
    if ( format != "HDF" && format != "hdf")
      COM_call_function( OUT_set, "format", "CGNS");
    else
#endif
      COM_call_function( OUT_set, "format", "HDF");
  }

  std::string fname = wname+"_"+timelevel;

  if ( !COMMPI_Initialized()) {
#if USE_CGNS
    if ( format != "HDF" && format != "hdf")
      fname.append(".cgns");
    else
#endif
      fname.append(".hdf");
  }
  else fname.append("_");

  COM_call_function( OUT_write, (char*)fname.c_str(),
                     &hdl, (char*)wname.c_str(), timelevel);
}

Here is the call graph for this function:

void print_usage	(	int	argc,
		char *	argv[]
	)

Definition at line 53 of file detectfeas.C.

                                          {
  if ( argc < 2) {
    cout << "Usage: " << argv[0] << " <surffile> [<controlfile>]" << std::endl;
    
    exit(-1);
  }
}

void read_control_file	(	const char *	fname,
		Control_parameter &	cp
	)

Definition at line 69 of file detectfeas.C.

References COM_assertion_msg, Control_parameter::fangle, Control_parameter::format, and Control_parameter::sangle.

                                                                  {
  /* Sample control file:
   * fangle: 15          # weak feature edge angle: between 0 and 180
   * sangle: 60          # strong feature edge angle: between 0 and 180
   * format: CGNS        # file format: CGNS or HDF
   */
  ifstream is(fname); COM_assertion_msg( is, "File does not exist");

  while ( !is.eof()) {
    char buf[255];
    is.get( buf, 255, ':');
    if ( buf[0] == '\0') { is.getline( buf, 255); continue; }

    istringstream istr(buf); 
    string keyword; istr >> keyword;
    is.getline( buf, 255, ':'); 

    if ( keyword == "fangle")
      is >> cp.fangle;
    else if ( keyword == "sangle")
      is >> cp.sangle;
    else if ( keyword == "format")
      is >> cp.format;
    else
      std::cerr << "Unknow keyword " << keyword << std::endl;
    is.getline( buf, 255); 
  }
}

std::string read_in_mesh

(

const char *

fname

)

Definition at line 120 of file detectfeas.C.

References COM_assertion_msg, rank, and IM_Reader::read_winmesh().

                                            {
  if ( rank==0) cout << "Reading surface mesh file \"" << fname << '"' << endl;

  std::string fname_str(fname);

  std::string::size_type pos = fname_str.find_first_of( ".");
  const string wname = fname_str.substr( 0, pos);

  if ( rank==0) cout << "Creating window \"" << wname << '"' << endl;

  IM_Reader im_reader;
  int npanes = im_reader.read_winmesh( fname, wname, false); 
  COM_assertion_msg( npanes>=0, "Failed to read in mesh file. File empty?");

  return wname;
}

Here is the call graph for this function:

Variable Documentation

int rank = 0

static

Definition at line 51 of file detectfeas.C.

Referenced by init_parameters(), main(), and read_in_mesh().