Dict_Coding/ormp.py at master · DomChennnn/Dict_Coding · GitHub

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
import os

import numpy

from utils import PROJECT_ROOT


class JavaORMP:
    def __init__(self, D, K, L):
        import jpype

        jvmPath = jpype.getDefaultJVMPath()  # the path of jvm.dll
        classpath = os.path.join(PROJECT_ROOT, "javaclasses")  # the path of PasswordCipher.class
        jvmArg = "-Djava.class.path=" + classpath
        if not jpype.isJVMStarted():  # test whether the JVM is started
            jpype.startJVM(jvmPath, jvmArg)  # start JVM

        SimpleMatrix = jpype.JClass("mpv2.SimpleMatrix")  # create the Java class
        MatchingPursuit = jpype.JClass("mpv2.MatchingPursuit")
        SymmetricMatrix = jpype.JClass("mpv2.SymmetricMatrix")

        jD = SimpleMatrix(D)
        if L == 1:
            self.jMP = MatchingPursuit(jD)
        else:
            jDD = SymmetricMatrix(K)
            jDD.eqInnerProductMatrix(jD)
            self.jMP = MatchingPursuit(jD, jDD)

        print("java ormp", jD.getN(), jDD.getN())

    def apply(self, ad, i, d1):
        return self.jMP.vsORMP(ad, i, d1)


class CppORMP:
    def __init__(self, D, K, L):
        import mpv2

        jD = mpv2.SimpleMatrix(mpv2.DoubleVectors.from_np_array(D))
        if L == 1:
            self.jMP = mpv2.MatchingPursuit(jD)
        else:
            jDD = mpv2.SymmetricMatrix(K)
            jDD.eqInnerProductMatrix(jD)

            self.jMP = mpv2.MatchingPursuit(jD, jDD)

        print("cpp ormp", jD.getN(), jDD.getN())

    def apply(self, ad, i, d1):
        import mpv2

        return numpy.array(self.jMP.vsORMP(mpv2.DoubleVector.from_np_array(ad), int(i), float(d1)))


def get_provider():
    try:
        import mpv2

        return CppORMP
    except:
        pass

    return JavaORMP