use std::sync::Arc;

use crate::joback::{JobackBinaryRecord, JobackParameters, JobackRecord};
use crate::parameter::*;
use crate::python::parameter::*;
use crate::{
    impl_binary_record, impl_json_handling, impl_parameter, impl_parameter_from_segments,
    impl_pure_record, impl_segment_record,
};
use numpy::{PyArray2, PyReadonlyArray2, ToPyArray};
use pyo3::exceptions::PyTypeError;
use pyo3::prelude::*;
use std::convert::{TryFrom, TryInto};

/// Create a set of Joback ideal gas heat capacity parameters
/// for a segment or a pure component.
///
/// The fourth order coefficient `e` is not present in the
/// orginial publication by Joback and Reid but is required
/// for correlations for some pure components that are modeled
/// using the same polynomial approach.
///
/// Parameters
/// ----------
/// a : float
///     zeroth order coefficient
/// b : float
///     first order coefficient
/// c : float
///     second order coefficient
/// d : float
///     third order coefficient
/// e : float
///     fourth order coefficient
///
/// Returns
/// -------
/// JobackRecord
#[pyclass(name = "JobackRecord")]
#[derive(Clone)]
pub struct PyJobackRecord(pub JobackRecord);

#[pymethods]
impl PyJobackRecord {
    #[new]
    fn new(a: f64, b: f64, c: f64, d: f64, e: f64) -> Self {
        Self(JobackRecord::new(a, b, c, d, e))
    }

    fn __repr__(&self) -> PyResult<String> {
        Ok(self.0.to_string())
    }
}

impl_json_handling!(PyJobackRecord);
impl_pure_record!(JobackRecord, PyJobackRecord);
impl_segment_record!(JobackRecord, PyJobackRecord);

#[pyclass(name = "JobackBinaryRecord")]
#[derive(Clone)]
pub struct PyJobackBinaryRecord(pub JobackBinaryRecord);

impl_binary_record!(JobackBinaryRecord, PyJobackBinaryRecord);

/// Create a set of Joback parameters from records.
///
/// Parameters
/// ----------
/// pure_records : List[PureRecord]
///     pure substance records.
/// substances : List[str], optional
///     The substances to use. Filters substances from `pure_records` according to
///     `search_option`.
///     When not provided, all entries of `pure_records` are used.
/// search_option : {'Name', 'Cas', 'Inchi', 'IupacName', 'Formula', 'Smiles'}, optional, defaults to 'Name'.
///     Identifier that is used to search substance.
///
/// Returns
/// -------
/// JobackParameters
#[pyclass(name = "JobackParameters")]
#[pyo3(text_signature = "(pure_records, substances=None, search_option='Name')")]
#[derive(Clone)]
pub struct PyJobackParameters(pub Arc<JobackParameters>);

impl_parameter!(
    JobackParameters,
    PyJobackParameters,
    PyJobackRecord,
    PyJobackBinaryRecord
);
impl_parameter_from_segments!(JobackParameters, PyJobackParameters);

#[pymethods]
impl PyJobackParameters {
    // fn _repr_markdown_(&self) -> String {
    //     self.0.to_markdown()
    // }
}