sness@618: // Copyright 2006-2010, Thomas Walters
sness@618: //
sness@618: // AIM-C: A C++ implementation of the Auditory Image Model
sness@618: // http://www.acousticscale.org/AIMC
sness@618: //
sness@618: // Licensed under the Apache License, Version 2.0 (the "License");
sness@618: // you may not use this file except in compliance with the License.
sness@618: // You may obtain a copy of the License at
sness@618: //
sness@618: //     http://www.apache.org/licenses/LICENSE-2.0
sness@618: //
sness@618: // Unless required by applicable law or agreed to in writing, software
sness@618: // distributed under the License is distributed on an "AS IS" BASIS,
sness@618: // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
sness@618: // See the License for the specific language governing permissions and
sness@618: // limitations under the License.
sness@618: 
sness@618: /*!
sness@618:  * \file
sness@618:  * \brief File output in the HTK format.
sness@618:  *
sness@618:  * \author Tom Walters <tom@acousticscale.org>
sness@618:  * \author Willem van Engen <cnbh@willem.engen.nl>
sness@618:  * \date created 2007/01/26
sness@618:  * \version \$Id: $
sness@618:  */
sness@618: 
sness@618: 
sness@618: #include <iostream>
sness@618: #include <fstream>
sness@618: #include <iomanip>
sness@618: 
sness@618: #include "Modules/Output/FileOutputJSON.h"
sness@618: 
sness@618: #ifdef _WINDOWS
sness@618: #  include <direct.h>  // for _mkdir & _rmdir
sness@618: #else
sness@618: #  include <sys/types.h>
sness@618: #  include <dirent.h>  // for opendir & friends
sness@618: #endif
sness@618: 
sness@618: #include <stdint.h>
sness@618: #include <stdio.h>
sness@618: #include <string.h>
sness@618: #include <cmath>
sness@618: #include <string>
sness@618: 
sness@618: namespace aimc {
sness@618: FileOutputJSON::FileOutputJSON(Parameters *params) : Module(params) {
sness@618:   module_description_ = "File output in JSON format";
sness@618:   module_identifier_ = "json_out";
sness@618:   module_type_ = "output";
sness@618:   module_version_ = "$Id: FileOutputJSON.cc 51 2010-03-30 22:06:24Z tomwalters $";
sness@618:   file_suffix_ = parameters_->DefaultString("file_suffix", ".json");
sness@618: 
sness@618:   header_written_ = false;
sness@618:   frame_period_ms_ = 0.0f;
sness@618: }
sness@618: 
sness@618: FileOutputJSON::~FileOutputJSON() {
sness@618:   if (file_handle_)
sness@618:     CloseFile();
sness@618: }
sness@618: 
sness@618: bool FileOutputJSON::OpenFile(string &filename) {
sness@618:   file_handle_.open(filename.c_str());
sness@618:   
sness@618:   frame_count_ = 0;
sness@618:   frame_period_ms_ = 0.0;
sness@618:   header_written_ = false;
sness@618: 
sness@618:   if (initialized_) {
sness@618:     WriteHeader();
sness@618:   }
sness@618:   
sness@618:   return true;
sness@618: }
sness@618: 
sness@618: bool FileOutputJSON::InitializeInternal(const SignalBank &input) {
sness@618:   channel_count_ = input.channel_count();
sness@618:   buffer_length_ = input.buffer_length();  
sness@618:   sample_rate_ = input.sample_rate();
sness@618:   ResetInternal();
sness@618:   if (file_handle_ == NULL) {
sness@618:     LOG_ERROR(_T("Couldn't initialize file output."));
sness@618:     return false;
sness@618:   }
sness@618:   return true;
sness@618: }
sness@618: 
sness@618: void FileOutputJSON::ResetInternal() {
sness@618:   if (file_handle_ != NULL && !header_written_) {
sness@618:     WriteHeader();
sness@618:   }
sness@618:   if (file_handle_ != NULL)
sness@618:     CloseFile();
sness@618:     
sness@618:   string out_filename;
sness@618:   out_filename = global_parameters_->GetString("output_filename_base") + file_suffix_;
sness@618:   OpenFile(out_filename);
sness@618: }
sness@618: 
sness@618: void FileOutputJSON::WriteHeader() {
sness@618:   if (header_written_)
sness@618:     return;
sness@618: 
sness@618:   uint32_t channels_out = channel_count_;
sness@618:   uint32_t samples_out = buffer_length_;
sness@618:   float sample_rate = sample_rate_;
sness@618: 
sness@618:   file_handle_ << std::setprecision(3);
sness@618:   file_handle_ << "{" << std::endl;
sness@618:   file_handle_ << "\"channels\" : " << channels_out << "," << std::endl;
sness@618:   file_handle_ << "\"samples\" : " << sizeof(samples_out) << "," << std::endl;
sness@618:   file_handle_ << "\"sample_rate\" : " << sizeof(sample_rate) << "," << std::endl;
sness@618: 
sness@618:   header_written_ = true;
sness@618: }
sness@618: 
sness@618: void FileOutputJSON::Process(const SignalBank &input) {
sness@618:   if (file_handle_ == NULL) {
sness@618:     LOG_ERROR(_T("Couldn't process file output. No file is open."
sness@618:                  "Please call FileOutputJSON::OpenFile first"));
sness@618:     return;
sness@618:   }
sness@618: 
sness@618:   if (!header_written_) {
sness@618:     LOG_ERROR(_T("No header has been written on the output file yet. Please "
sness@618:                  "call FileOutputJSON::Initialize() before calling "
sness@618:                  "FileOutputJSON::Process()"));
sness@618:     return;
sness@618:   }
sness@618:   float s;
sness@618: 
sness@618:   for (int ch = 0; ch < input.channel_count(); ch++) {
sness@618:     file_handle_ << "\"channel\" : [";
sness@618:     for (int i = 0; i < input.buffer_length(); i++) {
sness@618:       s = input.sample(ch, i);
sness@618:       file_handle_ << s;
sness@618:       if (i < input.buffer_length() - 1) {
sness@618:         file_handle_ << ",";
sness@618:       }
sness@618:     }
sness@618:     file_handle_ << "]" << std::endl;
sness@618:     if (ch < input.channel_count() - 1) {
sness@618:       file_handle_ << ",";
sness@618:     }
sness@618:   }
sness@618: 
sness@618:   frame_count_++;
sness@618: 
sness@618: }
sness@618: 
sness@618: bool FileOutputJSON::CloseFile() {
sness@618:   if (file_handle_ == NULL)
sness@618:     return false;
sness@618: 
sness@618:   uint32_t frame_count = frame_count_;
sness@618:   float sample_period_out = frame_period_ms_;
sness@618: 
sness@618:   file_handle_ << "\"frame_count\" : " << sizeof(frame_count) << std::endl;
sness@618:   file_handle_ << "\"samples\" : " << sizeof(sample_period_out) << std::endl;
sness@618:   file_handle_ << "}" << std::endl;
sness@618: 
sness@618: 
sness@618:   file_handle_.close();
sness@618:   header_written_ = false;
sness@618:   return true;
sness@618: }
sness@618: }  // namespace aimc
sness@618: