// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "shm_control_nn.h"
#include "logger/log.h"

#ifdef LOG_TAG
#undef LOG_TAG
#endif
#define LOG_TAG "shm_control_nn"

namespace rockchip {
namespace aiserver {

ShmNNController::ShmNNController() {
    shmWriteInited = 0;
    initialize();
}

ShmNNController::~ShmNNController() {

}

void ShmNNController::initialize() {
    if (shmWriteInited) {
        return;
    }

#ifdef ENABLE_SHM_SERVER
    shmc::SetLogHandler(shmc::kDebug, [](shmc::LogLevel lv, const char *s) {
        LOG_INFO("[%d] %s\n", lv, s);
    });
    shmWriteInited = shmNNQueue.InitForWrite(kShmNNKey, kNNQueueBufSize);
#endif
}

void ShmNNController::send(std::string &buf) {
    if (!shmWriteInited) {
        initialize();
    }

    if (shmWriteInited) {
        shmNNQueue.Push(buf);
    } else {
        LOG_ERROR("shm nn init for write failed");
    }
}

}
}
// namespace ShmControl