mirror of
https://github.com/aportelli/LatAnalyze.git
synced 2024-11-10 00:45:36 +00:00
Compare commits
5 Commits
fecc5690f3
...
49ae63ba61
Author | SHA1 | Date | |
---|---|---|---|
|
49ae63ba61 | ||
6739019c83 | |||
13fddf4947 | |||
1604b4712f | |||
|
269d0c338e |
@ -18,6 +18,7 @@
|
||||
*/
|
||||
|
||||
#include <LatAnalyze/Core/Math.hpp>
|
||||
#include <LatAnalyze/Numerical/GslFFT.hpp>
|
||||
#include <LatAnalyze/includes.hpp>
|
||||
#include <gsl/gsl_cdf.h>
|
||||
|
||||
@ -48,16 +49,42 @@ DMat MATH_NAMESPACE::corrToVar(const DMat &corr, const DVec &varDiag)
|
||||
return res;
|
||||
}
|
||||
|
||||
double MATH_NAMESPACE::svdDynamicRange(const DMat &mat)
|
||||
double MATH_NAMESPACE::conditionNumber(const DMat &mat)
|
||||
{
|
||||
DVec s = mat.singularValues();
|
||||
|
||||
return s.maxCoeff()/s.minCoeff();
|
||||
}
|
||||
|
||||
double MATH_NAMESPACE::svdDynamicRangeDb(const DMat &mat)
|
||||
double MATH_NAMESPACE::cdr(const DMat &mat)
|
||||
{
|
||||
return 10.*log10(svdDynamicRange(mat));
|
||||
return 10.*log10(conditionNumber(mat));
|
||||
}
|
||||
|
||||
template <typename FFT>
|
||||
double nsdr(const DMat &m)
|
||||
{
|
||||
Index n = m.rows();
|
||||
FFT fft(n);
|
||||
CMat buf(n, 1);
|
||||
|
||||
FOR_VEC(buf, i)
|
||||
{
|
||||
buf(i) = 0.;
|
||||
for (Index j = 0; j < n; ++j)
|
||||
{
|
||||
buf(i) += m(j, (i+j) % n);
|
||||
}
|
||||
buf(i) /= n;
|
||||
}
|
||||
fft(buf, FFT::Forward);
|
||||
|
||||
return 10.*log10(buf.real().maxCoeff()/buf.real().minCoeff());
|
||||
}
|
||||
|
||||
double MATH_NAMESPACE::nsdr(const DMat &mat)
|
||||
{
|
||||
return ::nsdr<GslFFT>(mat);
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
|
@ -73,8 +73,9 @@ namespace MATH_NAMESPACE
|
||||
DMat corrToVar(const DMat &corr, const DVec &varDiag);
|
||||
|
||||
// matrix SVD dynamic range
|
||||
double svdDynamicRange(const DMat &mat);
|
||||
double svdDynamicRangeDb(const DMat &mat);
|
||||
double conditionNumber(const DMat &mat);
|
||||
double cdr(const DMat &mat);
|
||||
double nsdr(const DMat &mat);
|
||||
|
||||
// Constants
|
||||
constexpr double pi = 3.1415926535897932384626433832795028841970;
|
||||
|
@ -217,146 +217,25 @@ void RunContext::reset(void)
|
||||
#define CODE_WIDTH 6
|
||||
#define CODE_MOD setw(CODE_WIDTH) << left
|
||||
|
||||
// Instruction operator ////////////////////////////////////////////////////////
|
||||
ostream &Latan::operator<<(ostream& out, const Instruction& ins)
|
||||
auto readConstant(Program::const_iterator ip)
|
||||
-> std::tuple<double, Program::const_iterator>
|
||||
{
|
||||
ins.print(out);
|
||||
double value = 0.0;
|
||||
std::copy(ip, ip + sizeof(double), reinterpret_cast<std::uint8_t*>(&value));
|
||||
|
||||
return out;
|
||||
return std::make_tuple(value, ip + sizeof(double));
|
||||
}
|
||||
|
||||
// Push constructors ///////////////////////////////////////////////////////////
|
||||
Push::Push(const double val)
|
||||
: type_(ArgType::Constant)
|
||||
, val_(val)
|
||||
, address_(0)
|
||||
, name_("")
|
||||
{}
|
||||
|
||||
Push::Push(const unsigned int address, const string &name)
|
||||
: type_(ArgType::Variable)
|
||||
, val_(0.0)
|
||||
, address_(address)
|
||||
, name_(name)
|
||||
{}
|
||||
|
||||
// Push execution //////////////////////////////////////////////////////////////
|
||||
void Push::operator()(RunContext &context) const
|
||||
auto readAddress(Program::const_iterator ip)
|
||||
-> std::tuple<unsigned int, Program::const_iterator>
|
||||
{
|
||||
if (type_ == ArgType::Constant)
|
||||
{
|
||||
context.stack().push(val_);
|
||||
}
|
||||
else
|
||||
{
|
||||
context.stack().push(context.getVariable(address_));
|
||||
}
|
||||
context.incrementInsIndex();
|
||||
unsigned int address = 0.0;
|
||||
const auto end = ip + sizeof(unsigned int);
|
||||
std::copy(ip, end, reinterpret_cast<std::uint8_t*>(&address));
|
||||
|
||||
return std::make_tuple(address, end);
|
||||
}
|
||||
|
||||
// Push print //////////////////////////////////////////////////////////////////
|
||||
void Push::print(ostream &out) const
|
||||
{
|
||||
out << CODE_MOD << "push";
|
||||
if (type_ == ArgType::Constant)
|
||||
{
|
||||
out << CODE_MOD << val_;
|
||||
}
|
||||
else
|
||||
{
|
||||
out << CODE_MOD << name_ << " @v" << address_;
|
||||
}
|
||||
}
|
||||
|
||||
// Pop constructor /////////////////////////////////////////////////////////////
|
||||
Pop::Pop(const unsigned int address, const string &name)
|
||||
: address_(address)
|
||||
, name_(name)
|
||||
{}
|
||||
|
||||
// Pop execution ///////////////////////////////////////////////////////////////
|
||||
void Pop::operator()(RunContext &context) const
|
||||
{
|
||||
if (!name_.empty())
|
||||
{
|
||||
context.setVariable(address_, context.stack().top());
|
||||
}
|
||||
context.stack().pop();
|
||||
context.incrementInsIndex();
|
||||
}
|
||||
|
||||
// Pop print ///////////////////////////////////////////////////////////////////
|
||||
void Pop::print(ostream &out) const
|
||||
{
|
||||
out << CODE_MOD << "pop" << CODE_MOD << name_ << " @v" << address_;
|
||||
}
|
||||
|
||||
// Store constructor ///////////////////////////////////////////////////////////
|
||||
Store::Store(const unsigned int address, const string &name)
|
||||
: address_(address)
|
||||
, name_(name)
|
||||
{}
|
||||
|
||||
// Store execution /////////////////////////////////////////////////////////////
|
||||
void Store::operator()(RunContext &context) const
|
||||
{
|
||||
if (!name_.empty())
|
||||
{
|
||||
context.setVariable(address_, context.stack().top());
|
||||
}
|
||||
context.incrementInsIndex();
|
||||
}
|
||||
|
||||
// Store print /////////////////////////////////////////////////////////////////
|
||||
void Store::print(ostream &out) const
|
||||
{
|
||||
out << CODE_MOD << "store" << CODE_MOD << name_ << " @v" << address_;
|
||||
}
|
||||
|
||||
// Call constructor ////////////////////////////////////////////////////////////
|
||||
Call::Call(const unsigned int address, const string &name)
|
||||
: address_(address)
|
||||
, name_(name)
|
||||
{}
|
||||
|
||||
// Call execution //////////////////////////////////////////////////////////////
|
||||
void Call::operator()(RunContext &context) const
|
||||
{
|
||||
context.stack().push((*context.getFunction(address_))(context.stack()));
|
||||
context.incrementInsIndex();
|
||||
}
|
||||
|
||||
// Call print //////////////////////////////////////////////////////////////////
|
||||
void Call::print(ostream &out) const
|
||||
{
|
||||
out << CODE_MOD << "call" << CODE_MOD << name_ << " @f" << address_;
|
||||
}
|
||||
|
||||
// Math operations /////////////////////////////////////////////////////////////
|
||||
#define DEF_OP(name, nArg, exp, insName)\
|
||||
void name::operator()(RunContext &context) const\
|
||||
{\
|
||||
double x[nArg];\
|
||||
for (int i = 0; i < nArg; ++i)\
|
||||
{\
|
||||
x[nArg-1-i] = context.stack().top();\
|
||||
context.stack().pop();\
|
||||
}\
|
||||
context.stack().push(exp);\
|
||||
context.incrementInsIndex();\
|
||||
}\
|
||||
void name::print(ostream &out) const\
|
||||
{\
|
||||
out << CODE_MOD << insName;\
|
||||
}
|
||||
|
||||
DEF_OP(Neg, 1, -x[0], "neg")
|
||||
DEF_OP(Add, 2, x[0] + x[1], "add")
|
||||
DEF_OP(Sub, 2, x[0] - x[1], "sub")
|
||||
DEF_OP(Mul, 2, x[0]*x[1], "mul")
|
||||
DEF_OP(Div, 2, x[0]/x[1], "div")
|
||||
DEF_OP(Pow, 2, pow(x[0],x[1]), "pow")
|
||||
|
||||
/******************************************************************************
|
||||
* ExprNode implementation *
|
||||
******************************************************************************/
|
||||
@ -442,29 +321,35 @@ ostream &Latan::operator<<(ostream &out, const ExprNode &n)
|
||||
return out;
|
||||
}
|
||||
|
||||
#define PUSH_INS(program, type, ...)\
|
||||
program.push_back(unique_ptr<type>(new type(__VA_ARGS__)))
|
||||
#define GET_ADDRESS(address, table, name)\
|
||||
try\
|
||||
{\
|
||||
address = (table).at(name);\
|
||||
}\
|
||||
catch (out_of_range)\
|
||||
{\
|
||||
address = (table).size();\
|
||||
(table)[(name)] = address;\
|
||||
}\
|
||||
// Bytecode helper functions ///////////////////////////////////////////////////
|
||||
void pushInstruction(Program &program, const Instruction instruction) {
|
||||
program.push_back(static_cast<std::uint8_t>(instruction));
|
||||
}
|
||||
|
||||
void pushAddress(Program &program, const unsigned int address) {
|
||||
const auto address_ptr = reinterpret_cast<const std::uint8_t*>(&address);
|
||||
const auto size = sizeof(unsigned int);
|
||||
program.insert(program.end(), address_ptr, address_ptr + size);
|
||||
}
|
||||
|
||||
void pushConstant(Program &program, const double value) {
|
||||
const auto value_ptr = reinterpret_cast<const std::uint8_t*>(&value);
|
||||
const auto size = sizeof(double);
|
||||
program.insert(program.end(), value_ptr, value_ptr + size);
|
||||
}
|
||||
|
||||
// VarNode compile /////////////////////////////////////////////////////////////
|
||||
void VarNode::compile(Program &program, RunContext &context) const
|
||||
{
|
||||
PUSH_INS(program, Push, context.getVariableAddress(getName()), getName());
|
||||
pushInstruction(program, Instruction::LOAD);
|
||||
pushAddress(program, context.getVariableAddress(getName()));
|
||||
}
|
||||
|
||||
// CstNode compile /////////////////////////////////////////////////////////////
|
||||
void CstNode::compile(Program &program, RunContext &context __dumb) const
|
||||
{
|
||||
PUSH_INS(program, Push, strTo<double>(getName()));
|
||||
pushInstruction(program, Instruction::CONST);
|
||||
pushConstant(program, strTo<double>(getName()));
|
||||
}
|
||||
|
||||
// SemicolonNode compile ///////////////////////////////////////////////////////
|
||||
@ -482,6 +367,10 @@ void SemicolonNode::compile(Program &program, RunContext &context) const
|
||||
{
|
||||
n[i].compile(program, context);
|
||||
}
|
||||
// Where a variable has just been assigned, pop it off the stack.
|
||||
if (isAssign) {
|
||||
pushInstruction(program, Instruction::POP);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -492,19 +381,10 @@ void AssignNode::compile(Program &program, RunContext &context) const
|
||||
|
||||
if (isDerivedFrom<VarNode>(&n[0]))
|
||||
{
|
||||
bool hasSemicolonParent = isDerivedFrom<SemicolonNode>(getParent());
|
||||
unsigned int address;
|
||||
|
||||
n[1].compile(program, context);
|
||||
address = context.addVariable(n[0].getName());
|
||||
if (hasSemicolonParent)
|
||||
{
|
||||
PUSH_INS(program, Pop, address, n[0].getName());
|
||||
}
|
||||
else
|
||||
{
|
||||
PUSH_INS(program, Store, address, n[0].getName());
|
||||
}
|
||||
const unsigned int address = context.addVariable(n[0].getName());
|
||||
pushInstruction(program, Instruction::STORE);
|
||||
pushAddress(program, address);
|
||||
}
|
||||
else
|
||||
{
|
||||
@ -519,19 +399,37 @@ void AssignNode::compile(Program &program, RunContext &context) const
|
||||
|
||||
void MathOpNode::compile(Program &program, RunContext &context) const
|
||||
{
|
||||
#define PUSH_BINARY_OP(op, instruction) \
|
||||
case op: \
|
||||
pushInstruction(program, Instruction::instruction); \
|
||||
break;
|
||||
|
||||
auto &n = *this;
|
||||
|
||||
for (Index i = 0; i < n.getNArg(); ++i)
|
||||
{
|
||||
n[i].compile(program, context);
|
||||
}
|
||||
IFNODE("-", 1) PUSH_INS(program, Neg,);
|
||||
ELIFNODE("+", 2) PUSH_INS(program, Add,);
|
||||
ELIFNODE("-", 2) PUSH_INS(program, Sub,);
|
||||
ELIFNODE("*", 2) PUSH_INS(program, Mul,);
|
||||
ELIFNODE("/", 2) PUSH_INS(program, Div,);
|
||||
ELIFNODE("^", 2) PUSH_INS(program, Pow,);
|
||||
ELSE LATAN_ERROR(Compilation, "unknown operator '" + getName() + "'");
|
||||
if (n.getName() == "-" and n.getNArg() == 1) {
|
||||
pushInstruction(program, Instruction::NEG);
|
||||
return;
|
||||
}
|
||||
|
||||
if (getNArg() != 2) {
|
||||
LATAN_ERROR(Compilation, "unknown operator '" + getName() + "'");
|
||||
}
|
||||
|
||||
switch (getName()[0]) {
|
||||
PUSH_BINARY_OP('+', ADD)
|
||||
PUSH_BINARY_OP('-', SUB)
|
||||
PUSH_BINARY_OP('*', MUL)
|
||||
PUSH_BINARY_OP('/', DIV)
|
||||
PUSH_BINARY_OP('^', POW)
|
||||
default:
|
||||
LATAN_ERROR(Compilation, "unknown operator '" + getName() + "'");
|
||||
}
|
||||
|
||||
#undef PUSH_BINARY_OP
|
||||
}
|
||||
|
||||
// FuncNode compile ////////////////////////////////////////////////////////////
|
||||
@ -543,7 +441,8 @@ void FuncNode::compile(Program &program, RunContext &context) const
|
||||
{
|
||||
n[i].compile(program, context);
|
||||
}
|
||||
PUSH_INS(program, Call, context.getFunctionAddress(getName()), getName());
|
||||
pushInstruction(program, Instruction::CALL);
|
||||
pushAddress(program, context.getFunctionAddress(getName()));
|
||||
}
|
||||
|
||||
// ReturnNode compile ////////////////////////////////////////////////////////////
|
||||
@ -552,7 +451,7 @@ void ReturnNode::compile(Program &program, RunContext &context) const
|
||||
auto &n = *this;
|
||||
|
||||
n[0].compile(program, context);
|
||||
program.push_back(nullptr);
|
||||
pushInstruction(program, Instruction::RET);
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
@ -582,7 +481,7 @@ MathInterpreter::MathInterpreter(const std::string &code)
|
||||
// access //////////////////////////////////////////////////////////////////////
|
||||
const Instruction * MathInterpreter::operator[](const Index i) const
|
||||
{
|
||||
return program_[i].get();
|
||||
return reinterpret_cast<const Instruction*>(&program_[i]);
|
||||
}
|
||||
|
||||
const ExprNode * MathInterpreter::getAST(void) const
|
||||
@ -592,7 +491,7 @@ const ExprNode * MathInterpreter::getAST(void) const
|
||||
|
||||
void MathInterpreter::push(const Instruction *i)
|
||||
{
|
||||
program_.push_back(unique_ptr<const Instruction>(i));
|
||||
pushInstruction(program_, *i);
|
||||
}
|
||||
|
||||
// initialization //////////////////////////////////////////////////////////////
|
||||
@ -695,7 +594,7 @@ void MathInterpreter::compile(RunContext &context)
|
||||
root_->compile(program_, context);
|
||||
for (unsigned int i = 0; i < program_.size(); ++i)
|
||||
{
|
||||
if (!program_[i])
|
||||
if (static_cast<Instruction>(program_[i]) == Instruction::RET)
|
||||
{
|
||||
gotReturn = true;
|
||||
program_.resize(i);
|
||||
@ -726,20 +625,145 @@ void MathInterpreter::operator()(RunContext &context)
|
||||
|
||||
void MathInterpreter::execute(RunContext &context) const
|
||||
{
|
||||
context.setInsIndex(0);
|
||||
while (context.getInsIndex() != program_.size())
|
||||
{
|
||||
(*(program_[context.getInsIndex()]))(context);
|
||||
#define BINARY_OP_CASE(instruction, expr) \
|
||||
case Instruction::instruction: { \
|
||||
const auto second = context.stack().top(); \
|
||||
context.stack().pop(); \
|
||||
const auto first = context.stack().top(); \
|
||||
context.stack().pop(); \
|
||||
context.stack().push(expr); \
|
||||
break; \
|
||||
}
|
||||
auto ip = program_.begin();
|
||||
|
||||
while (ip != program_.end()) {
|
||||
const auto instruction = static_cast<Instruction>(*ip);
|
||||
ip++;
|
||||
|
||||
switch (instruction) {
|
||||
BINARY_OP_CASE(ADD, first + second)
|
||||
BINARY_OP_CASE(SUB, first - second)
|
||||
BINARY_OP_CASE(MUL, first * second)
|
||||
BINARY_OP_CASE(DIV, first / second)
|
||||
BINARY_OP_CASE(POW, std::pow(first, second))
|
||||
case Instruction::NEG: {
|
||||
const auto operand = context.stack().top();
|
||||
context.stack().pop();
|
||||
context.stack().push(-operand);
|
||||
break;
|
||||
}
|
||||
case Instruction::CONST: {
|
||||
double value = 0.0;
|
||||
std::tie(value, ip) = readConstant(ip);
|
||||
context.stack().push(value);
|
||||
break;
|
||||
}
|
||||
case Instruction::POP:
|
||||
context.stack().pop();
|
||||
break;
|
||||
case Instruction::LOAD: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
context.stack().push(context.getVariable(address));
|
||||
break;
|
||||
}
|
||||
case Instruction::STORE: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
context.setVariable(address, context.stack().top());
|
||||
break;
|
||||
}
|
||||
case Instruction::CALL: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
auto& stack = context.stack();
|
||||
stack.push((*context.getFunction(address))(stack));
|
||||
break;
|
||||
}
|
||||
case Instruction::RET:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
#undef BINARY_OP_CASE
|
||||
}
|
||||
|
||||
Program::const_iterator instructionToStream(
|
||||
ostream &out, Program::const_iterator ip)
|
||||
{
|
||||
const auto instruction = static_cast<Instruction>(*ip);
|
||||
ip++;
|
||||
|
||||
switch (instruction) {
|
||||
case Instruction::ADD:
|
||||
out << "ADD";
|
||||
break;
|
||||
case Instruction::SUB:
|
||||
out << "SUB";
|
||||
break;
|
||||
case Instruction::MUL:
|
||||
out << "MUL";
|
||||
break;
|
||||
case Instruction::DIV:
|
||||
out << "DIV";
|
||||
break;
|
||||
case Instruction::POW:
|
||||
out << "POW";
|
||||
break;
|
||||
case Instruction::NEG:
|
||||
out << "NEG";
|
||||
break;
|
||||
case Instruction::CONST: {
|
||||
double value = 0.0;
|
||||
std::tie(value, ip) = readConstant(ip);
|
||||
out << CODE_MOD << setfill(' ') << "CONST" << value;
|
||||
break;
|
||||
}
|
||||
case Instruction::POP:
|
||||
out << "POP";
|
||||
break;
|
||||
case Instruction::LOAD: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
out << CODE_MOD << setfill(' ') << "LOAD" << address;
|
||||
break;
|
||||
}
|
||||
case Instruction::STORE: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
out << CODE_MOD << setfill(' ') << "STORE" << address;
|
||||
break;
|
||||
}
|
||||
case Instruction::CALL: {
|
||||
unsigned int address = 0;
|
||||
std::tie(address, ip) = readAddress(ip);
|
||||
out << CODE_MOD << setfill(' ') << "CALL" << address;
|
||||
break;
|
||||
}
|
||||
case Instruction::RET:
|
||||
out << "RET";
|
||||
break;
|
||||
}
|
||||
|
||||
return ip;
|
||||
}
|
||||
|
||||
ostream &programToStream(ostream &out, const Program &program)
|
||||
{
|
||||
auto ip = program.begin();
|
||||
|
||||
while (ip != program.end()) {
|
||||
const auto i = std::distance(program.begin(), ip);
|
||||
cout << setw(4) << setfill('0') << right << i << " ";
|
||||
ip = instructionToStream(out, ip);
|
||||
out << '\n';
|
||||
}
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
// IO //////////////////////////////////////////////////////////////////////////
|
||||
ostream &Latan::operator<<(ostream &out, const MathInterpreter &program)
|
||||
{
|
||||
for (unsigned int i = 0; i < program.program_.size(); ++i)
|
||||
{
|
||||
out << *(program.program_[i]) << endl;
|
||||
}
|
||||
|
||||
return out;
|
||||
return programToStream(out, program.program_);
|
||||
}
|
||||
|
@ -20,6 +20,8 @@
|
||||
#ifndef Latan_MathInterpreter_hpp_
|
||||
#define Latan_MathInterpreter_hpp_
|
||||
|
||||
#include <cstdint>
|
||||
|
||||
#include <LatAnalyze/Functional/Function.hpp>
|
||||
#include <LatAnalyze/Global.hpp>
|
||||
#include <LatAnalyze/Core/ParserState.hpp>
|
||||
@ -79,108 +81,26 @@ private:
|
||||
* Instruction classes *
|
||||
******************************************************************************/
|
||||
// Abstract base
|
||||
class Instruction
|
||||
{
|
||||
public:
|
||||
// destructor
|
||||
virtual ~Instruction(void) = default;
|
||||
// instruction execution
|
||||
virtual void operator()(RunContext &context) const = 0;
|
||||
friend std::ostream & operator<<(std::ostream &out, const Instruction &ins);
|
||||
private:
|
||||
virtual void print(std::ostream &out) const = 0;
|
||||
|
||||
enum class Instruction : std::uint8_t {
|
||||
ADD,
|
||||
SUB,
|
||||
MUL,
|
||||
DIV,
|
||||
POW,
|
||||
NEG,
|
||||
CONST,
|
||||
POP,
|
||||
LOAD,
|
||||
STORE,
|
||||
CALL,
|
||||
RET,
|
||||
};
|
||||
|
||||
std::ostream & operator<<(std::ostream &out, const Instruction &ins);
|
||||
|
||||
// Instruction container
|
||||
typedef std::vector<std::unique_ptr<const Instruction>> Program;
|
||||
|
||||
// Push
|
||||
class Push: public Instruction
|
||||
{
|
||||
private:
|
||||
enum class ArgType
|
||||
{
|
||||
Constant = 0,
|
||||
Variable = 1
|
||||
};
|
||||
public:
|
||||
//constructors
|
||||
explicit Push(const double val);
|
||||
explicit Push(const unsigned int address, const std::string &name);
|
||||
// instruction execution
|
||||
virtual void operator()(RunContext &context) const;
|
||||
private:
|
||||
virtual void print(std::ostream& out) const;
|
||||
private:
|
||||
ArgType type_;
|
||||
double val_;
|
||||
unsigned int address_;
|
||||
std::string name_;
|
||||
};
|
||||
|
||||
// Pop
|
||||
class Pop: public Instruction
|
||||
{
|
||||
public:
|
||||
//constructor
|
||||
explicit Pop(const unsigned int address, const std::string &name);
|
||||
// instruction execution
|
||||
virtual void operator()(RunContext &context) const;
|
||||
private:
|
||||
virtual void print(std::ostream& out) const;
|
||||
private:
|
||||
unsigned int address_;
|
||||
std::string name_;
|
||||
};
|
||||
|
||||
// Store
|
||||
class Store: public Instruction
|
||||
{
|
||||
public:
|
||||
//constructor
|
||||
explicit Store(const unsigned int address, const std::string &name);
|
||||
// instruction execution
|
||||
virtual void operator()(RunContext &context) const;
|
||||
private:
|
||||
virtual void print(std::ostream& out) const;
|
||||
private:
|
||||
unsigned int address_;
|
||||
std::string name_;
|
||||
};
|
||||
|
||||
// Call function
|
||||
class Call: public Instruction
|
||||
{
|
||||
public:
|
||||
//constructor
|
||||
explicit Call(const unsigned int address, const std::string &name);
|
||||
// instruction execution
|
||||
virtual void operator()(RunContext &context) const;
|
||||
private:
|
||||
virtual void print(std::ostream& out) const;
|
||||
private:
|
||||
unsigned int address_;
|
||||
std::string name_;
|
||||
};
|
||||
|
||||
// Floating point operations
|
||||
#define DECL_OP(name)\
|
||||
class name: public Instruction\
|
||||
{\
|
||||
public:\
|
||||
virtual void operator()(RunContext &context) const;\
|
||||
private:\
|
||||
virtual void print(std::ostream &out) const;\
|
||||
}
|
||||
|
||||
DECL_OP(Neg);
|
||||
DECL_OP(Add);
|
||||
DECL_OP(Sub);
|
||||
DECL_OP(Mul);
|
||||
DECL_OP(Div);
|
||||
DECL_OP(Pow);
|
||||
typedef std::vector<std::uint8_t> Program;
|
||||
|
||||
/******************************************************************************
|
||||
* Expression node classes *
|
||||
|
@ -515,14 +515,16 @@ void Dash::operator()(PlotOptions &option) const
|
||||
}
|
||||
|
||||
// LogScale constructor ////////////////////////////////////////////////////////
|
||||
LogScale::LogScale(const Axis axis)
|
||||
LogScale::LogScale(const Axis axis, const double basis)
|
||||
: axis_(axis)
|
||||
, basis_(basis)
|
||||
{}
|
||||
|
||||
// Logscale modifier ///////////////////////////////////////////////////////////
|
||||
void LogScale::operator()(PlotOptions &option) const
|
||||
{
|
||||
option.scaleMode[static_cast<int>(axis_)] |= Plot::Scale::log;
|
||||
option.logScaleBasis[static_cast<int>(axis_)] = basis_;
|
||||
}
|
||||
|
||||
// PlotRange constructors //////////////////////////////////////////////////////
|
||||
@ -915,11 +917,11 @@ ostream & Latan::operator<<(ostream &out, const Plot &plot)
|
||||
out << "unset log" << endl;
|
||||
if (plot.options_.scaleMode[x] & Plot::Scale::log)
|
||||
{
|
||||
out << "set log x" << endl;
|
||||
out << "set log x " << plot.options_.logScaleBasis[x] << endl;;
|
||||
}
|
||||
if (plot.options_.scaleMode[y] & Plot::Scale::log)
|
||||
{
|
||||
out << "set log y" << endl;
|
||||
out << "set log y " << plot.options_.logScaleBasis[y] << endl;
|
||||
}
|
||||
if (!plot.options_.label[x].empty())
|
||||
{
|
||||
|
@ -227,6 +227,7 @@ struct PlotOptions
|
||||
std::string caption;
|
||||
std::string title;
|
||||
unsigned int scaleMode[2];
|
||||
double logScaleBasis[2];
|
||||
Range scale[2];
|
||||
std::string label[2];
|
||||
std::string lineColor;
|
||||
@ -314,13 +315,14 @@ class LogScale: public PlotModifier
|
||||
{
|
||||
public:
|
||||
// constructor
|
||||
explicit LogScale(const Axis axis);
|
||||
explicit LogScale(const Axis axis, const double basis = 10);
|
||||
// destructor
|
||||
virtual ~LogScale(void) = default;
|
||||
// modifier
|
||||
virtual void operator()(PlotOptions &option) const;
|
||||
private:
|
||||
const Axis axis_;
|
||||
const double basis_;
|
||||
};
|
||||
|
||||
class PlotRange: public PlotModifier
|
||||
|
@ -135,3 +135,26 @@ DVec DWT::backward(const std::vector<DWTLevel>& dwt) const
|
||||
|
||||
return res;
|
||||
}
|
||||
|
||||
// concatenate levels //////////////////////////////////////////////////////////
|
||||
DVec DWT::concat(const std::vector<DWTLevel> &dwt, const int maxLevel, const bool dropLow)
|
||||
{
|
||||
unsigned int level = ((maxLevel >= 0) ? (maxLevel + 1) : dwt.size());
|
||||
Index nlast = dwt[level - 1].first.size();
|
||||
Index n = 2*dwt.front().first.size() - ((dropLow) ? nlast : 0);
|
||||
Index pt = n, nl;
|
||||
DVec res(n);
|
||||
|
||||
for (unsigned int l = 0; l < level; ++l)
|
||||
{
|
||||
nl = dwt[l].second.size();
|
||||
pt -= nl;
|
||||
res.segment(pt, nl) = dwt[l].second;
|
||||
}
|
||||
if (!dropLow)
|
||||
{
|
||||
res.segment(0, nl) = dwt[level-1].first;
|
||||
}
|
||||
|
||||
return res;
|
||||
}
|
||||
|
@ -46,6 +46,8 @@ public:
|
||||
// DWT
|
||||
std::vector<DWTLevel> forward(const DVec &data, const unsigned int level) const;
|
||||
DVec backward(const std::vector<DWTLevel>& dwt) const;
|
||||
// concatenate levels
|
||||
static DVec concat(const std::vector<DWTLevel>& dwt, const int maxLevel = -1, const bool dropLow = false);
|
||||
private:
|
||||
DWTFilter filter_;
|
||||
};
|
||||
|
@ -343,7 +343,7 @@ SampleFitResult XYSampleData::fit(std::vector<Minimizer *> &minimizer,
|
||||
result.nPar_ = sampleResult.getNPar();
|
||||
result.nDof_ = sampleResult.nDof_;
|
||||
result.parName_ = sampleResult.parName_;
|
||||
result.corrRangeDb_ = Math::svdDynamicRangeDb(getFitCorrMat());
|
||||
result.corrRangeDb_ = Math::cdr(getFitCorrMat());
|
||||
|
||||
return result;
|
||||
}
|
||||
|
@ -358,7 +358,7 @@ FitResult XYStatData::fit(vector<Minimizer *> &minimizer, const DVec &init,
|
||||
result = (*m)(chi2);
|
||||
totalInit = result;
|
||||
}
|
||||
result.corrRangeDb_ = Math::svdDynamicRangeDb(getFitCorrMat());
|
||||
result.corrRangeDb_ = Math::cdr(getFitCorrMat());
|
||||
result.chi2_ = chi2(result);
|
||||
result.nPar_ = nPar;
|
||||
result.nDof_ = layout.totalYSize - nPar;
|
||||
|
@ -17,6 +17,7 @@
|
||||
* along with LatAnalyze 3. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <LatAnalyze/Core/Math.hpp>
|
||||
#include <LatAnalyze/Core/OptParser.hpp>
|
||||
#include <LatAnalyze/Core/Plot.hpp>
|
||||
#include <LatAnalyze/Io/Io.hpp>
|
||||
@ -53,6 +54,12 @@ int main(int argc, char *argv[])
|
||||
cerr << "usage: " << argv[0];
|
||||
cerr << " <options> <input file>" << endl;
|
||||
cerr << endl << "Possible options:" << endl << opt << endl;
|
||||
cerr << "Available DWT filters:" << endl;
|
||||
for (auto &fv: DWTFilters::fromName)
|
||||
{
|
||||
cerr << fv.first << " ";
|
||||
}
|
||||
cerr << endl << endl;
|
||||
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
@ -68,22 +75,45 @@ int main(int argc, char *argv[])
|
||||
DMatSample in = Io::load<DMatSample>(inFilename), res;
|
||||
Index nSample = in.size(), n = in[central].rows();
|
||||
vector<DMatSample> out(ss ? 1 : level, DMatSample(nSample)),
|
||||
outh(ss ? 0 : level, DMatSample(nSample));
|
||||
outh(ss ? 0 : level, DMatSample(nSample)),
|
||||
concath(ss ? 0 : level, DMatSample(nSample));
|
||||
DMatSample concat(nSample, n, 1);
|
||||
DWT dwt(*DWTFilters::fromName.at(filterName));
|
||||
vector<DWT::DWTLevel> dataDWT(level);
|
||||
|
||||
FOR_STAT_ARRAY(in, s)
|
||||
{
|
||||
in[s].conservativeResize(n, 1);
|
||||
}
|
||||
if (!ss)
|
||||
{
|
||||
DMatSample buf(nSample);
|
||||
|
||||
cout << "-- compute discrete wavelet transform" << endl;
|
||||
cout << "filter '" << filterName << "' / " << level << " level(s)" << endl;
|
||||
FOR_STAT_ARRAY(in, s)
|
||||
{
|
||||
dataDWT = dwt.forward(in[s].col(0), level);
|
||||
dataDWT = dwt.forward(in[s], level);
|
||||
for (unsigned int l = 0; l < level; ++l)
|
||||
{
|
||||
out[l][s] = dataDWT[l].first;
|
||||
outh[l][s] = dataDWT[l].second;
|
||||
concath[l][s] = DWT::concat(dataDWT, l, true);
|
||||
}
|
||||
concat[s] = DWT::concat(dataDWT);
|
||||
}
|
||||
cout << "Data CDR " << Math::cdr(in.correlationMatrix()) << " dB" << endl;
|
||||
cout << "DWT CDR " << Math::cdr(concat.correlationMatrix()) << " dB" << endl;
|
||||
for (unsigned int l = 0; l < level; ++l)
|
||||
{
|
||||
cout << "DWT level " << l << " CDR: L= ";
|
||||
cout << Math::cdr(out[l].correlationMatrix()) << " dB / H= ";
|
||||
cout << Math::cdr(outh[l].correlationMatrix()) << " dB" << endl;
|
||||
}
|
||||
for (unsigned int l = 0; l < level; ++l)
|
||||
{
|
||||
cout << "DWT detail level " << l << " CDR: ";
|
||||
cout << Math::cdr(concath[l].correlationMatrix()) << " dB" << endl;
|
||||
}
|
||||
}
|
||||
else
|
||||
@ -102,7 +132,7 @@ int main(int argc, char *argv[])
|
||||
}
|
||||
FOR_STAT_ARRAY(in, s)
|
||||
{
|
||||
dataDWT.back().first = in[s].col(0);
|
||||
dataDWT.back().first = in[s];
|
||||
out[0][s] = dwt.backward(dataDWT);
|
||||
}
|
||||
}
|
||||
@ -115,7 +145,9 @@ int main(int argc, char *argv[])
|
||||
{
|
||||
Io::save<DMatSample>(out[l], outFilename + "/L" + strFrom(l) + ".h5");
|
||||
Io::save<DMatSample>(outh[l], outFilename + "/H" + strFrom(l) + ".h5");
|
||||
Io::save<DMatSample>(concath[l], outFilename + "/concatH" + strFrom(l) + ".h5");
|
||||
}
|
||||
Io::save<DMatSample>(concat, outFilename + "/concat.h5");
|
||||
}
|
||||
else
|
||||
{
|
||||
|
@ -68,7 +68,7 @@ int main(int argc, char *argv[])
|
||||
var = sample.varianceMatrix();
|
||||
corr = sample.correlationMatrix();
|
||||
|
||||
cout << "dynamic range " << Math::svdDynamicRangeDb(corr) << " dB" << endl;
|
||||
cout << "dynamic range " << Math::cdr(corr) << " dB" << endl;
|
||||
p << PlotCorrMatrix(corr);
|
||||
p.display();
|
||||
if (!outVarName.empty())
|
||||
|
Loading…
Reference in New Issue
Block a user