/*      Class ExampleProgramTwoImpedRegr

        Example Program for class ImpedSpecRegression

        Circuit Model supplied via the interface ImpedSpecModel

        Michael Thomas Flanagan  (http:\\www.ee.ucl.ac\~flanaga\java)

        May 2007

*/

import java.util.Vector;
import flanagan.complex.Complex;
import flanagan.io.*;
import flanagan.circuits.*;

public class ExampleProgramTwoImpedRegr{


    public static void main(String[] args){

        // Select data file
        FileChooser fin = new FileChooser();
        String inputFileName = fin.selectFile();

        // Create a default option name for the output file from the inut file name
        String outputFileName = fin.getStemName() + "Output.txt";

        // Request confirmation of or replacement name for the output file name
        outputFileName = Db.readLine("Output file name?", outputFileName);

        // Read the input data file
        String dataTitle = fin.readLine();                  // data title
        double appliedVoltage = fin.readDouble();           // applied voltage
        Complex referenceImpedance = fin.readComplex();     // reference impedance
        int numberOfFrequencies = fin.readInt();            // number of frequencies

        double[] frequencies = new double[numberOfFrequencies];
        double[] magnitudes = new double[numberOfFrequencies];
        double[] phases = new double[numberOfFrequencies];

        for(int i=0; i<numberOfFrequencies; i++){
            frequencies[i] = fin.readDouble();
            magnitudes[i] = fin.readDouble();
            phases[i] = fin.readDouble();
        }

        // close input file
        fin.close();

        // Create an instance of ImpedSpecRegression
        ImpedSpecRegression isr = new ImpedSpecRegression("Example Program Two");

        // Create an instance of the class, implementing the interface ImpedSpecModel,
        // that contains the impedance calculation for the supplied circuit model
        ExampleTwoModel circuit = new ExampleTwoModel();

        String[] symbols = {"R1", "C1", "R2", "C2", "R3"};                      // model parameters
        double[] initialEstimates = {3.0e4, 5.0e-8, 1.0e5, 1.0e-6, 1.0e3};      // initial estimates of the parameter values

        // enter the user supplied circuit model
        isr.setModel(circuit, symbols, initialEstimates);

        // enter the applied voltage and reference impedance
        isr.setAppliedVoltage(appliedVoltage);
        isr.setReferenceImpedance(referenceImpedance);

        // enter the experimental data
        isr.voltageDataAsPhasorDeg(frequencies, magnitudes, phases);

        Vector results = isr.plotVoltageMagnitudes();       // plot the voltage magnitude of the test circuit against frequency
        isr.plotVoltagePhases();                            // plot the voltage phase of the test circuit against frequency

        isr.plotImpedanceMagnitudes();                      // plot the impedance magnitude of the test circuit against frequency
        isr.plotImpedancePhases();                          // plot the impedance phase of the test circuit against frequency
        isr.plotColeCole();                                 // plot the Cole-Cole plot (Imag[z] against Real[z]

        isr.printToTextFile(outputFileName);                // print the results to a text file
        isr.printToExcelFile(outputFileName);               // print the results to an Excel file

    }
}

// Class containing the method that returns the complex impedance of the model circuit
class ExampleTwoModel extends Impedance implements ImpedSpecModel{

    public Complex modelImpedance(double[] parameter, double omega){

        // Impedance of R1 in parallel with C1 = Z1
        Complex z1 = Impedance.rInParallelWithC(parameter[0], parameter[1], omega);
        // Impedance of R2 in parallel with C2 = Z2
        Complex z2 = Impedance.rInParallelWithC(parameter[2], parameter[3], omega);
        // Impedance of Z1 in series with Z2  = Z3
        Complex z3 = z1.plus(z2);
        // Return impedance of Z3 in series with R3
        return z3.plus(parameter[4]);
    }
}