ruby_dsp 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d0352100ed3bc4d5984dd4b59ca7540eea006a7234eb2ed9a87ca88c61af3c91
-  data.tar.gz: 94b1b8182701c588618dbbe6e44f9135a73c62c9b8f174d745f5e807df379145
+  metadata.gz: d044ff97e82f0fdedd8749cfa07573fd350cd793094d9274700e623caf35e098
+  data.tar.gz: 2559e81b76c114cfa10dff50aa422bb29a933def9b75a2893c8f53c12a76f619
 SHA512:
-  metadata.gz: 17e977dd018f850f37210a0d4cc9abd209ad276f51f62792fb240ccb94a8d096f8f8ff7d59a201b21cab38e048b901ed93e4ff9c89809c281d2b923816bb2641
-  data.tar.gz: d807466791df6a0f6cac661dc77e763cd03c5aeb2dd251d969054ff2bda5d0321735f4615819233f9ed9fdb9bb51f3b8b0eaf2e886d7269f29c1595aa616bc0d
+  metadata.gz: 97c853452400908572b918d94460ded68831dbb7d550f648a8291147d6fbdcee5e75e8ba36715622ed0dd90f978b1547a9661523c1d80b93b1f0b8ed2139bd66
+  data.tar.gz: 1d6a9728f02760fa579858c83e2a81c28c486cfcd9dc556cf0f8d6106389a373c2f4444364b5a6acacb3821a16cabad396e7e9c9cc30f9c046dc4b91fa06f70d

data/README.md

@@ -1,8 +1,8 @@
-# [RubyDSP](https://github.com/cichrrad/rubyDSP) | [Documentation](https://www.rubydoc.info/gems/ruby_dsp/0.0.8)
+# [RubyDSP](https://github.com/cichrrad/rubyDSP) | [Documentation](https://www.rubydoc.info/gems/ruby_dsp/0.0.9)
 
 [![Ruby CI](https://github.com/cichrrad/rubyDSP/actions/workflows/test.yml/badge.svg)](https://github.com/cichrrad/rubyDSP/actions/workflows/test.yml)
 
-> 🚧 **Status:** This (***HOBBY***) project is currently in early development. It is hopefully functional, but API changes are expected. There is no warranty regarding anything 🗿.
+> 🚧 **Status:** This (***HOBBY***) project is currently in early development. It is functional, but API changes are expected. There is no warranty regarding anything 🗿.
 
 ---
 
@@ -30,7 +30,7 @@ I made this gem to try hands-on binding these two languages together, as I would
 
 ## Overview of Capabilities
 
-* **File Operations:** `save_track`
+* **File Operations:** `save_track`, `dup`
 * **Mutations:** `to_mono!`, `resample!`, `trim_silence!`, `normalize!`, `fade_in!`, `fade_out!`, `pad!`, `pad_to_duration!`, `clip!`
 * **DSP Filters:** `low_pass!`, `high_pass!`, `band_pass!`, `notch!`, `peak_eq!`, `low_shelf!`, `high_shelf!`
 * **Synthesis:** `add_wave!`

data/ext/ruby_dsp/ruby_dsp.cpp

@@ -34,9 +34,33 @@ struct AudioTrack
     std::vector<float> samples;
     unsigned long long sample_count = 0;
 
-    AudioTrack(std::string f, unsigned int target_channels = 0, unsigned int target_sample_rate = 0) : filename(f)
+    // static float get_file_duration(std::string filename)
+    // {
+    //     if (filename.empty())
+    //         return 0.0f;
+
+    //     ma_decoder decoder;
+    //     // NULL config means it reads the file's native format/rate
+    //     if (ma_decoder_init_file(filename.c_str(), NULL, &decoder) != MA_SUCCESS)
+    //     {
+    //         throw std::runtime_error("RubyDSP: Could not process audio file for duration: " + filename);
+    //     }
+
+    //     ma_uint64 totalFrames;
+    //     if (ma_decoder_get_length_in_pcm_frames(&decoder, &totalFrames) != MA_SUCCESS)
+    //     {
+    //         ma_decoder_uninit(&decoder);
+    //         throw std::runtime_error("RubyDSP: Could not determine track length.");
+    //     }
+
+    //     int sample_rate = decoder.outputSampleRate;
+    //     ma_decoder_uninit(&decoder);
+
+    //     return (float)totalFrames / sample_rate;
+    // }
+
+    AudioTrack(std::string f, unsigned int target_channels = 0, unsigned int target_sample_rate = 0, float start_sec = 0.0f, float duration_sec = 0.0f) : filename(f)
     {
-
         // empty constructor
         if (filename.empty())
         {
@@ -70,14 +94,49 @@ struct AudioTrack
             throw std::runtime_error("RubyDSP: Could not determine track length.");
         }
 
-        sample_count = totalFrames * channels;
+        ma_uint64 start_frame = (ma_uint64)(start_sec * sample_rate);
+        if (start_frame > totalFrames)
+        {
+            start_frame = totalFrames; // Prevent seeking past EOF
+        }
+
+        if (start_frame > 0)
+        {
+            if (ma_decoder_seek_to_pcm_frame(&decoder, start_frame) != MA_SUCCESS)
+            {
+                ma_decoder_uninit(&decoder);
+                throw std::runtime_error("RubyDSP: Could not seek to start frame.");
+            }
+        }
+
+        ma_uint64 framesToRead = totalFrames - start_frame;
+        if (duration_sec > 0.0f)
+        {
+            ma_uint64 requestedFrames = (ma_uint64)(duration_sec * sample_rate);
+            if (requestedFrames < framesToRead)
+            {
+                framesToRead = requestedFrames;
+            }
+        }
+
+        sample_count = framesToRead * channels;
         samples.resize(sample_count);
 
-        ma_uint64 framesRead;
-        if (ma_decoder_read_pcm_frames(&decoder, samples.data(), totalFrames, &framesRead) != MA_SUCCESS)
+        if (framesToRead > 0)
         {
-            ma_decoder_uninit(&decoder);
-            throw std::runtime_error("RubyDSP: Failed to read PCM data.");
+            ma_uint64 framesRead;
+            if (ma_decoder_read_pcm_frames(&decoder, samples.data(), framesToRead, &framesRead) != MA_SUCCESS)
+            {
+                ma_decoder_uninit(&decoder);
+                throw std::runtime_error("RubyDSP: Failed to read PCM data.");
+            }
+
+            // Shrink vector if we hit EOF earlier than expected
+            if (framesRead < framesToRead)
+            {
+                sample_count = framesRead * channels;
+                samples.resize(sample_count);
+            }
         }
 
         ma_decoder_uninit(&decoder);
@@ -904,6 +963,11 @@ struct AudioTrack
                << sample_rate << "Hz sample rate]";
         return stream.str();
     }
+
+    AudioTrack dup()
+    {
+        return *this;
+    }
 };
 
 extern "C"
@@ -916,10 +980,12 @@ extern "C"
 {
     Module rb_mRubyDSP = define_module("RubyDSP");
     Data_Type<AudioTrack> rb_cAudioTrack = define_class_under<AudioTrack>(rb_mRubyDSP, "AudioTrack")
-                                               .define_constructor(Constructor<AudioTrack, std::string, unsigned int, unsigned int>(),
+                                               .define_constructor(Constructor<AudioTrack, std::string, unsigned int, unsigned int, float, float>(),
                                                                    Arg("file_name") = (std::string) "",
                                                                    Arg("target_channels") = (unsigned int)0,
-                                                                   Arg("target_sample_rate") = (unsigned int)0)
+                                                                   Arg("target_sample_rate") = (unsigned int)0,
+                                                                   Arg("start_sec") = 0.0f,
+                                                                   Arg("duration_sec") = 0.0f)
                                                // attributes
                                                .define_attr("file_name", &AudioTrack::filename, Rice::AttrAccess::Read)
                                                .define_attr("channels", &AudioTrack::channels, Rice::AttrAccess::Read)
@@ -991,5 +1057,7 @@ extern "C"
                                                               Arg("cutoff_freq"),
                                                               Arg("gain_db"),
                                                               Arg("q") = 0.707)
-                                               .define_method("to_s", &AudioTrack::to_s);
+                                               .define_method("to_s", &AudioTrack::to_s)
+                                               .define_method("dup", &AudioTrack::dup)
+                                               .define_method("clone", &AudioTrack::dup);
 }