16121066516

Committed 07 Jul 2025 03:19PM UTC coverage: 74.129% (-0.07%) from 74.196%

Build # 16121066516

Build Type

push

github

Committed by

paulmthompson

Commit Message

delete intervals in datamanager widget

Run Details

0 of 20 new or added lines in 2 files covered. (0.0%)

139 existing lines in 5 files now uncovered.

13106 of 17680 relevant lines covered (74.13%)

1064.63 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/src/WhiskerToolbox/DataManager/Tensors/Tensor_Data.cpp

#include "Tensor_Data.hpp"

#include <filesystem>

// ========== Setters ==========

void TensorData::addTensorAtTime(TimeFrameIndex time, torch::Tensor const & tensor) {
    _data[time] = tensor;
    notifyObservers();
}

void TensorData::overwriteTensorAtTime(TimeFrameIndex time, torch::Tensor const & tensor) {
    _data[time] = tensor;
    notifyObservers();
}

// ========== Getters ==========

torch::Tensor TensorData::getTensorAtTime(TimeFrameIndex time) const {
    if (_data.find(time) != _data.end()) {
        return _data.at(time);
    }
    return torch::Tensor{};
}

std::vector<TimeFrameIndex> TensorData::getTimesWithTensors() const {
    std::vector<TimeFrameIndex> times;
    times.reserve(_data.size());
    for (const auto& [time, tensor] : _data) {
        times.push_back(time);
    }
    return times;
}

std::vector<float> TensorData::getChannelSlice(TimeFrameIndex time, int channel) const
{

    torch::Tensor const tensor = getTensorAtTime(time);

    //std::cout << "Tensor at time " << time << " with " << tensor.numel() << " elements" << std::endl;

    auto sub_tensor = tensor.index({
                                    torch::indexing::Slice(),
                                    torch::indexing::Slice(),
                                    channel});

    // Apply sigmoid
    sub_tensor = torch::sigmoid(sub_tensor);

    //std::cout << "Sub tensor with" << sub_tensor.numel() << " elements " << std::endl;

    std::vector<float> vec(sub_tensor.data_ptr<float>(), sub_tensor.data_ptr<float>() + sub_tensor.numel());

    //std::cout << "Tensor data has " << vec.size() << " elements" << std::endl;

    return vec;
}

1	#include "Tensor_Data.hpp"
2
3	#include <filesystem>
4
5	// ========== Setters ==========
6
UNCOV 7	void TensorData::addTensorAtTime(TimeFrameIndex time, torch::Tensor const & tensor) {	×
8	_data[time] = tensor;	×
9	notifyObservers();	×
10	}	×
11
UNCOV 12	void TensorData::overwriteTensorAtTime(TimeFrameIndex time, torch::Tensor const & tensor) {	×
13	_data[time] = tensor;	×
14	notifyObservers();	×
15	}	×
16
17	// ========== Getters ==========
18
19	torch::Tensor TensorData::getTensorAtTime(TimeFrameIndex time) const {	×
20	if (_data.find(time) != _data.end()) {	×
UNCOV 21	return _data.at(time);	×
22	}
UNCOV 23	return torch::Tensor{};	×
24	}
25
26	std::vector<TimeFrameIndex> TensorData::getTimesWithTensors() const {	×
27	std::vector<TimeFrameIndex> times;	×
28	times.reserve(_data.size());	×
29	for (const auto& [time, tensor] : _data) {	×
UNCOV 30	times.push_back(time);	×
31	}
32	return times;	×
UNCOV 33	}	×
34
UNCOV 35	std::vector<float> TensorData::getChannelSlice(TimeFrameIndex time, int channel) const	×
36	{
37
UNCOV 38	torch::Tensor const tensor = getTensorAtTime(time);	×
39
40	//std::cout << "Tensor at time " << time << " with " << tensor.numel() << " elements" << std::endl;
41
42	auto sub_tensor = tensor.index({	×
43	torch::indexing::Slice(),	×
44	torch::indexing::Slice(),	×
UNCOV 45	channel});	×
46
47	// Apply sigmoid
UNCOV 48	sub_tensor = torch::sigmoid(sub_tensor);	×
49
50	//std::cout << "Sub tensor with" << sub_tensor.numel() << " elements " << std::endl;
51
UNCOV 52	std::vector<float> vec(sub_tensor.data_ptr<float>(), sub_tensor.data_ptr<float>() + sub_tensor.numel());	×
53
54	//std::cout << "Tensor data has " << vec.size() << " elements" << std::endl;
55
56	return vec;	×
UNCOV 57	}	×

paulmthompson / WhiskerToolbox / 16121066516

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous