Code/Sandbox/Plugins/LodGeneratorPlugin/control/MeshBakerPopupPreview.cpp

   1 #include "StdAfx.h"
   2 #include <Cry3DEngine/IStatObj.h>
   3 #include "MeshBakerPopupPreview.h"
   4 #include "DisplayViewportAdapter.h"
   5 #include "Material/Material.h"
   6
   7
   8 CCMeshBakerPopupPreview::CCMeshBakerPopupPreview(QViewport * viewPort):m_ViewPort(viewPort)
   9 {
  10         m_pViewportAdapter.reset(new CDisplayViewportAdapter(m_ViewPort));
  11         m_pObject = NULL;
  12         m_pMat = NULL;
  13
  14         m_bPrecacheMaterial = false;
  15         m_bDrawWireFrame = false;
  16         m_bRotate = false;
  17         m_rotateAngle = 0;
  18
  19         CDLight l;
  20
  21         float L = 1.0f;
  22         l.m_fRadius = 10000;
  23         l.m_Flags |= DLF_SUN | DLF_DIRECTIONAL;
  24         l.SetLightColor(ColorF(L,L,L,1));
  25         l.SetPosition( Vec3(100,100,100) );
  26         m_lights.push_back( l );
  27
  28         m_bShowObject = true;
  29
  30         m_clearColor.set(0.5f, 0.5f, 0.5f, 1.0f);
  31         m_ambientColor.set(1.0f, 1.0f, 1.0f, 1.0f);
  32         m_ambientMultiplier = 0.5f;
  33 }
  34
  35
  36 void CCMeshBakerPopupPreview::OnKeyEvent(const SKeyEvent& sKeyEvent)
  37 {
  38
  39 }
  40
  41 void CCMeshBakerPopupPreview::OnMouseEvent(const SMouseEvent& sMouseEvent)
  42 {
  43
  44 }
  45
  46
  47
  48 void CCMeshBakerPopupPreview::SetObject(IStatObj * pObj)
  49 {
  50         m_pObject = pObj;
  51 }
  52
  53 void CCMeshBakerPopupPreview::ReleaseObject()
  54 {
  55         if (m_pObject != NULL)
  56         {
  57                 m_pObject = NULL;
  58         }
  59 }
  60
  61 void CCMeshBakerPopupPreview::SetMaterial(IMaterial* pMat)
  62 {
  63         m_pMat = (CMaterial*)pMat;
  64 }
  65
  66 void CCMeshBakerPopupPreview::SetRotate(bool rotate)
  67 {
  68         m_Rotate = rotate;
  69 }
  70
  71 void CCMeshBakerPopupPreview::SetWireframe(bool wireframe)
  72 {
  73         m_bDrawWireFrame = wireframe;
  74 }
  75
  76 void CCMeshBakerPopupPreview::SetGrid(bool grid)
  77 {
  78         m_Grid = grid;
  79 }
  80
  81
  82 void CCMeshBakerPopupPreview::OnRender(const SRenderContext& rc)
  83 {
  84         if (m_bDrawWireFrame)
  85         {
  86                 NoMaterialRender(rc);
  87         }
  88         else
  89         {
  90                 MaterialRender(rc);
  91         }
  92 }
  93
  94 bool CCMeshBakerPopupPreview::NoMaterialRender(const SRenderContext& rc)
  95 {
  96         IRenderAuxGeom* aux = gEnv->pRenderer->GetIRenderAuxGeom();
  97
  98         DisplayContext dc;
  99         m_pViewportAdapter->EnableXYViewport(true);
 100         dc.SetView(m_pViewportAdapter.get());
 101
 102         aux->SetRenderFlags(e_Mode3D|e_AlphaBlended|e_FillModeSolid|e_CullModeNone|e_DepthWriteOff|e_DepthTestOn);
 103
 104         dc.DepthTestOff();
 105
 106         RenderUnwrappedMesh(dc,rc);
 107
 108         dc.DepthTestOn();
 109
 110         return true;
 111 }
 112
 113 void CCMeshBakerPopupPreview::RenderUnwrappedMesh(DisplayContext& dc,const SRenderContext& rc)
 114 {
 115         if( !m_pObject )
 116                 return;
 117
 118 //      CEdGeometry* pCEdGeometry = m_pObject->GetGeometry();m_pObject->GetIndexedMesh(true);
 119 //      if (NULL == pCEdGeometry)
 120 //              return;
 121 //      IIndexedMesh* pIIndexedMesh = pCEdGeometry->GetIndexedMesh();
 122
 123
 124         IIndexedMesh* pIIndexedMesh = m_pObject->GetIndexedMesh(true);
 125         if (NULL == pIIndexedMesh)
 126                 return;
 127
 128         RenderUnwrappedLodPolygon(dc, pIIndexedMesh,rc);
 129 }
 130
 131
 132 void CCMeshBakerPopupPreview::RenderUnwrappedLodPolygon(DisplayContext& dc, IIndexedMesh* pIIndexedMesh,const SRenderContext& rc)
 133 {
 134         if (pIIndexedMesh->GetFaceCount() == 0)
 135         {
 136                 pIIndexedMesh->RestoreFacesFromIndices();
 137         }
 138
 139         IIndexedMesh::SMeshDescription sMeshDescription;
 140         pIIndexedMesh->GetMeshDescription(sMeshDescription);
 141
 142
 143         ColorB color = ColorB(255,0,0);
 144         dc.SetColor(color);
 145
 146         for (int i=0; i<sMeshDescription.m_nFaceCount; i++)
 147         {
 148                 const SMeshFace* pFace = &(sMeshDescription.m_pFaces[i]);
 149
 150                 const Vec3* pSMeshVert0 = &sMeshDescription.m_pVerts[pFace->v[0]];
 151                 const Vec3* pSMeshVert1 = &sMeshDescription.m_pVerts[pFace->v[1]];
 152                 const Vec3* pSMeshVert2 = &sMeshDescription.m_pVerts[pFace->v[2]];
 153                 dc.DrawLine(*pSMeshVert0, *pSMeshVert1);
 154                 dc.DrawLine(*pSMeshVert1, *pSMeshVert2);
 155                 dc.DrawLine(*pSMeshVert0, *pSMeshVert2);
 156
 157         }
 158
 159 }
 160
 161 bool CCMeshBakerPopupPreview::MaterialRender(const SRenderContext& rc)
 162 {
 163         if (m_pMat == NULL)
 164                 return false;
 165
 166         IRenderer* renderer = gEnv->pRenderer;
 167         int width = rc.viewport->width();
 168         int height = rc.viewport->height();
 169         if (height < 2 || width < 2)
 170                 return false;
 171
 172         SRenderingPassInfo passInfo = SRenderingPassInfo::CreateGeneralPassRenderingInfo(*rc.camera, SRenderingPassInfo::DEFAULT_FLAGS, true);
 173
 174         renderer->ChangeViewport(0,0,width,height);
 175         renderer->SetCamera( *rc.camera );
 176
 177         CScopedWireFrameMode scopedWireFrame(renderer, m_bDrawWireFrame?R_WIREFRAME_MODE:R_SOLID_MODE);
 178
 179         // Add lights.
 180         for (size_t i = 0; i < m_lights.size(); i++)
 181         {
 182                 renderer->EF_ADDDlight(&m_lights[i], passInfo);
 183         }
 184
 185         if(m_pMat)
 186                 m_pMat->DisableHighlight();
 187
 188         _smart_ptr<IMaterial> pMaterial;
 189         if (m_pMat)
 190                 pMaterial = m_pMat->GetMatInfo();
 191
 192         if (m_bPrecacheMaterial)
 193         {
 194                 IMaterial *pCurMat=pMaterial;
 195                 if (!pCurMat)
 196                 {
 197                         if (m_pObject)
 198                                 pCurMat=m_pObject->GetMaterial();
 199                         //                              else if (m_pEntity)
 200                         //                                      pCurMat=m_pEntity->GetMaterial();
 201                         //                              else if (m_pCharacter)
 202                         //                                      pCurMat=m_pCharacter->GetIMaterial();
 203                         //                              else if (m_pEmitter)
 204                         //                                      pCurMat=m_pEmitter->GetMaterial();
 205                 }
 206                 if (pCurMat)
 207                 {
 208                         pCurMat->PrecacheMaterial(0.0f, NULL, true, true);
 209                 }
 210         }
 211
 212         { // activate shader item
 213                 IMaterial *pCurMat=pMaterial;
 214                 if (!pCurMat)
 215                 {
 216                         if (m_pObject)
 217                                 pCurMat=m_pObject->GetMaterial();
 218                         //                              else if (m_pEntity)
 219                         //                                      pCurMat=m_pEntity->GetMaterial();
 220                         //                              else if (m_pCharacter)
 221                         //                                      pCurMat=m_pCharacter->GetIMaterial();
 222                         //                              else if (m_pEmitter)
 223                         //                                      pCurMat=m_pEmitter->GetMaterial();
 224                 }
 225
 226         }
 227
 228         if (m_bShowObject)
 229                 RenderObject(pMaterial, passInfo);
 230
 231
 232
 233
 234         return true;
 235 }
 236
 237
 238 void CCMeshBakerPopupPreview::RenderObject( IMaterial* pMaterial, SRenderingPassInfo& passInfo )
 239 {
 240         if (pMaterial == NULL)
 241                 return;
 242
 243         SRendParams rp;
 244         rp.dwFObjFlags = 0;
 245         //rp.nDLightMask = (1<<m_lights.size())-1;
 246         rp.AmbientColor = m_ambientColor * m_ambientMultiplier;
 247         rp.dwFObjFlags |= FOB_TRANS_MASK /*| FOB_GLOBAL_ILLUMINATION*/ | FOB_NO_FOG /*| FOB_ZPREPASS*/;
 248         rp.pMaterial = pMaterial;
 249
 250         Matrix34 tm;
 251         tm.SetIdentity();
 252         rp.pMatrix = &tm;
 253
 254         if (m_bRotate)
 255         {
 256                 tm.SetRotationXYZ(Ang3( 0,0,m_rotateAngle ));
 257                 m_rotateAngle += 0.1f;
 258         }
 259
 260         // visn test
 261 //      if (m_pObject)
 262 //              m_pObject->Render( rp, passInfo );
 263
 264 //      if (m_pEntity)
 265 //              m_pEntity->Render( rp, passInfo );
 266 //
 267 //      if (m_pCharacter)
 268 //              m_pCharacter->Render( rp, QuatTS(IDENTITY), passInfo );
 269 //
 270 //      if (m_pEmitter)
 271 //      {
 272 //              m_pEmitter->Update();
 273 //              m_pEmitter->Render( rp, passInfo );
 274 //      }
 275 }
 276
 277
 278 void CCMeshBakerPopupPreview::RenderUnwrappedLodPolygonUV(DisplayContext& dc, IIndexedMesh* pIIndexedMesh)
 279 {
 280         if (pIIndexedMesh->GetFaceCount() == 0)
 281         {
 282                 pIIndexedMesh->RestoreFacesFromIndices();
 283         }
 284
 285         IIndexedMesh::SMeshDescription sMeshDescription;
 286         pIIndexedMesh->GetMeshDescription(sMeshDescription);
 287
 288         ColorB color = ColorB(255,255,255);
 289         dc.SetColor(color);
 290
 291         for (int i=0; i<sMeshDescription.m_nFaceCount; i++)
 292         {
 293                 const SMeshFace* pFace = &(sMeshDescription.m_pFaces[i]);
 294                 for (int j=0; j<3; j++)
 295                 {
 296                         const SMeshTexCoord* pSMeshTexCoord0 = &sMeshDescription.m_pTexCoord[pFace->v[0]];
 297                         const SMeshTexCoord* pSMeshTexCoord1 = &sMeshDescription.m_pTexCoord[pFace->v[1]];
 298                         const SMeshTexCoord* pSMeshTexCoord2 = &sMeshDescription.m_pTexCoord[pFace->v[2]];
 299                         Vec2 uv0 = pSMeshTexCoord0->GetUV();
 300                         Vec2 uv1 = pSMeshTexCoord1->GetUV();
 301                         Vec2 uv2 = pSMeshTexCoord2->GetUV();
 302                         dc.DrawLine(Vec3(uv0.x,uv0.y,0.01f), Vec3(uv1.x,uv1.y,0.01f));
 303                         dc.DrawLine(Vec3(uv0.x,uv0.y,0.01f), Vec3(uv2.x,uv2.y,0.01f));
 304                         dc.DrawLine(Vec3(uv1.x,uv1.y,0.01f), Vec3(uv2.x,uv2.y,0.01f));
 305                 }
 306
 307         }
 308
 309 }