package lab1;

import java.sql.*;
import java.util.*;
import org.jfree.chart.*;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.category.*;
import org.jfree.ui.*;


public class StudentGrade extends ApplicationFrame{
        private static final String USERNAME = "lalala";
        private static final String PASSWORD = "11223344";
        private static final String URL = "jdbc:mysql://localhost:3306/lab1";
        public static ArrayList<String> s  = new <String> ArrayList() ;
        public static ArrayList<Double> d  = new <Double> ArrayList() ;
        public static Iterator op1 = s.iterator();
        public static Iterator op2 = d.iterator();

        public StudentGrade( String applicationTitle , String chartTitle )
   {
      super( applicationTitle );
      JFreeChart barChart = ChartFactory.createBarChart(chartTitle, "Category", "Score", createDataset(), PlotOrientation.VERTICAL, true, true, false);

      ChartPanel chartPanel = new ChartPanel( barChart );
      chartPanel.setPreferredSize(new java.awt.Dimension( 560 , 367 ) );
      setContentPane( chartPanel );
   }

     private static CategoryDataset createDataset( )
   {
    //some just dummy of variable
      final String fiat = "FIAT";
      final String audi = "AUDI";
      final String ford = "FORD";
      final String speed = "Speed";
      final String millage = "Millage";
      final String userrating = "User Rating";
      final String safety = "safety";
      DefaultCategoryDataset dataset = new DefaultCategoryDataset( );
            for (Iterator<String> iterator = s.iterator(); iterator.hasNext();){
                Object a= iterator.next();
                Object b= iterator.next();
                String c = (String) a;
                double d = (double) a;
                dataset.addValue(d , speed , c);
            }
//      dataset.addValue( 1.0 , fiat , speed );
//      dataset.addValue( 3.0 , fiat , userrating );
//      dataset.addValue( 5.0 , fiat , millage );
//      dataset.addValue( 5.0 , fiat , safety );


      return dataset;
   }

    public static void main(String [] args){
        Scanner input = new Scanner (System.in);
        Connection connect = null;
        Statement stmt = null;

        try{
            connect = DriverManager.getConnection(URL, USERNAME, PASSWORD);
            System.out.println("Connected");

            //Read from input and insert database
            System.out.print("Students in the course :");
            int repeat = input.nextInt();
                for(int i=0;i<repeat;i++){
                    System.out.print("Enter student name :");
                    String name = input.next();
                    System.out.print("Enter student marks :");
                    int marks= input.nextInt();
                    stmt = connect.createStatement();
                    String sql = "INSERT INTO score (Name, Marks)" + "VALUES (?, ?)";
                        PreparedStatement preparedStatement = connect.prepareStatement(sql);
                            preparedStatement.setString(1, name);
                            preparedStatement.setInt(2, marks);
                            preparedStatement.executeUpdate();
                }

         //Read from database
        String select = "SELECT * FROM score";
        stmt = connect.createStatement();
        ResultSet rs = stmt.executeQuery(select);
        System.out.println("\nThe list in score db :" );
        double avg=0;
//        avg=s.size();
        //print out the list
        while (rs.next()) {
        String name = rs.getString("Name");
        double marks = rs.getDouble("Marks");
        avg+= rs.getInt("Marks");
        s.add(name);
        d.add(marks);
        System.out.format("%s, %s \n",name, marks);
      }
        System.out.println("The average is "+(avg/repeat));
      stmt.close();
        }catch(SQLException e){
            System.out.println("Failed to connect");
        }
        //Call graph
      StudentGrade chart = new StudentGrade("Data Display", "Statistic for Students");
      chart.pack( );
      RefineryUtilities.centerFrameOnScreen( chart );
      chart.setVisible( true );

    }
}