Source/System/FileIO/OSB/OSGOSBTest.cpp

   1 /*---------------------------------------------------------------------------*\
   2  *                                OpenSG                                     *
   3  *                                                                           *
   4  *                                                                           *
   5  *               Copyright (C) 2000-2006 by the OpenSG Forum                 *
   6  *                                                                           *
   7  *   contact: dirk@opensg.org, gerrit.voss@vossg.org, jbehr@zgdv.de          *
   8  *                                                                           *
   9 \*---------------------------------------------------------------------------*/
  10 /*---------------------------------------------------------------------------*\
  11  *                                License                                    *
  12  *                                                                           *
  13  * This library is free software; you can redistribute it and/or modify it   *
  14  * under the terms of the GNU Library General Public License as published    *
  15  * by the Free Software Foundation, version 2.                               *
  16  *                                                                           *
  17  * This library is distributed in the hope that it will be useful, but       *
  18  * WITHOUT ANY WARRANTY; without even the implied warranty of                *
  19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU         *
  20  * Library General Public License for more details.                          *
  21  *                                                                           *
  22  * You should have received a copy of the GNU Library General Public         *
  23  * License along with this library; if not, write to the Free Software       *
  24  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.                 *
  25  *                                                                           *
  26 \*---------------------------------------------------------------------------*/
  27 /*---------------------------------------------------------------------------*\
  28  *                                Changes                                    *
  29  *                                                                           *
  30  *                                                                           *
  31  *                                                                           *
  32  *                                                                           *
  33  *                                                                           *
  34  *                                                                           *
  35 \*---------------------------------------------------------------------------*/
  36
  37 #include <UnitTest++.h>
  38
  39 // Unit tests for OSB file io.
  40
  41 #include "OSGNode.h"
  42 #include "OSGGroup.h"
  43 #include "OSGNameAttachment.h"
  44 #include "OSGSceneFileHandler.h"
  45
  46 #include "OSGGeometry.h"
  47 #include "OSGSimpleGeometry.h"
  48 #include "OSGImage.h"
  49 #include "OSGChunkMaterial.h"
  50 #include "OSGTextureObjChunk.h"
  51 #include "OSGTextureEnvChunk.h"
  52
  53 #ifndef OSG_DISABLE_DEPRECATED
  54 #include "OSGTextureChunk.h"
  55 #endif
  56
  57 #include <boost/filesystem/operations.hpp>
  58
  59 namespace bf = boost::filesystem;
  60
  61 SUITE(OSBTests)
  62 {
  63
  64 struct FileFixture
  65 {
  66    FileFixture()
  67    {
  68       test_file = bf::path("test_file.osb");
  69       bf::remove(test_file);
  70    }
  71
  72    ~FileFixture()
  73    {
  74       bf::remove(test_file);
  75    }
  76
  77    bf::path test_file;
  78 };
  79
  80
  81 TEST_FIXTURE(FileFixture, CreateOSBFile)
  82 {
  83    OSG::NodeUnrecPtr n = OSG::Node::create();
  84    n->setCore(OSG::Group::create());
  85
  86    CHECK(!bf::exists(test_file));
  87    OSG::SceneFileHandler::the()->write(n, test_file.native_file_string().c_str());
  88    CHECK(bf::exists(test_file));
  89
  90    OSG::NodeUnrecPtr new_n =
  91       OSG::SceneFileHandler::the()->read(test_file.native_file_string().c_str());
  92    CHECK(new_n != NULL);
  93 }
  94
  95 TEST_FIXTURE(FileFixture, CreateOSBTree)
  96 {
  97    // Test larger tree
  98    OSG::NodeUnrecPtr base_node  = OSG::Node::create();
  99    base_node->setCore(OSG::Group::create());
 100    OSG::NodeUnrecPtr child_node = OSG::Node::create();
 101    child_node->setCore(OSG::Group::create());
 102
 103    base_node->addChild(child_node);
 104
 105    CHECK(!bf::exists(test_file));
 106    OSG::SceneFileHandler::the()->write(base_node, test_file.native_file_string().c_str());
 107    CHECK(bf::exists(test_file));
 108
 109    OSG::NodeUnrecPtr new_n =
 110       OSG::SceneFileHandler::the()->read(test_file.native_file_string().c_str());
 111    CHECK(new_n != NULL);
 112    CHECK(new_n->getCore() != NULL);
 113    CHECK(new_n->getNChildren() == 1);
 114    CHECK(new_n->getChild(0) != NULL);
 115    CHECK(new_n->getChild(0)->getCore() != NULL);
 116 }
 117
 118
 119 TEST_FIXTURE(FileFixture, TestNameRetention)
 120 {
 121    std::string start_name("node");
 122    OSG::NodeUnrecPtr n = OSG::Node::create();
 123    OSG::setName(n, start_name);
 124    n->setCore(OSG::Group::create());
 125
 126    CHECK(!bf::exists(test_file));
 127    OSG::SceneFileHandler::the()->write(n, test_file.native_file_string().c_str());
 128    CHECK(bf::exists(test_file));
 129
 130    OSG::NodeUnrecPtr new_n =
 131       OSG::SceneFileHandler::the()->read(test_file.native_file_string().c_str());
 132    CHECK(new_n != NULL);
 133    CHECK(OSG::getName(new_n) != NULL);
 134    std::string cur_name = std::string(OSG::getName(new_n));
 135
 136    CHECK(cur_name == start_name);
 137 }
 138
 139 #ifndef OSG_DISABLE_DEPRECATED
 140 TEST_FIXTURE(FileFixture, TextureChunkIO)
 141 {
 142     OSG::NodeUnrecPtr          n   = OSG::Node::create();
 143     OSG::GeometryUnrecPtr      geo = OSG::makeBoxGeo(2.0, 2.0, 2.0, 1, 1, 1);
 144     OSG::ChunkMaterialUnrecPtr mat = OSG::ChunkMaterial::create();
 145
 146     OSG::ImageUnrecPtr img = OSG::Image::create();
 147     img->set(OSG::Image::OSG_RGBA_PF, 2, 2);
 148
 149     img->editData()[0 * 2 * 4 + 0 * 4 + 0] = 255;
 150     img->editData()[0 * 2 * 4 + 0 * 4 + 1] =   0;
 151     img->editData()[0 * 2 * 4 + 0 * 4 + 2] =   0;
 152     img->editData()[0 * 2 * 4 + 0 * 4 + 3] = 255;
 153
 154     img->editData()[0 * 2 * 4 + 1 * 4 + 0] = 255;
 155     img->editData()[0 * 2 * 4 + 1 * 4 + 1] = 255;
 156     img->editData()[0 * 2 * 4 + 1 * 4 + 2] =   0;
 157     img->editData()[0 * 2 * 4 + 1 * 4 + 3] = 255;
 158
 159     img->editData()[1 * 2 * 4 + 0 * 4 + 0] =   0;
 160     img->editData()[1 * 2 * 4 + 0 * 4 + 1] =   0;
 161     img->editData()[1 * 2 * 4 + 0 * 4 + 2] = 255;
 162     img->editData()[1 * 2 * 4 + 0 * 4 + 3] = 255;
 163
 164     img->editData()[1 * 2 * 4 + 1 * 4 + 0] =   0;
 165     img->editData()[1 * 2 * 4 + 1 * 4 + 1] = 255;
 166     img->editData()[1 * 2 * 4 + 1 * 4 + 2] = 255;
 167     img->editData()[1 * 2 * 4 + 1 * 4 + 3] = 255;
 168
 169     OSG::TextureChunkUnrecPtr tex = OSG::TextureChunk::create();
 170     tex->setImage(img);
 171
 172     mat->addChunk(tex);
 173     geo->setMaterial(mat);
 174
 175     n->setCore(geo);
 176
 177     OSG::SceneFileHandler::the()->write(n, test_file.native_file_string().c_str());
 178
 179     OSG::NodeUnrecPtr n2 =
 180         OSG::SceneFileHandler::the()->read(test_file.native_file_string().c_str());
 181
 182     CHECK(n2 != NULL);
 183
 184     CHECK(n2->getCore() != NULL);
 185     OSG::GeometryUnrecPtr      geo2 =
 186         dynamic_cast<OSG::Geometry *>(n2->getCore());
 187     CHECK(geo2 != NULL);
 188
 189     CHECK(geo2->getMaterial() != NULL);
 190     OSG::ChunkMaterialUnrecPtr mat2 =
 191         dynamic_cast<OSG::ChunkMaterial *>(geo2->getMaterial());
 192     CHECK(mat2 != NULL);
 193
 194     CHECK(mat2->getChunk(0) != NULL);
 195     const OSG::TextureChunk *tex2 =
 196         dynamic_cast<const OSG::TextureChunk *>(mat2->getChunk(0));
 197     const OSG::TextureObjChunk *texObj =
 198         dynamic_cast<const OSG::TextureObjChunk *>(mat2->getChunk(0));
 199     const OSG::TextureEnvChunk *texEnv =
 200         dynamic_cast<const OSG::TextureEnvChunk *>(mat2->getChunk(1));
 201     CHECK(tex2   == NULL);
 202     CHECK(texObj != NULL);
 203     CHECK(texEnv != NULL);
 204
 205     CHECK(texObj->getImage() != NULL);
 206     OSG::Image *img2 = texObj->getImage();
 207     CHECK(img2 != NULL);
 208
 209     CHECK_EQUAL(255, img2->getData()[0 * 2 * 4 + 0 * 4 + 0]);
 210     CHECK_EQUAL(  0, img2->getData()[0 * 2 * 4 + 0 * 4 + 1]);
 211     CHECK_EQUAL(  0, img2->getData()[0 * 2 * 4 + 0 * 4 + 2]);
 212     CHECK_EQUAL(255, img2->getData()[0 * 2 * 4 + 0 * 4 + 3]);
 213
 214     CHECK_EQUAL(255, img2->getData()[0 * 2 * 4 + 1 * 4 + 0]);
 215     CHECK_EQUAL(255, img2->getData()[0 * 2 * 4 + 1 * 4 + 1]);
 216     CHECK_EQUAL(  0, img2->getData()[0 * 2 * 4 + 1 * 4 + 2]);
 217     CHECK_EQUAL(255, img2->getData()[0 * 2 * 4 + 1 * 4 + 3]);
 218
 219     CHECK_EQUAL(  0, img2->getData()[1 * 2 * 4 + 0 * 4 + 0]);
 220     CHECK_EQUAL(  0, img2->getData()[1 * 2 * 4 + 0 * 4 + 1]);
 221     CHECK_EQUAL(255, img2->getData()[1 * 2 * 4 + 0 * 4 + 2]);
 222     CHECK_EQUAL(255, img2->getData()[1 * 2 * 4 + 0 * 4 + 3]);
 223
 224     CHECK_EQUAL(  0, img2->getData()[1 * 2 * 4 + 1 * 4 + 0]);
 225     CHECK_EQUAL(255, img2->getData()[1 * 2 * 4 + 1 * 4 + 1]);
 226     CHECK_EQUAL(255, img2->getData()[1 * 2 * 4 + 1 * 4 + 2]);
 227     CHECK_EQUAL(255, img2->getData()[1 * 2 * 4 + 1 * 4 + 3]);
 228 }
 229 #endif // OSG_DISABLE_DEPRECATED
 230
 231 } // SUITE