void VtkTest::check_field_attrib( const char* temp_file_name ) { MeshImpl mesh; MsqPrintError err(cout); mesh.read_vtk( temp_file_name, err ); remove( temp_file_name ); ASSERT_NO_ERROR(err); std::vector<Mesh::ElementHandle> elems; mesh.get_all_elements( elems, err ); CPPUNIT_ASSERT( !err ); CPPUNIT_ASSERT_EQUAL( elems.size(), (size_t)8 ); std::string name; Mesh::TagType type; unsigned tagsize; void* th = mesh.tag_get( "test_field elem_vects", err ); CPPUNIT_ASSERT( !err ); mesh.tag_properties( th, name, type, tagsize, err ); CPPUNIT_ASSERT( !err && type == Mesh::DOUBLE && tagsize == 3 ); double elem_data[24]; mesh.tag_get_element_data( th, 8, arrptr(elems), elem_data, err ); CPPUNIT_ASSERT( !err ); for (int i = 0; i < 8; ++i) CPPUNIT_ASSERT( Vector3D( elem_data+3*i ) == Vector3D( i+1, i+1, i+1 ) ); th = mesh.tag_get( "test_field elem_ids", err ); CPPUNIT_ASSERT( !err ); mesh.tag_properties( th, name, type, tagsize, err ); CPPUNIT_ASSERT( !err && type == Mesh::INT && tagsize == 1 ); int elem_ids[8]; mesh.tag_get_element_data( th, 8, arrptr(elems), elem_ids, err ); CPPUNIT_ASSERT( !err ); for (int i = 0; i < 8; ++i) CPPUNIT_ASSERT( elem_ids[i] == i+1 ); th = mesh.tag_get( "field1", err ); CPPUNIT_ASSERT( !err ); mesh.tag_properties( th, name, type, tagsize, err ); CPPUNIT_ASSERT( !err && type == Mesh::INT && tagsize == 1 ); int values[8]; mesh.tag_get_element_data( th, 8, arrptr(elems), values, err ); CPPUNIT_ASSERT( !err ); for (int i = 0; i < 8; ++i) CPPUNIT_ASSERT( values[i] == 8-i ); }
// Test reading Vtk vector attribute void VtkTest::test_read_vector_attrib() { MeshImpl mesh; MsqPrintError err(cout); FILE* file = fopen( temp_file_name, "w+" ); fputs( structured_3d_points_data, file ); fputs( simple_vector_attrib, file ); fclose( file ); mesh.read_vtk( temp_file_name, err ); remove( temp_file_name ); ASSERT_NO_ERROR(err); std::vector<Mesh::ElementHandle> elems; mesh.get_all_elements( elems, err ); CPPUNIT_ASSERT( !err ); CPPUNIT_ASSERT_EQUAL( elems.size(), (size_t)8 ); void* th = mesh.tag_get( "hexvect", err ); CPPUNIT_ASSERT( !err ); std::string name; Mesh::TagType type; unsigned tagsize; mesh.tag_properties( th, name, type, tagsize, err ); CPPUNIT_ASSERT( !err && type == Mesh::DOUBLE && tagsize == 3 ); double elem_data[24]; mesh.tag_get_element_data( th, 8, arrptr(elems), elem_data, err ); CPPUNIT_ASSERT( !err ); for (int i = 0; i < 8; ++i) CPPUNIT_ASSERT( Vector3D( elem_data+3*i ) == Vector3D( i+1, i+1, i+1 ) ); }
void VtkTest::test_read_quadratic( const char* filename ) { const size_t NUM_ELEM = 8; MeshImpl mesh; MsqPrintError err(cout); mesh.read_vtk( filename, err ); ASSERT_NO_ERROR(err); std::vector<Mesh::ElementHandle> elems(NUM_ELEM); mesh.get_all_elements( elems, err ); ASSERT_NO_ERROR(err); CPPUNIT_ASSERT_EQUAL(elems.size(), NUM_ELEM ); std::vector<Mesh::VertexHandle> conn; std::vector<size_t> offsets; mesh.elements_get_attached_vertices( arrptr(elems), elems.size(), conn, offsets, err ); ASSERT_NO_ERROR(err); CPPUNIT_ASSERT_EQUAL( conn.size(), (size_t)108 ); EntityTopology types[NUM_ELEM]; mesh.elements_get_topologies( arrptr(elems), types, NUM_ELEM, err ); ASSERT_NO_ERROR(err); static const double hex_corners[] = { 1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0 }; static const double tet_corners[] = { 1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 0.0, -1.0, 0.0, 0.0, 1.0 }; static const double pyr_corners[] = { 1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 0.0, 0.0, 1.0 }; static const double pri_corners[] = { -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0 }; static const unsigned hex_edges[] = { 0, 1, 1, 2, 2, 3, 3, 0, 0, 4, 1, 5, 2, 6, 3, 7, 4, 5, 5, 6, 6, 7, 7, 4 }; static const unsigned tet_edges[] = { 0, 1, 1, 2, 2, 0, 0, 3, 1, 3, 2, 3 }; static const unsigned pri_edges[] = { 0, 1, 1, 2, 2, 0, 0, 3, 1, 4, 2, 5, 3, 4, 4, 5, 5, 3 }; static const unsigned pyr_edges[] = { 0, 1, 1, 2, 2, 3, 3, 0, 0, 4, 1, 4, 2, 4, 3, 4 }; static const unsigned hex_faces[] = { 4, 0, 1, 5, 4, 4, 1, 2, 6, 5, 4, 2, 3, 7, 6, 4, 3, 0, 4, 7, 4, 3, 2, 1, 0, 4, 4, 5, 6, 7 }; static const struct { EntityTopology topology; unsigned num_corners; unsigned num_edges; unsigned num_faces; // if non-zero expect mid-face nodes unsigned num_region; // if non-zero expect mid-region node const double* corners; const unsigned* edges; const unsigned* faces; } expected_elems[NUM_ELEM] = { { Mesquite::HEXAHEDRON, 8, 12, 0, 0, hex_corners, hex_edges, hex_faces }, { Mesquite::HEXAHEDRON, 8, 12, 6, 1, hex_corners, hex_edges, hex_faces }, { Mesquite::TETRAHEDRON, 4, 6, 0, 0, tet_corners, tet_edges, 0 }, { Mesquite::QUADRILATERAL, 4, 4, 0, 0, hex_corners, hex_edges, 0 }, { Mesquite::QUADRILATERAL, 4, 4, 0, 1, hex_corners, hex_edges, 0 }, { Mesquite::TRIANGLE, 3, 3, 0, 0, tet_corners, tet_edges, 0 }, { Mesquite::PRISM, 6, 9, 0, 0, pri_corners, pri_edges, 0 }, { Mesquite::PYRAMID, 5, 8, 0, 0, pyr_corners, pyr_edges, 0 } }; MsqVertex have; std::vector<Mesh::VertexHandle>::iterator v_it = conn.begin(); for (unsigned i = 0; i < NUM_ELEM; ++i) { CPPUNIT_ASSERT_EQUAL( expected_elems[i].topology, types[i] ); size_t vtx_start = offsets[i]; size_t vtx_end = offsets[i+1]; size_t conn_len = expected_elems[i].num_corners + expected_elems[i].num_edges + expected_elems[i].num_faces + expected_elems[i].num_region; CPPUNIT_ASSERT_EQUAL( conn_len, vtx_end - vtx_start ); for (unsigned c = 0; c < expected_elems[i].num_corners; ++c, ++v_it) { Vector3D expected(expected_elems[i].corners + 3*c); mesh.vertices_get_coordinates( &*v_it, &have, 1, err ); ASSERT_NO_ERROR(err); expected -= have; CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, expected.length(), DBL_EPSILON ); } for (unsigned m = 0; m < expected_elems[i].num_edges; ++m, ++v_it) { unsigned start_idx = expected_elems[i].edges[2*m]; unsigned end_idx = expected_elems[i].edges[2*m+1]; Vector3D start( expected_elems[i].corners + 3*start_idx ); Vector3D end( expected_elems[i].corners + 3*end_idx ); Vector3D expected = 0.5 * (start + end); mesh.vertices_get_coordinates( &*v_it, &have, 1, err ); ASSERT_NO_ERROR(err); expected -= have; CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, expected.length(), DBL_EPSILON ); } const unsigned* f_it = expected_elems[i].faces; for (unsigned m = 0; m < expected_elems[i].num_faces; ++m, ++v_it) { Vector3D expected(0,0,0); const unsigned face_size = *f_it; ++f_it; CPPUNIT_ASSERT( face_size == 3u || face_size == 4u ); for (unsigned f = 0; f < face_size; ++f, ++f_it) expected += Vector3D( expected_elems[i].corners + 3 * *f_it ); expected /= face_size; mesh.vertices_get_coordinates( &*v_it, &have, 1, err ); ASSERT_NO_ERROR(err); expected -= have; CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, expected.length(), DBL_EPSILON ); } if (expected_elems[i].num_region) { CPPUNIT_ASSERT_EQUAL( 1u, expected_elems[i].num_region ); Vector3D expected(0,0,0); for (unsigned m = 0; m < expected_elems[i].num_corners; ++m) expected += Vector3D( expected_elems[i].corners + 3*m ); expected /= expected_elems[i].num_corners; mesh.vertices_get_coordinates( &*v_it, &have, 1, err ); ASSERT_NO_ERROR(err); expected -= have; CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, expected.length(), DBL_EPSILON ); ++v_it; } } }