76 lines
2.1 KiB
C++
76 lines
2.1 KiB
C++
#include <unistd.h>
|
|
#include <ios>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
|
|
#include "avail_mem.h"
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// process_mem_usage(double &, double &) - takes two doubles by reference,
|
|
// attempts to read the system-dependent data for a process' virtual memory
|
|
// size and resident set size, and return the results in KB.
|
|
//
|
|
// On failure, returns 0.0, 0.0
|
|
|
|
void process_mem_usage(double& vm_usage, double& resident_set)
|
|
{
|
|
using std::ios_base;
|
|
using std::ifstream;
|
|
using std::string;
|
|
|
|
vm_usage = 0.0;
|
|
resident_set = 0.0;
|
|
|
|
// 'file' stat seems to give the most reliable results
|
|
//
|
|
ifstream stat_stream("/proc/self/stat",ios_base::in);
|
|
|
|
// dummy vars for leading entries in stat that we don't care about
|
|
//
|
|
string pid, comm, state, ppid, pgrp, session, tty_nr;
|
|
string tpgid, flags, minflt, cminflt, majflt, cmajflt;
|
|
string utime, stime, cutime, cstime, priority, nice;
|
|
string O, itrealvalue, starttime;
|
|
|
|
// the two fields we want
|
|
//
|
|
unsigned long vsize;
|
|
long rss;
|
|
|
|
stat_stream >> pid >> comm >> state >> ppid >> pgrp >> session >> tty_nr
|
|
>> tpgid >> flags >> minflt >> cminflt >> majflt >> cmajflt
|
|
>> utime >> stime >> cutime >> cstime >> priority >> nice
|
|
>> O >> itrealvalue >> starttime >> vsize >> rss; // don't care about the rest
|
|
|
|
stat_stream.close();
|
|
|
|
long page_size_kb = sysconf(_SC_PAGE_SIZE) / 1024; // in case x86-64 is configured to use 2MB pages
|
|
vm_usage = vsize / 1024.0;
|
|
resident_set = rss * page_size_kb;
|
|
}
|
|
|
|
unsigned long long getTotalSystemMemory()
|
|
{
|
|
long pages = sysconf(_SC_PHYS_PAGES);
|
|
long page_size = sysconf(_SC_PAGE_SIZE);
|
|
return pages * page_size;
|
|
}
|
|
|
|
double get_alpha()
|
|
{
|
|
using std::cout;
|
|
using std::endl;
|
|
|
|
double vm, rss, alpha;
|
|
//vm and rss are KB
|
|
process_mem_usage(vm, rss);
|
|
//tot_mem is B
|
|
unsigned long long tot_mem = getTotalSystemMemory();
|
|
auto percent = (rss*1024)/tot_mem;
|
|
//cout << "Percent: " << percent << endl;
|
|
//cout << "VM: " << vm << "; RSS: " << rss << endl;
|
|
alpha = percent;
|
|
return alpha;
|
|
}
|