1/* example_cpp_encode_file - Simple FLAC file encoder using libFLAC 2 * Copyright (C) 2007 Josh Coalson 3 * 4 * This program is free software; you can redistribute it and/or 5 * modify it under the terms of the GNU General Public License 6 * as published by the Free Software Foundation; either version 2 7 * of the License, or (at your option) any later version. 8 * 9 * This program is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 * GNU General Public License for more details. 13 * 14 * You should have received a copy of the GNU General Public License 15 * along with this program; if not, write to the Free Software 16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 17 */ 18 19/* 20 * This example shows how to use libFLAC++ to encode a WAVE file to a FLAC 21 * file. It only supports 16-bit stereo files in canonical WAVE format. 22 * 23 * Complete API documentation can be found at: 24 * http://flac.sourceforge.net/api/ 25 */ 26 27#if HAVE_CONFIG_H 28# include <config.h> 29#endif 30 31#include <stdio.h> 32#include <stdlib.h> 33#include "FLAC++/metadata.h" 34#include "FLAC++/encoder.h" 35#include <string.h> 36 37class OurEncoder: public FLAC::Encoder::File { 38public: 39 OurEncoder(): FLAC::Encoder::File() { } 40protected: 41 virtual void progress_callback(FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate); 42}; 43 44#define READSIZE 1024 45 46static unsigned total_samples = 0; /* can use a 32-bit number due to WAVE size limitations */ 47static FLAC__byte buffer[READSIZE/*samples*/ * 2/*bytes_per_sample*/ * 2/*channels*/]; /* we read the WAVE data into here */ 48static FLAC__int32 pcm[READSIZE/*samples*/ * 2/*channels*/]; 49static FLAC__int32 *pcm_[2] = { pcm, pcm+READSIZE }; 50 51int main(int argc, char *argv[]) 52{ 53 bool ok = true; 54 OurEncoder encoder; 55 FLAC__StreamEncoderInitStatus init_status; 56 FLAC__StreamMetadata *metadata[2]; 57 FLAC__StreamMetadata_VorbisComment_Entry entry; 58 FILE *fin; 59 unsigned sample_rate = 0; 60 unsigned channels = 0; 61 unsigned bps = 0; 62 63 if(argc != 3) { 64 fprintf(stderr, "usage: %s infile.wav outfile.flac\n", argv[0]); 65 return 1; 66 } 67 68 if((fin = fopen(argv[1], "rb")) == NULL) { 69 fprintf(stderr, "ERROR: opening %s for output\n", argv[1]); 70 return 1; 71 } 72 73 /* read wav header and validate it */ 74 if( 75 fread(buffer, 1, 44, fin) != 44 || 76 memcmp(buffer, "RIFF", 4) || 77 memcmp(buffer+8, "WAVEfmt \020\000\000\000\001\000\002\000", 16) || 78 memcmp(buffer+32, "\004\000\020\000data", 8) 79 ) { 80 fprintf(stderr, "ERROR: invalid/unsupported WAVE file, only 16bps stereo WAVE in canonical form allowed\n"); 81 fclose(fin); 82 return 1; 83 } 84 sample_rate = ((((((unsigned)buffer[27] << 8) | buffer[26]) << 8) | buffer[25]) << 8) | buffer[24]; 85 channels = 2; 86 bps = 16; 87 total_samples = (((((((unsigned)buffer[43] << 8) | buffer[42]) << 8) | buffer[41]) << 8) | buffer[40]) / 4; 88 89 /* check the encoder */ 90 if(!encoder) { 91 fprintf(stderr, "ERROR: allocating encoder\n"); 92 fclose(fin); 93 return 1; 94 } 95 96 ok &= encoder.set_verify(true); 97 ok &= encoder.set_compression_level(5); 98 ok &= encoder.set_channels(channels); 99 ok &= encoder.set_bits_per_sample(bps); 100 ok &= encoder.set_sample_rate(sample_rate); 101 ok &= encoder.set_total_samples_estimate(total_samples); 102 103 /* now add some metadata; we'll add some tags and a padding block */ 104 if(ok) { 105 if( 106 (metadata[0] = FLAC__metadata_object_new(FLAC__METADATA_TYPE_VORBIS_COMMENT)) == NULL || 107 (metadata[1] = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING)) == NULL || 108 /* there are many tag (vorbiscomment) functions but these are convenient for this particular use: */ 109 !FLAC__metadata_object_vorbiscomment_entry_from_name_value_pair(&entry, "ARTIST", "Some Artist") || 110 !FLAC__metadata_object_vorbiscomment_append_comment(metadata[0], entry, /*copy=*/false) || /* copy=false: let metadata object take control of entry's allocated string */ 111 !FLAC__metadata_object_vorbiscomment_entry_from_name_value_pair(&entry, "YEAR", "1984") || 112 !FLAC__metadata_object_vorbiscomment_append_comment(metadata[0], entry, /*copy=*/false) 113 ) { 114 fprintf(stderr, "ERROR: out of memory or tag error\n"); 115 ok = false; 116 } 117 118 metadata[1]->length = 1234; /* set the padding length */ 119 120 ok = encoder.set_metadata(metadata, 2); 121 } 122 123 /* initialize encoder */ 124 if(ok) { 125 init_status = encoder.init(argv[2]); 126 if(init_status != FLAC__STREAM_ENCODER_INIT_STATUS_OK) { 127 fprintf(stderr, "ERROR: initializing encoder: %s\n", FLAC__StreamEncoderInitStatusString[init_status]); 128 ok = false; 129 } 130 } 131 132 /* read blocks of samples from WAVE file and feed to encoder */ 133 if(ok) { 134 size_t left = (size_t)total_samples; 135 while(ok && left) { 136 size_t need = (left>READSIZE? (size_t)READSIZE : (size_t)left); 137 if(fread(buffer, channels*(bps/8), need, fin) != need) { 138 fprintf(stderr, "ERROR: reading from WAVE file\n"); 139 ok = false; 140 } 141 else { 142 /* convert the packed little-endian 16-bit PCM samples from WAVE into an interleaved FLAC__int32 buffer for libFLAC */ 143 size_t i; 144 for(i = 0; i < need*channels; i++) { 145 /* inefficient but simple and works on big- or little-endian machines */ 146 pcm[i] = (FLAC__int32)(((FLAC__int16)(FLAC__int8)buffer[2*i+1] << 8) | (FLAC__int16)buffer[2*i]); 147 } 148 /* feed samples to encoder */ 149 ok = encoder.process_interleaved(pcm, need); 150 } 151 left -= need; 152 } 153 } 154 155 ok &= encoder.finish(); 156 157 fprintf(stderr, "encoding: %s\n", ok? "succeeded" : "FAILED"); 158 fprintf(stderr, " state: %s\n", encoder.get_state().resolved_as_cstring(encoder)); 159 160 /* now that encoding is finished, the metadata can be freed */ 161 FLAC__metadata_object_delete(metadata[0]); 162 FLAC__metadata_object_delete(metadata[1]); 163 164 fclose(fin); 165 166 return 0; 167} 168 169void OurEncoder::progress_callback(FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate) 170{ 171#ifdef _MSC_VER 172 fprintf(stderr, "wrote %I64u bytes, %I64u/%u samples, %u/%u frames\n", bytes_written, samples_written, total_samples, frames_written, total_frames_estimate); 173#else 174 fprintf(stderr, "wrote %llu bytes, %llu/%u samples, %u/%u frames\n", bytes_written, samples_written, total_samples, frames_written, total_frames_estimate); 175#endif 176} 177