llvm-project/lldb/tools/lldb-perf/lib/Results.cpp

276 lines
9.0 KiB
C++
Raw Normal View History

//===-- Results.cpp ---------------------------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "Results.h"
#include <assert.h>
#ifdef __APPLE__
#include "CFCMutableArray.h"
#include "CFCMutableDictionary.h"
#include "CFCReleaser.h"
#include "CFCString.h"
#endif
using namespace lldb_perf;
static void
AddResultToArray (CFCMutableArray &array, Results::Result *result);
static void
AddResultToDictionary (CFCMutableDictionary &parent_dict, const char *key, Results::Result *result);
static void
AddResultToArray (CFCMutableArray &parent_array, Results::Result *result)
{
switch (result->GetType())
{
case Results::Result::Type::Invalid:
break;
case Results::Result::Type::Array:
{
Results::Array *value = result->GetAsArray();
CFCMutableArray array;
value->ForEach([&array](const Results::ResultSP &value_sp) -> bool
{
AddResultToArray (array, value_sp.get());
return true;
});
parent_array.AppendValue(array.get(), true);
}
break;
case Results::Result::Type::Dictionary:
{
Results::Dictionary *value = result->GetAsDictionary();
CFCMutableDictionary dict;
value->ForEach([&dict](const std::string &key, const Results::ResultSP &value_sp) -> bool
{
AddResultToDictionary (dict, key.c_str(), value_sp.get());
return true;
});
if (result->GetDescription())
{
dict.AddValueCString(CFSTR("description"), result->GetDescription());
}
parent_array.AppendValue(dict.get(), true);
}
break;
case Results::Result::Type::Double:
{
double d = result->GetAsDouble()->GetValue();
CFCReleaser<CFNumberRef> cf_number(::CFNumberCreate (kCFAllocatorDefault, kCFNumberDoubleType, &d));
if (cf_number.get())
parent_array.AppendValue(cf_number.get(), true);
}
break;
case Results::Result::Type::String:
{
CFCString cfstr (result->GetAsString()->GetValue());
if (cfstr.get())
parent_array.AppendValue(cfstr.get(), true);
}
break;
case Results::Result::Type::Unsigned:
{
uint64_t uval64 = result->GetAsUnsigned()->GetValue();
CFCReleaser<CFNumberRef> cf_number(::CFNumberCreate (kCFAllocatorDefault, kCFNumberSInt64Type, &uval64));
if (cf_number.get())
parent_array.AppendValue(cf_number.get(), true);
}
break;
default:
assert (!"unhandled result");
break;
}
}
static void
AddResultToDictionary (CFCMutableDictionary &parent_dict, const char *key, Results::Result *result)
{
assert (key && key[0]);
CFCString cf_key(key);
switch (result->GetType())
{
case Results::Result::Type::Invalid:
break;
case Results::Result::Type::Array:
{
Results::Array *value = result->GetAsArray();
CFCMutableArray array;
value->ForEach([&array](const Results::ResultSP &value_sp) -> bool
{
AddResultToArray (array, value_sp.get());
return true;
});
parent_dict.AddValue(cf_key.get(), array.get(), true);
}
break;
case Results::Result::Type::Dictionary:
{
Results::Dictionary *value = result->GetAsDictionary();
CFCMutableDictionary dict;
value->ForEach([&dict](const std::string &key, const Results::ResultSP &value_sp) -> bool
{
AddResultToDictionary (dict, key.c_str(), value_sp.get());
return true;
});
if (result->GetDescription())
{
dict.AddValueCString(CFSTR("description"), result->GetDescription());
}
parent_dict.AddValue(cf_key.get(), dict.get(), true);
}
break;
case Results::Result::Type::Double:
{
parent_dict.SetValueDouble(cf_key.get(), result->GetAsDouble()->GetValue(), true);
}
break;
case Results::Result::Type::String:
{
parent_dict.SetValueCString(cf_key.get(), result->GetAsString()->GetValue(), true);
}
break;
case Results::Result::Type::Unsigned:
{
parent_dict.SetValueUInt64 (cf_key.get(), result->GetAsUnsigned()->GetValue(), true);
}
break;
default:
assert (!"unhandled result");
break;
}
}
void
Results::Write (const char *out_path)
{
#ifdef __APPLE__
CFCMutableDictionary dict;
m_results.ForEach([&dict](const std::string &key, const ResultSP &value_sp) -> bool
{
AddResultToDictionary (dict, key.c_str(), value_sp.get());
return true;
});
CFDataRef xmlData = CFPropertyListCreateData(kCFAllocatorDefault, dict.get(), kCFPropertyListXMLFormat_v1_0, 0, NULL);
if (out_path == NULL)
out_path = "/dev/stdout";
CFURLRef file = CFURLCreateFromFileSystemRepresentation(NULL, (const UInt8*)out_path, strlen(out_path), FALSE);
CFURLWriteDataAndPropertiesToResource(file, xmlData, NULL, NULL);
#endif
}
Results::ResultSP
Results::Dictionary::AddUnsigned (const char *name, const char *description, uint64_t value)
{
assert (name && name[0]);
if (description && description[0])
{
std::unique_ptr<Results::Dictionary> value_dict_ap (new Results::Dictionary ());
value_dict_ap->AddString("description", NULL, description);
value_dict_ap->AddUnsigned("value", NULL, value);
m_dictionary[std::string(name)] = ResultSP (value_dict_ap.release());
}
else
m_dictionary[std::string(name)] = ResultSP (new Unsigned (name, description, value));
return m_dictionary[std::string(name)];
}
Results::ResultSP
Results::Dictionary::AddDouble (const char *name, const char *description, double value)
{
assert (name && name[0]);
if (description && description[0])
{
std::unique_ptr<Results::Dictionary> value_dict_ap (new Results::Dictionary ());
value_dict_ap->AddString("description", NULL, description);
value_dict_ap->AddDouble("value", NULL, value);
m_dictionary[std::string(name)] = ResultSP (value_dict_ap.release());
}
else
m_dictionary[std::string(name)] = ResultSP (new Double (name, description, value));
return m_dictionary[std::string(name)];
}
Results::ResultSP
Results::Dictionary::AddString (const char *name, const char *description, const char *value)
{
assert (name && name[0]);
if (description && description[0])
{
std::unique_ptr<Results::Dictionary> value_dict_ap (new Results::Dictionary ());
value_dict_ap->AddString("description", NULL, description);
value_dict_ap->AddString("value", NULL, value);
m_dictionary[std::string(name)] = ResultSP (value_dict_ap.release());
}
else
m_dictionary[std::string(name)] = ResultSP (new String (name, description, value));
return m_dictionary[std::string(name)];
}
Results::ResultSP
Results::Dictionary::Add (const char *name, const char *description, const ResultSP &result_sp)
{
assert (name && name[0]);
if (description && description[0])
{
std::unique_ptr<Results::Dictionary> value_dict_ap (new Results::Dictionary ());
value_dict_ap->AddString("description", NULL, description);
value_dict_ap->Add("value", NULL, result_sp);
m_dictionary[std::string(name)] = ResultSP (value_dict_ap.release());
}
else
m_dictionary[std::string(name)] = result_sp;
return m_dictionary[std::string(name)];
}
void
Results::Dictionary::ForEach (const std::function <bool (const std::string &, const ResultSP &)> &callback)
{
collection::const_iterator pos, end = m_dictionary.end();
for (pos = m_dictionary.begin(); pos != end; ++pos)
{
if (callback (pos->first.c_str(), pos->second) == false)
return;
}
}
Results::ResultSP
Results::Array::Append (const ResultSP &result_sp)
{
m_array.push_back (result_sp);
return result_sp;
}
void
Results::Array::ForEach (const std::function <bool (const ResultSP &)> &callback)
{
collection::const_iterator pos, end = m_array.end();
for (pos = m_array.begin(); pos != end; ++pos)
{
if (callback (*pos) == false)
return;
}
}