threema-ios/Threema/RootSquareMean.m

//  _____ _
// |_   _| |_  _ _ ___ ___ _ __  __ _
//   | | | ' \| '_/ -_) -_) '  \/ _` |_
//   |_| |_||_|_| \___\___|_|_|_\__,_(_)
//
// Threema iOS Client
// Copyright (c) 2014-2020 Threema GmbH
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

#import "RootSquareMean.h"

@interface RootSquareMean ()

@property Float32 rms;
@property (nonatomic)  UInt32 count;

@end

@implementation RootSquareMean

- (instancetype)init
{
    self = [super init];
    if (self) {
        self.rms = 0.0f;
        self.count = 0;
    }
    
    return self;
}

- (void)addValue:(SInt16)value {
    _rms += value * value;
    _count ++;
}

- (Float32)getAndReset {
    Float32 result = sqrtf(_rms/_count);
    _rms = 0.0f;
    _count = 0;
    
    return result;
}

- (UInt32)count {
    return _count;
}

@end