Wrap title text according to plot width (open-edge-platform#769)

* Wrap title text according to plot width

Signed-off-by: Kim, Vinnam <[email protected]>

* Update CHANGELOG.md

Signed-off-by: Kim, Vinnam <[email protected]>

Signed-off-by: Kim, Vinnam <[email protected]>

Author: vinnamkim

CHANGELOG.md

@@ -41,6 +41,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   (<https://github.com/openvinotoolkit/datumaro/pull/767>)
 
 ### Changed
+- Wrap title text according to its plot width
+  (<https://github.com/openvinotoolkit/datumaro/pull/769>)
 - Get list of subsets and support only Image media type in visualizer
   (<https://github.com/openvinotoolkit/datumaro/pull/768>)
 

datumaro/components/visualizer.py

@@ -231,7 +231,18 @@ def vis_one_sample(
         img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         ax.imshow(img)
 
-        ax.set_title(f"ID: {id}, Subset={subset}")
+        width = ax.transAxes.transform_point((1, 0))[0] - ax.transAxes.transform_point((0, 0))[0]
+        text = ax.set_title(f"ID: {id}, Subset: {subset}", loc="center", wrap=True)
+        text.__get_wrapped_text = text._get_wrapped_text
+
+        def _get_wrapped_text():
+            wrapped_text = text.__get_wrapped_text()
+            text._text = wrapped_text
+            return wrapped_text
+
+        text._get_wrapped_text = _get_wrapped_text
+        text._get_wrap_line_width = lambda: width
+
         ax.set_axis_off()
 
         if self.draw_only_image: