//----------------------------------------------------------------------------//
//                           GNU GPL OS/K                                     //
//                                                                            //
//  Desc:  CPU detection                                                      //
//                                                                            //
//                                                                            //
//  Copyright © 2018-2019 The OS/K Team                                       //
//                                                                            //
//  This file is part of OS/K.                                                //
//                                                                            //
//  OS/K is free software: you can redistribute it and/or modify              //
//  it under the terms of the GNU General Public License as published by      //
//  the Free Software Foundation, either version 3 of the License, or         //
//  any later version.                                                        //
//                                                                            //
//  OS/K is distributed in the hope that it will be useful,                   //
//  but WITHOUT ANY WARRANTY//without even the implied warranty of            //
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the             //
//  GNU General Public License for more details.                              //
//                                                                            //
//  You should have received a copy of the GNU General Public License         //
//  along with OS/K.  If not, see <https://www.gnu.org/licenses/>.            //
//----------------------------------------------------------------------------//

#include <ke/cpuid.h>
#include <io/vga.h>

void KeGetCpuInfos(void)
{
    uint CpuVendorString[5] = {0};
    extern CpuInfo_t CpuInfo;
    uint eax = 0;

    CpuCpuidString(0, CpuVendorString);

    memmove(CpuInfo.vendorStr, (char *)&CpuVendorString[1], 12*sizeof(char));

    CpuCpuid(1, &eax, &CpuInfo.featureFlag);

    if (CpuInfo.featureFlag & FEAT_EDX_SSE) {
        KeActivateSSE();
    }

    DebugLog("\tCPU %s detected with features %#x and ?? speed %#d Hz ??\n",
             CpuInfo.vendorStr,
             CpuInfo.featureFlag,
             (long)KeGetCpuSpeed()
             );
}


double KeGetCpuSpeed(void)
{
    ulong flags = KePauseIRQs();

    IoWriteByteOnPort(0x43,0x34);   // set PIT channel 0 to single-shot mode
    IoWriteByteOnPort(0x40,0);
    IoWriteByteOnPort(0x40,0);      // program the counter will be
                                    // 0x10000 - n after n ticks
    long stsc = KeReadStsc();
    for (int i=0x9000;i>0;i--);
    long etsc= KeReadStsc();
    IoWriteByteOnPort(0x43,0x04);
    char lo=IoReadByteFromPort(0x40);
    char hi=IoReadByteFromPort(0x40);
    KeRestoreIRQs(flags);

    ulong ticks = (0x10000 - (hi*256+lo));
    return (etsc-stsc)*1193180.0 / ticks;
}