Merge pull request #408 from alibaba-damo-academy/hnluo-patch-1

zhifu gao

2023-04-24 331d57253ae25dd42c8e14930dee30cd8d2affa6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#include "precomp.h"
 
 
 
#ifdef __cplusplus
extern "C" {
#endif
 
 
 
 
 
    _RPPUNCAPI RPPUNC_HANDLE RapidPuncInit(const char* model_dir,int nThreadNum)
    {
        auto PuncOB =new CRapidPuncOnnx(model_dir, nThreadNum);
 
        return (void*)PuncOB;
 
    }
    _RPPUNCAPI void RapidPuncFinal(RPPUNC_HANDLE Handle)
    {
        if (Handle)
        {
            delete (CRapidPuncOnnx*)Handle;
            Handle = nullptr;
        }
 
    }
    _RPPUNCAPI RPPUNC_RESULT RapidPuncAddPunc(RPPUNC_HANDLE Handle, const char* szText)
    {
        if (!Handle)
            return nullptr;
        CRapidPuncOnnx* pPuncObj = (CRapidPuncOnnx*)Handle;
        auto Result = new string();
        (*Result) = pPuncObj->AddPunc(szText);
        return Result;
    }
 
    _RPPUNCAPI int RapidPuncGetResultLength(RPPUNC_RESULT Result)
    {
        if (!Result)
            return 0;
 
        return ((string*)Result)->size();
    }
    _RPPUNCAPI const char* RapidPuncGetResultText(RPPUNC_RESULT Result)
    {
        if (!Result)
            return 0;
 
        return ((string*)Result)->c_str();
    }
    _RPPUNCAPI void RapidPuncFree(RPPUNC_RESULT Result)
    {
 
        if (!Result)
            return;
 
        delete (string*)Result;
 
    }
 
 
 
 
 
 
 
#ifdef __cplusplus
}
#endif