examples/rendering-basic-pbr/rendering-basic-pbr-example.cpp

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  *
  16  */
  17
  18 // EXTERNAL INCLUDES
  19 #include <dali-toolkit/dali-toolkit.h>
  20
  21 #include <stdio.h>
  22 #include <sstream>
  23
  24 // INTERNAL INCLUDES
  25 #include "ktx-loader.h"
  26 #include "model-skybox.h"
  27 #include "model-pbr.h"
  28 #include <dali/integration-api/debug.h>
  29 #include <dali/devel-api/adaptor-framework/file-stream.h>
  30
  31 using namespace Dali;
  32 using namespace Toolkit;
  33
  34 /*
  35  *
  36  * "papermill_pmrem.ktx" and "papermill_E_diffuse-64.ktx are image files generated from
  37  * image file downloaded from "http://www.hdrlabs.com/sibl/archive.html" with title
  38  * "Papermill Ruins E" by Blochi and is licensed under Creative Commons License
  39  * http://creativecommons.org/licenses/by-nc-sa/3.0/us/
  40  *
  41 */
  42
  43 namespace
  44 {
  45
  46 const char* NORMAL_ROUGH_TEXTURE_URL = DEMO_IMAGE_DIR "Test_100_normal_roughness.png";
  47 const char* ALBEDO_METAL_TEXTURE_URL = DEMO_IMAGE_DIR "Test_wblue_100_albedo_metal.png";
  48 const char* CUBEMAP_SPECULAR_TEXTURE_URL = DEMO_IMAGE_DIR "papermill_pmrem.ktx";
  49 const char* CUBEMAP_DIFFUSE_TEXTURE_URL = DEMO_IMAGE_DIR "papermill_E_diffuse-64.ktx";
  50
  51 const char* SPHERE_URL = DEMO_MODEL_DIR "sphere.obj";
  52 const char* TEAPOT_URL = DEMO_MODEL_DIR "teapot.obj";
  53
  54 const char* VERTEX_SHADER_URL = DEMO_SHADER_DIR "pbr_shader.vsh";
  55 const char* FRAGMENT_SHADER_URL = DEMO_SHADER_DIR "pbr_shader.fsh";
  56
  57 const Vector3 SKYBOX_SCALE( 1.0f, 1.0f, 1.0f );
  58 const Vector3 SPHERE_SCALE( 1.5f, 1.5f, 1.5f );
  59 const Vector3 TEAPOT_SCALE( 2.0f, 2.0f, 2.0f );
  60
  61 const float CAMERA_DEFAULT_FOV(    60.0f );
  62 const float CAMERA_DEFAULT_NEAR(    0.1f );
  63 const float CAMERA_DEFAULT_FAR(  1000.0f );
  64 const Vector3 CAMERA_DEFAULT_POSITION( 0.0f, 0.0f, 3.5f );
  65
  66 }
  67
  68 /*
  69  *
  70  * This example shows a Physically Based Rendering illumination model.
  71  *
  72  * - Double-tap to toggle between 3D models (teapot & cube)
  73  * - Pan up/down on left side of screen to change roughness
  74  * - Pan up/down on right side of screen to change metalness
  75  * - Pan anywhere else to rotate scene
  76  *
  77 */
  78
  79 class BasicPbrController : public ConnectionTracker
  80 {
  81 public:
  82
  83   BasicPbrController( Application& application )
  84   : mApplication( application ),
  85     mLabel(),
  86     m3dRoot(),
  87     mUiRoot(),
  88     mDoubleTapTime(),
  89     mModelOrientation(),
  90     mRoughness( 1.f ),
  91     mMetalness( 0.f ),
  92     mDoubleTap(false),
  93     mTeapotView(true)
  94   {
  95     // Connect to the Application's Init signal
  96     mApplication.InitSignal().Connect( this, &BasicPbrController::Create );
  97   }
  98
  99   ~BasicPbrController()
 100   {
 101     // Nothing to do here;
 102   }
 103
 104   // The Init signal is received once (only) during the Application lifetime
 105   void Create( Application& application )
 106   {
 107     // Get a handle to the stage
 108     Stage stage = Stage::GetCurrent();
 109     stage.SetBackgroundColor( Color::BLACK );
 110     mAnimation = Animation::New( 1.0f );
 111     mLabel = TextLabel::New( "R:1 M:0" );
 112     mLabel.SetProperty( Actor::Property::ANCHOR_POINT, AnchorPoint::TOP_CENTER );
 113     mLabel.SetProperty( Actor::Property::PARENT_ORIGIN, ParentOrigin::TOP_CENTER );
 114     mLabel.SetProperty( Actor::Property::SIZE, Vector2( stage.GetSize().width * 0.5f, stage.GetSize().height * 0.083f ) );
 115     mLabel.SetProperty( TextLabel::Property::HORIZONTAL_ALIGNMENT, "CENTER" );
 116     mLabel.SetProperty( TextLabel::Property::VERTICAL_ALIGNMENT, "CENTER" );
 117     mLabel.SetProperty( TextLabel::Property::TEXT_COLOR, Color::WHITE );
 118     mLabel.SetProperty( Actor::Property::COLOR_ALPHA, 0.0f );
 119     // Step 1. Create shader
 120     CreateModelShader();
 121
 122     // Step 2. Create texture
 123     CreateTexture();
 124
 125     // Step 3. Initialise Main Actor
 126     InitActors();
 127
 128     // Respond to a click anywhere on the stage
 129     stage.GetRootLayer().TouchSignal().Connect( this, &BasicPbrController::OnTouch );
 130
 131     // Respond to key events
 132     stage.KeyEventSignal().Connect( this, &BasicPbrController::OnKeyEvent );
 133
 134     mDoubleTapTime = Timer::New(150);
 135     mDoubleTapTime.TickSignal().Connect( this, &BasicPbrController::OnDoubleTapTime );
 136   }
 137
 138
 139   bool OnDoubleTapTime()
 140   {
 141     mDoubleTap = false;
 142     return true;
 143   }
 144
 145   /**
 146    * This function will change the material Roughness, Metalness or the model orientation when touched
 147    */
 148   bool OnTouch( Actor actor, const TouchData& touch )
 149   {
 150     const PointState::Type state = touch.GetState( 0 );
 151
 152     switch( state )
 153     {
 154       case PointState::DOWN:
 155       {
 156         if(mDoubleTap)
 157         {
 158           mTeapotView = !mTeapotView;
 159           mModel[0].GetActor().SetProperty(Dali::Actor::Property::VISIBLE, !mTeapotView);
 160           mModel[1].GetActor().SetProperty(Dali::Actor::Property::VISIBLE, mTeapotView);
 161         }
 162         mDoubleTapTime.Stop();
 163         mStartTouch = touch.GetScreenPosition(0);
 164         mPointZ = mStartTouch;
 165         mAnimation.Stop();
 166         mLabel.SetProperty( Actor::Property::COLOR_ALPHA, 1.0f );
 167         break;
 168       }
 169       case PointState::MOTION:
 170       {
 171         const Stage stage = Stage::GetCurrent();
 172         const Size size = stage.GetSize();
 173         const float scaleX = size.width;
 174         const float scaleY = size.height;
 175         const Vector2 point = touch.GetScreenPosition(0);
 176         bool process = false;
 177         if( ( mStartTouch.x < ( scaleX * 0.3f ) ) && ( point.x < ( scaleX * 0.3f ) ) )
 178         {
 179           mRoughness += ( mStartTouch.y - point.y ) / ( scaleY * 0.9f );
 180
 181           //Clamp Roughness to 0.0 to 1.0
 182           mRoughness = std::max( 0.f, std::min( 1.f, mRoughness ) );
 183
 184           mShader.SetProperty( mShader.GetPropertyIndex( "uRoughness" ), mRoughness );
 185           std::ostringstream oss;
 186           oss.precision(2);
 187           oss << " R:" << mRoughness<< "," << " M:" << mMetalness;
 188           mLabel.SetProperty( TextLabel::Property::TEXT, oss.str() );
 189           mStartTouch = point;
 190           process = true;
 191         }
 192         if( ( mStartTouch.x > ( scaleX * 0.7f ) ) && ( point.x > ( scaleX * 0.7f ) ) )
 193         {
 194           mMetalness += ( mStartTouch.y - point.y ) / ( scaleY * 0.9f );
 195
 196           //Clamp Metalness to 0.0 to 1.0
 197           mMetalness = std::max( 0.f, std::min( 1.f, mMetalness ) );
 198
 199           mShader.SetProperty( mShader.GetPropertyIndex( "uMetallic" ), mMetalness );
 200           std::ostringstream oss;
 201           oss.precision(2);
 202           oss << " R:" << mRoughness<< "," << " M:" << mMetalness;
 203           mLabel.SetProperty( TextLabel::Property::TEXT, oss.str() );
 204           mStartTouch = point;
 205           process = true;
 206         }
 207         //If the touch is not processed above, then change the model orientation
 208         if( !process )
 209         {
 210           process = true;
 211           const float angle1 = ( ( mPointZ.y - point.y ) / scaleY );
 212           const float angle2 = ( ( mPointZ.x - point.x ) / scaleX );
 213           Actor actor1 = mModel[0].GetActor();
 214           Actor actor2 = mModel[1].GetActor();
 215
 216           Quaternion modelOrientation = mModelOrientation;
 217           modelOrientation.Conjugate();
 218           const Quaternion pitchRot(Radian(Degree(angle1 * -200.0f)), modelOrientation.Rotate( Vector3::XAXIS ) );
 219           const Quaternion yawRot(Radian(Degree(angle2 * -200.0f)), modelOrientation.Rotate( Vector3::YAXIS ) );
 220
 221           mModelOrientation = mModelOrientation * yawRot * pitchRot ;
 222           mSkybox.GetActor().SetProperty( Actor::Property::ORIENTATION, mModelOrientation );
 223           actor1.SetProperty( Actor::Property::ORIENTATION, mModelOrientation );
 224           actor2.SetProperty( Actor::Property::ORIENTATION, mModelOrientation );
 225
 226           mPointZ = point;
 227         }
 228         break;
 229       }
 230       case PointState::UP:
 231       {
 232         mDoubleTapTime.Start();
 233         mDoubleTap = true;
 234         mAnimation.AnimateTo( Property( mLabel, Actor::Property::COLOR_ALPHA ), 0.0f, TimePeriod( 0.5f, 1.0f ) );
 235         mAnimation.Play();
 236         break;
 237       }
 238
 239       default:
 240       {
 241         break;
 242       }
 243     }
 244     return true;
 245   }
 246
 247   /**
 248    * @brief Called when any key event is received
 249    *
 250    * Will use this to quit the application if Back or the Escape key is received
 251    * @param[in] event The key event information
 252    */
 253   void OnKeyEvent( const KeyEvent& event )
 254   {
 255     if( event.state == KeyEvent::Down )
 256     {
 257       if( IsKey( event, Dali::DALI_KEY_ESCAPE ) || IsKey( event, Dali::DALI_KEY_BACK ) )
 258       {
 259         mApplication.Quit();
 260       }
 261     }
 262   }
 263
 264   /**
 265    * Creates new main actor
 266    */
 267   void InitActors()
 268   {
 269     Stage stage = Stage::GetCurrent();
 270
 271     mSkybox.Init( SKYBOX_SCALE );
 272     mModel[0].Init( mShader, SPHERE_URL, Vector3::ZERO, SPHERE_SCALE );
 273     mModel[1].Init( mShader, TEAPOT_URL, Vector3::ZERO, TEAPOT_SCALE );
 274
 275     // Hide the model according with mTeapotView variable
 276     mModel[0].GetActor().SetProperty(Dali::Actor::Property::VISIBLE, !mTeapotView);
 277     mModel[1].GetActor().SetProperty(Dali::Actor::Property::VISIBLE,  mTeapotView);
 278
 279     // Creating root and camera actor for rendertask for 3D Scene rendering
 280     mUiRoot = Actor::New();
 281     m3dRoot = Actor::New();
 282     CameraActor cameraUi = CameraActor::New(stage.GetSize());
 283     cameraUi.SetProperty( Actor::Property::ANCHOR_POINT,AnchorPoint::CENTER);
 284     cameraUi.SetProperty( Actor::Property::PARENT_ORIGIN,ParentOrigin::CENTER);
 285
 286     RenderTask rendertask = Stage::GetCurrent().GetRenderTaskList().CreateTask();
 287     rendertask.SetCameraActor( cameraUi );
 288     rendertask.SetSourceActor( mUiRoot );
 289
 290     mUiRoot.SetProperty( Actor::Property::ANCHOR_POINT,AnchorPoint::TOP_LEFT);
 291     mUiRoot.SetProperty( Actor::Property::PARENT_ORIGIN,ParentOrigin::TOP_LEFT);
 292     mUiRoot.SetProperty( Actor::Property::SIZE, stage.GetSize());
 293
 294     m3dRoot.SetProperty( Actor::Property::ANCHOR_POINT,AnchorPoint::CENTER);
 295     m3dRoot.SetProperty( Actor::Property::PARENT_ORIGIN,ParentOrigin::CENTER);
 296
 297     // Setting camera parameters for 3D Scene
 298     mSkybox.GetActor().SetProperty( Actor::Property::POSITION, CAMERA_DEFAULT_POSITION );
 299     CameraActor camera3d = stage.GetRenderTaskList().GetTask(0).GetCameraActor();
 300     camera3d.SetInvertYAxis( true );
 301     camera3d.SetProperty( Actor::Property::POSITION, CAMERA_DEFAULT_POSITION );
 302     camera3d.SetNearClippingPlane( CAMERA_DEFAULT_NEAR );
 303     camera3d.SetFarClippingPlane( CAMERA_DEFAULT_FAR );
 304     camera3d.SetFieldOfView( Radian( Degree( CAMERA_DEFAULT_FOV ) ) );
 305
 306     stage.Add( cameraUi );
 307     stage.Add( mUiRoot );
 308     stage.Add( m3dRoot );
 309
 310     m3dRoot.Add( mSkybox.GetActor() );
 311     m3dRoot.Add( mModel[0].GetActor() );
 312     m3dRoot.Add( mModel[1].GetActor() );
 313
 314
 315     if( (stage.GetSize().x > 360.0f) && (stage.GetSize().y > 360.0f) )
 316     {
 317       mUiRoot.Add( mLabel );
 318     }
 319
 320   }
 321
 322   /**
 323    * Creates a shader using file path
 324    */
 325   void CreateModelShader()
 326   {
 327     const std::string mCurrentVShaderFile( VERTEX_SHADER_URL );
 328     const std::string mCurrentFShaderFile( FRAGMENT_SHADER_URL );
 329
 330     mShader = LoadShaders( mCurrentVShaderFile, mCurrentFShaderFile );
 331
 332     // Initialise shader uniforms
 333     // Level 8 because the environment texture has 6 levels plus 2 are missing (2x2 and 1x1)
 334     mShader.RegisterProperty( "uMaxLOD", 8.0f );
 335     mShader.RegisterProperty( "uRoughness", 1.0f );
 336     mShader.RegisterProperty( "uMetallic" , 0.0f );
 337   }
 338
 339   /**
 340    * Create Textures
 341    */
 342   void CreateTexture()
 343   {
 344     PixelData albeldoPixelData = SyncImageLoader::Load( ALBEDO_METAL_TEXTURE_URL );
 345     Texture textureAlbedoMetal = Texture::New( TextureType::TEXTURE_2D, albeldoPixelData.GetPixelFormat(), albeldoPixelData.GetWidth(), albeldoPixelData.GetHeight() );
 346     textureAlbedoMetal.Upload( albeldoPixelData, 0, 0, 0, 0, albeldoPixelData.GetWidth(), albeldoPixelData.GetHeight() );
 347
 348     PixelData normalPixelData = SyncImageLoader::Load( NORMAL_ROUGH_TEXTURE_URL );
 349     Texture textureNormalRough = Texture::New( TextureType::TEXTURE_2D, normalPixelData.GetPixelFormat(), normalPixelData.GetWidth(), normalPixelData.GetHeight() );
 350     textureNormalRough.Upload( normalPixelData, 0, 0, 0, 0, normalPixelData.GetWidth(), normalPixelData.GetHeight() );
 351
 352     // This texture should have 6 faces and only one mipmap
 353     PbrDemo::CubeData diffuse;
 354     PbrDemo::LoadCubeMapFromKtxFile( CUBEMAP_DIFFUSE_TEXTURE_URL, diffuse );
 355
 356     Texture diffuseTexture = Texture::New( TextureType::TEXTURE_CUBE, diffuse.img[0][0].GetPixelFormat(), diffuse.img[0][0].GetWidth(), diffuse.img[0][0].GetHeight() );
 357     for( unsigned int midmapLevel = 0; midmapLevel < diffuse.img[0].size(); ++midmapLevel )
 358     {
 359       for( unsigned int i = 0; i < diffuse.img.size(); ++i )
 360       {
 361         diffuseTexture.Upload( diffuse.img[i][midmapLevel], CubeMapLayer::POSITIVE_X + i, midmapLevel, 0, 0, diffuse.img[i][midmapLevel].GetWidth(), diffuse.img[i][midmapLevel].GetHeight() );
 362       }
 363     }
 364
 365     // This texture should have 6 faces and 6 mipmaps
 366     PbrDemo::CubeData specular;
 367     PbrDemo::LoadCubeMapFromKtxFile( CUBEMAP_SPECULAR_TEXTURE_URL, specular);
 368
 369     Texture specularTexture = Texture::New( TextureType::TEXTURE_CUBE, specular.img[0][0].GetPixelFormat(), specular.img[0][0].GetWidth(), specular.img[0][0].GetHeight() );
 370     for( unsigned int midmapLevel = 0; midmapLevel < specular.img[0].size(); ++midmapLevel )
 371     {
 372       for( unsigned int i = 0; i < specular.img.size(); ++i )
 373       {
 374         specularTexture.Upload( specular.img[i][midmapLevel], CubeMapLayer::POSITIVE_X + i, midmapLevel, 0, 0, specular.img[i][midmapLevel].GetWidth(), specular.img[i][midmapLevel].GetHeight() );
 375       }
 376     }
 377
 378     mModel[0].InitTexture( textureAlbedoMetal, textureNormalRough, diffuseTexture, specularTexture );
 379     mModel[1].InitTexture( textureAlbedoMetal, textureNormalRough, diffuseTexture, specularTexture );
 380     mSkybox.InitTexture( specularTexture );
 381   }
 382
 383   /**
 384   * @brief Load a shader source file
 385   * @param[in] The path of the source file
 386   * @param[out] The contents of file
 387   * @return True if the source was read successfully
 388   */
 389   bool LoadShaderCode( const std::string& fullpath, std::vector<char>& output )
 390   {
 391     Dali::FileStream fileStream( fullpath, FileStream::READ | FileStream::BINARY );
 392     FILE* file = fileStream.GetFile();
 393     if( NULL == file )
 394     {
 395       return false;
 396     }
 397
 398     bool retValue = false;
 399     if( ! fseek( file, 0, SEEK_END ) )
 400     {
 401       long int size = ftell( file );
 402
 403       if( ( size != -1L ) &&
 404         ( ! fseek( file, 0, SEEK_SET ) ) )
 405       {
 406         output.resize( size + 1 );
 407         std::fill( output.begin(), output.end(), 0 );
 408         ssize_t result = fread( output.data(), size, 1, file );
 409
 410         retValue = ( result >= 0 );
 411       }
 412     }
 413
 414     return retValue;
 415   }
 416
 417   /**
 418   * @brief Load vertex and fragment shader source
 419   * @param[in] shaderVertexFileName is the filepath of Vertex shader
 420   * @param[in] shaderFragFileName is the filepath of Fragment shader
 421   * @return the Dali::Shader object
 422   */
 423   Shader LoadShaders( const std::string& shaderVertexFileName, const std::string& shaderFragFileName )
 424   {
 425     Shader shader;
 426     std::vector<char> bufV, bufF;
 427
 428     if( LoadShaderCode( shaderVertexFileName.c_str(), bufV ) )
 429     {
 430       if( LoadShaderCode( shaderFragFileName.c_str(), bufF ) )
 431       {
 432         shader = Shader::New( bufV.data() , bufF.data() );
 433       }
 434     }
 435     return shader;
 436   }
 437
 438 private:
 439   Application& mApplication;
 440   TextLabel mLabel;
 441   Actor m3dRoot;
 442   Actor mUiRoot;
 443   Shader mShader;
 444   Animation mAnimation;
 445   Timer mDoubleTapTime;
 446
 447   ModelSkybox mSkybox;
 448   ModelPbr mModel[2];
 449
 450   Vector2 mPointZ;
 451   Vector2 mStartTouch;
 452
 453   Quaternion mModelOrientation;
 454   float mRoughness;
 455   float mMetalness;
 456   bool mDoubleTap;
 457   bool mTeapotView;
 458
 459 };
 460
 461 int DALI_EXPORT_API main( int argc, char **argv )
 462 {
 463   Application application = Application::New( &argc, &argv);
 464   BasicPbrController test( application );
 465   application.MainLoop();
 466   return 0;
 467 }