#!/usr/bin/env python3
# ==========================================================================================
# Copyright (c) 2020 AIIT XUOS Lab
# XiOS is licensed under Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#        http://license.coscl.org.cn/MulanPSL2
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
# See the Mulan PSL v2 for more details.

#  @file:    mnist-c-digit.py
#  @brief:   print image digit at command line
#  @version: 1.0
#  @author:  AIIT XUOS Lab
#  @date:    2021/4/30
# ==========================================================================================


import tensorflow as tf

print("TensorFlow version %s" % (tf.__version__))

def show(image):
    for i in range(28):
        for j in range(28):
            if image[i][j] > 0.3:
                print('#', end = '')
            else:
                print('.', end = '')
        print()

digit_file_path = 'digit.h'
digit_content = '''const float mnist_digit[] = {
  %s
};
const int mnist_label = %d;
'''

if __name__ == '__main__':
    mnist = tf.keras.datasets.mnist
    (_, _), (test_images, test_labels) = mnist.load_data()
    index = 0
    shape = 28
    image = test_images[index].astype('float32')/255
    label = test_labels[index]
    print('label: %d' % label)
    #show(image)
    digit_data = (',\n  ').join([ (', ').join([ '%.2f' % image[row][col] for col in range(shape)]) for row in range(shape)])
    digit_file = open(digit_file_path, 'w')
    digit_file.write(digit_content % (digit_data, label))
    digit_file.close()