/*
* This program source code file is part of KiCad, a free EDA CAD application.
*
* Copyright The KiCad Developers, see AUTHORS.txt for contributors.
*
* This program 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.
*
* This program 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 this program. If not, see .
*/
#include
#include
#include
#include
#include
#include
namespace calc_parser
{
thread_local ERROR_COLLECTOR* g_errorCollector = nullptr;
class DATE_UTILS
{
private:
static constexpr int epochYear = 1970;
static constexpr std::array daysInMonth = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
static constexpr std::array monthNames = { "January", "February", "March", "April",
"May", "June", "July", "August",
"September", "October", "November", "December" };
static constexpr std::array monthAbbrev = { "Jan", "Feb", "Mar", "Apr", "May", "Jun",
"Jul", "Aug", "Sep", "Oct", "Nov", "Dec" };
static constexpr std::array weekdayNames = { "Monday", "Tuesday", "Wednesday", "Thursday",
"Friday", "Saturday", "Sunday" };
static auto isLeapYear( int aYear ) -> bool
{
return ( aYear % 4 == 0 && aYear % 100 != 0 ) || ( aYear % 400 == 0 );
}
static auto daysInYear( int aYear ) -> int { return isLeapYear( aYear ) ? 366 : 365; }
static auto daysInMonthForYear( int aMonth, int aYear ) -> int
{
if( aMonth == 2 && isLeapYear( aYear ) )
return 29;
return daysInMonth[aMonth - 1];
}
public:
static auto DaysToYmd( int aDaysSinceEpoch ) -> std::tuple
{
int year = epochYear;
int remainingDays = aDaysSinceEpoch;
if( remainingDays >= 0 )
{
while( remainingDays >= daysInYear( year ) )
{
remainingDays -= daysInYear( year );
year++;
}
}
else
{
while( remainingDays < 0 )
{
year--;
remainingDays += daysInYear( year );
}
}
int month = 1;
while( month <= 12 && remainingDays >= daysInMonthForYear( month, year ) )
{
remainingDays -= daysInMonthForYear( month, year );
month++;
}
int day = remainingDays + 1;
return { year, month, day };
}
static auto YmdToDays( int aYear, int aMonth, int aDay ) -> int
{
int totalDays = 0;
if( aYear >= epochYear )
{
for( int y = epochYear; y < aYear; ++y )
totalDays += daysInYear( y );
}
else
{
for( int y = aYear; y < epochYear; ++y )
totalDays -= daysInYear( y );
}
for( int m = 1; m < aMonth; ++m )
totalDays += daysInMonthForYear( m, aYear );
totalDays += aDay - 1;
return totalDays;
}
static auto ParseDate( const std::string& aDateStr ) -> std::optional
{
std::istringstream iss( aDateStr );
std::string token;
std::vector parts;
char separator = 0;
bool isCjkFormat = false;
// Check for CJK date formats first (Chinese, Korean, or mixed)
bool hasChineseYear = aDateStr.find( "年" ) != std::string::npos;
bool hasChineseMonth = aDateStr.find( "月" ) != std::string::npos;
bool hasChineseDay = aDateStr.find( "日" ) != std::string::npos;
bool hasKoreanYear = aDateStr.find( "년" ) != std::string::npos;
bool hasKoreanMonth = aDateStr.find( "월" ) != std::string::npos;
bool hasKoreanDay = aDateStr.find( "일" ) != std::string::npos;
// Check if we have any CJK date format (pure or mixed)
if( ( hasChineseYear || hasKoreanYear ) && ( hasChineseMonth || hasKoreanMonth )
&& ( hasChineseDay || hasKoreanDay ) )
{
// CJK format: Support pure Chinese, pure Korean, or mixed formats
isCjkFormat = true;
size_t yearPos, monthPos, dayPos;
// Find year position and marker
if( hasChineseYear )
yearPos = aDateStr.find( "年" );
else
yearPos = aDateStr.find( "년" );
// Find month position and marker
if( hasChineseMonth )
monthPos = aDateStr.find( "月" );
else
monthPos = aDateStr.find( "월" );
// Find day position and marker
if( hasChineseDay )
dayPos = aDateStr.find( "日" );
else
dayPos = aDateStr.find( "일" );
try
{
int year = std::stoi( aDateStr.substr( 0, yearPos ) );
int month = std::stoi(
aDateStr.substr( yearPos + 3, monthPos - yearPos - 3 ) ); // 3 bytes for CJK year marker
int day = std::stoi(
aDateStr.substr( monthPos + 3, dayPos - monthPos - 3 ) ); // 3 bytes for CJK month marker
parts = { year, month, day };
}
catch( ... )
{
return std::nullopt;
}
}
else if( aDateStr.find( '-' ) != std::string::npos )
separator = '-';
else if( aDateStr.find( '/' ) != std::string::npos )
separator = '/';
else if( aDateStr.find( '.' ) != std::string::npos )
separator = '.';
if( separator )
{
while( std::getline( iss, token, separator ) )
{
try
{
parts.push_back( std::stoi( token ) );
}
catch( ... )
{
return std::nullopt;
}
}
}
else if( !isCjkFormat && aDateStr.length() == 8 )
{
try
{
int dateNum = std::stoi( aDateStr );
int year = dateNum / 10000;
int month = ( dateNum / 100 ) % 100;
int day = dateNum % 100;
return YmdToDays( year, month, day );
}
catch( ... )
{
return std::nullopt;
}
}
else if( !isCjkFormat )
{
return std::nullopt;
}
if( parts.empty() || parts.size() > 3 )
return std::nullopt;
int year, month, day;
if( parts.size() == 1 )
{
year = parts[0];
month = 1;
day = 1;
}
else if( parts.size() == 2 )
{
year = parts[0];
month = parts[1];
day = 1;
}
else
{
if( isCjkFormat )
{
// CJK formats are always in YYYY年MM月DD日 or YYYY년 MM월 DD일 order
year = parts[0];
month = parts[1];
day = parts[2];
}
else if( separator == '/' && parts[0] <= 12 && parts[1] <= 31 )
{
month = parts[0];
day = parts[1];
year = parts[2];
}
else if( separator == '/' && parts[1] <= 12 )
{
day = parts[0];
month = parts[1];
year = parts[2];
}
else
{
year = parts[0];
month = parts[1];
day = parts[2];
}
}
if( month < 1 || month > 12 )
return std::nullopt;
if( day < 1 || day > daysInMonthForYear( month, year ) )
return std::nullopt;
return YmdToDays( year, month, day );
}
static auto FormatDate( int aDaysSinceEpoch, const std::string& aFormat ) -> std::string
{
auto [year, month, day] = DaysToYmd( aDaysSinceEpoch );
if( aFormat == "ISO" || aFormat == "iso" )
return fmt::format( "{:04d}-{:02d}-{:02d}", year, month, day );
else if( aFormat == "US" || aFormat == "us" )
return fmt::format( "{:02d}/{:02d}/{:04d}", month, day, year );
else if( aFormat == "EU" || aFormat == "european" )
return fmt::format( "{:02d}/{:02d}/{:04d}", day, month, year );
else if( aFormat == "long" )
return fmt::format( "{} {}, {}", monthNames[month - 1], day, year );
else if( aFormat == "short" )
return fmt::format( "{} {}, {}", monthAbbrev[month - 1], day, year );
else if( aFormat == "Chinese" || aFormat == "chinese" || aFormat == "CN" || aFormat == "cn"
|| aFormat == "中文" )
return fmt::format( "{}年{:02d}月{:02d}日", year, month, day );
else if( aFormat == "Japanese" || aFormat == "japanese" || aFormat == "JP" || aFormat == "jp"
|| aFormat == "日本語" )
return fmt::format( "{}年{:02d}月{:02d}日", year, month, day );
else if( aFormat == "Korean" || aFormat == "korean" || aFormat == "KR" || aFormat == "kr"
|| aFormat == "한국어" )
return fmt::format( "{}년 {:02d}월 {:02d}일", year, month, day );
else
return fmt::format( "{:04d}-{:02d}-{:02d}", year, month, day );
}
static auto GetWeekdayName( int aDaysSinceEpoch ) -> std::string
{
int weekday = ( ( aDaysSinceEpoch + 3 ) % 7 ); // +3 because epoch was Thursday (Monday = 0)
if( weekday < 0 )
weekday += 7;
return std::string{ weekdayNames[weekday] };
}
static auto GetCurrentDays() -> int
{
auto now = std::chrono::system_clock::now();
auto timeT = std::chrono::system_clock::to_time_t( now );
return static_cast( timeT / ( 24 * 3600 ) );
}
static auto GetCurrentTimestamp() -> double
{
auto now = std::chrono::system_clock::now();
auto timeT = std::chrono::system_clock::to_time_t( now );
return static_cast( timeT );
}
};
class ESERIES_UTILS
{
private:
// E24 series values in 100-999 decade (2 significant figures)
static constexpr std::array s_e24 = {
100, 110, 120, 130, 150, 160, 180, 200, 220, 240, 270, 300,
330, 360, 390, 430, 470, 510, 560, 620, 680, 750, 820, 910
};
// E192 series values in 100-999 decade (3 significant figures)
static constexpr std::array s_e192 = {
100, 101, 102, 104, 105, 106, 107, 109, 110, 111, 113, 114, 115, 117, 118, 120, 121, 123,
124, 126, 127, 129, 130, 132, 133, 135, 137, 138, 140, 142, 143, 145, 147, 149, 150, 152,
154, 156, 158, 160, 162, 164, 165, 167, 169, 172, 174, 176, 178, 180, 182, 184, 187, 189,
191, 193, 196, 198, 200, 203, 205, 208, 210, 213, 215, 218, 221, 223, 226, 229, 232, 234,
237, 240, 243, 246, 249, 252, 255, 258, 261, 264, 267, 271, 274, 277, 280, 284, 287, 291,
294, 298, 301, 305, 309, 312, 316, 320, 324, 328, 332, 336, 340, 344, 348, 352, 357, 361,
365, 370, 374, 379, 383, 388, 392, 397, 402, 407, 412, 417, 422, 427, 432, 437, 442, 448,
453, 459, 464, 470, 475, 481, 487, 493, 499, 505, 511, 517, 523, 530, 536, 542, 549, 556,
562, 569, 576, 583, 590, 597, 604, 612, 619, 626, 634, 642, 649, 657, 665, 673, 681, 690,
698, 706, 715, 723, 732, 741, 750, 759, 768, 777, 787, 796, 806, 816, 825, 835, 845, 856,
866, 876, 887, 898, 909, 920, 931, 942, 953, 965, 976, 988
};
static auto parseSeriesString( const std::string& aSeries ) -> int
{
if( aSeries == "E3" || aSeries == "e3" )
return 3;
else if( aSeries == "E6" || aSeries == "e6" )
return 6;
else if( aSeries == "E12" || aSeries == "e12" )
return 12;
else if( aSeries == "E24" || aSeries == "e24" )
return 24;
else if( aSeries == "E48" || aSeries == "e48" )
return 48;
else if( aSeries == "E96" || aSeries == "e96" )
return 96;
else if( aSeries == "E192" || aSeries == "e192" )
return 192;
else
return -1; // Invalid series
}
static auto getSeriesValue( int aSeries, size_t aIndex ) -> uint16_t
{
// E1, E3, E6, E12, E24 are derived from E24
if( aSeries <= 24 )
{
const size_t skipValue = 24 / aSeries;
return s_e24[aIndex * skipValue];
}
// E48, E96, E192 are derived from E192
else
{
const size_t skipValue = 192 / aSeries;
return s_e192[aIndex * skipValue];
}
}
static auto getSeriesSize( int aSeries ) -> size_t
{
return static_cast( aSeries );
}
public:
static auto FindNearest( double aValue, const std::string& aSeries ) -> std::optional
{
const int series = parseSeriesString( aSeries );
if( series < 0 )
return std::nullopt;
if( aValue <= 0.0 )
return std::nullopt;
// Scale value to 100-999 decade
const double logValue = std::log10( aValue );
const int decade = static_cast( std::floor( logValue ) );
const double scaledValue = aValue / std::pow( 10.0, decade );
const double normalized = scaledValue * 100.0;
// Find nearest value in series
const size_t seriesSize = getSeriesSize( series );
double minDiff = std::numeric_limits::max();
uint16_t nearest = 100;
for( size_t i = 0; i < seriesSize; ++i )
{
const uint16_t val = getSeriesValue( series, i );
const double diff = std::abs( normalized - val );
if( diff < minDiff )
{
minDiff = diff;
nearest = val;
}
}
// Scale back to original decade
return ( nearest / 100.0 ) * std::pow( 10.0, decade );
}
static auto FindUp( double aValue, const std::string& aSeries ) -> std::optional
{
const int series = parseSeriesString( aSeries );
if( series < 0 )
return std::nullopt;
if( aValue <= 0.0 )
return std::nullopt;
// Scale value to 100-999 decade
const double logValue = std::log10( aValue );
const int decade = static_cast( std::floor( logValue ) );
const double scaledValue = aValue / std::pow( 10.0, decade );
const double normalized = scaledValue * 100.0;
// Find next higher value in series
const size_t seriesSize = getSeriesSize( series );
// Check current decade
for( size_t i = 0; i < seriesSize; ++i )
{
const uint16_t val = getSeriesValue( series, i );
if( val > normalized )
return ( val / 100.0 ) * std::pow( 10.0, decade );
}
// Wrap to next decade
const uint16_t firstVal = getSeriesValue( series, 0 );
return ( firstVal / 100.0 ) * std::pow( 10.0, decade + 1 );
}
static auto FindDown( double aValue, const std::string& aSeries ) -> std::optional
{
const int series = parseSeriesString( aSeries );
if( series < 0 )
return std::nullopt;
if( aValue <= 0.0 )
return std::nullopt;
// Scale value to 100-999 decade
const double logValue = std::log10( aValue );
const int decade = static_cast( std::floor( logValue ) );
const double scaledValue = aValue / std::pow( 10.0, decade );
const double normalized = scaledValue * 100.0;
// Find next lower value in series
const size_t seriesSize = getSeriesSize( series );
// Check current decade (search backwards)
for( int i = seriesSize - 1; i >= 0; --i )
{
const uint16_t val = getSeriesValue( series, i );
if( val < normalized )
return ( val / 100.0 ) * std::pow( 10.0, decade );
}
// Wrap to previous decade
const uint16_t lastVal = getSeriesValue( series, seriesSize - 1 );
return ( lastVal / 100.0 ) * std::pow( 10.0, decade - 1 );
}
};
EVAL_VISITOR::EVAL_VISITOR( VariableCallback aVariableCallback, ERROR_COLLECTOR& aErrorCollector ) :
m_variableCallback( std::move( aVariableCallback ) ),
m_errors( aErrorCollector ),
m_gen( m_rd() )
{
}
auto EVAL_VISITOR::operator()( const NODE& aNode ) const -> Result
{
switch( aNode.type )
{
case NodeType::Number: return MakeValue( std::get( aNode.data ) );
case NodeType::String: return MakeValue( std::get( aNode.data ) );
case NodeType::Var:
{
const auto& varName = std::get( aNode.data );
// Use callback to resolve variable
if( m_variableCallback )
return m_variableCallback( varName );
return MakeError( fmt::format( "No variable resolver configured for: {}", varName ) );
}
case NodeType::BinOp:
{
const auto& binop = std::get( aNode.data );
auto leftResult = binop.left->Accept( *this );
if( !leftResult )
return leftResult;
auto rightResult = binop.right ? binop.right->Accept( *this ) : MakeValue( 0.0 );
if( !rightResult )
return rightResult;
// Special handling for string concatenation with +
if( binop.op == '+' )
{
const auto& leftVal = leftResult.GetValue();
const auto& rightVal = rightResult.GetValue();
// If either operand is a string, concatenate
if( std::holds_alternative( leftVal ) || std::holds_alternative( rightVal ) )
{
return MakeValue( VALUE_UTILS::ConcatStrings( leftVal, rightVal ) );
}
}
// Special handling for string comparisons with == and !=
if( binop.op == 3 || binop.op == 4 ) // == or !=
{
const auto& leftVal = leftResult.GetValue();
const auto& rightVal = rightResult.GetValue();
// If both operands are strings, do string comparison
if( std::holds_alternative( leftVal ) && std::holds_alternative( rightVal ) )
{
bool equal = std::get( leftVal ) == std::get( rightVal );
double result = ( binop.op == 3 ) ? ( equal ? 1.0 : 0.0 ) : ( equal ? 0.0 : 1.0 );
return MakeValue( result );
}
}
// Otherwise, perform arithmetic
return VALUE_UTILS::ArithmeticOp( leftResult.GetValue(), rightResult.GetValue(), binop.op );
}
case NodeType::Function:
{
const auto& func = std::get( aNode.data );
return evaluateFunction( func );
}
default: return MakeError( "Cannot evaluate this node type" );
}
}
auto EVAL_VISITOR::evaluateFunction( const FUNC_DATA& aFunc ) const -> Result
{
const auto& name = aFunc.name;
const auto& args = aFunc.args;
// Zero-argument functions
if( args.empty() )
{
if( name == "today" )
return MakeValue( static_cast( DATE_UTILS::GetCurrentDays() ) );
else if( name == "now" )
return MakeValue( DATE_UTILS::GetCurrentTimestamp() );
else if( name == "random" )
{
std::uniform_real_distribution dis( 0.0, 1.0 );
return MakeValue( dis( m_gen ) );
}
}
// Evaluate arguments to mixed types
std::vector argValues;
argValues.reserve( args.size() );
for( const auto& arg : args )
{
auto result = arg->Accept( *this );
if( !result )
return result;
argValues.push_back( result.GetValue() );
}
const auto argc = argValues.size();
// String formatting functions (return strings!)
if( name == "format" && argc >= 1 )
{
auto numResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !numResult )
return MakeError( numResult.GetError() );
const auto value = numResult.GetValue();
int decimals = 2;
if( argc > 1 )
{
auto decResult = VALUE_UTILS::ToDouble( argValues[1] );
if( decResult )
decimals = static_cast( decResult.GetValue() );
}
return MakeValue( fmt::format( "{:.{}f}", value, decimals ) );
}
else if( name == "currency" && argc >= 1 )
{
auto numResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !numResult )
return MakeError( numResult.GetError() );
const auto amount = numResult.GetValue();
const auto symbol = argc > 1 ? VALUE_UTILS::ToString( argValues[1] ) : "$";
return MakeValue( fmt::format( "{}{:.2f}", symbol, amount ) );
}
else if( name == "fixed" && argc >= 1 )
{
auto numResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !numResult )
return MakeError( numResult.GetError() );
const auto value = numResult.GetValue();
int decimals = 2;
if( argc > 1 )
{
auto decResult = VALUE_UTILS::ToDouble( argValues[1] );
if( decResult )
decimals = static_cast( decResult.GetValue() );
}
return MakeValue( fmt::format( "{:.{}f}", value, decimals ) );
}
// Date formatting functions (return strings!)
else if( name == "dateformat" && argc >= 1 )
{
auto dateResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !dateResult )
return MakeError( dateResult.GetError() );
const auto days = static_cast( dateResult.GetValue() );
const auto format = argc > 1 ? VALUE_UTILS::ToString( argValues[1] ) : "ISO";
return MakeValue( DATE_UTILS::FormatDate( days, format ) );
}
else if( name == "datestring" && argc == 1 )
{
auto dateStr = VALUE_UTILS::ToString( argValues[0] );
auto daysResult = DATE_UTILS::ParseDate( dateStr );
if( !daysResult )
return MakeError( "Invalid date format: " + dateStr );
return MakeValue( static_cast( daysResult.value() ) );
}
else if( name == "weekdayname" && argc == 1 )
{
auto dateResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !dateResult )
return MakeError( dateResult.GetError() );
const auto days = static_cast( dateResult.GetValue() );
return MakeValue( DATE_UTILS::GetWeekdayName( days ) );
}
// VCS functions (return strings!)
// Empty results from the VCS layer mean "not in a repository" or "no data available"
auto vcsResult = []( const std::string& aResult ) -> std::string
{
return aResult.empty() ? "" : aResult;
};
if( name == "vcsidentifier" && argc <= 1 )
{
int length = 40; // Full identifier by default
if( argc == 1 )
{
auto lenResult = VALUE_UTILS::ToDouble( argValues[0] );
if( lenResult )
length = static_cast( lenResult.GetValue() );
}
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitHash( ".", length ) ) );
}
else if( name == "vcsnearestlabel" && argc <= 2 )
{
std::string match;
bool anyTags = false;
if( argc >= 1 )
match = VALUE_UTILS::ToString( argValues[0] );
if( argc >= 2 )
{
auto tagsResult = VALUE_UTILS::ToDouble( argValues[1] );
if( tagsResult )
anyTags = tagsResult.GetValue() != 0.0;
}
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetNearestTag( match, anyTags ) ) );
}
else if( name == "vcslabeldistance" && argc <= 2 )
{
std::string match;
bool anyTags = false;
if( argc >= 1 )
match = VALUE_UTILS::ToString( argValues[0] );
if( argc >= 2 )
{
auto tagsResult = VALUE_UTILS::ToDouble( argValues[1] );
if( tagsResult )
anyTags = tagsResult.GetValue() != 0.0;
}
return MakeValue( std::to_string( TEXT_EVAL_VCS::GetDistanceFromTag( match, anyTags ) ) );
}
else if( name == "vcsdirty" && argc <= 1 )
{
bool includeUntracked = false;
if( argc == 1 )
{
auto utResult = VALUE_UTILS::ToDouble( argValues[0] );
if( utResult )
includeUntracked = utResult.GetValue() != 0.0;
}
return MakeValue( TEXT_EVAL_VCS::IsDirty( includeUntracked ) ? "1" : "0" );
}
else if( name == "vcsdirtysuffix" && argc <= 2 )
{
std::string suffix = "-dirty";
bool includeUntracked = false;
if( argc >= 1 )
suffix = VALUE_UTILS::ToString( argValues[0] );
if( argc >= 2 )
{
auto utResult = VALUE_UTILS::ToDouble( argValues[1] );
if( utResult )
includeUntracked = utResult.GetValue() != 0.0;
}
return MakeValue( TEXT_EVAL_VCS::IsDirty( includeUntracked ) ? suffix : "" );
}
else if( name == "vcsauthor" && argc == 0 )
{
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetAuthor( "." ) ) );
}
else if( name == "vcsauthoremail" && argc == 0 )
{
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetAuthorEmail( "." ) ) );
}
else if( name == "vcscommitter" && argc == 0 )
{
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitter( "." ) ) );
}
else if( name == "vcscommitteremail" && argc == 0 )
{
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitterEmail( "." ) ) );
}
else if( name == "vcsbranch" && argc == 0 )
{
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetBranch() ) );
}
else if( name == "vcscommitdate" && argc <= 1 )
{
std::string format = "ISO";
if( argc == 1 )
format = VALUE_UTILS::ToString( argValues[0] );
int64_t timestamp = TEXT_EVAL_VCS::GetCommitTimestamp( "." );
if( timestamp == 0 )
return MakeValue( vcsResult( std::string() ) );
int days = static_cast( timestamp / ( 24 * 3600 ) );
return MakeValue( DATE_UTILS::FormatDate( days, format ) );
}
// VCS file functions (file-specific versions)
else if( name == "vcsfileidentifier" && argc >= 1 && argc <= 2 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
int length = 40;
if( argc == 2 )
{
auto lenResult = VALUE_UTILS::ToDouble( argValues[1] );
if( lenResult )
length = static_cast( lenResult.GetValue() );
}
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitHash( filePath, length ) ) );
}
else if( name == "vcsfileauthor" && argc == 1 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetAuthor( filePath ) ) );
}
else if( name == "vcsfileauthoremail" && argc == 1 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetAuthorEmail( filePath ) ) );
}
else if( name == "vcsfilecommitter" && argc == 1 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitter( filePath ) ) );
}
else if( name == "vcsfilecommitteremail" && argc == 1 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
return MakeValue( vcsResult( TEXT_EVAL_VCS::GetCommitterEmail( filePath ) ) );
}
else if( name == "vcsfilecommitdate" && argc >= 1 && argc <= 2 )
{
std::string filePath = VALUE_UTILS::ToString( argValues[0] );
std::string format = "ISO";
if( argc == 2 )
format = VALUE_UTILS::ToString( argValues[1] );
int64_t timestamp = TEXT_EVAL_VCS::GetCommitTimestamp( filePath );
if( timestamp == 0 )
return MakeValue( vcsResult( std::string() ) );
int days = static_cast( timestamp / ( 24 * 3600 ) );
return MakeValue( DATE_UTILS::FormatDate( days, format ) );
}
// String functions (return strings!)
else if( name == "upper" && argc == 1 )
{
auto str = VALUE_UTILS::ToString( argValues[0] );
std::transform( str.begin(), str.end(), str.begin(), ::toupper );
return MakeValue( str );
}
else if( name == "lower" && argc == 1 )
{
auto str = VALUE_UTILS::ToString( argValues[0] );
std::transform( str.begin(), str.end(), str.begin(), ::tolower );
return MakeValue( str );
}
else if( name == "concat" && argc >= 2 )
{
std::string result;
for( const auto& val : argValues )
result += VALUE_UTILS::ToString( val );
return MakeValue( result );
}
else if( name == "beforefirst" && argc == 2 )
{
wxString result = VALUE_UTILS::ToString( argValues[0] );
result = result.BeforeFirst( VALUE_UTILS::ToChar( argValues[1] ) );
return MakeValue( result.ToStdString() );
}
else if( name == "beforelast" && argc == 2 )
{
wxString result = VALUE_UTILS::ToString( argValues[0] );
result = result.BeforeLast( VALUE_UTILS::ToChar( argValues[1] ) );
return MakeValue( result.ToStdString() );
}
else if( name == "afterfirst" && argc == 2 )
{
wxString result = VALUE_UTILS::ToString( argValues[0] );
result = result.AfterFirst( VALUE_UTILS::ToChar( argValues[1] ) );
return MakeValue( result.ToStdString() );
}
else if( name == "afterlast" && argc == 2 )
{
wxString result = VALUE_UTILS::ToString( argValues[0] );
result = result.AfterLast( VALUE_UTILS::ToChar( argValues[1] ) );
return MakeValue( result.ToStdString() );
}
// Conditional functions (handle mixed types)
if( name == "if" && argc == 3 )
{
// Convert only the condition to a number
auto conditionResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !conditionResult )
return MakeError( conditionResult.GetError() );
const auto condition = conditionResult.GetValue() != 0.0;
return MakeValue( condition ? argValues[1] : argValues[2] );
}
// E-series functions (handle value as number, series as string)
else if( ( name == "enearest" || name == "eup" || name == "edown" ) && argc >= 1 && argc <= 2 )
{
auto valueResult = VALUE_UTILS::ToDouble( argValues[0] );
if( !valueResult )
return MakeError( valueResult.GetError() );
const auto value = valueResult.GetValue();
const auto series = argc > 1 ? VALUE_UTILS::ToString( argValues[1] ) : "E24";
std::optional result;
if( name == "enearest" )
result = ESERIES_UTILS::FindNearest( value, series );
else if( name == "eup" )
result = ESERIES_UTILS::FindUp( value, series );
else if( name == "edown" )
result = ESERIES_UTILS::FindDown( value, series );
if( !result )
return MakeError( fmt::format( "Invalid E-series: {}", series ) );
return MakeValue( result.value() );
}
// Mathematical functions (return numbers) - convert args to doubles first
std::vector numArgs;
for( const auto& val : argValues )
{
auto numResult = VALUE_UTILS::ToDouble( val );
if( !numResult )
return MakeError( numResult.GetError() );
numArgs.push_back( numResult.GetValue() );
}
// Mathematical function implementations
if( name == "abs" && argc == 1 )
return MakeValue( std::abs( numArgs[0] ) );
else if( name == "sum" && argc >= 1 )
return MakeValue( std::accumulate( numArgs.begin(), numArgs.end(), 0.0 ) );
else if( name == "round" && argc >= 1 )
{
const auto value = numArgs[0];
const auto precision = argc > 1 ? static_cast( numArgs[1] ) : 0;
const auto multiplier = std::pow( 10.0, precision );
return MakeValue( std::round( value * multiplier ) / multiplier );
}
else if( name == "sqrt" && argc == 1 )
{
if( numArgs[0] < 0 )
return MakeError( "Square root of negative number" );
return MakeValue( std::sqrt( numArgs[0] ) );
}
else if( name == "pow" && argc == 2 )
return MakeValue( std::pow( numArgs[0], numArgs[1] ) );
else if( name == "floor" && argc == 1 )
return MakeValue( std::floor( numArgs[0] ) );
else if( name == "ceil" && argc == 1 )
return MakeValue( std::ceil( numArgs[0] ) );
else if( name == "min" && argc >= 1 )
return MakeValue( *std::min_element( numArgs.begin(), numArgs.end() ) );
else if( name == "max" && argc >= 1 )
return MakeValue( *std::max_element( numArgs.begin(), numArgs.end() ) );
else if( name == "avg" && argc >= 1 )
{
const auto sum = std::accumulate( numArgs.begin(), numArgs.end(), 0.0 );
return MakeValue( sum / static_cast( argc ) );
}
else if( name == "shunt" && argc == 2 )
{
const auto r1 = numArgs[0];
const auto r2 = numArgs[1];
const auto sum = r1 + r2;
// Calculate parallel resistance: (r1*r2)/(r1+r2)
// If sum is not positive, return 0.0 (handles edge cases like shunt(0,0))
if( sum > 0.0 )
return MakeValue( ( r1 * r2 ) / sum );
else
return MakeValue( 0.0 );
}
else if( name == "db" && argc == 1 )
{
// Power ratio to dB: 10*log10(ratio)
if( numArgs[0] <= 0.0 )
return MakeError( "db() argument must be positive" );
return MakeValue( 10.0 * std::log10( numArgs[0] ) );
}
else if( name == "dbv" && argc == 1 )
{
// Voltage/current ratio to dB: 20*log10(ratio)
if( numArgs[0] <= 0.0 )
return MakeError( "dbv() argument must be positive" );
return MakeValue( 20.0 * std::log10( numArgs[0] ) );
}
else if( name == "fromdb" && argc == 1 )
{
// dB to power ratio: 10^(dB/10)
return MakeValue( std::pow( 10.0, numArgs[0] / 10.0 ) );
}
else if( name == "fromdbv" && argc == 1 )
{
// dB to voltage/current ratio: 10^(dB/20)
return MakeValue( std::pow( 10.0, numArgs[0] / 20.0 ) );
}
return MakeError( fmt::format( "Unknown function: {} with {} arguments", name, argc ) );
}
auto DOC_PROCESSOR::Process( const DOC& aDoc, VariableCallback aVariableCallback ) -> std::pair
{
std::string result;
auto localErrors = ERROR_COLLECTOR{};
EVAL_VISITOR evaluator{ std::move( aVariableCallback ), localErrors };
bool hadErrors = aDoc.HasErrors();
for( const auto& node : aDoc.GetNodes() )
{
switch( node->type )
{
case NodeType::Text: result += std::get( node->data ); break;
case NodeType::Calc:
{
const auto& calcData = std::get( node->data );
auto evalResult = calcData.left->Accept( evaluator );
if( evalResult )
result += VALUE_UTILS::ToString( evalResult.GetValue() );
else
{
// Don't add error formatting to result - errors go to error vector only
// The higher level will return original input unchanged if there are errors
hadErrors = true;
}
break;
}
default:
result += "[Unknown node type]";
hadErrors = true;
break;
}
}
return { std::move( result ), hadErrors || localErrors.HasErrors() };
}
auto DOC_PROCESSOR::ProcessWithDetails( const DOC& aDoc, VariableCallback aVariableCallback )
-> std::tuple, bool>
{
auto [result, hadErrors] = Process( aDoc, std::move( aVariableCallback ) );
auto allErrors = aDoc.GetErrors();
return { std::move( result ), std::move( allErrors ), hadErrors };
}
} // namespace calc_parser