transitsong 0.0.1__tar.gz

transitsong-0.0.1/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: transitsong
+Version: 0.0.1
+Requires-Dist: numpy
+Requires-Dist: sounddevice
+Requires-Dist: lightkurve
+Requires-Dist: matplotlib
+Requires-Dist: scipy
+Requires-Dist: moviepy

transitsong-0.0.1/README.md

@@ -0,0 +1,2 @@
+# sounds
+sonification of transits

transitsong-0.0.1/pyproject.toml

@@ -0,0 +1,21 @@
+[build-system]
+requires = [
+    "setuptools", 
+    "numpy",
+    "cython", 
+    "wheel"
+]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "transitsong"
+version = "0.0.1"
+
+dependencies = [
+    "numpy",
+    "sounddevice",
+    "lightkurve",
+    "matplotlib",
+    "scipy",
+    "moviepy"
+]

transitsong-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

transitsong-0.0.1/tests/tests.py

@@ -0,0 +1,12 @@
+from transitsong.main import Transit
+
+# test that lightcurve downloads ok! 
+
+def test_lc_download():
+    tic = 124029677 
+    sector = 33
+
+    transit = Transit(tic, sector)
+
+    assert len(transit.norm_flux) == len(transit.time), "Time and flux arrays are not the same length"
+    assert len(transit.norm_flux) > 0, "norm flux array, time array is of length 0"

transitsong-0.0.1/transitsong/main.py

@@ -0,0 +1,243 @@
+import lightkurve as lk
+import matplotlib.pyplot as plt 
+import numpy as np
+from scipy.io.wavfile import write 
+import sounddevice as sd
+import matplotlib.animation as animation
+from moviepy import *
+from moviepy.video.fx import MultiplySpeed
+import os 
+from pathlib import Path
+
+# set matplotlib preferences
+#plt.style.use(path + 'text.mplstyle')
+
+
+class Transit:
+    """A program that makes an animation with audio for an input TESS lightcurve.
+
+    Attributes:
+        tic (int or str): TESS Input Catalog number
+        sector (int): TESS sector number
+        window (list of length 2): start and end days of sector
+    """
+
+    def __init__(self, tic, sector, window = None):
+        self.tic = tic
+        self.sector = sector
+        self.window = window
+
+        try:
+            search_result = lk.search_lightcurve(
+                f"TIC {tic}",
+                mission="TESS",
+                author="SPOC",
+                sector =[sector] )
+            
+            lc = search_result.download()
+            self.success = True
+
+        except Exception as e:
+            self.success = False
+            raise ValueError("Failed to load lightcurve for TIC {} in sector {}: {}".format(tic, sector, str(e)))
+
+        print("Successfully loaded lightcurve for TIC {} in sector {}".format(tic, sector))
+
+        lc_binned = lc.bin(time_bin_size=0.01)
+
+        lc_binned = lc_binned.normalize()
+
+        lc_fluxes = lc_binned.flux.value
+        lc_times = lc_binned.time.value
+
+        #med_value = np.nanmedian(lc_fluxes)
+        
+        #lc_normflux = lc_fluxes/med_value
+
+        self.time = lc_times
+        self.norm_flux = lc_fluxes
+
+
+        if window is not None:
+            new_times = []
+            new_fluxes = []
+
+            for i in range(len(self.time)):
+                if self.time[i] > window[0] and self.time[i] < window[1]:
+                    new_times.append(self.time[i])
+                    new_fluxes.append(self.norm_flux[i])
+            
+            self.time = np.array(new_times)
+            self.norm_flux = np.array(new_fluxes)
+
+    
+    def make_sound_arr(self, max_val=900, min_val=200):
+        """Sound array
+
+        Make sound array as .wav file, higher pitch corresponds to higher flux. Saves to subdirectory "song."
+
+        Args:
+            max_val (int or float): maximum Hz frequency
+            min_val (int or float): minimum Hz frequency
+        
+        Returns:
+            Nothing.
+        """
+
+        mapped_flux = (self.norm_flux - np.nanmin(self.norm_flux)) / (np.nanmax(self.norm_flux) - np.nanmin(self.norm_flux)) * (max_val - min_val) + min_val
+
+        self.mapped_flux = mapped_flux
+
+        audio = []
+        len_points = []
+
+        samplerate = 44100
+        duration = 0.05
+
+        for flux in mapped_flux: 
+            frequency = flux 
+            t = np.linspace(0, duration, int(samplerate * duration), endpoint=False)
+            
+            audio_signal = np.sin(2*np.pi*frequency*t)
+            audio_signal = audio_signal[~np.isnan(audio_signal)]
+            len_points.append(len(audio_signal))
+            
+            audio.extend(audio_signal)
+
+        audio_arr = np.array(audio)
+        self.audio_arr = audio_arr
+
+        song_path = os.getcwd() + "/song/"
+        self.song_path = song_path
+        if not os.path.isdir(song_path):
+            directory = Path(song_path)
+            directory.mkdir(parents=True, exist_ok=True)
+
+        write(song_path + f"TIC{self.tic}_S{self.sector}_SONG.wav", samplerate, audio_arr)
+
+        # colors
+        cmap = plt.cm.magma
+
+        sorted_args = np.argsort(self.norm_flux)
+
+        values = np.linspace(0, 1, len(self.norm_flux))
+        colors = np.empty((len(self.norm_flux), 4))  # 4 for RGBA
+        colors[sorted_args] = cmap(values)
+
+        self.colors= colors 
+
+        #sd.play(audio_arr, samplerate)
+        #sd.wait()
+
+
+    def make_video(self):
+        """Video file
+
+        Makes animated plot of lightcurve, lighter color corresponds to higher flux. Saves as .mp4 in subdirectory "dance".
+        
+        Args:
+            None.
+
+        Returns:
+            Nothing.
+        """
+        # Original time
+        x_raw = self.time
+        y_raw = self.norm_flux
+
+        # Removing nans
+        y = y_raw[~np.isnan(y_raw)]
+        x = x_raw[~np.isnan(y_raw)]
+
+        
+        fig, ax = plt.subplots()
+        line, = ax.plot([], [], marker='o', linestyle='', color='b') 
+        ax.set_xlabel('Time (days)')
+        ax.set_ylabel('Normalized Flux')
+        ax.set_xlim(self.time[0], self.time[-1])
+        ax.set_ylim(np.nanmin(self.norm_flux)-0.005, np.nanmax(self.norm_flux)+0.005)
+        ax.set_title(f"TIC {self.tic} - Sector {self.sector}")
+
+        
+        def update(i):
+            
+            # line.set_data(x[:frame], y[:frame])
+            # line.set_color(self.colors[frame])
+
+            plt.scatter(x[:i], y[:i], color=self.colors[i], edgecolor="k", linewidth=0.5)
+
+           # return line,
+
+        
+        ani = animation.FuncAnimation(
+            fig, update, frames=len(x), interval=50, repeat=True
+        )
+
+        dance_path = os.getcwd() + "/dance/"
+        self.dance_path = dance_path
+        if not os.path.isdir(dance_path):
+            directory = Path(dance_path)
+            directory.mkdir(parents=True, exist_ok=True)
+
+        ani.save(dance_path + f"TIC{self.tic}_S{self.sector}_DANCE.mp4", writer='ffmpeg', fps=30)
+        #plt.show()
+
+    def combine(self):
+        video_clip = VideoFileClip(self.dance_path + f"TIC{self.tic}_S{self.sector}_DANCE.mp4", audio=False)
+        audio_clip = AudioFileClip(self.song_path + f"TIC{self.tic}_S{self.sector}_SONG.wav")
+
+        video_factor = video_clip.duration / audio_clip.duration
+
+        
+        video_clip = video_clip.with_effects([MultiplySpeed(video_factor)])
+
+        
+        final_clip = video_clip.with_audio(audio_clip)
+
+        song_and_dance_path = os.getcwd() + "/song_and_dance/"
+        if not os.path.isdir(song_and_dance_path):
+            directory = Path(song_and_dance_path)
+            directory.mkdir(parents=True, exist_ok=True)
+
+        final_clip.write_videofile(
+            song_and_dance_path + f"TIC{self.tic}_S{self.sector}_FINAL.mp4", 
+            codec="libx264", 
+            audio_codec="aac", 
+            temp_audiofile="temp-audio.m4a",  
+            remove_temp=True,                 
+        )
+
+       
+        video_clip.close()
+        audio_clip.close()
+        final_clip.close()
+
+#def play_song(Transit):
+    # here is where we do the simultaneous thing?
+#tic = 124029677 
+#sector = 33
+#window = [2217, 2220]
+
+
+
+# tic = 55652896
+# sector = 63 #originally did 38
+# window = [2340, 2341]
+
+# tic = 149601126
+# sector = 32 #96
+# window = [2196, 2198.5]
+
+# tic = 263930790
+# sector = 73
+# window = [3293, 3299]
+
+# planet = Transit(tic, sector, window=window)
+# planet.make_sound_arr()
+# planet.make_video()
+# planet.combine()
+
+
+#55652896, 38, 63
+
+

transitsong-0.0.1/transitsong.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: transitsong
+Version: 0.0.1
+Requires-Dist: numpy
+Requires-Dist: sounddevice
+Requires-Dist: lightkurve
+Requires-Dist: matplotlib
+Requires-Dist: scipy
+Requires-Dist: moviepy

transitsong-0.0.1/transitsong.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+tests/tests.py
+transitsong/__init__.py
+transitsong/main.py
+transitsong.egg-info/PKG-INFO
+transitsong.egg-info/SOURCES.txt
+transitsong.egg-info/dependency_links.txt
+transitsong.egg-info/requires.txt
+transitsong.egg-info/top_level.txt

transitsong-0.0.1/transitsong.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

transitsong-0.0.1/transitsong.egg-info/requires.txt

@@ -0,0 +1,6 @@
+numpy
+sounddevice
+lightkurve
+matplotlib
+scipy
+moviepy

transitsong-0.0.1/transitsong.egg-info/top_level.txt

@@ -0,0 +1 @@
+transitsong