Source/System/FileIO/Collada/OSGColladaTexture.cpp

   1 /*---------------------------------------------------------------------------*\
   2  *                                OpenSG                                     *
   3  *                                                                           *
   4  *                                                                           *
   5  *                Copyright (C) 2009 by the OpenSG Forum                     *
   6  *                                                                           *
   7  *                            www.opensg.org                                 *
   8  *                                                                           *
   9  *   contact: dirk@opensg.org, gerrit.voss@vossg.org, jbehr@zgdv.de          *
  10  *                                                                           *
  11 \*---------------------------------------------------------------------------*/
  12 /*---------------------------------------------------------------------------*\
  13  *                                License                                    *
  14  *                                                                           *
  15  * This library is free software; you can redistribute it and/or modify it   *
  16  * under the terms of the GNU Library General Public License as published    *
  17  * by the Free Software Foundation, version 2.                               *
  18  *                                                                           *
  19  * This library is distributed in the hope that it will be useful, but       *
  20  * WITHOUT ANY WARRANTY; without even the implied warranty of                *
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU         *
  22  * Library General Public License for more details.                          *
  23  *                                                                           *
  24  * You should have received a copy of the GNU Library General Public         *
  25  * License along with this library; if not, write to the Free Software       *
  26  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.                 *
  27  *                                                                           *
  28 \*---------------------------------------------------------------------------*/
  29 /*---------------------------------------------------------------------------*\
  30  *                                Changes                                    *
  31  *                                                                           *
  32  *                                                                           *
  33  *                                                                           *
  34  *                                                                           *
  35  *                                                                           *
  36  *                                                                           *
  37 \*---------------------------------------------------------------------------*/
  38
  39 #if __GNUC__ >= 4 || __GNUC_MINOR__ >=3
  40 #pragma GCC diagnostic ignored "-Wold-style-cast"
  41 #endif
  42
  43 #include "OSGColladaTexture.h"
  44
  45 #if defined(OSG_WITH_COLLADA) || defined(OSG_DO_DOC)
  46
  47 #include "OSGColladaLog.h"
  48 #include "OSGColladaGlobal.h"
  49 #include "OSGColladaEffect.h"
  50 #include "OSGColladaSurface.h"
  51 #include "OSGColladaImage.h"
  52 #include "OSGColladaExtraHandler.h"
  53 #include "OSGColladaHandlerFactory.h"
  54
  55 #include <dom/domFx_sampler2D_common.h>
  56
  57 OSG_BEGIN_NAMESPACE
  58
  59 ColladaElementRegistrationHelper ColladaTexture::_regHelper(
  60     &ColladaTexture::create,
  61      "texture");
  62
  63
  64 ColladaElementTransitPtr ColladaTexture::create(daeElement    *elem,
  65                                                 ColladaGlobal *global)
  66 {
  67     return ColladaElementTransitPtr(new ColladaTexture(elem, global));
  68 }
  69
  70 void ColladaTexture::read(ColladaElement *colElemParent)
  71 {
  72     OSG_COLLADA_LOG(("OSGColladaTexture::read\n"));
  73
  74     domCommon_color_or_texture_type::domTextureRef texture =
  75         dynamic_cast< domCommon_color_or_texture_type::domTexture * >(
  76             this->getDOMElement());
  77
  78     if(_colEffect == NULL)
  79     {
  80         SWARNING << "ColladaTexture::read: No effect set, can not resolve "
  81                  << "sampler." << std::endl;
  82         return;
  83     }
  84
  85     xsNCName          texId      =  texture->getTexture();
  86     ColladaElement   *colElem    = _colEffect->findParam(texId);
  87     ColladaSampler2D *colSampler =  dynamic_cast<ColladaSampler2D *>(colElem);
  88
  89     if(colSampler != NULL)
  90     {
  91         this->readSampler(colSampler);
  92     }
  93     else
  94     {
  95         this->readImage(texId);
  96     }
  97
  98
  99     const domExtraRef &extra = texture->getExtra();
 100
 101     if(extra == NULL || _extraHandlers.size() == 0)
 102         return;
 103
 104     ExtraHandlerStoreIt      ehIt  = _extraHandlers.begin();
 105     ExtraHandlerStoreConstIt ehEnd = _extraHandlers.end  ();
 106
 107     for(; ehIt != ehEnd; ++ehIt)
 108     {
 109         (*ehIt)->readTextureExtraElements(this, extra);
 110     }
 111
 112     if(_texObj == NULL)
 113     {
 114         SWARNING << "ColladaTexture::read: can not resolve "
 115                  << "image source." << std::endl;
 116     }
 117 }
 118
 119 ColladaEffect *ColladaTexture::getEffect(void) const
 120 {
 121     return _colEffect;
 122 }
 123
 124 void ColladaTexture::setEffect(ColladaEffect *colEffect)
 125 {
 126     _colEffect = colEffect;
 127 }
 128
 129 TextureObjChunk *ColladaTexture::getTexture(void) const
 130 {
 131     return _texObj;
 132 }
 133
 134 TextureEnvChunk *ColladaTexture::getTexEnv(void) const
 135 {
 136     return _texEnv;
 137 }
 138
 139 TextureEnvChunk *ColladaTexture::editTexEnv(void)
 140 {
 141     if(_texEnv == NULL)
 142         _texEnv = TextureEnvChunk::create();
 143
 144     return _texEnv;
 145 }
 146
 147 TextureTransformChunk *ColladaTexture::getTexTransform (void) const
 148 {
 149     return _texTransform;
 150 }
 151
 152 TextureTransformChunk *ColladaTexture::editTexTransform(void)
 153 {
 154     if(_texTransform == NULL)
 155         _texTransform = TextureTransformChunk::create();
 156
 157     return _texTransform;
 158 }
 159
 160 bool ColladaTexture::hasAlpha(void) const
 161 {
 162     bool retVal = false;
 163
 164     if(_texObj != NULL && _texObj->getImage() != NULL)
 165         retVal = _texObj->getImage()->hasAlphaChannel();
 166
 167     return retVal;
 168 }
 169
 170 bool ColladaTexture::hasBinaryAlpha(void) const
 171 {
 172     bool retVal = false;
 173
 174     if(_texObj != NULL && _texObj->getImage() != NULL)
 175     {
 176         retVal = _texObj->getImage()->isAlphaBinary    () ||
 177                  _texObj->getImage()->calcIsAlphaBinary();
 178     }
 179
 180     return retVal;
 181 }
 182
 183 const std::string ColladaTexture::getSemantic(void) const
 184 {
 185     domCommon_color_or_texture_type::domTextureRef texture =
 186         dynamic_cast< domCommon_color_or_texture_type::domTexture * >(
 187             this->getDOMElement());
 188
 189     if(texture != NULL)
 190         return texture->getTexcoord();
 191
 192     return std::string();
 193 }
 194
 195 ColladaTexture::ColladaTexture(daeElement    *elem,
 196                                ColladaGlobal *global) :
 197      Inherited    (elem,
 198                    global),
 199     _colEffect    (NULL  ),
 200     _texObj       (NULL  ),
 201     _texEnv       (NULL  ),
 202     _texTransform (NULL  ),
 203     _extraHandlers(      )
 204 {
 205     ColladaHandlerFactory::the()->createExtraHandlers(_extraHandlers);
 206 }
 207
 208 ColladaTexture::~ColladaTexture(void)
 209 {
 210     _extraHandlers.clear();
 211 }
 212
 213 void ColladaTexture::readSampler(ColladaSampler2D *colSampler)
 214 {
 215     ColladaSurface *colSurface = colSampler->getSurface();
 216
 217     if(colSurface == NULL)
 218     {
 219         SWARNING << "ColladaTexture::read: Could not resolve surface"
 220                  << std::endl;
 221
 222         return;
 223     }
 224
 225     _texObj = TextureObjChunk::create();
 226     _texObj->setImage(colSurface->getImage());
 227
 228     getGlobal()->getStatCollector()->getElem(
 229         ColladaGlobal::statNTextureCreated)->inc();
 230
 231     colSampler->setupTexObj(_texObj);
 232 }
 233
 234 void ColladaTexture::readImage(const Char8 *texId)
 235 {
 236     domCOLLADA *pDoc = this->getGlobal()->getDocRoot();
 237
 238     const domLibrary_images_Array &imagesLib = pDoc->getLibrary_images_array();
 239
 240     for(UInt32 i = 0; i < imagesLib.getCount(); ++i)
 241     {
 242         domImage_Array &imgArray = imagesLib[i]->getImage_array();
 243
 244         for(UInt32 j = 0; j < imgArray.getCount(); ++j)
 245         {
 246             domImageRef pImg = imgArray[j];
 247
 248             if(osgStringCmp(pImg->getId(), texId) == 0)
 249             {
 250                 ColladaImageRefPtr colImage  =
 251                     getUserDataAs<ColladaImage>(pImg);
 252
 253                 if(colImage == NULL)
 254                 {
 255                     colImage = dynamic_pointer_cast<ColladaImage>(
 256                         ColladaElementFactory::the()->create(pImg,
 257                                                              getGlobal()));
 258
 259                     colImage->read(this);
 260                 }
 261
 262                 _texObj = TextureObjChunk::create();
 263                 _texObj->setImage(colImage->getImage());
 264
 265                 break;
 266             }
 267         }
 268     }
 269 }
 270
 271 OSG_END_NAMESPACE
 272
 273 #endif // OSG_WITH_COLLADA