addpconn.C

Go to the documentation of this file.

/* *******************************************************************
 * Rocstar Simulation Suite                                          *
 * Copyright@2015, Illinois Rocstar LLC. All rights reserved.        *
 *                                                                   *
 * Illinois Rocstar LLC                                              *
 * Champaign, IL                                                     *
 * www.illinoisrocstar.com                                           *
 * sales@illinoisrocstar.com                                         *
 *                                                                   *
 * License: See LICENSE file in top level of distribution package or *
 * http://opensource.org/licenses/NCSA                               *
 *********************************************************************/
/* *******************************************************************
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,   *
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES   *
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND          *
 * NONINFRINGEMENT.  IN NO EVENT SHALL THE CONTRIBUTORS OR           *
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER       *
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,   *
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE    *
 * USE OR OTHER DEALINGS WITH THE SOFTWARE.                          *
 *********************************************************************/
// $Id: addpconn.C,v 1.6 2009/01/22 23:56:54 gzagaris Exp $

#include <iostream>
#include <cstring>
#include <cstdlib>
#include "roccom.h"
#include "mapbasic.h"

COM_EXTERN_MODULE( Rocin);
COM_EXTERN_MODULE( Rocout);
COM_EXTERN_MODULE( Rocmap);

using namespace std;

int main(int argc, char *argv[]) {
  COM_init( &argc, &argv);

  if ( argc < 3) {
    std::cout << "Add pane connectivity to a given mesh." << std::endl;
    std::cout << "Usage:\n\tTo run in serial: " << argv[0] 
              << " <input filename patterns or Rocin control file> <output file prefix> " << endl;
    std::cout << "\n\tTo run in parallel: <mpirun-command> " << argv[0] 
              << " -com-mpi <Rocin control file> <output file prefix> " << endl;
    std::cout << "If the second argument does not end with .hdf, then each pane is written into a separate file." << std::endl;
    exit(-1);
  }

  std::cout << "Reading mesh file \"" << argv[1] << '"' << endl;

  std::string fname(argv[1]), wname;
  string::size_type n0 = fname.find_last_of( "/");

  if ( n0 != std::string::npos) 
    fname = fname.substr( n0+1, fname.size());

  string::size_type ni;
  ni = fname.find_first_of( ".:-*[]?\\\"\'0123456789");
  COM_assertion_msg(ni, "File name must start with a letter");

  if ( ni == std::string::npos) {
    wname = fname;
  }
  else {
    while (fname[ni-1]=='_') --ni; // Remove the '_' at the end.
    wname = fname.substr( 0, ni);
  }

  std::cout << "Creating window \"" << wname << '"' << endl;

  // Read in HDF format
  COM_LOAD_MODULE_STATIC_DYNAMIC( Rocin, "IN");
  std::cout << "Reading window " << endl;

  const char *lastdot=std::strrchr( argv[1], '.');
  int IN_read;
  if ( lastdot && std::strcmp( lastdot, ".txt")==0)
    IN_read = COM_get_function_handle( "IN.read_by_control_file");
  else
    IN_read = COM_get_function_handle( "IN.read_window");
                     
  COM_call_function( IN_read, argv[1], wname.c_str());

  int IN_obtain = COM_get_function_handle( "IN.obtain_attribute");
  int mesh_hdl = COM_get_attribute_handle((wname+".mesh").c_str());
  std::cout << "Obtaining the mesh " << endl;
  COM_call_function( IN_obtain, &mesh_hdl, &mesh_hdl);

  // Change the memory layout to contiguous.
  std::cout << "Resizing the array " << endl;
  COM_resize_array( (wname+".mesh").c_str(), 0, NULL, 0);
  // Delete all attributes and leave alone the mesh.
  std::cout << "deleting the attribute" << endl;
  COM_delete_attribute( (wname+".atts").c_str());

  int npanes; COM_get_panes( wname.c_str(), &npanes, NULL);

  COM_assertion( npanes>=0);

  std::cout << "finishing up window initialization" << endl;
  COM_window_init_done( wname.c_str());

  std::cout << "Computing connectivity map... " << endl;

  COM_LOAD_MODULE_STATIC_DYNAMIC( Rocmap, "MAP");
  // Invoke compute_pconn to obtain pane connectivity.
  int MAP_compute_pconn = COM_get_function_handle( "MAP.compute_pconn");
  const string pconn = wname+".pconn";
  int pconn_hdl = COM_get_attribute_handle( pconn.c_str());
  COM_call_function( MAP_compute_pconn, &mesh_hdl, &pconn_hdl);

  std::cout << "Output window into file..." << endl;

  std::cout << "loading Rocout" << endl;
  COM_LOAD_MODULE_STATIC_DYNAMIC(Rocout, "OUT");

  // Output pconn
  int OUT_set = COM_get_function_handle( "OUT.set_option");
  int OUT_write = COM_get_function_handle( "OUT.write_attribute");

  COM_call_function( OUT_set, "mode", "w");
  COM_call_function( OUT_set, "rankwidth", "0");
  COM_call_function( OUT_set, "pnidwidth", "6");

  int all_hdl = COM_get_attribute_handle( (wname+".all").c_str());
  COM_call_function( OUT_write, argv[2], &all_hdl, 
                     (char*)wname.c_str(), "000");
  
  COM_finalize();
}