path: root/tools/tnet_to_nerv.cpp

#include <cstdio>
#include <fstream>
#include <string>
#include <cstring>
char token[1024];
char output[1024];
double mat[4096][4096];
int main() {
    std::ofstream fout;
    fout.open("converted.nerv");
    int cnt = 0;
    while (scanf("%s", token) != EOF)
    {
        int nrow, ncol;
        int i, j;
        if (strcmp(token, "<biasedlinearity>") == 0)
        {
            scanf("%d %d", &ncol, &nrow);
            scanf("%s %d %d", token, &ncol, &nrow);
            printf("%d %d\n", nrow, ncol);
            for (j = 0; j < ncol; j++)
                for (i = 0; i < nrow; i++)
                    scanf("%lf", mat[i] + j);
            long base = fout.tellp();
            sprintf(output, "%16d", 0);
            fout << output;
            sprintf(output, "{type=\"nerv.LinearTransParam\",id=\"affine%d_ltp\"}\n",
                            cnt);
            fout << output;
            sprintf(output, "%d %d\n", nrow, ncol);
            fout << output;
            for (i = 0; i < nrow; i++)
            {
                for (j = 0; j < ncol; j++)
                    fout << mat[i][j] << " ";
                fout << std::endl;
            }
            long length = fout.tellp() - base;
            fout.seekp(base);
            sprintf(output, "[%13lu]\n", length);
            fout << output;
            fout.seekp(0, std::ios_base::end);
            if (scanf("%s %d", token, &ncol) == 2 && *token == 'v')
            {
                base = fout.tellp();
                for (j = 0; j < ncol; j++)
                    scanf("%lf", mat[0] + j);
                sprintf(output, "%16d", 0);
                fout << output;
                sprintf(output, "{type=\"nerv.BiasParam\",id=\"affine%d_bp\"}\n",
                        cnt);
                fout << output;
                sprintf(output, "1 %d\n", ncol);
                fout << output;
                for (j = 0; j < ncol; j++)
                    fout << mat[0][j] << " ";
                fout << std::endl;
                length = fout.tellp() - base;
                fout.seekp(base);
                sprintf(output, "[%13lu]\n", length);
                fout << output;
                fout.seekp(0, std::ios_base::end);
                cnt++;
            }
        }
    }
    return 0;
}