SMF_Reader.h

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: SMF_Reader.h,v 1.5 2008/12/06 08:43:24 mtcampbe Exp $

#ifndef SMF_READER_H
#define SMF_READER_H

#include "../Rocsurf/include/surfbasic.h"
#include <vector>
#include <string>
#include <fstream>
#include <cstdlib>
#include <cstdio>

COM_EXTERN_MODULE( Rocin);

// This class provides an interface for reading in a surface mesh 
// from the ".smf" or ".obj" file.
class SMF_Reader {
public:
  // Constructor from an MPI communicator and a scale factor.
  SMF_Reader()  {}

  // Assuming the window has been created, read in the local panes from
  // given file and return the number of local panes.
  int read_mesh( const char *fname, const std::string &wname) {
    std::ifstream is( fname); 
    if ( is == NULL) {
      std::cerr << "Error: Could not open file " << fname << std::endl;
      exit(-1);
    }

    // Create the window if not yet exist
    int h=COM_get_window_handle(wname.c_str());
    if ( h<=0) COM_new_window( wname.c_str());

    std::cout << "Reading file " << fname << std::endl;

    read_pane_coors( is, wname);
    read_pane_elems( is, wname);

    return 1;
  }

private:
  void get_nextline( std::istream &is, char *str) {
    char  accpet[]={'0','1','2','3','4','5','6','7','8','9',
                    '.','+','-','e','E'};

    do { 
      if ( is.eof()) { str[0] = '\0'; return; }
      str[0] = '\1'; str[1] = '\0';
      is.getline( str, MAXLEN); 
      if ( str[0] == '\1') return;
    } while ( (str[0]!='v' && str[0]!='f') || strpbrk( str,accpet)==NULL);
  }

  // Read in coordinates of a pane into a window if local
  void read_pane_coors( std::istream &is, const std::string &wname) {
    std::vector<SURF::Point_3<double> >  coors;

    int n=0;
    for (;;) {
      get_nextline( is, buf);

      if ( buf[0] == 'v') {
        if ( buf[1] == 't' || buf[1] == 'n') continue;
        SURF::Point_3<double> p;

        std::sscanf( &buf[2], "%lf %lf %lf", &p[0], &p[1], &p[2]);
        coors.push_back( p);

        ++n;
      }
      else 
        break;
    }
    COM_assertion_msg( n>0, "Did not find any nodes");
    std::cout << "Read in " << n << " nodes" << std::endl;

    COM_set_size( (wname+".nc").c_str(), 1, n);
    SURF::Point_3<double> *p;
    COM_allocate_array( (wname+".nc").c_str(), 1, &(void*&)p);

    std::copy( coors.begin(), coors.end(), p);
  }

  // Read in element connectivity of a pane into a window if local
  void read_pane_elems( std::istream &is, const std::string &wname){
    std::vector<SURF::Vector_3<int> >  elems;

    // When entering this routine, buf[0] should start with 'f'
    int n=0;
    for (;;) {
      if ( buf[0] == 'f') {
        if ( buf[1] != 't' && buf[1] != 'n') {
          SURF::Vector_3<int> p;

          std::sscanf( &buf[2], "%d %d %d", &p[0], &p[1], &p[2]);
          elems.push_back( p);

          ++n;
        }
        get_nextline( is, buf);
      }
      else break;
    }
    COM_assertion_msg( n>0, "Did not find any faces");
    std::cout << "Read in " << n << " faces" << std::endl;

    COM_set_size( (wname+".:t3:").c_str(), 1, n);
    SURF::Vector_3<int> *p;
    COM_allocate_array( (wname+".:t3:").c_str(), 1, &(void*&)p);
    
    std::copy( elems.begin(), elems.end(), p);
  }

private:
  enum  {MAXLEN=255};
  char  buf[MAXLEN+1];
};

#endif