#!/usr/bin/python3
# Copyright (C) 2018 spdf
# 
# This file is part of Elektron.
# 
# Elektron 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
# (at your option) any later version.
# 
# Elektron 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 Elektron.  If not, see <http://www.gnu.org/licenses/>.
# 

import sys
import os
import math

def main():
   POWER_CAP_DIR = "/sys/class/powercap/"

   if not len(sys.argv) == 2:
      print(sys.argv[0] + " <throttle percent>")
      exit(-1)

   if not os.path.exists(POWER_CAP_DIR):
      print("Powercap framework not installed exist")
      exit(-1)

   throttle_percent = float(sys.argv[1])/100.0

   if throttle_percent > 1.0 or throttle_percent < 0:
      print("Percent must be between 0 and 100")
      exit(-1)

#   print(throttle_percent)



   for directory in os.listdir(POWER_CAP_DIR):
      if len(directory.split(':')) == 2:
         max_watts = open(POWER_CAP_DIR + directory + '/constraint_0_max_power_uw', 'r')
         rapl_cap_watts = open(POWER_CAP_DIR + directory + '/constraint_0_power_limit_uw', 'w') #0=longer window, 1=shorter window

         rapl_cap_watts.write(str(math.ceil(float(max_watts.read())*throttle_percent)))
   

main()