cortex/export.h at main · fractal-solutions/cortex · 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
#ifndef EXPORT_H
#define EXPORT_H

#include "DQN.h"

// Define cross-platform export macros
#if defined(_WIN32) || defined(_WIN64)
    #define DLL_EXPORT __declspec(dllexport)
    #define DLL_CALL __stdcall
#else
    #define DLL_EXPORT __attribute__((visibility("default")))
    #define DLL_CALL
#endif

// Export functions for the DLL
extern "C" {

DLL_EXPORT DQN* DLL_CALL CreateDQN(int stateSize, int actionSize, int* hiddenLayers, int hiddenLayersSize);
DLL_EXPORT void DLL_CALL DestroyDQN(DQN* dqn);
DLL_EXPORT int DLL_CALL SelectAction(DQN* dqn, double* state, int stateSize, double epsilon);
DLL_EXPORT void DLL_CALL Train(DQN* dqn, double* state, int stateSize, int action, double reward, double* nextState, int nextStateSize, double gamma, double epsilonDecay);
DLL_EXPORT void DLL_CALL UpdateTargetNetwork(DQN* dqn);
DLL_EXPORT double DLL_CALL GetGamma(DQN* dqn);
DLL_EXPORT double DLL_CALL GetEpsilon(DQN* dqn);
DLL_EXPORT double DLL_CALL GetEpsilonDecay(DQN* dqn);
DLL_EXPORT double DLL_CALL GetQNetLoss(DQN* dqn);
DLL_EXPORT double DLL_CALL GetTargetNetLoss(DQN* dqn);
DLL_EXPORT int DLL_CALL SaveModel(DQN* dqn, const char* filepath);
DLL_EXPORT int DLL_CALL LoadModel(DQN* dqn, const char* filepath);

}

#endif // EXPORT_H