py2ls 0.1.0__py3-none-any.whl

py2ls/version.py

@@ -0,0 +1 @@
+version = "0.0.1"

py2ls/wb_detector.py

@@ -0,0 +1,169 @@
+# Author: Jianfeng Liu (andyandhope@gmail.com)
+# This script contains functions for detecting white balance bands in images and calculating the area under the curve for each band.
+
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+from scipy.signal import find_peaks
+from scipy.integrate import simps
+import cv2
+import os
+from PIL import Image
+
+def find_pks(data_valid, n_bands, idx_corr=0, threshold=np.arange(0.8, 1, 0.01)):
+    # This function finds peaks in the data_valid array based on specified parameters.
+    # It adjusts the threshold until it finds the desired number of peaks.
+
+    # Find initial peaks using a peak finding function (not provided)
+    pks_pre, _ = find_peaks(data_valid)
+
+    # Iterate through different threshold values
+    for thr in threshold:
+        # Calculate distance between peaks based on threshold
+        distance = ((pks_pre[-1] - pks_pre[0]) // n_bands) * thr
+
+        # Find peaks using adjusted threshold and other parameters
+        peaks, _ = find_peaks(
+            (np.max(data_valid) - data_valid),
+            distance=distance,
+            height=(np.max(data_valid) - np.min(data_valid)) * 0.1,
+        )
+
+        # Check if the number of found peaks matches the expected number of bands
+        if peaks.shape[0] == (n_bands - 1):
+            break
+
+        # If the threshold reaches its last value and the desired number of peaks is not found, print a message
+        if thr == threshold[-1]:
+            print("did not find the perfect threshold")
+
+    # Adjust the peaks indices to reflect their position in the original data
+    peaks += idx_valid[0][0]
+
+    # Print information about found peaks, threshold, and distance
+    print(f"peaks={peaks}\npeaks_shape={len(peaks)}\nthr={thr}\ndistance={distance}")
+
+    # Return the peaks indices
+    return peaks
+
+
+def cal_area(data, peaks, idx_valid_bands, x_range):
+    # This function calculates the area under the curve between adjacent peaks for each band.
+    area_values = []
+    for iband in idx_valid_bands:
+        if iband + 1 >= len(peaks):
+            break
+        # Define the x and y values within the band
+        x_band = x_range[peaks[iband] : peaks[iband + 1]]
+        y_band = data[peaks[iband] : peaks[iband + 1]]
+        # Integrate the y-values within the band to find the area
+        area = simps(y_band, x_band)
+        area_values.append(area)
+    return area_values
+
+def check_load_img(dir_data): 
+    if isinstance(dir_data,str):
+        rgb_image = cv2.imread(dir_data)
+        data = cv2.cvtColor(rgb_image, cv2.COLOR_BGR2GRAY)
+    elif isinstance(dir_data, Image.Image):
+        # convert it
+        rgb_image = np.array(dir_data) 
+        data = cv2.cvtColor(rgb_image, cv2.COLOR_RGB2BGR)
+    else:
+        raise TypeError(f"{dir_data} is neither a directory nor a PIL.Image.Image instance.")
+    return data
+
+def detect_wb_bands(dir_data, n_bands=24, weight=0.25, outlier=None):
+    # This function detects the white balance bands in an image.
+    """ param:
+        dir_data, [path]
+        n_bands [int], default 24 lanes. how many bands
+        weight=0.25, [float], 0.01 means keep more light bands, 0.5 remove 
+            light band. the greater, the less sensitivity
+        outlier=None, [list], e.g., [0,3] means the 1st and 4th bands will 
+            be removed, will not taken into account
+    """
+    # Load the image and convert it to grayscale 
+    data = check_load_img(dir_data)
+
+    # Determine the bit depth of the image
+    if data.dtype == np.uint16:
+        bit_depth = 16
+    elif data.dtype == np.uint8:
+        bit_depth = 8
+    print(f"bit_depth:{bit_depth}")
+
+    # Invert the grayscale image
+    data_ivrt = 2**bit_depth - data
+    data_cum = np.sum(data_ivrt, axis=0)
+    x_range = np.arange(data.shape[1])  # Generate x-axis values
+
+    # Identify baseline and normalize cumulative intensity values
+    bsln = np.min(data_cum)
+    data_cum_bsln = data_cum - bsln
+    data_valid = np.zeros((data_cum_bsln.shape))  # Initialize valid data array
+    idx_valid = np.where(data_cum_bsln > np.max(data_cum_bsln) * weight)[
+        0
+    ]  # Determine valid indices
+    data_valid[idx_valid] = data_cum_bsln[idx_valid]  # Assign valid data values
+
+    threshold = np.arange(0.5, 1, 0.01)  # Define threshold range
+    for thr in threshold:
+        # Calculate distance between peaks based on threshold
+        distance = ((idx_valid[-1] - idx_valid[0]) // n_bands) * thr
+
+        # Find peaks using adjusted threshold
+        peaks, _ = find_peaks(
+            (np.max(data_valid) - data_valid),
+            distance=distance,
+        )
+
+        # Check if the number of found peaks matches the expected number of bands
+        if peaks.shape[0] == (n_bands - 1):
+            break
+
+        # If the threshold reaches its last value and the desired number of peaks is not found, print a message
+        if thr == threshold[-1]:
+            print("did not find the perfect threshold")
+
+    # Adjust the peaks indices to include the start and end of the data
+    peaks_tail = [idx_valid[0]]
+    for i in peaks:
+        peaks_tail.append(i)
+    peaks_tail.append(idx_valid[-1])
+
+    # Handle outlier bands if specified
+    if outlier is None:
+        idx_valid_bands = np.arange(n_bands)
+    else:
+        idx_valid_bands = np.delete(np.arange(n_bands), outlier)
+
+    # Calculate area under the curve for each band
+    area_values = cal_area(data_valid, peaks_tail, idx_valid_bands, x_range)
+
+    # Plot the detected bands on the image
+    fig, axs = plt.subplots(1, 1)
+    axs.imshow(data_ivrt)
+    axs.set_ylim([0, data.shape[0]])
+    axr = axs.twinx()
+    ylim_ = axr.get_ylim()
+    for iband in idx_valid_bands:
+        axr.fill_between(
+            x_range[peaks_tail[iband] : peaks_tail[iband + 1]],
+            0,
+            data_cum_bsln[peaks_tail[iband] : peaks_tail[iband + 1]],
+            alpha=0.5,
+        )
+        axr.text(
+            peaks_tail[iband],
+            np.min(ylim_) - (np.max(ylim_) - np.min(ylim_)) * 0.05,
+            str(iband + 1),
+        )
+
+    axs.set_title("WB raw image")
+    axs.set_ylabel("Height pixels")
+    axs.set_xlabel("Width pixels")
+    axr.set_ylabel("Integrated intensity")
+
+    # Return area values and the plotted figure
+    return area_values, fig

py2ls-0.1.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.1
+Name: py2ls
+Version: 0.1.0
+Summary: 
+Author: Jianfeng
+Author-email: Jianfeng.Liu0413@gmail.com
+Requires-Python: >=3.12,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Description-Content-Type: text/markdown
+
+

py2ls-0.1.0.dist-info/RECORD

@@ -0,0 +1,64 @@
+py2ls/.git/COMMIT_EDITMSG,sha256=EZlkQViEF6J68QzXxear21cPGLHAAEj7OaLgd1J5c2w,15
+py2ls/.git/FETCH_HEAD,sha256=GW4RIooDnHicCnDuRlc37IaJnJNpdCkBud1P9ZR6KPY,100
+py2ls/.git/HEAD,sha256=KNJb-Cr0wOK3L1CVmyvrhZ4-YLljCl6MYD2tTdsrboA,21
+py2ls/.git/config,sha256=XswTg1Ts7_7IBDlKHh4OF_0Tq7v4wW7BXb6xSVInSec,345
+py2ls/.git/description,sha256=ZzMxc0Ca26m45Twn1DDnOHqin5VHEZ9uOTBrScIXSjE,16
+py2ls/.git/hooks/applypatch-msg.sample,sha256=AiNJeguLAzqlijpSG4YphpOGz3qw4vEBlj0yiqYhk_c,478
+py2ls/.git/hooks/commit-msg.sample,sha256=H3TV6SkpebVz69WXQdRsuT_zkazdCD00C5Q3B1PZJDc,896
+py2ls/.git/hooks/fsmonitor-watchman.sample,sha256=4FSZZOk4l7UZvY4zPAN-Uf_w-IuhPghqMxWSv4AfodA,4726
+py2ls/.git/hooks/post-update.sample,sha256=gXZa8trvMjBh3LxeYfwWSBy3SzusmtihdLGGUjWG9sU,189
+py2ls/.git/hooks/pre-applypatch.sample,sha256=4VxbRp6j4KaVvqbyyCvPjmKCEHSTnd2Ft34AB_8WVHU,424
+py2ls/.git/hooks/pre-commit.sample,sha256=-a99lesSMezy66l3D-36jUeXoSsC1yQOmNVoIBJRJEo,1643
+py2ls/.git/hooks/pre-merge-commit.sample,sha256=04JacDN5QOu9ClwHKYThMkWSDN-ImL0iXI0npt_Jy1M,416
+py2ls/.git/hooks/pre-push.sample,sha256=7M6cfgTT9d2ditqBdT3R1UmpY0smdwBCtY3aACF9CGo,1374
+py2ls/.git/hooks/pre-rebase.sample,sha256=T-vOhneQBSM4B29OZsxH77FIedGAl9HWHIJhhZ6qp7M,4898
+py2ls/.git/hooks/pre-receive.sample,sha256=pMPSuce7P9jRRBwxvU7nGlldZrRPz0ndsxAlIyAWmYk,544
+py2ls/.git/hooks/prepare-commit-msg.sample,sha256=6d3KpBif3dJe2X_Ix4nsp7bKFjkLI5KuMnbwyOGqRhk,1492
+py2ls/.git/hooks/push-to-checkout.sample,sha256=pT0HQXmLKHxt16-mSu5HPzBeZdP0lGO7nXQI7DsSv18,2783
+py2ls/.git/hooks/update.sample,sha256=jV8vqD4QPPCLV-qmdSHfkZT0XL28s32lKtWGCXoU0QY,3650
+py2ls/.git/index,sha256=UiAHONBax9obxHeBw2XnO3CA5lQhLSiwBPj8Ovw0Ah0,1338
+py2ls/.git/info/exclude,sha256=ZnH-g7egfIky7okWTR8nk7IxgFjri5jcXAbuClo7DsE,240
+py2ls/.git/logs/HEAD,sha256=LmciHIgChKHvNdm7QcAGSSuJwCsh4x0R5qVCOKqs9Ow,170
+py2ls/.git/logs/refs/heads/main,sha256=LmciHIgChKHvNdm7QcAGSSuJwCsh4x0R5qVCOKqs9Ow,170
+py2ls/.git/logs/refs/remotes/origin/HEAD,sha256=CDj-_Vd1xu0AY9atNFqscjp8cDKa4GNMxJVinI1v3_Q,162
+py2ls/.git/logs/refs/remotes/origin/main,sha256=gj8FTeA4KEjnjDE5H16EdDZm_Myl0IREl_iqX54-GqY,152
+py2ls/.git/objects/25/b796accd261b9135fd32a2c00785f68edf6c46,sha256=4ic5vOwEdfbGL8oARSVEeAnSoDs14-gggGZEL-61nYE,564
+py2ls/.git/objects/36/b4a1b7403abc6c360f8fe2cb656ab945254971,sha256=X18sHo17gQTxusNptoILoWxSfU0ulrsZvbETlBtV5aY,2327
+py2ls/.git/objects/3f/d6561300938afbb3d11976cf9c8f29549280d9,sha256=91oqbTWfUE1d_hT_1ptYmRUb5pOQ1X4oxQxpF6NXjKU,8501
+py2ls/.git/objects/58/20a729045d4dc7e37ccaf8aa8eec126850afe2,sha256=3Pf6NS8OTK4EdHZGVeJ421BtK7w4WJncQDBauZI_wW4,34
+py2ls/.git/objects/60/f273eb1c412d916fa3f11318a7da7a9911b52a,sha256=aJD9iF_LmYSrqDepXFBZKN1yMYbQczVkN_wnrDosBdI,5620
+py2ls/.git/objects/61/570cec8c061abe74121f27f5face6c69b98f99,sha256=IQZi5MkbRu3ToRUPsRcXuh1Xa3pkAz_HDRCVhNL89ds,5753
+py2ls/.git/objects/69/13c452ca319f7cbf6a0836dc10a5bb033c84e4,sha256=NYLQQZTfd0htZst42ALS2dmryv1q_l1N19ZfHEbz_38,3193
+py2ls/.git/objects/78/3d4167bc95c9d2175e0df03ef1c1c880ba75ab,sha256=SK2QDjDBiDhVMG1I5p19g4RbEm2Rax7mYnxawmVZYxs,15523
+py2ls/.git/objects/79/7ae089b2212a937840e215276005ce76881307,sha256=lQOKF2pb1JvipI3eT79X0-TuMGWsy1A-Yw4BCgKZNOM,33472
+py2ls/.git/objects/7e/5956c806b5edc344d46dab599dec337891ba1f,sha256=sfqJBiSNj-gyJo4D7xkmRAo76mC2ztjqeZZsl4ifULA,162
+py2ls/.git/objects/8e/55a7d2b96184030211f20c9b9af201eefcac82,sha256=yW-jVYeCTWR-nX3JJgA1g9YLPjzNsKlDmEOH290Ywx0,1221
+py2ls/.git/objects/91/c69ad88fe0ba94aa7859fb5f7edac5e6f1a3f7,sha256=Kk2MWCO1OcShYuABGzp2O9LiWGDfDkcZtd0oy4nY6RU,9529
+py2ls/.git/objects/b0/56be4be89ba6b76949dd641df45bb7036050c8,sha256=8Y7z30eNceBd5QIx09QfMp5cYBbrgUllmats0kvJEJ4,132
+py2ls/.git/objects/b0/9cd7856d58590578ee1a4f3ad45d1310a97f87,sha256=82dx4hIdMpdcB64e5PU1s2gZFVkTvrj1cPwwJ_kasNU,4444
+py2ls/.git/objects/d9/005f2cc7fc4e65f14ed5518276007c08cf2fd0,sha256=IJIoz93V7pf9yx43U1JdN8gBq_LWtw8A9Z2YMPnq_B0,1450
+py2ls/.git/objects/df/e0770424b2a19faf507a501ebfc23be8f54e7b,sha256=vCdlxwEidekh8i-5TVMVgSLGk9DPZCZAbWqvGYSKQ9c,76
+py2ls/.git/objects/e9/391ffe371f1cc43b42ef09b705d9c767c2e14f,sha256=RWTy2n8L2XxZQknBFyPczA0Aa_4gSG_Ybcr8e8v4ccc,10264
+py2ls/.git/objects/fc/292e793ecfd42240ac43be407023bd731fa9e7,sha256=hGIYoxKWNT3IPwk3DE4l3FLBbUYF-kXcHcx7KrH9uS0,1971
+py2ls/.git/refs/heads/main,sha256=uKGNQVhljreyO2Y49vGbb1FAz1yOtOwulP83EwO84I0,41
+py2ls/.git/refs/remotes/origin/HEAD,sha256=K7aiSqD8bEhBAPXVGim7rYQc0sdV9dk_qiBOXbtOsrQ,30
+py2ls/.git/refs/remotes/origin/main,sha256=uKGNQVhljreyO2Y49vGbb1FAz1yOtOwulP83EwO84I0,41
+py2ls/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb6ZAFs,66
+py2ls/.gitignore,sha256=y7GvbD_zZkjPVVIue8AyiuFkDMuUbvMaV65Lgu89To8,2763
+py2ls/LICENSE,sha256=UOZ1F5fFDe3XXvG4oNnkL1-Ecun7zpHzRxjp-XsMeAo,11324
+py2ls/README.md,sha256=uc1-9UZhOWvFQSamsVqhnD1vR4T_50zOiVDbz-tuubo,9726
+py2ls/__init__.py,sha256=k05sUEKwWnAaOAn4VKCiPopcWrbmLLsZOuAt7vmUKWc,260
+py2ls/brain_atlas.py,sha256=w1o5EelRjq89zuFJUNSz4Da8HnTCwAwDAZ4NU4a-bAY,5486
+py2ls/correlators.py,sha256=RbOaJIPLCHJtUm5SFi_4dCJ7VFUPWR0PErfK3K26ad4,18243
+py2ls/dbhandler.py,sha256=i9dNrpHyx0oIaFieHI4X4tsrCdN-aFxudPTDOgy9Ppo,3574
+py2ls/freqanalysis.py,sha256=F4218VSPbgL5tnngh6xNCYuNnfR-F_QjECUUxrPYZss,32594
+py2ls/internet_finder.py,sha256=I-jPepbBhpDaOAsD2MqbKMe1CBN8w1PYo0CjNFkaeeU,19149
+py2ls/ips.py,sha256=E5u8kBybs2DnHzBTj_lQy8taNRiQpAY1uRRV4dy3Owg,119286
+py2ls/netfinder.py,sha256=Woilmx31B1Uh3we19WLVa63ptw2YU4Z99KUqlq90eAY,29774
+py2ls/sleep_events_detectors.py,sha256=36MCuRrpurn0Uvzpo3p3b3_JlVsRNHSWCXbJxCGM3mg,51546
+py2ls/translator.py,sha256=QfDUO0-pXHGMBFZBefiBHzOrC93-__N5sUQY_VP4wes,29734
+py2ls/version.py,sha256=CactNZqrHHYTPrkHKccy2WKXmaiUdtTgPqSjFyVXnJk,18
+py2ls/wb_detector.py,sha256=7y6TmBUj9exCZeIgBAJ_9hwuhkDh1x_-yg4dvNY1_GQ,6284
+py2ls-0.1.0.dist-info/METADATA,sha256=75PMsOFYdSIj5l3VAyw-KsuK3ZBPWhxJrAL8kAgouJ0,287
+py2ls-0.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+py2ls-0.1.0.dist-info/RECORD,,

py2ls-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.0
+Root-Is-Purelib: true
+Tag: py3-none-any