0033731: Visualization - Transparent background for rendered image files

- add flag to 3D view dump for removing background based color keying.
- add color keying function to Image_PixMap

diff --git a/src/Image/Image_PixMap.cxx b/src/Image/Image_PixMap.cxx
index c99971aefa6729d495714d9db6b94f105050c8bb..f967d9cb22bf72f229f749834ce2146108dcd00e 100644 (file)
--- a/src/Image/Image_PixMap.cxx
+++ b/src/Image/Image_PixMap.cxx
@@ -17,6 +17,7 @@
 
 #include <NCollection_AlignedAllocator.hxx>
 #include <Standard_ProgramError.hxx>
+#include <Message.hxx>
 
 #include <algorithm>
 
@@ -829,6 +830,42 @@ void Image_PixMap::ToBlackWhite (Image_PixMap& theImage)
   }
 }
 
+// =======================================================================
+// function : ColorKeying
+// purpose  :
+// =======================================================================
+void Image_PixMap::ColorKeying(const Quantity_Color& theKey, Image_PixMap& theImage)
+{
+  if (theImage.myImgFormat != Image_Format_RGBA && theImage.myImgFormat != Image_Format_BGRA)
+  {
+    Message::SendWarning("Warning: Image format should support alpha channel");
+  }
+
+  const Standard_Byte aRed = (Standard_Byte) (theKey.Red() * 255);
+  const Standard_Byte aGreen = (Standard_Byte) (theKey.Green() * 255);
+  const Standard_Byte aBlue = (Standard_Byte) (theKey.Blue() * 255);
+
+  if (theImage.myImgFormat == Image_Format_RGBA)
+  {
+    for (Standard_Size i = 0; i < theImage.myData.Size(); i += 4)
+    {
+      if (theImage.myData.ChangeData()[i] == aRed && theImage.myData.ChangeData()[i + 1] == aGreen && theImage.myData.ChangeData()[i + 2] == aBlue)
+      {
+        theImage.myData.ChangeData()[i + 3] = 0;
+      }
+    }
+  }
+  if (theImage.myImgFormat == Image_Format_BGRA)
+  {
+    for (Standard_Size i = 0; i < theImage.myData.Size(); i += 4)
+    {
+      if (theImage.myData.ChangeData()[i] == aBlue && theImage.myData.ChangeData()[i + 1] == aGreen && theImage.myData.ChangeData()[i + 2] == aRed)
+      {
+        theImage.myData.ChangeData()[i + 3] = 0;
+      }
+    }
+  }
+}
 // =======================================================================
 // function : FlipY
 // purpose  :

diff --git a/src/Image/Image_PixMap.hxx b/src/Image/Image_PixMap.hxx
index b4bef537b0418398dea1dd4adae7d4b1d2fb0795..e9207591b8e69ed5361bd2afe13c90930ca4f3ae 100644 (file)
--- a/src/Image/Image_PixMap.hxx
+++ b/src/Image/Image_PixMap.hxx
@@ -51,6 +51,9 @@ public:
   //! Convert image to Black/White.
   Standard_EXPORT static void ToBlackWhite (Image_PixMap& theImage);
 
+  //! Isolating a foreground object against a background area of uniform color 
+  Standard_EXPORT static void ColorKeying(const Quantity_Color& theKey, Image_PixMap& theImage);
+
   //! Reverse line order as it draws it from bottom to top.
   Standard_EXPORT static bool FlipY (Image_PixMap& theImage);
 

diff --git a/src/V3d/V3d_ImageDumpOptions.hxx b/src/V3d/V3d_ImageDumpOptions.hxx
index 368f7c881d639b2e96bb23f7951c1af83f5b58bf..77cfc969f8caa9b2f0fcdd7c76c96d1ceaa6514a 100644 (file)
--- a/src/V3d/V3d_ImageDumpOptions.hxx
+++ b/src/V3d/V3d_ImageDumpOptions.hxx
@@ -21,12 +21,13 @@
 struct V3d_ImageDumpOptions
 {
 
-  Standard_Integer      Width;          //!< width  of image dump to allocate an image, 0 by default (meaning that image should be already allocated)
-  Standard_Integer      Height;         //!< height of image dump to allocate an image, 0 by default (meaning that image should be already allocated)
-  Graphic3d_BufferType  BufferType;     //!< which buffer to dump (color / depth), Graphic3d_BT_RGB by default
-  V3d_StereoDumpOptions StereoOptions;  //!< dumping stereoscopic camera, V3d_SDO_MONO by default (middle-point monographic projection)
-  Standard_Integer      TileSize;       //!< the view dimension limited for tiled dump, 0 by default (automatic tiling depending on hardware capabilities)
-  Standard_Boolean      ToAdjustAspect; //!< flag to override active view aspect ratio by (Width / Height) defined for image dump (TRUE by default)
+  Standard_Integer      Width;              //!< width  of image dump to allocate an image, 0 by default (meaning that image should be already allocated)
+  Standard_Integer      Height;             //!< height of image dump to allocate an image, 0 by default (meaning that image should be already allocated)
+  Graphic3d_BufferType  BufferType;         //!< which buffer to dump (color / depth), Graphic3d_BT_RGB by default
+  V3d_StereoDumpOptions StereoOptions;      //!< dumping stereoscopic camera, V3d_SDO_MONO by default (middle-point monographic projection)
+  Standard_Integer      TileSize;           //!< the view dimension limited for tiled dump, 0 by default (automatic tiling depending on hardware capabilities)
+  Standard_Boolean      ToAdjustAspect;     //!< flag to override active view aspect ratio by (Width / Height) defined for image dump (TRUE by default)
+  Standard_Boolean      ToRemoveBackground; //!< flag to override background with chroma key background
 
 public:
 

diff --git a/src/V3d/V3d_View.cxx b/src/V3d/V3d_View.cxx
index 7658ea79842fcff2bbafee0c277ebbb36d22cf72..ae22147685efb8db867614b6b8894ec76933e8d9 100644 (file)
--- a/src/V3d/V3d_View.cxx
+++ b/src/V3d/V3d_View.cxx
@@ -2941,6 +2941,17 @@ Standard_Boolean V3d_View::ToPixMap (Image_PixMap&               theImage,
     aCamera->SetAspect (Standard_Real(aTargetSize.x()) / Standard_Real(aTargetSize.y()));
   }
 
+  // backup background
+  Aspect_Background theBackground = myView->Background();
+  Quantity_Color aChroma = Quantity_NOC_MAGENTA;
+  // backup antialiasing settings
+  Standard_Boolean aIsAntialiasingEnabled = myView->RenderingParams().IsAntialiasingEnabled;
+  if (theParams.ToRemoveBackground)
+  {
+    myView->SetBackground(Aspect_Background(aChroma));
+    myView->ChangeRenderingParams().IsAntialiasingEnabled = Standard_False;
+  }
+
   // render immediate structures into back buffer rather than front
   const Standard_Boolean aPrevImmediateMode = myView->SetImmediateModeDrawToFront (Standard_False);
 
@@ -3012,6 +3023,16 @@ Standard_Boolean V3d_View::ToPixMap (Image_PixMap&               theImage,
     }
   }
 
+  if(theParams.ToRemoveBackground)
+  {
+    // color keying
+    Image_PixMap::ColorKeying(aChroma, theImage);
+    // restore original background
+    myView->SetBackground(theBackground);
+    // restore Antialiasing settings
+    myView->ChangeRenderingParams().IsAntialiasingEnabled = aIsAntialiasingEnabled;
+  }
+  
   // restore state
   myView->SetImmediateModeDrawToFront (aPrevImmediateMode);
   aCamera->Copy (aStoreMapping);

diff --git a/src/V3d/V3d_View.hxx b/src/V3d/V3d_View.hxx
index 22847902e993e4815eb548aa5bd8ae78b1f58868..11c71dc235484e865a4e4b9669678d490c86cd2d 100644 (file)
--- a/src/V3d/V3d_View.hxx
+++ b/src/V3d/V3d_View.hxx
@@ -856,7 +856,8 @@ public:
                              const Standard_Integer theHeight,
                              const Graphic3d_BufferType& theBufferType     = Graphic3d_BT_RGB,
                              const Standard_Boolean      theToAdjustAspect = Standard_True,
-                             const V3d_StereoDumpOptions theStereoOptions  = V3d_SDO_MONO)
+                             const V3d_StereoDumpOptions theStereoOptions  = V3d_SDO_MONO,
+                             const Standard_Boolean theRemoveBackground = Standard_False)
   {
     V3d_ImageDumpOptions aParams;
     aParams.Width  = theWidth;
@@ -864,6 +865,7 @@ public:
     aParams.BufferType = theBufferType;
     aParams.StereoOptions  = theStereoOptions;
     aParams.ToAdjustAspect = theToAdjustAspect;
+    aParams.ToRemoveBackground = theRemoveBackground;
     return ToPixMap (theImage, aParams);
   }