def __init__(self, doc: 'Drawing' = None):
self.doc = doc
self.image_key_generator = ImageKeyGenerator()
self.underlay_key_generator = UnderlayKeyGenerator()
def __init__(self, drawing: 'Drawing'):
super(ModernDXFFactory, self).__init__(drawing)
self.ENTITY_WRAPPERS.update(UPDATE_ENTITY_WRAPPERS)
self.image_key_generator = ImageKeyGenerator()
self.underlay_key_generator = UnderlayKeyGenerator()
class ModernDXFFactory(LegacyDXFFactory):
"""
DXf factory for DXF version AC1015 and later.
"""
DEFAULT_WRAPPER = graphics.ModernGraphicEntity
def __init__(self, drawing: 'Drawing'):
super(ModernDXFFactory, self).__init__(drawing)
self.ENTITY_WRAPPERS.update(UPDATE_ENTITY_WRAPPERS)
self.image_key_generator = ImageKeyGenerator()
self.underlay_key_generator = UnderlayKeyGenerator()
@property
def rootdict(self) -> dxfdict.DXFDictionary:
return self.drawing.rootdict
@property
def block_records(self) -> 'Table':
return self.drawing.sections.tables.block_records
def create_block_entry_in_block_records_table(
self, block_layout: BlockLayout) -> None:
# required for DXFVERSION > ac1009: Entry in the BLOCK_RECORDS section
block_record = self.block_records.new(block_layout.name)
block_layout.set_block_record_handle(block_record.dxf.handle)
def get_layouts(self) -> Layouts:
return Layouts(self.drawing)
def new_block_layout(self, block_handle: str,
endblk_handle: str) -> BlockLayout:
# Warning: Do not call create_block_entry_in_block_records_table() from this point, this will not work!
return BlockLayout(self.entitydb, self, block_handle, endblk_handle)
def copy_layout(self, source_entity: 'DXFEntity',
target_entity: 'DXFEntity'):
"""
Place `target_entity` in same layout as `source_entity`
"""
target_entity.dxf.paperspace = source_entity.dxf.paperspace
target_entity.dxf.owner = source_entity.dxf.owner
def next_image_key(self, checkfunc=lambda k: True) -> str:
while True:
key = self.image_key_generator.next()
if checkfunc(key):
return key
def next_underlay_key(self, checkfunc=lambda k: True) -> str:
while True:
key = self.underlay_key_generator.next()
if checkfunc(key):
return key
def get_layout_for_entity(
self, entity: 'DXFEntity') -> Optional['GenericLayoutType']:
if entity.dxf.owner not in self.entitydb:
return None
dwg = self.drawing
try:
layout = dwg.layouts.get_layout_for_entity(entity)
except DXFKeyError:
block_rec = dwg.dxffactory.wrap_handle(entity.dxf.owner)
block_name = block_rec.dxf.name
layout = dwg.blocks.get(block_name)
return layout
def __init__(self, doc: "Drawing", entities: Iterable["DXFObject"] = None):
self.doc = doc
self.underlay_key_generator = UnderlayKeyGenerator()
self._entity_space = EntitySpace()
if entities is not None:
self._build(iter(entities))
class ObjectsSection:
def __init__(self, doc: "Drawing", entities: Iterable["DXFObject"] = None):
self.doc = doc
self.underlay_key_generator = UnderlayKeyGenerator()
self._entity_space = EntitySpace()
if entities is not None:
self._build(iter(entities))
@property
def entitydb(self) -> "EntityDB":
"""Returns drawing entity database. (internal API)"""
return self.doc.entitydb
def get_entity_space(self) -> "EntitySpace":
"""Returns entity space. (internal API)"""
return self._entity_space
def next_underlay_key(self, checkfunc=lambda k: True) -> str:
while True:
key = self.underlay_key_generator.next()
if checkfunc(key):
return key
def _build(self, entities: Iterator["DXFObject"]) -> None:
section_head = cast("DXFTagStorage", next(entities))
if section_head.dxftype() != "SECTION" or section_head.base_class[
1
] != (2, "OBJECTS"):
raise const.DXFStructureError(
"Critical structure error in the OBJECTS section."
)
for entity in entities:
# No check for valid entities here:
# Use the audit- or the recover module to fix invalid DXF files!
self._entity_space.add(entity)
def export_dxf(self, tagwriter: "TagWriter") -> None:
"""Export DXF entity by `tagwriter`. (internal API)"""
tagwriter.write_str(" 0\nSECTION\n 2\nOBJECTS\n")
self._entity_space.export_dxf(tagwriter)
tagwriter.write_tag2(0, "ENDSEC")
def new_entity(self, _type: str, dxfattribs: dict) -> "DXFObject":
"""Create new DXF object, add it to the entity database and to the
entity space.
Args:
_type: DXF type like `DICTIONARY`
dxfattribs: DXF attributes as dict
(internal API)
"""
dxf_entity = factory.create_db_entry(_type, dxfattribs, self.doc)
self._entity_space.add(dxf_entity)
return dxf_entity # type: ignore
def delete_entity(self, entity: "DXFObject") -> None:
"""Remove `entity` from entity space and destroy object. (internal API)"""
self._entity_space.remove(entity)
self.entitydb.delete_entity(entity)
def delete_all_entities(self) -> None:
"""Delete all DXF objects. (internal API)"""
db = self.entitydb
for entity in self._entity_space:
db.delete_entity(entity)
self._entity_space.clear()
def setup_rootdict(self) -> Dictionary:
"""Create a root dictionary. Has to be the first object in the objects
section. (internal API)"""
if len(self):
raise const.DXFStructureError(
"Can not create root dictionary in none empty objects section."
)
logger.debug("Creating ROOT dictionary.")
# root directory has no owner
return self.add_dictionary(owner="0", hard_owned=False)
def setup_object_management_tables(self, rootdict: Dictionary) -> None:
"""Setup required management tables. (internal API)"""
def setup_plot_style_name_table():
plot_style_name_dict = self.add_dictionary_with_default(
owner=rootdict.dxf.handle,
hard_owned=False,
)
placeholder = self.add_placeholder(
owner=plot_style_name_dict.dxf.handle
)
plot_style_name_dict.set_default(placeholder)
plot_style_name_dict["Normal"] = placeholder
rootdict["ACAD_PLOTSTYLENAME"] = plot_style_name_dict
def restore_table_handle(table_name: str, handle: str) -> None:
# The original object table does not exist, but the handle is maybe
# used by some DXF entities. Try to restore the original table
# handle.
new_table = rootdict.get(table_name)
if isinstance(
new_table, Dictionary
) and self.doc.entitydb.reset_handle(new_table, handle):
logger.debug(f"reset handle of table {table_name} to #{handle}")
# check required object tables:
for name in _OBJECT_TABLE_NAMES:
table: Union[Dictionary, str, None] = rootdict.get(name) # type: ignore
# Dictionary: existing table object
# str: handle of not existing table object
# None: no entry in the rootdict for "name"
if isinstance(table, Dictionary):
continue # skip existing tables
logger.info(f"creating {name} dictionary")
if name == "ACAD_PLOTSTYLENAME":
setup_plot_style_name_table()
else:
rootdict.add_new_dict(name, hard_owned=False)
if isinstance(table, str) and validator.is_handle(table):
restore_table_handle(name, handle=table)
def add_object(self, entity: "DXFObject") -> None:
"""Add `entity` to OBJECTS section. (internal API)"""
if is_dxf_object(entity):
self._entity_space.add(entity)
else:
raise const.DXFTypeError(
f"invalid DXF type {entity.dxftype()} for OBJECTS section"
)
def add_dxf_object_with_reactor(
self, dxftype: str, dxfattribs: dict
) -> "DXFObject":
"""Add DXF object with reactor. (internal API)"""
dxfobject = self.new_entity(dxftype, dxfattribs)
dxfobject.set_reactors([dxfattribs["owner"]])
return dxfobject
def purge(self):
self._entity_space.purge()
# start of public interface
@property
def rootdict(self) -> Dictionary:
"""Returns the root DICTIONARY, or as AutoCAD calls it:
the named DICTIONARY.
"""
if len(self):
return self._entity_space[0] # type: ignore
else:
return self.setup_rootdict()
def __len__(self) -> int:
"""Returns count of DXF objects."""
return len(self._entity_space)
def __iter__(self):
"""Returns iterable of all DXF objects in the OBJECTS section."""
return iter(self._entity_space)
def __getitem__(self, index) -> "DXFObject":
"""Get entity at `index`.
The underlying data structure for storing DXF objects is organized like
a standard Python list, therefore `index` can be any valid list indexing
or slicing term, like a single index ``objects[-1]`` to get the last
entity, or an index slice ``objects[:10]`` to get the first 10 or less
objects as ``List[DXFObject]``.
"""
return self._entity_space[index] # type: ignore
def __contains__(self, entity):
"""Returns ``True`` if `entity` stored in OBJECTS section.
Args:
entity: :class:`DXFObject` or handle as hex string
"""
if isinstance(entity, str):
try:
entity = self.entitydb[entity]
except KeyError:
return False
return entity in self._entity_space
def query(self, query: str = "*") -> EntityQuery:
"""Get all DXF objects matching the :ref:`entity query string`."""
return EntityQuery(iter(self), query)
def audit(self, auditor: Auditor) -> None:
"""Audit and repair OBJECTS section.
.. important::
Do not delete entities while auditing process, because this
would alter the entity database while iterating, instead use::
auditor.trash(entity)
to delete invalid entities after auditing automatically.
"""
assert self.doc is auditor.doc, "Auditor for different DXF document."
for entity in self._entity_space:
if not is_dxf_object(entity):
auditor.fixed_error(
code=AuditError.REMOVED_INVALID_DXF_OBJECT,
message=f"Removed invalid DXF entity {str(entity)} "
f"from OBJECTS section.",
)
auditor.trash(entity)
def add_dictionary(
self, owner: str = "0", hard_owned: bool = True
) -> Dictionary:
"""Add new :class:`~ezdxf.entities.Dictionary` object.
Args:
owner: handle to owner as hex string.
hard_owned: ``True`` to treat entries as hard owned.
"""
entity = self.new_entity(
"DICTIONARY",
dxfattribs={
"owner": owner,
"hard_owned": hard_owned,
},
)
return cast(Dictionary, entity)
def add_dictionary_with_default(
self, owner="0", default="0", hard_owned: bool = True
) -> "DictionaryWithDefault":
"""Add new :class:`~ezdxf.entities.DictionaryWithDefault` object.
Args:
owner: handle to owner as hex string.
default: handle to default entry.
hard_owned: ``True`` to treat entries as hard owned.
"""
entity = self.new_entity(
"ACDBDICTIONARYWDFLT",
dxfattribs={
"owner": owner,
"default": default,
"hard_owned": hard_owned,
},
)
return cast("DictionaryWithDefault", entity)
def add_dictionary_var(
self, owner: str = "0", value: str = ""
) -> "DictionaryVar":
"""Add a new :class:`~ezdxf.entities.DictionaryVar` object.
Args:
owner: handle to owner as hex string.
value: value as string
"""
return self.new_entity( # type: ignore
"DICTIONARYVAR", dxfattribs={"owner": owner, "value": value}
)
def add_xrecord(self, owner: str = "0") -> "XRecord":
"""Add a new :class:`~ezdxf.entities.XRecord` object.
Args:
owner: handle to owner as hex string.
"""
return self.new_entity( # type: ignore
"XRECORD", dxfattribs={"owner": owner}
)
def add_placeholder(self, owner: str = "0") -> "Placeholder":
"""Add a new :class:`~ezdxf.entities.Placeholder` object.
Args:
owner: handle to owner as hex string.
"""
return self.new_entity( # type: ignore
"ACDBPLACEHOLDER", dxfattribs={"owner": owner}
)
# end of public interface
def set_raster_variables(
self, frame: int = 0, quality: int = 1, units: str = "m"
) -> None:
"""Set raster variables.
Args:
frame: ``0`` = do not show image frame; ``1`` = show image frame
quality: ``0`` = draft; ``1`` = high
units: units for inserting images. This defines the real world unit for one drawing unit for the purpose of
inserting and scaling images with an associated resolution.
===== ===========================
mm Millimeter
cm Centimeter
m Meter (ezdxf default)
km Kilometer
in Inch
ft Foot
yd Yard
mi Mile
===== ===========================
(internal API), public interface :meth:`~ezdxf.drawing.Drawing.set_raster_variables`
"""
units_: int = const.RASTER_UNITS.get(units, 0)
try:
raster_vars = self.rootdict["ACAD_IMAGE_VARS"]
except const.DXFKeyError:
raster_vars = self.add_dxf_object_with_reactor(
"RASTERVARIABLES",
dxfattribs={
"owner": self.rootdict.dxf.handle,
"frame": frame,
"quality": quality,
"units": units_,
},
)
self.rootdict["ACAD_IMAGE_VARS"] = raster_vars
else:
raster_vars.dxf.frame = frame
raster_vars.dxf.quality = quality
raster_vars.dxf.units = units_
def set_wipeout_variables(self, frame: int = 0) -> None:
"""Set wipeout variables.
Args:
frame: ``0`` = do not show image frame; ``1`` = show image frame
(internal API)
"""
try:
wipeout_vars = self.rootdict["ACAD_WIPEOUT_VARS"]
except const.DXFKeyError:
wipeout_vars = self.add_dxf_object_with_reactor(
"WIPEOUTVARIABLES",
dxfattribs={
"owner": self.rootdict.dxf.handle,
"frame": int(frame),
},
)
self.rootdict["ACAD_WIPEOUT_VARS"] = wipeout_vars
else:
wipeout_vars.dxf.frame = int(frame)
def add_image_def(
self, filename: str, size_in_pixel: Tuple[int, int], name=None
) -> "ImageDef":
"""Add an image definition to the objects section.
Add an :class:`~ezdxf.entities.image.ImageDef` entity to the drawing
(objects section). `filename` is the image file name as relative or
absolute path and `size_in_pixel` is the image size in pixel as (x, y)
tuple. To avoid dependencies to external packages, `ezdxf` can not
determine the image size by itself. Returns a :class:`~ezdxf.entities.image.ImageDef`
entity which is needed to create an image reference. `name` is the
internal image name, if set to ``None``, name is auto-generated.
Absolute image paths works best for AutoCAD but not really good, you
have to update external references manually in AutoCAD, which is not
possible in TrueView. If the drawing units differ from 1 meter, you also
have to use: :meth:`set_raster_variables`.
Args:
filename: image file name (absolute path works best for AutoCAD)
size_in_pixel: image size in pixel as (x, y) tuple
name: image name for internal use, None for using filename as name
(best for AutoCAD)
"""
if name is None:
name = filename
image_dict = self.rootdict.get_required_dict("ACAD_IMAGE_DICT")
image_def = self.add_dxf_object_with_reactor(
"IMAGEDEF",
dxfattribs={
"owner": image_dict.dxf.handle,
"filename": filename,
"image_size": size_in_pixel,
},
)
image_dict[name] = image_def.dxf.handle
return cast("ImageDef", image_def)
def add_image_def_reactor(self, image_handle: str) -> "ImageDefReactor":
"""Add required IMAGEDEF_REACTOR object for IMAGEDEF object.
(internal API)
"""
image_def_reactor = self.new_entity(
"IMAGEDEF_REACTOR",
dxfattribs={
"owner": image_handle,
"image_handle": image_handle,
},
)
return cast("ImageDefReactor", image_def_reactor)
def add_underlay_def(
self, filename: str, fmt: str = "pdf", name: str = None
) -> "UnderlayDefinition":
"""Add an :class:`~ezdxf.entities.underlay.UnderlayDefinition` entity
to the drawing (OBJECTS section). `filename` is the underlay file name
as relative or absolute path and `fmt` as string (pdf, dwf, dgn).
The underlay definition is required to create an underlay reference.
Args:
filename: underlay file name
fmt: file format as string ``'pdf'|'dwf'|'dgn'``
name: pdf format = page number to display; dgn format = ``'default'``; dwf: ????
"""
fmt = fmt.upper()
if fmt in ("PDF", "DWF", "DGN"):
underlay_dict_name = f"ACAD_{fmt}DEFINITIONS"
underlay_def_entity = f"{fmt}DEFINITION"
else:
raise const.DXFValueError(f"Unsupported file format: '{fmt}'")
if name is None:
if fmt == "PDF":
name = "1" # Display first page by default
elif fmt == "DGN":
name = "default"
else:
name = "Model" # Display model space for DWF ???
underlay_dict = self.rootdict.get_required_dict(underlay_dict_name)
underlay_def = self.new_entity(
underlay_def_entity,
dxfattribs={
"owner": underlay_dict.dxf.handle,
"filename": filename,
"name": name,
},
)
# auto-generated underlay key
key = self.next_underlay_key(lambda k: k not in underlay_dict)
underlay_dict[key] = underlay_def.dxf.handle
return cast("UnderlayDefinition", underlay_def)
def add_geodata(
self, owner: str = "0", dxfattribs: dict = None
) -> "GeoData":
"""Creates a new :class:`GeoData` entity and replaces existing ones.
The GEODATA entity resides in the OBJECTS section and NOT in the layout
entity space and it is linked to the layout by an extension dictionary
located in BLOCK_RECORD of the layout.
The GEODATA entity requires DXF version R2010+. The DXF Reference does
not document if other layouts than model space supports geo referencing,
so getting/setting geo data may only make sense for the model space
layout, but it is also available in paper space layouts.
Args:
owner: handle to owner as hex string
dxfattribs: DXF attributes for :class:`~ezdxf.entities.GeoData` entity
"""
if dxfattribs is None:
dxfattribs = {}
dxfattribs["owner"] = owner
return cast(
"GeoData", self.add_dxf_object_with_reactor("GEODATA", dxfattribs)
)
class ObjectsSection:
def __init__(self, doc: 'Drawing', entities: Iterable['DXFObject'] = None):
self.doc = doc
self.underlay_key_generator = UnderlayKeyGenerator()
self._entity_space = EntitySpace()
if entities is not None:
self._build(iter(entities))
@property
def entitydb(self) -> 'EntityDB':
""" Returns drawing entity database. (internal API)"""
return self.doc.entitydb
def get_entity_space(self) -> 'EntitySpace':
""" Returns entity space. (internal API) """
return self._entity_space
def next_underlay_key(self, checkfunc=lambda k: True) -> str:
while True:
key = self.underlay_key_generator.next()
if checkfunc(key):
return key
def _build(self, entities: Iterator['DXFObject']) -> None:
section_head = next(entities) # type: DXFTagStorage
if section_head.dxftype(
) != 'SECTION' or section_head.base_class[1] != (2, 'OBJECTS'):
raise DXFStructureError(
"Critical structure error in 'OBJECTS' section.")
for entity in entities:
self._entity_space.add(entity)
def export_dxf(self, tagwriter: 'TagWriter') -> None:
""" Export DXF entity by `tagwriter`. (internal API) """
tagwriter.write_str(" 0\nSECTION\n 2\nOBJECTS\n")
self._entity_space.export_dxf(tagwriter)
tagwriter.write_tag2(0, "ENDSEC")
def new_entity(self, _type: str, dxfattribs: dict) -> 'DXFObject':
""" Create new DXF object, add it to the entity database and to the entity space.
Args:
_type: DXF type like `DICTIONARY`
dxfattribs: DXF attributes as dict
(internal API)
"""
dxf_entity = factory.create_db_entry(_type, dxfattribs, self.doc)
self._entity_space.add(dxf_entity)
return dxf_entity
def delete_entity(self, entity: 'DXFObject') -> None:
""" Remove `entity` from entity space and destroy object. (internal API) """
self._entity_space.remove(entity)
self.entitydb.delete_entity(entity)
def delete_all_entities(self) -> None:
""" Delete all DXF objects. (internal API) """
db = self.entitydb
for entity in self._entity_space:
db.delete_entity(entity)
self._entity_space.clear()
def setup_rootdict(self) -> Dictionary:
""" Create a root dictionary. Has to be the first object in the objects section. (internal API) """
if len(self):
raise DXFStructureError(
"Can not create root dictionary in none empty objects section."
)
logger.debug('Creating ROOT dictionary.')
# root directory has no owner
return self.add_dictionary(owner='0')
def setup_objects_management_tables(self, rootdict: Dictionary) -> None:
""" Setup required management tables. (internal API)"""
def setup_plot_style_name_table():
plot_style_name_dict = self.add_dictionary_with_default(
owner=rootdict.dxf.handle)
placeholder = self.add_placeholder(
owner=plot_style_name_dict.dxf.handle)
plot_style_name_dict.set_default(placeholder)
plot_style_name_dict['Normal'] = placeholder
rootdict['ACAD_PLOTSTYLENAME'] = plot_style_name_dict
for name in _OBJECT_TABLE_NAMES:
if name in rootdict:
continue # just create not existing tables
logger.info('creating {} dictionary'.format(name))
if name == "ACAD_PLOTSTYLENAME":
setup_plot_style_name_table()
else:
rootdict.add_new_dict(name)
def add_object(self, entity: 'DXFObject') -> None:
""" Add `entity` to OBJECTS section. (internal API) """
self._entity_space.add(entity)
def add_dxf_object_with_reactor(self, dxftype: str,
dxfattribs: dict) -> 'DXFObject':
""" Add DXF object with reactor. (internal API) """
dxfobject = self.new_entity(dxftype, dxfattribs)
dxfobject.set_reactors([dxfattribs['owner']])
return dxfobject
def purge(self):
self._entity_space.purge()
# start of public interface
@property
def rootdict(self) -> Dictionary:
""" Root dictionary. """
if len(self):
return self._entity_space[0]
else:
return self.setup_rootdict()
def __len__(self) -> int:
""" Returns count of DXF objects. """
return len(self._entity_space)
def __iter__(self) -> Iterable['DXFObject']:
""" Returns iterable of all DXF objects in the OBJECTS section. """
return iter(self._entity_space)
def __getitem__(self, index) -> 'DXFObject':
""" Get entity at `index`.
The underlying data structure for storing DXF objects is organized like a standard Python list, therefore `index`
can be any valid list indexing or slicing term, like a single index ``objects[-1]`` to get the last entity, or
an index slice ``objects[:10]`` to get the first 10 or less objects as ``List[DXFObject]``.
"""
return self._entity_space[index] # type: DXFObject
def __contains__(self, entity: Union['DXFObject', str]) -> bool:
""" Returns ``True`` if `entity` stored in OBJECTS section.
Args:
entity: :class:`DXFObject` or handle as hex string
"""
if isinstance(entity, str):
try:
entity = self.entitydb[entity]
except KeyError:
return False
return entity in self._entity_space
def query(self, query: str = '*') -> EntityQuery:
"""
Get all DXF objects matching the :ref:`entity query string`.
"""
return EntityQuery(iter(self), query)
def add_dictionary(self,
owner: str = '0',
hard_owned: bool = False) -> Dictionary:
""" Add new :class:`~ezdxf.entities.Dictionary` object.
Args:
owner: handle to owner as hex string.
hard_owned: ``True`` to treat entries as hard owned.
"""
entity = self.new_entity('DICTIONARY',
dxfattribs={
'owner': owner,
'hard_owned': hard_owned,
})
return cast(Dictionary, entity)
def add_dictionary_with_default(
self,
owner='0',
default='0',
hard_owned: bool = False) -> 'DictionaryWithDefault':
""" Add new :class:`~ezdxf.entities.DictionaryWithDefault` object.
Args:
owner: handle to owner as hex string.
default: handle to default entry.
hard_owned: ``True`` to treat entries as hard owned.
"""
entity = self.new_entity('ACDBDICTIONARYWDFLT',
dxfattribs={
'owner': owner,
'default': default,
'hard_owned': hard_owned,
})
return cast('DictionaryWithDefault', entity)
def add_dictionary_var(self,
owner: str = '0',
value: str = '') -> 'DictionaryVar':
"""
Add a new :class:`~ezdxf.entities.DictionaryVar` object.
Args:
owner: handle to owner as hex string.
value: value as string
"""
return self.new_entity('DICTIONARYVAR',
dxfattribs={
'owner': owner,
'value': value
})
def add_xrecord(self, owner: str = '0') -> 'XRecord':
"""
Add a new :class:`~ezdxf.entities.XRecord` object.
Args:
owner: handle to owner as hex string.
"""
return self.new_entity('XRECORD', dxfattribs={'owner': owner})
def add_placeholder(self, owner: str = '0') -> 'Placeholder':
"""
Add a new :class:`~ezdxf.entities.Placeholder` object.
Args:
owner: handle to owner as hex string.
"""
return self.new_entity('ACDBPLACEHOLDER', dxfattribs={'owner': owner})
# end of public interface
def set_raster_variables(self,
frame: int = 0,
quality: int = 1,
units: str = 'm') -> None:
"""
Set raster variables.
Args:
frame: ``0`` = do not show image frame; ``1`` = show image frame
quality: ``0`` = draft; ``1`` = high
units: units for inserting images. This defines the real world unit for one drawing unit for the purpose of
inserting and scaling images with an associated resolution.
===== ===========================
mm Millimeter
cm Centimeter
m Meter (ezdxf default)
km Kilometer
in Inch
ft Foot
yd Yard
mi Mile
===== ===========================
(internal API), public interface :meth:`~ezdxf.drawing.Drawing.set_raster_variables`
"""
units = RASTER_UNITS.get(units, 0)
try:
raster_vars = self.rootdict['ACAD_IMAGE_VARS']
except DXFKeyError:
raster_vars = self.add_dxf_object_with_reactor(
'RASTERVARIABLES',
dxfattribs={
'owner': self.rootdict.dxf.handle,
'frame': frame,
'quality': quality,
'units': units,
})
self.rootdict['ACAD_IMAGE_VARS'] = raster_vars
else:
raster_vars.dxf.frame = frame
raster_vars.dxf.quality = quality
raster_vars.dxf.units = units
def set_wipeout_variables(self, frame: int = 0) -> None:
"""
Set wipeout variables.
Args:
frame: ``0`` = do not show image frame; ``1`` = show image frame
(internal API), public interface :meth:`~ezdxf.drawing.Drawing.set_wipeout_variables`
"""
try:
wipeout_vars = self.rootdict['ACAD_WIPEOUT_VARS']
except DXFKeyError:
wipeout_vars = self.add_dxf_object_with_reactor(
'WIPEOUTVARIABLES',
dxfattribs={
'owner': self.rootdict.dxf.handle,
'frame': int(frame),
})
self.rootdict['ACAD_WIPEOUT_VARS'] = wipeout_vars
else:
wipeout_vars.dxf.frame = int(frame)
def add_image_def(self,
filename: str,
size_in_pixel: Tuple[int, int],
name=None) -> 'ImageDef':
"""
Add an image definition to the objects section.
Add an :class:`~ezdxf.entities.image.ImageDef` entity to the drawing (objects section). `filename` is the image
file name as relative or absolute path and `size_in_pixel` is the image size in pixel as (x, y) tuple. To avoid
dependencies to external packages, `ezdxf` can not determine the image size by itself. Returns a
:class:`~ezdxf.entities.image.ImageDef` entity which is needed to create an image reference. `name` is the
internal image name, if set to ``None``, name is auto-generated.
Absolute image paths works best for AutoCAD but not really good, you have to update external references manually
in AutoCAD, which is not possible in TrueView. If the drawing units differ from 1 meter, you also have to use:
:meth:`set_raster_variables`.
Args:
filename: image file name (absolute path works best for AutoCAD)
size_in_pixel: image size in pixel as (x, y) tuple
name: image name for internal use, None for using filename as name (best for AutoCAD)
"""
# removed auto-generated name
# use absolute image paths for filename and AutoCAD loads images automatically
if name is None:
name = filename
image_dict = self.rootdict.get_required_dict('ACAD_IMAGE_DICT')
image_def = self.add_dxf_object_with_reactor('IMAGEDEF',
dxfattribs={
'owner':
image_dict.dxf.handle,
'filename':
filename,
'image_size':
size_in_pixel,
})
image_dict[name] = image_def.dxf.handle
return cast('ImageDef', image_def)
def add_image_def_reactor(self, image_handle: str) -> 'ImageDefReactor':
""" Add required IMAGEDEF_REACTOR object for IMAGEDEF object. (internal API) """
image_def_reactor = self.new_entity('IMAGEDEF_REACTOR',
dxfattribs={
'owner': image_handle,
'image_handle': image_handle,
})
return cast('ImageDefReactor', image_def_reactor)
def add_underlay_def(self,
filename: str,
format: str = 'pdf',
name: str = None) -> 'UnderlayDef':
"""
Add an :class:`~ezdxf.entities.underlay.UnderlayDef` entity to the drawing (OBJECTS section).
`filename` is the underlay file name as relative or absolute path and `format` as string (pdf, dwf, dgn).
The underlay definition is required to create an underlay reference.
Args:
filename: underlay file name
format: file format as string ``'pdf'|'dwf'|'dgn'`` or ``'ext'`` for getting file format from filename extension
name: pdf format = page number to display; dgn format = ``'default'``; dwf: ????
"""
fmt = format.upper()
if fmt in ('PDF', 'DWF', 'DGN'):
underlay_dict_name = 'ACAD_{}DEFINITIONS'.format(fmt)
underlay_def_entity = "{}DEFINITION".format(fmt)
else:
raise DXFValueError("Unsupported file format: '{}'".format(fmt))
if name is None:
if fmt == 'PDF':
name = '1' # Display first page by default
elif fmt == 'DGN':
name = 'default'
else:
name = 'Model' # Display model space for DWF ???
underlay_dict = self.rootdict.get_required_dict(underlay_dict_name)
underlay_def = self.new_entity(underlay_def_entity,
dxfattribs={
'owner': underlay_dict.dxf.handle,
'filename': filename,
'name': name,
})
# auto-generated underlay key
key = self.next_underlay_key(lambda k: k not in underlay_dict)
underlay_dict[key] = underlay_def.dxf.handle
return cast('UnderlayDef', underlay_def)
def add_geodata(self,
owner: str = '0',
dxfattribs: dict = None) -> 'GeoData':
"""
Creates a new :class:`GeoData` entity and replaces existing ones. The GEODATA entity resides in the OBJECTS section
and NOT in the layout entity space and it is linked to the layout by an extension dictionary located in BLOCK_RECORD
of the layout.
The GEODATA entity requires DXF version R2010+. The DXF Reference does not document if other layouts than model
space supports geo referencing, so getting/setting geo data may only make sense for the model space layout, but
it is also available in paper space layouts.
Args:
owner: handle to owner as hex string
dxfattribs: DXF attributes for :class:`~ezdxf.entities.GeoData` entity
"""
if dxfattribs is None:
dxfattribs = {}
dxfattribs['owner'] = owner
return cast('GeoData',
self.add_dxf_object_with_reactor('GEODATA', dxfattribs))