DynamiSpectra 1.0.3__py3-none-any.whl → 1.0.5__py3-none-any.whl

dynamispectra/SecondaryStructure.py

@@ -0,0 +1,201 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from matplotlib.patches import Patch  # Para criar elementos de legenda personalizados
+import os
+
+# Mapeamento dos estados secundários para números
+state_mapping = {
+    'H': 1,  # Hélice-α
+    'E': 2,  # Folha-β
+    'C': 0,  # Loop/Coil
+    'T': 3,  # Turn
+    'S': 4,  # Bend
+    'G': 5,  # 3-Helix
+    '~': -1, # Sem estrutura definida
+    'B': -1, # Tratar como sem estrutura
+}
+
+# Nomes das estruturas secundárias
+state_names = {
+    0: 'Loop/Coil',
+    1: 'α-Helix',
+    2: 'β-Sheet',
+    3: 'Turn',
+    4: 'Bend',
+    5: '3-Helix',
+}
+
+def read_ss(file):
+    """
+    Reads secondary structure data from a .dat file.
+
+    Parameters:
+    -----------
+    file : str
+        Path to the .dat file.
+
+    Returns:
+    --------
+    ss_data : numpy.ndarray
+        Array of secondary structure data.
+    """
+    try:
+        print(f"Reading file: {file}")
+        
+        # Open the file and process line by line
+        ss_data = []
+        
+        with open(file, 'r') as f:
+            for line in f:
+                # Skip comment lines and empty lines
+                if line.startswith(('#', '@', ';')) or line.strip() == '':
+                    continue
+                
+                # Try to extract the secondary structure data
+                try:
+                    ss_line = [state_mapping.get(char, -1) for char in line.strip()]
+                    ss_data.append(ss_line)
+                except ValueError:
+                    # Skip lines that cannot be converted to numbers
+                    print(f"Error processing line: {line.strip()}")
+                    continue
+        
+        # Check if the data is valid
+        if len(ss_data) == 0:
+            raise ValueError(f"File {file} does not contain valid data.")
+        
+        # Convert lists to numpy arrays
+        ss_data = np.array(ss_data)
+        
+        return ss_data
+    
+    except Exception as e:
+        print(f"Error reading file {file}: {e}")
+        return None
+
+def calculate_probabilities(ss_data):
+    """
+    Calculates the probability of each secondary structure.
+
+    Parameters:
+    -----------
+    ss_data : numpy.ndarray
+        Array of secondary structure data.
+
+    Returns:
+    --------
+    probabilities : dict
+        Dictionary with probabilities for each secondary structure.
+    """
+    probabilities = {state: [] for state in state_names.values()}
+    total_frames = ss_data.shape[0]  # Total de frames
+
+    for estado, nome in state_names.items():
+        probabilities[nome] = np.sum(ss_data == estado, axis=1) / ss_data.shape[1]  # Probabilidade por frame
+
+    return probabilities
+
+def plot_ss_boxplot(probabilities_list, labels, colors, output_folder):
+    """
+    Generates a boxplot for secondary structure probabilities.
+
+    Parameters:
+    -----------
+    probabilities_list : list of dict
+        List of dictionaries with probabilities for each simulation.
+    labels : list of str
+        Labels for each simulation.
+    colors : list of str
+        Colors for each simulation.
+    output_folder : str
+        Output folder to save the plots.
+    """
+    # Preparar dados para o boxplot
+    x_labels = list(state_names.values())  # Estruturas secundárias
+    x = np.arange(len(x_labels))  # Posições no eixo X
+
+    # Criar figura para o boxplot
+    plt.figure(figsize=(7, 6))
+    plt.plot()
+
+    # Função para plotar boxplots
+    def plot_boxplot(data, positions, color, label):
+        box = plt.boxplot(data, positions=positions, widths=0.4, patch_artist=True, labels=[label] * len(positions), showfliers=False)  # Remover outliers
+        for boxplot in box['boxes']:
+            boxplot.set_facecolor(color)
+            boxplot.set_alpha(0.7)
+        for median in box['medians']:
+            median.set_color('black')
+        return box
+
+    # Plotar boxplots para cada simulação
+    for i, (probabilities, label, color) in enumerate(zip(probabilities_list, labels, colors)):
+        data = [probabilities[name] * 100 for name in x_labels]
+        plot_boxplot(data, x - 0.25 + i * 0.25, color, label)
+
+    # Configuração do gráfico
+    plt.xlabel('', fontsize=12)
+    plt.ylabel('Probability (%)', fontsize=12)  # Eixo y em percentual
+    plt.title('', fontsize=14, fontweight='bold')
+    plt.xticks(x, x_labels, rotation=45, fontsize=10)
+    plt.yticks(fontsize=10)
+    plt.grid(False)  # Remover a grade
+
+    # Cores e rótulos para a legenda
+    legend_elements = [
+        Patch(facecolor=color, edgecolor='black', linewidth=1.2, alpha=0.7, label=label)
+        for label, color in zip(labels, colors)
+    ]
+
+    # Posicionar a legenda
+    plt.legend(handles=legend_elements, frameon=False, fontsize=10, loc='upper right', edgecolor='black')  # Borda ao redor da legenda
+
+    # Ajustar layout para evitar cortes
+    plt.tight_layout()
+
+    # Salvar o gráfico em 300 dpi (PNG e TIFF)
+    os.makedirs(output_folder, exist_ok=True)  # Create the folder if it doesn't exist
+
+    # Salvar como PNG
+    plt.savefig(os.path.join(output_folder, 'secondary_structure_boxplot.png'), dpi=300, format='png', bbox_inches='tight')
+
+    # Salvar como TIFF
+    plt.savefig(os.path.join(output_folder, 'secondary_structure_boxplot.tiff'), dpi=300, format='tiff', bbox_inches='tight')
+
+    # Mostrar gráfico
+    plt.show()
+
+def ss_analysis(output_folder, *simulation_files_groups):
+    r"""
+    Main function to generate secondary structure analysis and plots.
+
+    Parameters:
+    -----------
+    output_folder : str
+        Output folder to save the plots.
+    \*simulation_files_groups : list of str
+        List of paths to .dat files for each simulation group.
+        You can pass 1, 2, or 3 groups.
+    """
+    # Helper function to process a group of files
+    def process_group(file_paths):
+        ss_data = []
+        for file in file_paths:
+            data = read_ss(file)
+            ss_data.append(data)
+        return ss_data
+    
+    # Process each group of files
+    probabilities_list = []
+    labels = []
+    colors = ['#333333', '#6A9EDA', '#54b36a']  # Cores para cada simulação
+
+    for i, group in enumerate(simulation_files_groups):
+        if group:  # Check if the list is not empty
+            ss_data = process_group(group)
+            probabilities = calculate_probabilities(ss_data[0])  # Assume only one file per group
+            probabilities_list.append(probabilities)
+            labels.append(f'Simulation {i + 1}')
+
+    # Generate plots
+    plot_ss_boxplot(probabilities_list, labels, colors[:len(probabilities_list)], output_folder)

dynamispectra/__init__.py

@@ -0,0 +1,19 @@
+# src/__init__.py
+from .Hbond import read_hbond, hbond_analysis
+from .RMSD import rmsd_analysis
+from .RMSF import read_rmsf, rmsf_analysis
+from .SASA import read_sasa, sasa_analysis
+from .Rg import read_rg, rg_analysis
+from .PCA import pca_analysis
+from .SecondaryStructure import (
+    read_ss,
+    calculate_probabilities,
+    plot_ss_boxplot,
+    ss_analysis,
+    state_mapping,
+    state_names
+)
+from .FractionSS import fractions_ss_analysis
+
+
+__all__ = ['read_hbond', 'hbond_analysis', 'read_rmsd', 'rmsd_analysis', 'read_rmsf', 'rmsf_analysis', 'read_sasa', 'sasa_analysis', 'read_rg', 'rg_analysis', 'pca_analysis', 'read_ss', 'calculate_probabilities', 'plot_ss_boxplot', 'ss_analysis', 'state_mapping', 'state_names', 'fractions_ss_analysis']

dynamispectra/main.py

@@ -0,0 +1,4 @@
+# src/dynamispectra/main.py
+
+def main():
+    print("DynamiSpectra is working!")

{dynamispectra-1.0.3.dist-info → dynamispectra-1.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: DynamiSpectra
-Version: 1.0.3
+Version: 1.0.5
 Summary: Scripts for Molecular dynamics analysis
 Home-page: https://github.com/Conradoou/DynamiSpectra
 Author: Iverson Conrado-Bezerra

dynamispectra-1.0.5.dist-info/RECORD

@@ -0,0 +1,16 @@
+dynamispectra/FractionSS.py,sha256=-4GiRsvTiwyn5rN17hNKfbmzfevnEOqOm1TxMl8gQOs,3647
+dynamispectra/Hbond.py,sha256=-Mb4g-tITqxEG1VbvYae4TgmTsIRyMis3NEm3Ey9Zu0,8658
+dynamispectra/PCA.py,sha256=nwQ7jEMvuhRAIvRRiGboQ8Cl0qrm15xSgHYCX0HlKiU,3482
+dynamispectra/RMSD.py,sha256=EUSv1Ql0f7P3VRhx6175NxjEn-2cIXx4lbVfeI88Gbk,8622
+dynamispectra/RMSF.py,sha256=M2b-KfvogfFbLUtv6PgZB4_nxWBeg9XbBpY0HjNQ9fw,8545
+dynamispectra/Rg.py,sha256=4F_vjdgcRG9LNqNCig1kdaUpMIRZrtQppYwL2OpJFYk,8557
+dynamispectra/SASA.py,sha256=o6_hcsmeY9yjAKUgsd0RpMikwC9VhoKAmQ1JLtPuXos,8661
+dynamispectra/SecondaryStructure.py,sha256=7jiYnqRgDaiR-16Zk40aCyv060y-1R__m44AVkhvm9g,6764
+dynamispectra/__init__.py,sha256=r13WozJ-uCGqA2QH84LzO-wzsv3HiMoVpZoIBydrPWk,771
+dynamispectra/cly.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dynamispectra/main.py,sha256=YhivGFXZb1qwRCPUNmST5LeejqTyHmthMhzJs9uebJY,84
+dynamispectra-1.0.5.dist-info/METADATA,sha256=PvTAdyQx1tNbIMq5F0RREWuIoC7BMkg4nddVF6-KNQ8,3807
+dynamispectra-1.0.5.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+dynamispectra-1.0.5.dist-info/entry_points.txt,sha256=aaQocKWsZi_lJ7MZMsxHmc2DpQF4BSY0fwfE1s0CLxo,37
+dynamispectra-1.0.5.dist-info/top_level.txt,sha256=MDRYQBL4n7V0YVNSoOO_yf7iv_Kz3JGoVUEHiq_zT6U,14
+dynamispectra-1.0.5.dist-info/RECORD,,

dynamispectra-1.0.5.dist-info/top_level.txt

@@ -0,0 +1 @@
+dynamispectra

dynamispectra-1.0.3.dist-info/RECORD

@@ -1,5 +0,0 @@
-dynamispectra-1.0.3.dist-info/METADATA,sha256=vjVXE6ffTuM-Z7jcabSwIHBSV8_WMvafk85Js2gfEMo,3807
-dynamispectra-1.0.3.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-dynamispectra-1.0.3.dist-info/entry_points.txt,sha256=aaQocKWsZi_lJ7MZMsxHmc2DpQF4BSY0fwfE1s0CLxo,37
-dynamispectra-1.0.3.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-dynamispectra-1.0.3.dist-info/RECORD,,

dynamispectra-1.0.3.dist-info/top_level.txt

@@ -1 +0,0 @@
-