@ -80,11 +80,32 @@ class RasterView:
p = point2d - offset
return self . image_x_dir_inv * p . x + self . image_y_dir_inv * p . y + self . image_z_dir_inv * p . z
def _flatten_objects_tree ( self , obj_list ) :
result = [ ]
for obj in obj_list :
if obj . TypeId == ' Part::FeaturePython ' and hasattr ( obj , ' LinkedObject ' ) : # variant link
result . extend ( self . _flatten_objects_tree ( obj . Group ) )
elif obj . TypeId in [ ' App::Link ' ] :
result . extend ( self . _flatten_objects_tree ( [ obj . LinkedObject ] ) )
elif obj . TypeId in [ ' App::Part ' , ' App::DocumentObjectGroup ' ] :
result . extend ( self . _flatten_objects_tree ( obj . Group ) )
elif obj . TypeId in [ ' Part::Feature ' , ' Part::FeaturePython ' , ' PartDesign::Body ' , ' PartDesign::CoordinateSystem ' , ' PartDesign::Line ' , ' Part::Mirroring ' , ' Part::Cut ' ] :
result . append ( obj )
if hasattr ( obj , ' Group ' ) :
result . extend ( self . _flatten_objects_tree ( obj . Group ) )
return result
def _should_render ( self , obj ) :
return obj . TypeId in [ ' Part::Feature ' , ' Part::FeaturePython ' , ' PartDesign::Body ' , ' Part::Mirroring ' , ' Part::Cut ' ]
def render ( self ) :
from pivy import coin
import os
from PIL import Image , ImageDraw , ImageChops
workbench = Gui . getWorkbench ( " AssemblyHandbookWorkbench " ) #: :type workbench: AssemblyHandbookWorkbench
view = self . source_view
self . init_image ( )
@ -96,6 +117,9 @@ class RasterView:
os . makedirs ( dir )
doc = App . newDocument ( ' tmp_raster ' , hidden = False , temp = False )
objects_to_reset = { }
try :
for part in view . XSource :
link = doc . addObject ( ' App::Link ' , part . Name )
link . Label = part . Label
@ -108,6 +132,42 @@ class RasterView:
else :
link . LinkedObject = part
new_part = workbench . techDrawExtensions . isNewPartInView ( view , part )
link . ViewObject . OverrideMaterial = True
link . ViewObject . ShapeMaterial . DiffuseColor = ( 0.0 , 0.0 , 0.0 , 0.0 ) if new_part else ( 0.5 , 0.5 , 0.5 , 0.0 ) # this actually changes the line color
# in current version of freecad, link override material does not allow to override all material properties, for example emissive color, so we have to change material of the linked object
for obj in self . _flatten_objects_tree ( [ link ] ) :
if obj in objects_to_reset . keys ( ) :
continue
if self . _should_render ( obj ) :
objects_to_reset [ obj ] = (
obj . ViewObject . LineColor ,
obj . ViewObject . ShapeMaterial . AmbientColor ,
obj . ViewObject . ShapeMaterial . DiffuseColor ,
obj . ViewObject . ShapeMaterial . SpecularColor ,
obj . ViewObject . ShapeMaterial . EmissiveColor ,
obj . ViewObject . LineWidth
)
obj . ViewObject . LineColor = ( 0.0 , 0.0 , 0.0 , 0.0 )
obj . ViewObject . ShapeMaterial . AmbientColor = ( 0.0 , 0.0 , 0.0 , 0.0 )
obj . ViewObject . ShapeMaterial . DiffuseColor = ( 0.0 , 0.0 , 0.0 , 0.0 )
obj . ViewObject . ShapeMaterial . SpecularColor = ( 0.0 , 0.0 , 0.0 , 0.0 )
obj . ViewObject . ShapeMaterial . EmissiveColor = ( 1.0 , 1.0 , 1.0 , 0.0 )
# We need to set two different values otherwise freecad does not always update LineWidth of sub-elements
obj . ViewObject . LineWidth = 1.0
obj . ViewObject . LineWidth = 2.0
else :
objects_to_reset [ obj ] = (
obj . ViewObject . Visibility ,
)
obj . ViewObject . Visibility = False
docView = Gui . getDocument ( doc . Name ) . mdiViewsOfType ( ' Gui::View3DInventor ' ) [ 0 ]
cam = docView . getCameraNode ( )
@ -125,6 +185,18 @@ class RasterView:
docView . saveImage ( self . image_file_name , 4096 , 4096 , " #ffffff " )
finally :
for obj , props in objects_to_reset . items ( ) :
if self . _should_render ( obj ) :
obj . ViewObject . LineColor = props [ 0 ]
obj . ViewObject . ShapeMaterial . AmbientColor = props [ 1 ]
obj . ViewObject . ShapeMaterial . DiffuseColor = props [ 2 ]
obj . ViewObject . ShapeMaterial . SpecularColor = props [ 3 ]
obj . ViewObject . ShapeMaterial . EmissiveColor = props [ 4 ]
obj . ViewObject . LineWidth = props [ 5 ]
else :
obj . ViewObject . Visibility = props [ 0 ]
with Image . open ( self . image_file_name ) as img :
original_size = img . size