0016-add-C-API-for-mtmd_input_text.patch 1.45 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
From: Gabe Goodhart <ghart@us.ibm.com>
Date: Tue, 24 Jun 2025 16:55:31 -0600
Subject: [PATCH] add C API for mtmd_input_text

Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
---
 tools/mtmd/mtmd.cpp | 10 ++++++++++
 tools/mtmd/mtmd.h   |  3 +++
 2 files changed, 13 insertions(+)

diff --git a/tools/mtmd/mtmd.cpp b/tools/mtmd/mtmd.cpp
13
index d06fa42e6..0f5712e21 100644
14
15
--- a/tools/mtmd/mtmd.cpp
+++ b/tools/mtmd/mtmd.cpp
Daniel Hiltgen's avatar
Daniel Hiltgen committed
16
@@ -87,6 +87,16 @@ enum mtmd_slice_tmpl {
17
     MTMD_SLICE_TMPL_IDEFICS3,
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 };
 
+mtmd_input_text* mtmd_input_text_init(const char * text, bool add_special, bool parse_special) {
+    return new mtmd_input_text{text, add_special, parse_special};
+}
+
+void mtmd_input_text_free(mtmd_input_text* input_text) {
+    if (input_text) {
+        delete input_text;
+    }
+}
+
 const char * mtmd_default_marker() {
     return "<__media__>";
 }
diff --git a/tools/mtmd/mtmd.h b/tools/mtmd/mtmd.h
34
index b3df24c29..a6a1af3b8 100644
35
36
37
38
39
40
41
42
43
44
45
46
--- a/tools/mtmd/mtmd.h
+++ b/tools/mtmd/mtmd.h
@@ -75,6 +75,9 @@ typedef struct mtmd_input_chunk  mtmd_input_chunk;
 typedef struct mtmd_input_chunks mtmd_input_chunks;
 typedef struct mtmd_input_text   mtmd_input_text;
 
+MTMD_API mtmd_input_text* mtmd_input_text_init(const char * text, bool add_special, bool parse_special);
+MTMD_API void mtmd_input_text_free(mtmd_input_text* input_text);
+
 struct mtmd_context_params {
     bool use_gpu;
     bool print_timings;