import numpy as np
import matplotlib.pyplot as plt

# Fungsi untuk menghitung flow berdasarkan DP
def calculate_flow(dp_values, C=1.0):
    return C * np.sqrt(dp_values)

# Fungsi untuk menggambar grafik hubungan DP dan Flow
def plot_flow_vs_dp(C=1.0, dp_max=100, num_points=500):
    dp_values = np.linspace(0, dp_max, num_points)
    flow_values = calculate_flow(dp_values, C)

    plt.figure(figsize=(10, 6))
    plt.plot(dp_values, flow_values, label=f'C = {C}')
    plt.title('Hubungan Differential Pressure (DP) dan Flow pada Orifice Flowmeter')
    plt.xlabel('Differential Pressure (DP)')
    plt.ylabel('Flow Rate (Q)')
    plt.grid(True, linestyle='--', alpha=0.6)
    plt.legend()
    # Simpan grafik jika diperlukan
    # plt.savefig('dp_vs_flow.svg', dpi=300)
    plt.show()

# Contoh penggunaan
if __name__ == "__main__":
    # Anda dapat mengubah nilai C dan dp_max
    plot_flow_vs_dp(C=1.0, dp_max=100)