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