The JUCE cross-platform C++ framework, with DISTRHO/KXStudio specific changes
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1842 lines
82KB

  1. /*
  2. ==============================================================================
  3. This file is part of the JUCE library.
  4. Copyright (c) 2017 - ROLI Ltd.
  5. JUCE is an open source library subject to commercial or open-source
  6. licensing.
  7. By using JUCE, you agree to the terms of both the JUCE 5 End-User License
  8. Agreement and JUCE 5 Privacy Policy (both updated and effective as of the
  9. 27th April 2017).
  10. End User License Agreement: www.juce.com/juce-5-licence
  11. Privacy Policy: www.juce.com/juce-5-privacy-policy
  12. Or: You may also use this code under the terms of the GPL v3 (see
  13. www.gnu.org/licenses).
  14. JUCE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL WARRANTIES, WHETHER
  15. EXPRESSED OR IMPLIED, INCLUDING MERCHANTABILITY AND FITNESS FOR PURPOSE, ARE
  16. DISCLAIMED.
  17. ==============================================================================
  18. */
  19. static const char* const wavFormatName = "WAV file";
  20. //==============================================================================
  21. const char* const WavAudioFormat::bwavDescription = "bwav description";
  22. const char* const WavAudioFormat::bwavOriginator = "bwav originator";
  23. const char* const WavAudioFormat::bwavOriginatorRef = "bwav originator ref";
  24. const char* const WavAudioFormat::bwavOriginationDate = "bwav origination date";
  25. const char* const WavAudioFormat::bwavOriginationTime = "bwav origination time";
  26. const char* const WavAudioFormat::bwavTimeReference = "bwav time reference";
  27. const char* const WavAudioFormat::bwavCodingHistory = "bwav coding history";
  28. StringPairArray WavAudioFormat::createBWAVMetadata (const String& description,
  29. const String& originator,
  30. const String& originatorRef,
  31. Time date,
  32. const int64 timeReferenceSamples,
  33. const String& codingHistory)
  34. {
  35. StringPairArray m;
  36. m.set (bwavDescription, description);
  37. m.set (bwavOriginator, originator);
  38. m.set (bwavOriginatorRef, originatorRef);
  39. m.set (bwavOriginationDate, date.formatted ("%Y-%m-%d"));
  40. m.set (bwavOriginationTime, date.formatted ("%H:%M:%S"));
  41. m.set (bwavTimeReference, String (timeReferenceSamples));
  42. m.set (bwavCodingHistory, codingHistory);
  43. return m;
  44. }
  45. const char* const WavAudioFormat::acidOneShot = "acid one shot";
  46. const char* const WavAudioFormat::acidRootSet = "acid root set";
  47. const char* const WavAudioFormat::acidStretch = "acid stretch";
  48. const char* const WavAudioFormat::acidDiskBased = "acid disk based";
  49. const char* const WavAudioFormat::acidizerFlag = "acidizer flag";
  50. const char* const WavAudioFormat::acidRootNote = "acid root note";
  51. const char* const WavAudioFormat::acidBeats = "acid beats";
  52. const char* const WavAudioFormat::acidDenominator = "acid denominator";
  53. const char* const WavAudioFormat::acidNumerator = "acid numerator";
  54. const char* const WavAudioFormat::acidTempo = "acid tempo";
  55. const char* const WavAudioFormat::riffInfoArchivalLocation = "IARL";
  56. const char* const WavAudioFormat::riffInfoArtist = "IART";
  57. const char* const WavAudioFormat::riffInfoBaseURL = "IBSU";
  58. const char* const WavAudioFormat::riffInfoCinematographer = "ICNM";
  59. const char* const WavAudioFormat::riffInfoComment = "CMNT";
  60. const char* const WavAudioFormat::riffInfoComments = "COMM";
  61. const char* const WavAudioFormat::riffInfoCommissioned = "ICMS";
  62. const char* const WavAudioFormat::riffInfoCopyright = "ICOP";
  63. const char* const WavAudioFormat::riffInfoCostumeDesigner = "ICDS";
  64. const char* const WavAudioFormat::riffInfoCountry = "ICNT";
  65. const char* const WavAudioFormat::riffInfoCropped = "ICRP";
  66. const char* const WavAudioFormat::riffInfoDateCreated = "ICRD";
  67. const char* const WavAudioFormat::riffInfoDateTimeOriginal = "IDIT";
  68. const char* const WavAudioFormat::riffInfoDefaultAudioStream = "ICAS";
  69. const char* const WavAudioFormat::riffInfoDimension = "IDIM";
  70. const char* const WavAudioFormat::riffInfoDirectory = "DIRC";
  71. const char* const WavAudioFormat::riffInfoDistributedBy = "IDST";
  72. const char* const WavAudioFormat::riffInfoDotsPerInch = "IDPI";
  73. const char* const WavAudioFormat::riffInfoEditedBy = "IEDT";
  74. const char* const WavAudioFormat::riffInfoEighthLanguage = "IAS8";
  75. const char* const WavAudioFormat::riffInfoEncodedBy = "CODE";
  76. const char* const WavAudioFormat::riffInfoEndTimecode = "TCDO";
  77. const char* const WavAudioFormat::riffInfoEngineer = "IENG";
  78. const char* const WavAudioFormat::riffInfoFifthLanguage = "IAS5";
  79. const char* const WavAudioFormat::riffInfoFirstLanguage = "IAS1";
  80. const char* const WavAudioFormat::riffInfoFourthLanguage = "IAS4";
  81. const char* const WavAudioFormat::riffInfoGenre = "GENR";
  82. const char* const WavAudioFormat::riffInfoKeywords = "IKEY";
  83. const char* const WavAudioFormat::riffInfoLanguage = "LANG";
  84. const char* const WavAudioFormat::riffInfoLength = "TLEN";
  85. const char* const WavAudioFormat::riffInfoLightness = "ILGT";
  86. const char* const WavAudioFormat::riffInfoLocation = "LOCA";
  87. const char* const WavAudioFormat::riffInfoLogoIconURL = "ILIU";
  88. const char* const WavAudioFormat::riffInfoLogoURL = "ILGU";
  89. const char* const WavAudioFormat::riffInfoMedium = "IMED";
  90. const char* const WavAudioFormat::riffInfoMoreInfoBannerImage = "IMBI";
  91. const char* const WavAudioFormat::riffInfoMoreInfoBannerURL = "IMBU";
  92. const char* const WavAudioFormat::riffInfoMoreInfoText = "IMIT";
  93. const char* const WavAudioFormat::riffInfoMoreInfoURL = "IMIU";
  94. const char* const WavAudioFormat::riffInfoMusicBy = "IMUS";
  95. const char* const WavAudioFormat::riffInfoNinthLanguage = "IAS9";
  96. const char* const WavAudioFormat::riffInfoNumberOfParts = "PRT2";
  97. const char* const WavAudioFormat::riffInfoOrganisation = "TORG";
  98. const char* const WavAudioFormat::riffInfoPart = "PRT1";
  99. const char* const WavAudioFormat::riffInfoProducedBy = "IPRO";
  100. const char* const WavAudioFormat::riffInfoProductionDesigner = "IPDS";
  101. const char* const WavAudioFormat::riffInfoProductionStudio = "ISDT";
  102. const char* const WavAudioFormat::riffInfoRate = "RATE";
  103. const char* const WavAudioFormat::riffInfoRated = "AGES";
  104. const char* const WavAudioFormat::riffInfoRating = "IRTD";
  105. const char* const WavAudioFormat::riffInfoRippedBy = "IRIP";
  106. const char* const WavAudioFormat::riffInfoSecondaryGenre = "ISGN";
  107. const char* const WavAudioFormat::riffInfoSecondLanguage = "IAS2";
  108. const char* const WavAudioFormat::riffInfoSeventhLanguage = "IAS7";
  109. const char* const WavAudioFormat::riffInfoSharpness = "ISHP";
  110. const char* const WavAudioFormat::riffInfoSixthLanguage = "IAS6";
  111. const char* const WavAudioFormat::riffInfoSoftware = "ISFT";
  112. const char* const WavAudioFormat::riffInfoSoundSchemeTitle = "DISP";
  113. const char* const WavAudioFormat::riffInfoSource = "ISRC";
  114. const char* const WavAudioFormat::riffInfoSourceFrom = "ISRF";
  115. const char* const WavAudioFormat::riffInfoStarring_ISTR = "ISTR";
  116. const char* const WavAudioFormat::riffInfoStarring_STAR = "STAR";
  117. const char* const WavAudioFormat::riffInfoStartTimecode = "TCOD";
  118. const char* const WavAudioFormat::riffInfoStatistics = "STAT";
  119. const char* const WavAudioFormat::riffInfoSubject = "ISBJ";
  120. const char* const WavAudioFormat::riffInfoTapeName = "TAPE";
  121. const char* const WavAudioFormat::riffInfoTechnician = "ITCH";
  122. const char* const WavAudioFormat::riffInfoThirdLanguage = "IAS3";
  123. const char* const WavAudioFormat::riffInfoTimeCode = "ISMP";
  124. const char* const WavAudioFormat::riffInfoTitle = "INAM";
  125. const char* const WavAudioFormat::riffInfoTrackNumber = "TRCK";
  126. const char* const WavAudioFormat::riffInfoURL = "TURL";
  127. const char* const WavAudioFormat::riffInfoVegasVersionMajor = "VMAJ";
  128. const char* const WavAudioFormat::riffInfoVegasVersionMinor = "VMIN";
  129. const char* const WavAudioFormat::riffInfoVersion = "TVER";
  130. const char* const WavAudioFormat::riffInfoWatermarkURL = "IWMU";
  131. const char* const WavAudioFormat::riffInfoWrittenBy = "IWRI";
  132. const char* const WavAudioFormat::riffInfoYear = "YEAR";
  133. const char* const WavAudioFormat::ISRC = "ISRC";
  134. const char* const WavAudioFormat::tracktionLoopInfo = "tracktion loop info";
  135. //==============================================================================
  136. namespace WavFileHelpers
  137. {
  138. inline int chunkName (const char* const name) noexcept { return (int) ByteOrder::littleEndianInt (name); }
  139. inline size_t roundUpSize (size_t sz) noexcept { return (sz + 3) & ~3u; }
  140. #if JUCE_MSVC
  141. #pragma pack (push, 1)
  142. #endif
  143. struct BWAVChunk
  144. {
  145. char description [256];
  146. char originator [32];
  147. char originatorRef [32];
  148. char originationDate [10];
  149. char originationTime [8];
  150. uint32 timeRefLow;
  151. uint32 timeRefHigh;
  152. uint16 version;
  153. uint8 umid[64];
  154. uint8 reserved[190];
  155. char codingHistory[1];
  156. void copyTo (StringPairArray& values, const int totalSize) const
  157. {
  158. values.set (WavAudioFormat::bwavDescription, String::fromUTF8 (description, sizeof (description)));
  159. values.set (WavAudioFormat::bwavOriginator, String::fromUTF8 (originator, sizeof (originator)));
  160. values.set (WavAudioFormat::bwavOriginatorRef, String::fromUTF8 (originatorRef, sizeof (originatorRef)));
  161. values.set (WavAudioFormat::bwavOriginationDate, String::fromUTF8 (originationDate, sizeof (originationDate)));
  162. values.set (WavAudioFormat::bwavOriginationTime, String::fromUTF8 (originationTime, sizeof (originationTime)));
  163. const uint32 timeLow = ByteOrder::swapIfBigEndian (timeRefLow);
  164. const uint32 timeHigh = ByteOrder::swapIfBigEndian (timeRefHigh);
  165. const int64 time = (((int64) timeHigh) << 32) + timeLow;
  166. values.set (WavAudioFormat::bwavTimeReference, String (time));
  167. values.set (WavAudioFormat::bwavCodingHistory,
  168. String::fromUTF8 (codingHistory, totalSize - (int) offsetof (BWAVChunk, codingHistory)));
  169. }
  170. static MemoryBlock createFrom (const StringPairArray& values)
  171. {
  172. MemoryBlock data (roundUpSize (sizeof (BWAVChunk) + values [WavAudioFormat::bwavCodingHistory].getNumBytesAsUTF8()));
  173. data.fillWith (0);
  174. BWAVChunk* b = (BWAVChunk*) data.getData();
  175. // Allow these calls to overwrite an extra byte at the end, which is fine as long
  176. // as they get called in the right order..
  177. values [WavAudioFormat::bwavDescription] .copyToUTF8 (b->description, 257);
  178. values [WavAudioFormat::bwavOriginator] .copyToUTF8 (b->originator, 33);
  179. values [WavAudioFormat::bwavOriginatorRef] .copyToUTF8 (b->originatorRef, 33);
  180. values [WavAudioFormat::bwavOriginationDate].copyToUTF8 (b->originationDate, 11);
  181. values [WavAudioFormat::bwavOriginationTime].copyToUTF8 (b->originationTime, 9);
  182. const int64 time = values [WavAudioFormat::bwavTimeReference].getLargeIntValue();
  183. b->timeRefLow = ByteOrder::swapIfBigEndian ((uint32) (time & 0xffffffff));
  184. b->timeRefHigh = ByteOrder::swapIfBigEndian ((uint32) (time >> 32));
  185. values [WavAudioFormat::bwavCodingHistory].copyToUTF8 (b->codingHistory, 0x7fffffff);
  186. if (b->description[0] != 0
  187. || b->originator[0] != 0
  188. || b->originationDate[0] != 0
  189. || b->originationTime[0] != 0
  190. || b->codingHistory[0] != 0
  191. || time != 0)
  192. {
  193. return data;
  194. }
  195. return {};
  196. }
  197. } JUCE_PACKED;
  198. //==============================================================================
  199. AudioChannelSet canonicalWavChannelSet (int numChannels)
  200. {
  201. if (numChannels == 1) return AudioChannelSet::mono();
  202. if (numChannels == 2) return AudioChannelSet::stereo();
  203. if (numChannels == 3) return AudioChannelSet::createLCR();
  204. if (numChannels == 4) return AudioChannelSet::quadraphonic();
  205. if (numChannels == 5) return AudioChannelSet::create5point0();
  206. if (numChannels == 6) return AudioChannelSet::create5point1();
  207. if (numChannels == 7) return AudioChannelSet::create7point0SDDS();
  208. if (numChannels == 8) return AudioChannelSet::create7point1SDDS();
  209. return AudioChannelSet::discreteChannels (numChannels);
  210. }
  211. //==============================================================================
  212. struct SMPLChunk
  213. {
  214. struct SampleLoop
  215. {
  216. uint32 identifier;
  217. uint32 type; // these are different in AIFF and WAV
  218. uint32 start;
  219. uint32 end;
  220. uint32 fraction;
  221. uint32 playCount;
  222. } JUCE_PACKED;
  223. uint32 manufacturer;
  224. uint32 product;
  225. uint32 samplePeriod;
  226. uint32 midiUnityNote;
  227. uint32 midiPitchFraction;
  228. uint32 smpteFormat;
  229. uint32 smpteOffset;
  230. uint32 numSampleLoops;
  231. uint32 samplerData;
  232. SampleLoop loops[1];
  233. template <typename NameType>
  234. static void setValue (StringPairArray& values, NameType name, uint32 val)
  235. {
  236. values.set (name, String (ByteOrder::swapIfBigEndian (val)));
  237. }
  238. static void setValue (StringPairArray& values, int prefix, const char* name, uint32 val)
  239. {
  240. setValue (values, "Loop" + String (prefix) + name, val);
  241. }
  242. void copyTo (StringPairArray& values, const int totalSize) const
  243. {
  244. setValue (values, "Manufacturer", manufacturer);
  245. setValue (values, "Product", product);
  246. setValue (values, "SamplePeriod", samplePeriod);
  247. setValue (values, "MidiUnityNote", midiUnityNote);
  248. setValue (values, "MidiPitchFraction", midiPitchFraction);
  249. setValue (values, "SmpteFormat", smpteFormat);
  250. setValue (values, "SmpteOffset", smpteOffset);
  251. setValue (values, "NumSampleLoops", numSampleLoops);
  252. setValue (values, "SamplerData", samplerData);
  253. for (int i = 0; i < (int) numSampleLoops; ++i)
  254. {
  255. if ((uint8*) (loops + (i + 1)) > ((uint8*) this) + totalSize)
  256. break;
  257. setValue (values, i, "Identifier", loops[i].identifier);
  258. setValue (values, i, "Type", loops[i].type);
  259. setValue (values, i, "Start", loops[i].start);
  260. setValue (values, i, "End", loops[i].end);
  261. setValue (values, i, "Fraction", loops[i].fraction);
  262. setValue (values, i, "PlayCount", loops[i].playCount);
  263. }
  264. }
  265. template <typename NameType>
  266. static uint32 getValue (const StringPairArray& values, NameType name, const char* def)
  267. {
  268. return ByteOrder::swapIfBigEndian ((uint32) values.getValue (name, def).getIntValue());
  269. }
  270. static uint32 getValue (const StringPairArray& values, int prefix, const char* name, const char* def)
  271. {
  272. return getValue (values, "Loop" + String (prefix) + name, def);
  273. }
  274. static MemoryBlock createFrom (const StringPairArray& values)
  275. {
  276. MemoryBlock data;
  277. const int numLoops = jmin (64, values.getValue ("NumSampleLoops", "0").getIntValue());
  278. data.setSize (roundUpSize (sizeof (SMPLChunk) + (size_t) (jmax (0, numLoops - 1)) * sizeof (SampleLoop)), true);
  279. auto s = static_cast<SMPLChunk*> (data.getData());
  280. s->manufacturer = getValue (values, "Manufacturer", "0");
  281. s->product = getValue (values, "Product", "0");
  282. s->samplePeriod = getValue (values, "SamplePeriod", "0");
  283. s->midiUnityNote = getValue (values, "MidiUnityNote", "60");
  284. s->midiPitchFraction = getValue (values, "MidiPitchFraction", "0");
  285. s->smpteFormat = getValue (values, "SmpteFormat", "0");
  286. s->smpteOffset = getValue (values, "SmpteOffset", "0");
  287. s->numSampleLoops = ByteOrder::swapIfBigEndian ((uint32) numLoops);
  288. s->samplerData = getValue (values, "SamplerData", "0");
  289. for (int i = 0; i < numLoops; ++i)
  290. {
  291. auto& loop = s->loops[i];
  292. loop.identifier = getValue (values, i, "Identifier", "0");
  293. loop.type = getValue (values, i, "Type", "0");
  294. loop.start = getValue (values, i, "Start", "0");
  295. loop.end = getValue (values, i, "End", "0");
  296. loop.fraction = getValue (values, i, "Fraction", "0");
  297. loop.playCount = getValue (values, i, "PlayCount", "0");
  298. }
  299. return data;
  300. }
  301. } JUCE_PACKED;
  302. //==============================================================================
  303. struct InstChunk
  304. {
  305. int8 baseNote;
  306. int8 detune;
  307. int8 gain;
  308. int8 lowNote;
  309. int8 highNote;
  310. int8 lowVelocity;
  311. int8 highVelocity;
  312. static void setValue (StringPairArray& values, const char* name, int val)
  313. {
  314. values.set (name, String (val));
  315. }
  316. void copyTo (StringPairArray& values) const
  317. {
  318. setValue (values, "MidiUnityNote", baseNote);
  319. setValue (values, "Detune", detune);
  320. setValue (values, "Gain", gain);
  321. setValue (values, "LowNote", lowNote);
  322. setValue (values, "HighNote", highNote);
  323. setValue (values, "LowVelocity", lowVelocity);
  324. setValue (values, "HighVelocity", highVelocity);
  325. }
  326. static int8 getValue (const StringPairArray& values, const char* name, const char* def)
  327. {
  328. return (int8) values.getValue (name, def).getIntValue();
  329. }
  330. static MemoryBlock createFrom (const StringPairArray& values)
  331. {
  332. MemoryBlock data;
  333. auto& keys = values.getAllKeys();
  334. if (keys.contains ("LowNote", true) && keys.contains ("HighNote", true))
  335. {
  336. data.setSize (8, true);
  337. auto* inst = static_cast<InstChunk*> (data.getData());
  338. inst->baseNote = getValue (values, "MidiUnityNote", "60");
  339. inst->detune = getValue (values, "Detune", "0");
  340. inst->gain = getValue (values, "Gain", "0");
  341. inst->lowNote = getValue (values, "LowNote", "0");
  342. inst->highNote = getValue (values, "HighNote", "127");
  343. inst->lowVelocity = getValue (values, "LowVelocity", "1");
  344. inst->highVelocity = getValue (values, "HighVelocity", "127");
  345. }
  346. return data;
  347. }
  348. } JUCE_PACKED;
  349. //==============================================================================
  350. struct CueChunk
  351. {
  352. struct Cue
  353. {
  354. uint32 identifier;
  355. uint32 order;
  356. uint32 chunkID;
  357. uint32 chunkStart;
  358. uint32 blockStart;
  359. uint32 offset;
  360. } JUCE_PACKED;
  361. uint32 numCues;
  362. Cue cues[1];
  363. static void setValue (StringPairArray& values, int prefix, const char* name, uint32 val)
  364. {
  365. values.set ("Cue" + String (prefix) + name, String (ByteOrder::swapIfBigEndian (val)));
  366. }
  367. void copyTo (StringPairArray& values, const int totalSize) const
  368. {
  369. values.set ("NumCuePoints", String (ByteOrder::swapIfBigEndian (numCues)));
  370. for (int i = 0; i < (int) numCues; ++i)
  371. {
  372. if ((uint8*) (cues + (i + 1)) > ((uint8*) this) + totalSize)
  373. break;
  374. setValue (values, i, "Identifier", cues[i].identifier);
  375. setValue (values, i, "Order", cues[i].order);
  376. setValue (values, i, "ChunkID", cues[i].chunkID);
  377. setValue (values, i, "ChunkStart", cues[i].chunkStart);
  378. setValue (values, i, "BlockStart", cues[i].blockStart);
  379. setValue (values, i, "Offset", cues[i].offset);
  380. }
  381. }
  382. static MemoryBlock createFrom (const StringPairArray& values)
  383. {
  384. MemoryBlock data;
  385. const int numCues = values.getValue ("NumCuePoints", "0").getIntValue();
  386. if (numCues > 0)
  387. {
  388. data.setSize (roundUpSize (sizeof (CueChunk) + (size_t) (numCues - 1) * sizeof (Cue)), true);
  389. auto c = static_cast<CueChunk*> (data.getData());
  390. c->numCues = ByteOrder::swapIfBigEndian ((uint32) numCues);
  391. const String dataChunkID (chunkName ("data"));
  392. int nextOrder = 0;
  393. #if JUCE_DEBUG
  394. Array<uint32> identifiers;
  395. #endif
  396. for (int i = 0; i < numCues; ++i)
  397. {
  398. auto prefix = "Cue" + String (i);
  399. auto identifier = (uint32) values.getValue (prefix + "Identifier", "0").getIntValue();
  400. #if JUCE_DEBUG
  401. jassert (! identifiers.contains (identifier));
  402. identifiers.add (identifier);
  403. #endif
  404. auto order = values.getValue (prefix + "Order", String (nextOrder)).getIntValue();
  405. nextOrder = jmax (nextOrder, order) + 1;
  406. auto& cue = c->cues[i];
  407. cue.identifier = ByteOrder::swapIfBigEndian ((uint32) identifier);
  408. cue.order = ByteOrder::swapIfBigEndian ((uint32) order);
  409. cue.chunkID = ByteOrder::swapIfBigEndian ((uint32) values.getValue (prefix + "ChunkID", dataChunkID).getIntValue());
  410. cue.chunkStart = ByteOrder::swapIfBigEndian ((uint32) values.getValue (prefix + "ChunkStart", "0").getIntValue());
  411. cue.blockStart = ByteOrder::swapIfBigEndian ((uint32) values.getValue (prefix + "BlockStart", "0").getIntValue());
  412. cue.offset = ByteOrder::swapIfBigEndian ((uint32) values.getValue (prefix + "Offset", "0").getIntValue());
  413. }
  414. }
  415. return data;
  416. }
  417. } JUCE_PACKED;
  418. //==============================================================================
  419. namespace ListChunk
  420. {
  421. static int getValue (const StringPairArray& values, const String& name)
  422. {
  423. return values.getValue (name, "0").getIntValue();
  424. }
  425. static int getValue (const StringPairArray& values, const String& prefix, const char* name)
  426. {
  427. return getValue (values, prefix + name);
  428. }
  429. static void appendLabelOrNoteChunk (const StringPairArray& values, const String& prefix,
  430. const int chunkType, MemoryOutputStream& out)
  431. {
  432. auto label = values.getValue (prefix + "Text", prefix);
  433. auto labelLength = (int) label.getNumBytesAsUTF8() + 1;
  434. auto chunkLength = 4 + labelLength + (labelLength & 1);
  435. out.writeInt (chunkType);
  436. out.writeInt (chunkLength);
  437. out.writeInt (getValue (values, prefix, "Identifier"));
  438. out.write (label.toUTF8(), (size_t) labelLength);
  439. if ((out.getDataSize() & 1) != 0)
  440. out.writeByte (0);
  441. }
  442. static void appendExtraChunk (const StringPairArray& values, const String& prefix, MemoryOutputStream& out)
  443. {
  444. auto text = values.getValue (prefix + "Text", prefix);
  445. auto textLength = (int) text.getNumBytesAsUTF8() + 1; // include null terminator
  446. int chunkLength = textLength + 20 + (textLength & 1);
  447. out.writeInt (chunkName ("ltxt"));
  448. out.writeInt (chunkLength);
  449. out.writeInt (getValue (values, prefix, "Identifier"));
  450. out.writeInt (getValue (values, prefix, "SampleLength"));
  451. out.writeInt (getValue (values, prefix, "Purpose"));
  452. out.writeShort ((short) getValue (values, prefix, "Country"));
  453. out.writeShort ((short) getValue (values, prefix, "Language"));
  454. out.writeShort ((short) getValue (values, prefix, "Dialect"));
  455. out.writeShort ((short) getValue (values, prefix, "CodePage"));
  456. out.write (text.toUTF8(), (size_t) textLength);
  457. if ((out.getDataSize() & 1) != 0)
  458. out.writeByte (0);
  459. }
  460. static MemoryBlock createFrom (const StringPairArray& values)
  461. {
  462. auto numCueLabels = getValue (values, "NumCueLabels");
  463. auto numCueNotes = getValue (values, "NumCueNotes");
  464. auto numCueRegions = getValue (values, "NumCueRegions");
  465. MemoryOutputStream out;
  466. if (numCueLabels + numCueNotes + numCueRegions > 0)
  467. {
  468. out.writeInt (chunkName ("adtl"));
  469. for (int i = 0; i < numCueLabels; ++i)
  470. appendLabelOrNoteChunk (values, "CueLabel" + String (i), chunkName ("labl"), out);
  471. for (int i = 0; i < numCueNotes; ++i)
  472. appendLabelOrNoteChunk (values, "CueNote" + String (i), chunkName ("note"), out);
  473. for (int i = 0; i < numCueRegions; ++i)
  474. appendExtraChunk (values, "CueRegion" + String (i), out);
  475. }
  476. return out.getMemoryBlock();
  477. }
  478. }
  479. //==============================================================================
  480. /** Reads a RIFF List Info chunk from a stream positioned just after the size byte. */
  481. namespace ListInfoChunk
  482. {
  483. static const char* const types[] =
  484. {
  485. WavAudioFormat::riffInfoArchivalLocation,
  486. WavAudioFormat::riffInfoArtist,
  487. WavAudioFormat::riffInfoBaseURL,
  488. WavAudioFormat::riffInfoCinematographer,
  489. WavAudioFormat::riffInfoComment,
  490. WavAudioFormat::riffInfoComments,
  491. WavAudioFormat::riffInfoCommissioned,
  492. WavAudioFormat::riffInfoCopyright,
  493. WavAudioFormat::riffInfoCostumeDesigner,
  494. WavAudioFormat::riffInfoCountry,
  495. WavAudioFormat::riffInfoCropped,
  496. WavAudioFormat::riffInfoDateCreated,
  497. WavAudioFormat::riffInfoDateTimeOriginal,
  498. WavAudioFormat::riffInfoDefaultAudioStream,
  499. WavAudioFormat::riffInfoDimension,
  500. WavAudioFormat::riffInfoDirectory,
  501. WavAudioFormat::riffInfoDistributedBy,
  502. WavAudioFormat::riffInfoDotsPerInch,
  503. WavAudioFormat::riffInfoEditedBy,
  504. WavAudioFormat::riffInfoEighthLanguage,
  505. WavAudioFormat::riffInfoEncodedBy,
  506. WavAudioFormat::riffInfoEndTimecode,
  507. WavAudioFormat::riffInfoEngineer,
  508. WavAudioFormat::riffInfoFifthLanguage,
  509. WavAudioFormat::riffInfoFirstLanguage,
  510. WavAudioFormat::riffInfoFourthLanguage,
  511. WavAudioFormat::riffInfoGenre,
  512. WavAudioFormat::riffInfoKeywords,
  513. WavAudioFormat::riffInfoLanguage,
  514. WavAudioFormat::riffInfoLength,
  515. WavAudioFormat::riffInfoLightness,
  516. WavAudioFormat::riffInfoLocation,
  517. WavAudioFormat::riffInfoLogoIconURL,
  518. WavAudioFormat::riffInfoLogoURL,
  519. WavAudioFormat::riffInfoMedium,
  520. WavAudioFormat::riffInfoMoreInfoBannerImage,
  521. WavAudioFormat::riffInfoMoreInfoBannerURL,
  522. WavAudioFormat::riffInfoMoreInfoText,
  523. WavAudioFormat::riffInfoMoreInfoURL,
  524. WavAudioFormat::riffInfoMusicBy,
  525. WavAudioFormat::riffInfoNinthLanguage,
  526. WavAudioFormat::riffInfoNumberOfParts,
  527. WavAudioFormat::riffInfoOrganisation,
  528. WavAudioFormat::riffInfoPart,
  529. WavAudioFormat::riffInfoProducedBy,
  530. WavAudioFormat::riffInfoProductionDesigner,
  531. WavAudioFormat::riffInfoProductionStudio,
  532. WavAudioFormat::riffInfoRate,
  533. WavAudioFormat::riffInfoRated,
  534. WavAudioFormat::riffInfoRating,
  535. WavAudioFormat::riffInfoRippedBy,
  536. WavAudioFormat::riffInfoSecondaryGenre,
  537. WavAudioFormat::riffInfoSecondLanguage,
  538. WavAudioFormat::riffInfoSeventhLanguage,
  539. WavAudioFormat::riffInfoSharpness,
  540. WavAudioFormat::riffInfoSixthLanguage,
  541. WavAudioFormat::riffInfoSoftware,
  542. WavAudioFormat::riffInfoSoundSchemeTitle,
  543. WavAudioFormat::riffInfoSource,
  544. WavAudioFormat::riffInfoSourceFrom,
  545. WavAudioFormat::riffInfoStarring_ISTR,
  546. WavAudioFormat::riffInfoStarring_STAR,
  547. WavAudioFormat::riffInfoStartTimecode,
  548. WavAudioFormat::riffInfoStatistics,
  549. WavAudioFormat::riffInfoSubject,
  550. WavAudioFormat::riffInfoTapeName,
  551. WavAudioFormat::riffInfoTechnician,
  552. WavAudioFormat::riffInfoThirdLanguage,
  553. WavAudioFormat::riffInfoTimeCode,
  554. WavAudioFormat::riffInfoTitle,
  555. WavAudioFormat::riffInfoTrackNumber,
  556. WavAudioFormat::riffInfoURL,
  557. WavAudioFormat::riffInfoVegasVersionMajor,
  558. WavAudioFormat::riffInfoVegasVersionMinor,
  559. WavAudioFormat::riffInfoVersion,
  560. WavAudioFormat::riffInfoWatermarkURL,
  561. WavAudioFormat::riffInfoWrittenBy,
  562. WavAudioFormat::riffInfoYear
  563. };
  564. static bool isMatchingTypeIgnoringCase (const int value, const char* const name) noexcept
  565. {
  566. for (int i = 0; i < 4; ++i)
  567. if ((juce_wchar) name[i] != CharacterFunctions::toUpperCase ((juce_wchar) ((value >> (i * 8)) & 0xff)))
  568. return false;
  569. return true;
  570. }
  571. static void addToMetadata (StringPairArray& values, InputStream& input, int64 chunkEnd)
  572. {
  573. while (input.getPosition() < chunkEnd)
  574. {
  575. const int infoType = input.readInt();
  576. int64 infoLength = chunkEnd - input.getPosition();
  577. if (infoLength > 0)
  578. {
  579. infoLength = jmin (infoLength, (int64) input.readInt());
  580. if (infoLength <= 0)
  581. return;
  582. for (int i = 0; i < numElementsInArray (types); ++i)
  583. {
  584. if (isMatchingTypeIgnoringCase (infoType, types[i]))
  585. {
  586. MemoryBlock mb;
  587. input.readIntoMemoryBlock (mb, (ssize_t) infoLength);
  588. values.set (types[i], String::createStringFromData ((const char*) mb.getData(),
  589. (int) mb.getSize()));
  590. break;
  591. }
  592. }
  593. }
  594. }
  595. }
  596. static bool writeValue (const StringPairArray& values, MemoryOutputStream& out, const char* paramName)
  597. {
  598. auto value = values.getValue (paramName, {});
  599. if (value.isEmpty())
  600. return false;
  601. const int valueLength = (int) value.getNumBytesAsUTF8() + 1;
  602. const int chunkLength = valueLength + (valueLength & 1);
  603. out.writeInt (chunkName (paramName));
  604. out.writeInt (chunkLength);
  605. out.write (value.toUTF8(), (size_t) valueLength);
  606. if ((out.getDataSize() & 1) != 0)
  607. out.writeByte (0);
  608. return true;
  609. }
  610. static MemoryBlock createFrom (const StringPairArray& values)
  611. {
  612. MemoryOutputStream out;
  613. out.writeInt (chunkName ("INFO"));
  614. bool anyParamsDefined = false;
  615. for (int i = 0; i < numElementsInArray (types); ++i)
  616. if (writeValue (values, out, types[i]))
  617. anyParamsDefined = true;
  618. return anyParamsDefined ? out.getMemoryBlock() : MemoryBlock();
  619. }
  620. }
  621. //==============================================================================
  622. struct AcidChunk
  623. {
  624. /** Reads an acid RIFF chunk from a stream positioned just after the size byte. */
  625. AcidChunk (InputStream& input, size_t length)
  626. {
  627. zerostruct (*this);
  628. input.read (this, (int) jmin (sizeof (*this), length));
  629. }
  630. AcidChunk (const StringPairArray& values)
  631. {
  632. zerostruct (*this);
  633. flags = getFlagIfPresent (values, WavAudioFormat::acidOneShot, 0x01)
  634. | getFlagIfPresent (values, WavAudioFormat::acidRootSet, 0x02)
  635. | getFlagIfPresent (values, WavAudioFormat::acidStretch, 0x04)
  636. | getFlagIfPresent (values, WavAudioFormat::acidDiskBased, 0x08)
  637. | getFlagIfPresent (values, WavAudioFormat::acidizerFlag, 0x10);
  638. if (values[WavAudioFormat::acidRootSet].getIntValue() != 0)
  639. rootNote = ByteOrder::swapIfBigEndian ((uint16) values[WavAudioFormat::acidRootNote].getIntValue());
  640. numBeats = ByteOrder::swapIfBigEndian ((uint32) values[WavAudioFormat::acidBeats].getIntValue());
  641. meterDenominator = ByteOrder::swapIfBigEndian ((uint16) values[WavAudioFormat::acidDenominator].getIntValue());
  642. meterNumerator = ByteOrder::swapIfBigEndian ((uint16) values[WavAudioFormat::acidNumerator].getIntValue());
  643. if (values.containsKey (WavAudioFormat::acidTempo))
  644. tempo = swapFloatByteOrder (values[WavAudioFormat::acidTempo].getFloatValue());
  645. }
  646. static MemoryBlock createFrom (const StringPairArray& values)
  647. {
  648. return AcidChunk (values).toMemoryBlock();
  649. }
  650. MemoryBlock toMemoryBlock() const
  651. {
  652. return (flags != 0 || rootNote != 0 || numBeats != 0 || meterDenominator != 0 || meterNumerator != 0)
  653. ? MemoryBlock (this, sizeof (*this)) : MemoryBlock();
  654. }
  655. void addToMetadata (StringPairArray& values) const
  656. {
  657. setBoolFlag (values, WavAudioFormat::acidOneShot, 0x01);
  658. setBoolFlag (values, WavAudioFormat::acidRootSet, 0x02);
  659. setBoolFlag (values, WavAudioFormat::acidStretch, 0x04);
  660. setBoolFlag (values, WavAudioFormat::acidDiskBased, 0x08);
  661. setBoolFlag (values, WavAudioFormat::acidizerFlag, 0x10);
  662. if (flags & 0x02) // root note set
  663. values.set (WavAudioFormat::acidRootNote, String (ByteOrder::swapIfBigEndian (rootNote)));
  664. values.set (WavAudioFormat::acidBeats, String (ByteOrder::swapIfBigEndian (numBeats)));
  665. values.set (WavAudioFormat::acidDenominator, String (ByteOrder::swapIfBigEndian (meterDenominator)));
  666. values.set (WavAudioFormat::acidNumerator, String (ByteOrder::swapIfBigEndian (meterNumerator)));
  667. values.set (WavAudioFormat::acidTempo, String (swapFloatByteOrder (tempo)));
  668. }
  669. void setBoolFlag (StringPairArray& values, const char* name, uint32 mask) const
  670. {
  671. values.set (name, (flags & ByteOrder::swapIfBigEndian (mask)) ? "1" : "0");
  672. }
  673. static uint32 getFlagIfPresent (const StringPairArray& values, const char* name, uint32 flag)
  674. {
  675. return values[name].getIntValue() != 0 ? ByteOrder::swapIfBigEndian (flag) : 0;
  676. }
  677. static float swapFloatByteOrder (const float x) noexcept
  678. {
  679. #ifdef JUCE_BIG_ENDIAN
  680. union { uint32 asInt; float asFloat; } n;
  681. n.asFloat = x;
  682. n.asInt = ByteOrder::swap (n.asInt);
  683. return n.asFloat;
  684. #else
  685. return x;
  686. #endif
  687. }
  688. uint32 flags;
  689. uint16 rootNote;
  690. uint16 reserved1;
  691. float reserved2;
  692. uint32 numBeats;
  693. uint16 meterDenominator;
  694. uint16 meterNumerator;
  695. float tempo;
  696. } JUCE_PACKED;
  697. //==============================================================================
  698. struct TracktionChunk
  699. {
  700. static MemoryBlock createFrom (const StringPairArray& values)
  701. {
  702. MemoryOutputStream out;
  703. const String s (values[WavAudioFormat::tracktionLoopInfo]);
  704. if (s.isNotEmpty())
  705. {
  706. out.writeString (s);
  707. if ((out.getDataSize() & 1) != 0)
  708. out.writeByte (0);
  709. }
  710. return out.getMemoryBlock();
  711. }
  712. };
  713. //==============================================================================
  714. namespace AXMLChunk
  715. {
  716. static void addToMetadata (StringPairArray& destValues, const String& source)
  717. {
  718. ScopedPointer<XmlElement> xml (XmlDocument::parse (source));
  719. if (xml != nullptr && xml->hasTagName ("ebucore:ebuCoreMain"))
  720. {
  721. if (auto* xml2 = xml->getChildByName ("ebucore:coreMetadata"))
  722. {
  723. if (auto* xml3 = xml2->getChildByName ("ebucore:identifier"))
  724. {
  725. if (auto* xml4 = xml3->getChildByName ("dc:identifier"))
  726. {
  727. auto ISRCCode = xml4->getAllSubText().fromFirstOccurrenceOf ("ISRC:", false, true);
  728. if (ISRCCode.isNotEmpty())
  729. destValues.set (WavAudioFormat::ISRC, ISRCCode);
  730. }
  731. }
  732. }
  733. }
  734. }
  735. static MemoryBlock createFrom (const StringPairArray& values)
  736. {
  737. auto ISRC = values.getValue (WavAudioFormat::ISRC, {});
  738. MemoryOutputStream xml;
  739. if (ISRC.isNotEmpty())
  740. {
  741. xml << "<ebucore:ebuCoreMain xmlns:dc=\" http://purl.org/dc/elements/1.1/\" "
  742. "xmlns:ebucore=\"urn:ebu:metadata-schema:ebuCore_2012\">"
  743. "<ebucore:coreMetadata>"
  744. "<ebucore:identifier typeLabel=\"GUID\" "
  745. "typeDefinition=\"Globally Unique Identifier\" "
  746. "formatLabel=\"ISRC\" "
  747. "formatDefinition=\"International Standard Recording Code\" "
  748. "formatLink=\"http://www.ebu.ch/metadata/cs/ebu_IdentifierTypeCodeCS.xml#3.7\">"
  749. "<dc:identifier>ISRC:" << ISRC << "</dc:identifier>"
  750. "</ebucore:identifier>"
  751. "</ebucore:coreMetadata>"
  752. "</ebucore:ebuCoreMain>";
  753. xml.writeRepeatedByte (0, xml.getDataSize()); // ensures even size, null termination and room for future growing
  754. }
  755. return xml.getMemoryBlock();
  756. }
  757. };
  758. //==============================================================================
  759. struct ExtensibleWavSubFormat
  760. {
  761. uint32 data1;
  762. uint16 data2;
  763. uint16 data3;
  764. uint8 data4[8];
  765. bool operator== (const ExtensibleWavSubFormat& other) const noexcept { return memcmp (this, &other, sizeof (*this)) == 0; }
  766. bool operator!= (const ExtensibleWavSubFormat& other) const noexcept { return ! operator== (other); }
  767. } JUCE_PACKED;
  768. static const ExtensibleWavSubFormat pcmFormat = { 0x00000001, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 } };
  769. static const ExtensibleWavSubFormat IEEEFloatFormat = { 0x00000003, 0x0000, 0x0010, { 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 } };
  770. static const ExtensibleWavSubFormat ambisonicFormat = { 0x00000001, 0x0721, 0x11d3, { 0x86, 0x44, 0xC8, 0xC1, 0xCA, 0x00, 0x00, 0x00 } };
  771. struct DataSize64Chunk // chunk ID = 'ds64' if data size > 0xffffffff, 'JUNK' otherwise
  772. {
  773. uint32 riffSizeLow; // low 4 byte size of RF64 block
  774. uint32 riffSizeHigh; // high 4 byte size of RF64 block
  775. uint32 dataSizeLow; // low 4 byte size of data chunk
  776. uint32 dataSizeHigh; // high 4 byte size of data chunk
  777. uint32 sampleCountLow; // low 4 byte sample count of fact chunk
  778. uint32 sampleCountHigh; // high 4 byte sample count of fact chunk
  779. uint32 tableLength; // number of valid entries in array 'table'
  780. } JUCE_PACKED;
  781. #if JUCE_MSVC
  782. #pragma pack (pop)
  783. #endif
  784. }
  785. //==============================================================================
  786. class WavAudioFormatReader : public AudioFormatReader
  787. {
  788. public:
  789. WavAudioFormatReader (InputStream* const in)
  790. : AudioFormatReader (in, wavFormatName)
  791. {
  792. using namespace WavFileHelpers;
  793. uint64 len = 0;
  794. uint64 end = 0;
  795. int cueNoteIndex = 0;
  796. int cueLabelIndex = 0;
  797. int cueRegionIndex = 0;
  798. const int firstChunkType = input->readInt();
  799. if (firstChunkType == chunkName ("RF64"))
  800. {
  801. input->skipNextBytes (4); // size is -1 for RF64
  802. isRF64 = true;
  803. }
  804. else if (firstChunkType == chunkName ("RIFF"))
  805. {
  806. len = (uint64) (uint32) input->readInt();
  807. end = len + (uint64) input->getPosition();
  808. }
  809. else
  810. {
  811. return;
  812. }
  813. auto startOfRIFFChunk = input->getPosition();
  814. if (input->readInt() == chunkName ("WAVE"))
  815. {
  816. if (isRF64 && input->readInt() == chunkName ("ds64"))
  817. {
  818. const uint32 length = (uint32) input->readInt();
  819. if (length < 28)
  820. return;
  821. const int64 chunkEnd = input->getPosition() + length + (length & 1);
  822. len = (uint64) input->readInt64();
  823. end = len + (uint64) startOfRIFFChunk;
  824. dataLength = input->readInt64();
  825. input->setPosition (chunkEnd);
  826. }
  827. while ((uint64) input->getPosition() < end && ! input->isExhausted())
  828. {
  829. const int chunkType = input->readInt();
  830. uint32 length = (uint32) input->readInt();
  831. const int64 chunkEnd = input->getPosition() + length + (length & 1);
  832. if (chunkType == chunkName ("fmt "))
  833. {
  834. // read the format chunk
  835. auto format = (unsigned short) input->readShort();
  836. numChannels = (unsigned int) input->readShort();
  837. sampleRate = input->readInt();
  838. auto bytesPerSec = input->readInt();
  839. input->skipNextBytes (2);
  840. bitsPerSample = (unsigned int) (int) input->readShort();
  841. if (bitsPerSample > 64)
  842. {
  843. bytesPerFrame = bytesPerSec / (int) sampleRate;
  844. bitsPerSample = 8 * (unsigned int) bytesPerFrame / numChannels;
  845. }
  846. else
  847. {
  848. bytesPerFrame = numChannels * bitsPerSample / 8;
  849. }
  850. if (format == 3)
  851. {
  852. usesFloatingPointData = true;
  853. }
  854. else if (format == 0xfffe /*WAVE_FORMAT_EXTENSIBLE*/)
  855. {
  856. if (length < 40) // too short
  857. {
  858. bytesPerFrame = 0;
  859. }
  860. else
  861. {
  862. input->skipNextBytes (4); // skip over size and bitsPerSample
  863. auto channelMask = input->readInt();
  864. metadataValues.set ("ChannelMask", String (channelMask));
  865. // AudioChannelSet and wav's dwChannelMask are compatible
  866. BigInteger channelBits (channelMask);
  867. for (auto bit = channelBits.findNextSetBit (0); bit >= 0; bit = channelBits.findNextSetBit (bit + 1))
  868. channelLayout.addChannel (static_cast<AudioChannelSet::ChannelType> (bit + 1));
  869. // channel layout and number of channels do not match
  870. jassert (channelLayout.size() == static_cast<int> (numChannels));
  871. ExtensibleWavSubFormat subFormat;
  872. subFormat.data1 = (uint32) input->readInt();
  873. subFormat.data2 = (uint16) input->readShort();
  874. subFormat.data3 = (uint16) input->readShort();
  875. input->read (subFormat.data4, sizeof (subFormat.data4));
  876. if (subFormat == IEEEFloatFormat)
  877. usesFloatingPointData = true;
  878. else if (subFormat != pcmFormat && subFormat != ambisonicFormat)
  879. bytesPerFrame = 0;
  880. }
  881. }
  882. else if (format != 1)
  883. {
  884. bytesPerFrame = 0;
  885. }
  886. }
  887. else if (chunkType == chunkName ("data"))
  888. {
  889. if (! isRF64) // data size is expected to be -1, actual data size is in ds64 chunk
  890. dataLength = length;
  891. dataChunkStart = input->getPosition();
  892. lengthInSamples = (bytesPerFrame > 0) ? (dataLength / bytesPerFrame) : 0;
  893. }
  894. else if (chunkType == chunkName ("bext"))
  895. {
  896. bwavChunkStart = input->getPosition();
  897. bwavSize = length;
  898. HeapBlock<BWAVChunk> bwav;
  899. bwav.calloc (jmax ((size_t) length + 1, sizeof (BWAVChunk)), 1);
  900. input->read (bwav, (int) length);
  901. bwav->copyTo (metadataValues, (int) length);
  902. }
  903. else if (chunkType == chunkName ("smpl"))
  904. {
  905. HeapBlock<SMPLChunk> smpl;
  906. smpl.calloc (jmax ((size_t) length + 1, sizeof (SMPLChunk)), 1);
  907. input->read (smpl, (int) length);
  908. smpl->copyTo (metadataValues, (int) length);
  909. }
  910. else if (chunkType == chunkName ("inst") || chunkType == chunkName ("INST")) // need to check which...
  911. {
  912. HeapBlock<InstChunk> inst;
  913. inst.calloc (jmax ((size_t) length + 1, sizeof (InstChunk)), 1);
  914. input->read (inst, (int) length);
  915. inst->copyTo (metadataValues);
  916. }
  917. else if (chunkType == chunkName ("cue "))
  918. {
  919. HeapBlock<CueChunk> cue;
  920. cue.calloc (jmax ((size_t) length + 1, sizeof (CueChunk)), 1);
  921. input->read (cue, (int) length);
  922. cue->copyTo (metadataValues, (int) length);
  923. }
  924. else if (chunkType == chunkName ("axml"))
  925. {
  926. MemoryBlock axml;
  927. input->readIntoMemoryBlock (axml, (ssize_t) length);
  928. AXMLChunk::addToMetadata (metadataValues, axml.toString());
  929. }
  930. else if (chunkType == chunkName ("LIST"))
  931. {
  932. const int subChunkType = input->readInt();
  933. if (subChunkType == chunkName ("info") || subChunkType == chunkName ("INFO"))
  934. {
  935. ListInfoChunk::addToMetadata (metadataValues, *input, chunkEnd);
  936. }
  937. else if (subChunkType == chunkName ("adtl"))
  938. {
  939. while (input->getPosition() < chunkEnd)
  940. {
  941. const int adtlChunkType = input->readInt();
  942. const uint32 adtlLength = (uint32) input->readInt();
  943. const int64 adtlChunkEnd = input->getPosition() + (adtlLength + (adtlLength & 1));
  944. if (adtlChunkType == chunkName ("labl") || adtlChunkType == chunkName ("note"))
  945. {
  946. String prefix;
  947. if (adtlChunkType == chunkName ("labl"))
  948. prefix << "CueLabel" << cueLabelIndex++;
  949. else if (adtlChunkType == chunkName ("note"))
  950. prefix << "CueNote" << cueNoteIndex++;
  951. const uint32 identifier = (uint32) input->readInt();
  952. const int stringLength = (int) adtlLength - 4;
  953. MemoryBlock textBlock;
  954. input->readIntoMemoryBlock (textBlock, stringLength);
  955. metadataValues.set (prefix + "Identifier", String (identifier));
  956. metadataValues.set (prefix + "Text", textBlock.toString());
  957. }
  958. else if (adtlChunkType == chunkName ("ltxt"))
  959. {
  960. const String prefix ("CueRegion" + String (cueRegionIndex++));
  961. const uint32 identifier = (uint32) input->readInt();
  962. const uint32 sampleLength = (uint32) input->readInt();
  963. const uint32 purpose = (uint32) input->readInt();
  964. const uint16 country = (uint16) input->readInt();
  965. const uint16 language = (uint16) input->readInt();
  966. const uint16 dialect = (uint16) input->readInt();
  967. const uint16 codePage = (uint16) input->readInt();
  968. const uint32 stringLength = adtlLength - 20;
  969. MemoryBlock textBlock;
  970. input->readIntoMemoryBlock (textBlock, (int) stringLength);
  971. metadataValues.set (prefix + "Identifier", String (identifier));
  972. metadataValues.set (prefix + "SampleLength", String (sampleLength));
  973. metadataValues.set (prefix + "Purpose", String (purpose));
  974. metadataValues.set (prefix + "Country", String (country));
  975. metadataValues.set (prefix + "Language", String (language));
  976. metadataValues.set (prefix + "Dialect", String (dialect));
  977. metadataValues.set (prefix + "CodePage", String (codePage));
  978. metadataValues.set (prefix + "Text", textBlock.toString());
  979. }
  980. input->setPosition (adtlChunkEnd);
  981. }
  982. }
  983. }
  984. else if (chunkType == chunkName ("acid"))
  985. {
  986. AcidChunk (*input, length).addToMetadata (metadataValues);
  987. }
  988. else if (chunkType == chunkName ("Trkn"))
  989. {
  990. MemoryBlock tracktion;
  991. input->readIntoMemoryBlock (tracktion, (ssize_t) length);
  992. metadataValues.set (WavAudioFormat::tracktionLoopInfo, tracktion.toString());
  993. }
  994. else if (chunkEnd <= input->getPosition())
  995. {
  996. break;
  997. }
  998. input->setPosition (chunkEnd);
  999. }
  1000. }
  1001. if (cueLabelIndex > 0) metadataValues.set ("NumCueLabels", String (cueLabelIndex));
  1002. if (cueNoteIndex > 0) metadataValues.set ("NumCueNotes", String (cueNoteIndex));
  1003. if (cueRegionIndex > 0) metadataValues.set ("NumCueRegions", String (cueRegionIndex));
  1004. if (metadataValues.size() > 0) metadataValues.set ("MetaDataSource", "WAV");
  1005. }
  1006. //==============================================================================
  1007. bool readSamples (int** destSamples, int numDestChannels, int startOffsetInDestBuffer,
  1008. int64 startSampleInFile, int numSamples) override
  1009. {
  1010. clearSamplesBeyondAvailableLength (destSamples, numDestChannels, startOffsetInDestBuffer,
  1011. startSampleInFile, numSamples, lengthInSamples);
  1012. if (numSamples <= 0)
  1013. return true;
  1014. input->setPosition (dataChunkStart + startSampleInFile * bytesPerFrame);
  1015. while (numSamples > 0)
  1016. {
  1017. const int tempBufSize = 480 * 3 * 4; // (keep this a multiple of 3)
  1018. char tempBuffer [tempBufSize];
  1019. const int numThisTime = jmin (tempBufSize / bytesPerFrame, numSamples);
  1020. const int bytesRead = input->read (tempBuffer, numThisTime * bytesPerFrame);
  1021. if (bytesRead < numThisTime * bytesPerFrame)
  1022. {
  1023. jassert (bytesRead >= 0);
  1024. zeromem (tempBuffer + bytesRead, (size_t) (numThisTime * bytesPerFrame - bytesRead));
  1025. }
  1026. copySampleData (bitsPerSample, usesFloatingPointData,
  1027. destSamples, startOffsetInDestBuffer, numDestChannels,
  1028. tempBuffer, (int) numChannels, numThisTime);
  1029. startOffsetInDestBuffer += numThisTime;
  1030. numSamples -= numThisTime;
  1031. }
  1032. return true;
  1033. }
  1034. static void copySampleData (unsigned int bitsPerSample, const bool usesFloatingPointData,
  1035. int* const* destSamples, int startOffsetInDestBuffer, int numDestChannels,
  1036. const void* sourceData, int numChannels, int numSamples) noexcept
  1037. {
  1038. switch (bitsPerSample)
  1039. {
  1040. case 8: ReadHelper<AudioData::Int32, AudioData::UInt8, AudioData::LittleEndian>::read (destSamples, startOffsetInDestBuffer, numDestChannels, sourceData, numChannels, numSamples); break;
  1041. case 16: ReadHelper<AudioData::Int32, AudioData::Int16, AudioData::LittleEndian>::read (destSamples, startOffsetInDestBuffer, numDestChannels, sourceData, numChannels, numSamples); break;
  1042. case 24: ReadHelper<AudioData::Int32, AudioData::Int24, AudioData::LittleEndian>::read (destSamples, startOffsetInDestBuffer, numDestChannels, sourceData, numChannels, numSamples); break;
  1043. case 32: if (usesFloatingPointData) ReadHelper<AudioData::Float32, AudioData::Float32, AudioData::LittleEndian>::read (destSamples, startOffsetInDestBuffer, numDestChannels, sourceData, numChannels, numSamples);
  1044. else ReadHelper<AudioData::Int32, AudioData::Int32, AudioData::LittleEndian>::read (destSamples, startOffsetInDestBuffer, numDestChannels, sourceData, numChannels, numSamples);
  1045. break;
  1046. default: jassertfalse; break;
  1047. }
  1048. }
  1049. AudioChannelSet getChannelLayout() override
  1050. {
  1051. if (channelLayout.size() == static_cast<int> (numChannels))
  1052. return channelLayout;
  1053. return WavFileHelpers::canonicalWavChannelSet (static_cast<int> (numChannels));
  1054. }
  1055. int64 bwavChunkStart = 0, bwavSize = 0;
  1056. int64 dataChunkStart = 0, dataLength = 0;
  1057. int bytesPerFrame = 0;
  1058. bool isRF64 = false;
  1059. AudioChannelSet channelLayout;
  1060. private:
  1061. JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (WavAudioFormatReader)
  1062. };
  1063. //==============================================================================
  1064. class WavAudioFormatWriter : public AudioFormatWriter
  1065. {
  1066. public:
  1067. WavAudioFormatWriter (OutputStream* const out, const double rate,
  1068. const AudioChannelSet& channelLayoutToUse, const unsigned int bits,
  1069. const StringPairArray& metadataValues)
  1070. : AudioFormatWriter (out, wavFormatName, rate, channelLayoutToUse, bits)
  1071. {
  1072. using namespace WavFileHelpers;
  1073. if (metadataValues.size() > 0)
  1074. {
  1075. // The meta data should have been sanitised for the WAV format.
  1076. // If it was originally sourced from an AIFF file the MetaDataSource
  1077. // key should be removed (or set to "WAV") once this has been done
  1078. jassert (metadataValues.getValue ("MetaDataSource", "None") != "AIFF");
  1079. bwavChunk = BWAVChunk::createFrom (metadataValues);
  1080. axmlChunk = AXMLChunk::createFrom (metadataValues);
  1081. smplChunk = SMPLChunk::createFrom (metadataValues);
  1082. instChunk = InstChunk::createFrom (metadataValues);
  1083. cueChunk = CueChunk ::createFrom (metadataValues);
  1084. listChunk = ListChunk::createFrom (metadataValues);
  1085. listInfoChunk = ListInfoChunk::createFrom (metadataValues);
  1086. acidChunk = AcidChunk::createFrom (metadataValues);
  1087. trckChunk = TracktionChunk::createFrom (metadataValues);
  1088. }
  1089. headerPosition = out->getPosition();
  1090. writeHeader();
  1091. }
  1092. ~WavAudioFormatWriter()
  1093. {
  1094. writeHeader();
  1095. }
  1096. //==============================================================================
  1097. bool write (const int** data, int numSamples) override
  1098. {
  1099. jassert (numSamples >= 0);
  1100. jassert (data != nullptr && *data != nullptr); // the input must contain at least one channel!
  1101. if (writeFailed)
  1102. return false;
  1103. const size_t bytes = numChannels * (size_t) numSamples * bitsPerSample / 8;
  1104. tempBlock.ensureSize (bytes, false);
  1105. switch (bitsPerSample)
  1106. {
  1107. case 8: WriteHelper<AudioData::UInt8, AudioData::Int32, AudioData::LittleEndian>::write (tempBlock.getData(), (int) numChannels, data, numSamples); break;
  1108. case 16: WriteHelper<AudioData::Int16, AudioData::Int32, AudioData::LittleEndian>::write (tempBlock.getData(), (int) numChannels, data, numSamples); break;
  1109. case 24: WriteHelper<AudioData::Int24, AudioData::Int32, AudioData::LittleEndian>::write (tempBlock.getData(), (int) numChannels, data, numSamples); break;
  1110. case 32: WriteHelper<AudioData::Int32, AudioData::Int32, AudioData::LittleEndian>::write (tempBlock.getData(), (int) numChannels, data, numSamples); break;
  1111. default: jassertfalse; break;
  1112. }
  1113. if (! output->write (tempBlock.getData(), bytes))
  1114. {
  1115. // failed to write to disk, so let's try writing the header.
  1116. // If it's just run out of disk space, then if it does manage
  1117. // to write the header, we'll still have a useable file..
  1118. writeHeader();
  1119. writeFailed = true;
  1120. return false;
  1121. }
  1122. bytesWritten += bytes;
  1123. lengthInSamples += (uint64) numSamples;
  1124. return true;
  1125. }
  1126. bool flush() override
  1127. {
  1128. auto lastWritePos = output->getPosition();
  1129. writeHeader();
  1130. if (output->setPosition (lastWritePos))
  1131. return true;
  1132. // if this fails, you've given it an output stream that can't seek! It needs
  1133. // to be able to seek back to write the header
  1134. jassertfalse;
  1135. return false;
  1136. }
  1137. private:
  1138. MemoryBlock tempBlock, bwavChunk, axmlChunk, smplChunk, instChunk, cueChunk, listChunk, listInfoChunk, acidChunk, trckChunk;
  1139. uint64 lengthInSamples = 0, bytesWritten = 0;
  1140. int64 headerPosition = 0;
  1141. bool writeFailed = false;
  1142. void writeHeader()
  1143. {
  1144. if ((bytesWritten & 1) != 0) // pad to an even length
  1145. output->writeByte (0);
  1146. using namespace WavFileHelpers;
  1147. if (headerPosition != output->getPosition() && ! output->setPosition (headerPosition))
  1148. {
  1149. // if this fails, you've given it an output stream that can't seek! It needs to be
  1150. // able to seek back to go back and write the header after the data has been written.
  1151. jassertfalse;
  1152. return;
  1153. }
  1154. const size_t bytesPerFrame = numChannels * bitsPerSample / 8;
  1155. uint64 audioDataSize = bytesPerFrame * lengthInSamples;
  1156. auto channelMask = getChannelMaskFromChannelLayout (channelLayout);
  1157. const bool isRF64 = (bytesWritten >= 0x100000000LL);
  1158. const bool isWaveFmtEx = isRF64 || (channelMask != 0);
  1159. int64 riffChunkSize = (int64) (4 /* 'RIFF' */ + 8 + 40 /* WAVEFORMATEX */
  1160. + 8 + audioDataSize + (audioDataSize & 1)
  1161. + chunkSize (bwavChunk)
  1162. + chunkSize (axmlChunk)
  1163. + chunkSize (smplChunk)
  1164. + chunkSize (instChunk)
  1165. + chunkSize (cueChunk)
  1166. + chunkSize (listChunk)
  1167. + chunkSize (listInfoChunk)
  1168. + chunkSize (acidChunk)
  1169. + chunkSize (trckChunk)
  1170. + (8 + 28)); // (ds64 chunk)
  1171. riffChunkSize += (riffChunkSize & 1);
  1172. if (isRF64)
  1173. writeChunkHeader (chunkName ("RF64"), -1);
  1174. else
  1175. writeChunkHeader (chunkName ("RIFF"), (int) riffChunkSize);
  1176. output->writeInt (chunkName ("WAVE"));
  1177. if (! isRF64)
  1178. {
  1179. #if ! JUCE_WAV_DO_NOT_PAD_HEADER_SIZE
  1180. /* NB: This junk chunk is added for padding, so that the header is a fixed size
  1181. regardless of whether it's RF64 or not. That way, we can begin recording a file,
  1182. and when it's finished, can go back and write either a RIFF or RF64 header,
  1183. depending on whether more than 2^32 samples were written.
  1184. The JUCE_WAV_DO_NOT_PAD_HEADER_SIZE macro allows you to disable this feature in case
  1185. you need to create files for crappy WAV players with bugs that stop them skipping chunks
  1186. which they don't recognise. But DO NOT USE THIS option unless you really have no choice,
  1187. because it means that if you write more than 2^32 samples to the file, you'll corrupt it.
  1188. */
  1189. writeChunkHeader (chunkName ("JUNK"), 28 + (isWaveFmtEx? 0 : 24));
  1190. output->writeRepeatedByte (0, 28 /* ds64 */ + (isWaveFmtEx? 0 : 24));
  1191. #endif
  1192. }
  1193. else
  1194. {
  1195. #if JUCE_WAV_DO_NOT_PAD_HEADER_SIZE
  1196. // If you disable padding, then you MUST NOT write more than 2^32 samples to a file.
  1197. jassertfalse;
  1198. #endif
  1199. writeChunkHeader (chunkName ("ds64"), 28); // chunk size for uncompressed data (no table)
  1200. output->writeInt64 (riffChunkSize);
  1201. output->writeInt64 ((int64) audioDataSize);
  1202. output->writeRepeatedByte (0, 12);
  1203. }
  1204. if (isWaveFmtEx)
  1205. {
  1206. writeChunkHeader (chunkName ("fmt "), 40);
  1207. output->writeShort ((short) (uint16) 0xfffe); // WAVE_FORMAT_EXTENSIBLE
  1208. }
  1209. else
  1210. {
  1211. writeChunkHeader (chunkName ("fmt "), 16);
  1212. output->writeShort (bitsPerSample < 32 ? (short) 1 /*WAVE_FORMAT_PCM*/
  1213. : (short) 3 /*WAVE_FORMAT_IEEE_FLOAT*/);
  1214. }
  1215. output->writeShort ((short) numChannels);
  1216. output->writeInt ((int) sampleRate);
  1217. output->writeInt ((int) (bytesPerFrame * sampleRate)); // nAvgBytesPerSec
  1218. output->writeShort ((short) bytesPerFrame); // nBlockAlign
  1219. output->writeShort ((short) bitsPerSample); // wBitsPerSample
  1220. if (isWaveFmtEx)
  1221. {
  1222. output->writeShort (22); // cbSize (size of the extension)
  1223. output->writeShort ((short) bitsPerSample); // wValidBitsPerSample
  1224. output->writeInt (channelMask);
  1225. const ExtensibleWavSubFormat& subFormat = bitsPerSample < 32 ? pcmFormat : IEEEFloatFormat;
  1226. output->writeInt ((int) subFormat.data1);
  1227. output->writeShort ((short) subFormat.data2);
  1228. output->writeShort ((short) subFormat.data3);
  1229. output->write (subFormat.data4, sizeof (subFormat.data4));
  1230. }
  1231. writeChunk (bwavChunk, chunkName ("bext"));
  1232. writeChunk (axmlChunk, chunkName ("axml"));
  1233. writeChunk (smplChunk, chunkName ("smpl"));
  1234. writeChunk (instChunk, chunkName ("inst"), 7);
  1235. writeChunk (cueChunk, chunkName ("cue "));
  1236. writeChunk (listChunk, chunkName ("LIST"));
  1237. writeChunk (listInfoChunk, chunkName ("LIST"));
  1238. writeChunk (acidChunk, chunkName ("acid"));
  1239. writeChunk (trckChunk, chunkName ("Trkn"));
  1240. writeChunkHeader (chunkName ("data"), isRF64 ? -1 : (int) (lengthInSamples * bytesPerFrame));
  1241. usesFloatingPointData = (bitsPerSample == 32);
  1242. }
  1243. static size_t chunkSize (const MemoryBlock& data) noexcept { return data.getSize() > 0 ? (8 + data.getSize()) : 0; }
  1244. void writeChunkHeader (int chunkType, int size) const
  1245. {
  1246. output->writeInt (chunkType);
  1247. output->writeInt (size);
  1248. }
  1249. void writeChunk (const MemoryBlock& data, int chunkType, int size = 0) const
  1250. {
  1251. if (data.getSize() > 0)
  1252. {
  1253. writeChunkHeader (chunkType, size != 0 ? size : (int) data.getSize());
  1254. *output << data;
  1255. }
  1256. }
  1257. static int getChannelMaskFromChannelLayout (const AudioChannelSet& channelLayout)
  1258. {
  1259. if (channelLayout.isDiscreteLayout())
  1260. return 0;
  1261. // Don't add an extended format chunk for mono and stereo. Basically, all wav players
  1262. // interpret a wav file with only one or two channels to be mono or stereo anyway.
  1263. if (channelLayout == AudioChannelSet::mono() || channelLayout == AudioChannelSet::stereo())
  1264. return 0;
  1265. auto channels = channelLayout.getChannelTypes();
  1266. auto wavChannelMask = 0;
  1267. for (auto channel : channels)
  1268. {
  1269. int wavChannelBit = static_cast<int> (channel) - 1;
  1270. jassert (wavChannelBit >= 0 && wavChannelBit <= 31);
  1271. wavChannelMask |= (1 << wavChannelBit);
  1272. }
  1273. return wavChannelMask;
  1274. }
  1275. JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (WavAudioFormatWriter)
  1276. };
  1277. //==============================================================================
  1278. class MemoryMappedWavReader : public MemoryMappedAudioFormatReader
  1279. {
  1280. public:
  1281. MemoryMappedWavReader (const File& wavFile, const WavAudioFormatReader& reader)
  1282. : MemoryMappedAudioFormatReader (wavFile, reader, reader.dataChunkStart,
  1283. reader.dataLength, reader.bytesPerFrame)
  1284. {
  1285. }
  1286. bool readSamples (int** destSamples, int numDestChannels, int startOffsetInDestBuffer,
  1287. int64 startSampleInFile, int numSamples) override
  1288. {
  1289. clearSamplesBeyondAvailableLength (destSamples, numDestChannels, startOffsetInDestBuffer,
  1290. startSampleInFile, numSamples, lengthInSamples);
  1291. if (map == nullptr || ! mappedSection.contains (Range<int64> (startSampleInFile, startSampleInFile + numSamples)))
  1292. {
  1293. jassertfalse; // you must make sure that the window contains all the samples you're going to attempt to read.
  1294. return false;
  1295. }
  1296. WavAudioFormatReader::copySampleData (bitsPerSample, usesFloatingPointData,
  1297. destSamples, startOffsetInDestBuffer, numDestChannels,
  1298. sampleToPointer (startSampleInFile), (int) numChannels, numSamples);
  1299. return true;
  1300. }
  1301. void getSample (int64 sample, float* result) const noexcept override
  1302. {
  1303. auto num = (int) numChannels;
  1304. if (map == nullptr || ! mappedSection.contains (sample))
  1305. {
  1306. jassertfalse; // you must make sure that the window contains all the samples you're going to attempt to read.
  1307. zeromem (result, sizeof (float) * (size_t) num);
  1308. return;
  1309. }
  1310. auto dest = &result;
  1311. auto source = sampleToPointer (sample);
  1312. switch (bitsPerSample)
  1313. {
  1314. case 8: ReadHelper<AudioData::Float32, AudioData::UInt8, AudioData::LittleEndian>::read (dest, 0, 1, source, 1, num); break;
  1315. case 16: ReadHelper<AudioData::Float32, AudioData::Int16, AudioData::LittleEndian>::read (dest, 0, 1, source, 1, num); break;
  1316. case 24: ReadHelper<AudioData::Float32, AudioData::Int24, AudioData::LittleEndian>::read (dest, 0, 1, source, 1, num); break;
  1317. case 32: if (usesFloatingPointData) ReadHelper<AudioData::Float32, AudioData::Float32, AudioData::LittleEndian>::read (dest, 0, 1, source, 1, num);
  1318. else ReadHelper<AudioData::Float32, AudioData::Int32, AudioData::LittleEndian>::read (dest, 0, 1, source, 1, num);
  1319. break;
  1320. default: jassertfalse; break;
  1321. }
  1322. }
  1323. void readMaxLevels (int64 startSampleInFile, int64 numSamples, Range<float>* results, int numChannelsToRead) override
  1324. {
  1325. numSamples = jmin (numSamples, lengthInSamples - startSampleInFile);
  1326. if (map == nullptr || numSamples <= 0 || ! mappedSection.contains (Range<int64> (startSampleInFile, startSampleInFile + numSamples)))
  1327. {
  1328. jassert (numSamples <= 0); // you must make sure that the window contains all the samples you're going to attempt to read.
  1329. for (int i = 0; i < numChannelsToRead; ++i)
  1330. results[i] = Range<float>();
  1331. return;
  1332. }
  1333. switch (bitsPerSample)
  1334. {
  1335. case 8: scanMinAndMax<AudioData::UInt8> (startSampleInFile, numSamples, results, numChannelsToRead); break;
  1336. case 16: scanMinAndMax<AudioData::Int16> (startSampleInFile, numSamples, results, numChannelsToRead); break;
  1337. case 24: scanMinAndMax<AudioData::Int24> (startSampleInFile, numSamples, results, numChannelsToRead); break;
  1338. case 32: if (usesFloatingPointData) scanMinAndMax<AudioData::Float32> (startSampleInFile, numSamples, results, numChannelsToRead);
  1339. else scanMinAndMax<AudioData::Int32> (startSampleInFile, numSamples, results, numChannelsToRead);
  1340. break;
  1341. default: jassertfalse; break;
  1342. }
  1343. }
  1344. private:
  1345. template <typename SampleType>
  1346. void scanMinAndMax (int64 startSampleInFile, int64 numSamples, Range<float>* results, int numChannelsToRead) const noexcept
  1347. {
  1348. for (int i = 0; i < numChannelsToRead; ++i)
  1349. results[i] = scanMinAndMaxInterleaved<SampleType, AudioData::LittleEndian> (i, startSampleInFile, numSamples);
  1350. }
  1351. JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (MemoryMappedWavReader)
  1352. };
  1353. //==============================================================================
  1354. WavAudioFormat::WavAudioFormat() : AudioFormat (wavFormatName, ".wav .bwf") {}
  1355. WavAudioFormat::~WavAudioFormat() {}
  1356. Array<int> WavAudioFormat::getPossibleSampleRates()
  1357. {
  1358. const int rates[] = { 8000, 11025, 12000, 16000, 22050, 32000, 44100,
  1359. 48000, 88200, 96000, 176400, 192000, 352800, 384000 };
  1360. return Array<int> (rates, numElementsInArray (rates));
  1361. }
  1362. Array<int> WavAudioFormat::getPossibleBitDepths()
  1363. {
  1364. const int depths[] = { 8, 16, 24, 32 };
  1365. return Array<int> (depths, numElementsInArray (depths));
  1366. }
  1367. bool WavAudioFormat::canDoStereo() { return true; }
  1368. bool WavAudioFormat::canDoMono() { return true; }
  1369. bool WavAudioFormat::isChannelLayoutSupported (const AudioChannelSet& channelSet)
  1370. {
  1371. auto channelTypes = channelSet.getChannelTypes();
  1372. // When
  1373. if (channelSet.isDiscreteLayout())
  1374. return true;
  1375. // WAV supports all channel types from left ... topRearRight
  1376. for (auto channel : channelTypes)
  1377. if (channel < AudioChannelSet::left || channel > AudioChannelSet::topRearRight)
  1378. return false;
  1379. return true;
  1380. }
  1381. AudioFormatReader* WavAudioFormat::createReaderFor (InputStream* sourceStream,
  1382. const bool deleteStreamIfOpeningFails)
  1383. {
  1384. ScopedPointer<WavAudioFormatReader> r (new WavAudioFormatReader (sourceStream));
  1385. if (r->sampleRate > 0 && r->numChannels > 0 && r->bytesPerFrame > 0)
  1386. return r.release();
  1387. if (! deleteStreamIfOpeningFails)
  1388. r->input = nullptr;
  1389. return nullptr;
  1390. }
  1391. MemoryMappedAudioFormatReader* WavAudioFormat::createMemoryMappedReader (const File& file)
  1392. {
  1393. return createMemoryMappedReader (file.createInputStream());
  1394. }
  1395. MemoryMappedAudioFormatReader* WavAudioFormat::createMemoryMappedReader (FileInputStream* fin)
  1396. {
  1397. if (fin != nullptr)
  1398. {
  1399. WavAudioFormatReader reader (fin);
  1400. if (reader.lengthInSamples > 0)
  1401. return new MemoryMappedWavReader (fin->getFile(), reader);
  1402. }
  1403. return nullptr;
  1404. }
  1405. AudioFormatWriter* WavAudioFormat::createWriterFor (OutputStream* out, double sampleRate,
  1406. unsigned int numChannels, int bitsPerSample,
  1407. const StringPairArray& metadataValues, int qualityOptionIndex)
  1408. {
  1409. return createWriterFor (out, sampleRate, WavFileHelpers::canonicalWavChannelSet (static_cast<int> (numChannels)),
  1410. bitsPerSample, metadataValues, qualityOptionIndex);
  1411. }
  1412. AudioFormatWriter* WavAudioFormat::createWriterFor (OutputStream* out,
  1413. double sampleRate,
  1414. const AudioChannelSet& channelLayout,
  1415. int bitsPerSample,
  1416. const StringPairArray& metadataValues,
  1417. int /*qualityOptionIndex*/)
  1418. {
  1419. if (out != nullptr && getPossibleBitDepths().contains (bitsPerSample) && isChannelLayoutSupported (channelLayout))
  1420. return new WavAudioFormatWriter (out, sampleRate, channelLayout,
  1421. (unsigned int) bitsPerSample, metadataValues);
  1422. return nullptr;
  1423. }
  1424. namespace WavFileHelpers
  1425. {
  1426. static bool slowCopyWavFileWithNewMetadata (const File& file, const StringPairArray& metadata)
  1427. {
  1428. TemporaryFile tempFile (file);
  1429. WavAudioFormat wav;
  1430. if (ScopedPointer<AudioFormatReader> reader = wav.createReaderFor (file.createInputStream(), true))
  1431. {
  1432. if (ScopedPointer<OutputStream> outStream = tempFile.getFile().createOutputStream())
  1433. {
  1434. if (ScopedPointer<AudioFormatWriter> writer = wav.createWriterFor (outStream, reader->sampleRate,
  1435. reader->numChannels, (int) reader->bitsPerSample,
  1436. metadata, 0))
  1437. {
  1438. outStream.release();
  1439. bool ok = writer->writeFromAudioReader (*reader, 0, -1);
  1440. writer = nullptr;
  1441. reader = nullptr;
  1442. return ok && tempFile.overwriteTargetFileWithTemporary();
  1443. }
  1444. }
  1445. }
  1446. return false;
  1447. }
  1448. }
  1449. bool WavAudioFormat::replaceMetadataInFile (const File& wavFile, const StringPairArray& newMetadata)
  1450. {
  1451. using namespace WavFileHelpers;
  1452. if (ScopedPointer<WavAudioFormatReader> reader = static_cast<WavAudioFormatReader*> (createReaderFor (wavFile.createInputStream(), true)))
  1453. {
  1454. auto bwavPos = reader->bwavChunkStart;
  1455. auto bwavSize = reader->bwavSize;
  1456. reader = nullptr;
  1457. if (bwavSize > 0)
  1458. {
  1459. MemoryBlock chunk (BWAVChunk::createFrom (newMetadata));
  1460. if (chunk.getSize() <= (size_t) bwavSize)
  1461. {
  1462. // the new one will fit in the space available, so write it directly..
  1463. auto oldSize = wavFile.getSize();
  1464. {
  1465. FileOutputStream out (wavFile);
  1466. if (! out.failedToOpen())
  1467. {
  1468. out.setPosition (bwavPos);
  1469. out << chunk;
  1470. out.setPosition (oldSize);
  1471. }
  1472. }
  1473. jassert (wavFile.getSize() == oldSize);
  1474. return true;
  1475. }
  1476. }
  1477. }
  1478. return slowCopyWavFileWithNewMetadata (wavFile, newMetadata);
  1479. }
  1480. //==============================================================================
  1481. #if JUCE_UNIT_TESTS
  1482. struct WaveAudioFormatTests : public UnitTest
  1483. {
  1484. WaveAudioFormatTests() : UnitTest ("Wave audio format tests") {}
  1485. void runTest() override
  1486. {
  1487. beginTest ("Setting up metadata");
  1488. StringPairArray metadataValues = WavAudioFormat::createBWAVMetadata ("description",
  1489. "originator",
  1490. "originatorRef",
  1491. Time::getCurrentTime(),
  1492. numTestAudioBufferSamples,
  1493. "codingHistory");
  1494. for (int i = numElementsInArray (WavFileHelpers::ListInfoChunk::types); --i >= 0;)
  1495. metadataValues.set (WavFileHelpers::ListInfoChunk::types[i],
  1496. WavFileHelpers::ListInfoChunk::types[i]);
  1497. if (metadataValues.size() > 0)
  1498. metadataValues.set ("MetaDataSource", "WAV");
  1499. metadataValues.addArray (createDefaultSMPLMetadata());
  1500. WavAudioFormat format;
  1501. MemoryBlock memoryBlock;
  1502. {
  1503. beginTest ("Creating a basic wave writer");
  1504. ScopedPointer<AudioFormatWriter> writer (format.createWriterFor (new MemoryOutputStream (memoryBlock, false),
  1505. 44100.0, numTestAudioBufferChannels,
  1506. 32, metadataValues, 0));
  1507. expect (writer != nullptr);
  1508. AudioSampleBuffer buffer (numTestAudioBufferChannels, numTestAudioBufferSamples);
  1509. buffer.clear();
  1510. beginTest ("Writing audio data to the basic wave writer");
  1511. expect (writer->writeFromAudioSampleBuffer (buffer, 0, numTestAudioBufferSamples));
  1512. }
  1513. {
  1514. beginTest ("Creating a basic wave reader");
  1515. ScopedPointer<AudioFormatReader> reader (format.createReaderFor (new MemoryInputStream (memoryBlock, false), false));
  1516. expect (reader != nullptr);
  1517. expect (reader->metadataValues == metadataValues, "Somehow, the metadata is different!");
  1518. }
  1519. }
  1520. private:
  1521. enum
  1522. {
  1523. numTestAudioBufferChannels = 2,
  1524. numTestAudioBufferSamples = 256
  1525. };
  1526. StringPairArray createDefaultSMPLMetadata() const
  1527. {
  1528. StringPairArray m;
  1529. m.set ("Manufacturer", "0");
  1530. m.set ("Product", "0");
  1531. m.set ("SamplePeriod", "0");
  1532. m.set ("MidiUnityNote", "60");
  1533. m.set ("MidiPitchFraction", "0");
  1534. m.set ("SmpteFormat", "0");
  1535. m.set ("SmpteOffset", "0");
  1536. m.set ("NumSampleLoops", "0");
  1537. m.set ("SamplerData", "0");
  1538. return m;
  1539. }
  1540. JUCE_DECLARE_NON_COPYABLE (WaveAudioFormatTests)
  1541. };
  1542. static const WaveAudioFormatTests waveAudioFormatTests;
  1543. #endif