1*a58d3d2aSXin Li /* Copyright (c) 2023 Amazon */
2*a58d3d2aSXin Li /*
3*a58d3d2aSXin Li Redistribution and use in source and binary forms, with or without
4*a58d3d2aSXin Li modification, are permitted provided that the following conditions
5*a58d3d2aSXin Li are met:
6*a58d3d2aSXin Li
7*a58d3d2aSXin Li - Redistributions of source code must retain the above copyright
8*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer.
9*a58d3d2aSXin Li
10*a58d3d2aSXin Li - Redistributions in binary form must reproduce the above copyright
11*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer in the
12*a58d3d2aSXin Li documentation and/or other materials provided with the distribution.
13*a58d3d2aSXin Li
14*a58d3d2aSXin Li THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
15*a58d3d2aSXin Li ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
16*a58d3d2aSXin Li LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
17*a58d3d2aSXin Li A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
18*a58d3d2aSXin Li CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19*a58d3d2aSXin Li EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20*a58d3d2aSXin Li PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21*a58d3d2aSXin Li PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
22*a58d3d2aSXin Li LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
23*a58d3d2aSXin Li NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24*a58d3d2aSXin Li SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25*a58d3d2aSXin Li */
26*a58d3d2aSXin Li
27*a58d3d2aSXin Li #ifdef HAVE_CONFIG_H
28*a58d3d2aSXin Li #include "config.h"
29*a58d3d2aSXin Li #endif
30*a58d3d2aSXin Li
31*a58d3d2aSXin Li #include <stdio.h>
32*a58d3d2aSXin Li #include <string.h>
33*a58d3d2aSXin Li #include <stddef.h>
34*a58d3d2aSXin Li #include "nnet.h"
35*a58d3d2aSXin Li #include "os_support.h"
36*a58d3d2aSXin Li #include "arch.h"
37*a58d3d2aSXin Li
38*a58d3d2aSXin Li /* This is a bit of a hack because we need to build nnet_data.c and plc_data.c without USE_WEIGHTS_FILE,
39*a58d3d2aSXin Li but USE_WEIGHTS_FILE is defined in config.h. */
40*a58d3d2aSXin Li #undef HAVE_CONFIG_H
41*a58d3d2aSXin Li #ifdef USE_WEIGHTS_FILE
42*a58d3d2aSXin Li #undef USE_WEIGHTS_FILE
43*a58d3d2aSXin Li #endif
44*a58d3d2aSXin Li #include "pitchdnn_data.c"
45*a58d3d2aSXin Li #include "fargan_data.c"
46*a58d3d2aSXin Li #include "plc_data.c"
47*a58d3d2aSXin Li #include "dred_rdovae_enc_data.c"
48*a58d3d2aSXin Li #include "dred_rdovae_dec_data.c"
49*a58d3d2aSXin Li #ifdef ENABLE_OSCE
50*a58d3d2aSXin Li #include "lace_data.c"
51*a58d3d2aSXin Li #include "nolace_data.c"
52*a58d3d2aSXin Li #endif
53*a58d3d2aSXin Li
write_weights(const WeightArray * list,FILE * fout)54*a58d3d2aSXin Li void write_weights(const WeightArray *list, FILE *fout)
55*a58d3d2aSXin Li {
56*a58d3d2aSXin Li int i=0;
57*a58d3d2aSXin Li unsigned char zeros[WEIGHT_BLOCK_SIZE] = {0};
58*a58d3d2aSXin Li while (list[i].name != NULL) {
59*a58d3d2aSXin Li WeightHead h;
60*a58d3d2aSXin Li if (strlen(list[i].name) >= sizeof(h.name) - 1) {
61*a58d3d2aSXin Li printf("[write_weights] warning: name %s too long\n", list[i].name);
62*a58d3d2aSXin Li }
63*a58d3d2aSXin Li memcpy(h.head, "DNNw", 4);
64*a58d3d2aSXin Li h.version = WEIGHT_BLOB_VERSION;
65*a58d3d2aSXin Li h.type = list[i].type;
66*a58d3d2aSXin Li h.size = list[i].size;
67*a58d3d2aSXin Li h.block_size = (h.size+WEIGHT_BLOCK_SIZE-1)/WEIGHT_BLOCK_SIZE*WEIGHT_BLOCK_SIZE;
68*a58d3d2aSXin Li OPUS_CLEAR(h.name, sizeof(h.name));
69*a58d3d2aSXin Li strncpy(h.name, list[i].name, sizeof(h.name));
70*a58d3d2aSXin Li h.name[sizeof(h.name)-1] = 0;
71*a58d3d2aSXin Li celt_assert(sizeof(h) == WEIGHT_BLOCK_SIZE);
72*a58d3d2aSXin Li fwrite(&h, 1, WEIGHT_BLOCK_SIZE, fout);
73*a58d3d2aSXin Li fwrite(list[i].data, 1, h.size, fout);
74*a58d3d2aSXin Li fwrite(zeros, 1, h.block_size-h.size, fout);
75*a58d3d2aSXin Li i++;
76*a58d3d2aSXin Li }
77*a58d3d2aSXin Li }
78*a58d3d2aSXin Li
main(void)79*a58d3d2aSXin Li int main(void)
80*a58d3d2aSXin Li {
81*a58d3d2aSXin Li FILE *fout = fopen("weights_blob.bin", "w");
82*a58d3d2aSXin Li write_weights(pitchdnn_arrays, fout);
83*a58d3d2aSXin Li write_weights(fargan_arrays, fout);
84*a58d3d2aSXin Li write_weights(plcmodel_arrays, fout);
85*a58d3d2aSXin Li write_weights(rdovaeenc_arrays, fout);
86*a58d3d2aSXin Li write_weights(rdovaedec_arrays, fout);
87*a58d3d2aSXin Li #ifdef ENABLE_OSCE
88*a58d3d2aSXin Li #ifndef DISABLE_LACE
89*a58d3d2aSXin Li write_weights(lacelayers_arrays, fout);
90*a58d3d2aSXin Li #endif
91*a58d3d2aSXin Li #ifndef DISABLE_NOLACE
92*a58d3d2aSXin Li write_weights(nolacelayers_arrays, fout);
93*a58d3d2aSXin Li #endif
94*a58d3d2aSXin Li #endif
95*a58d3d2aSXin Li fclose(fout);
96*a58d3d2aSXin Li return 0;
97*a58d3d2aSXin Li }
98