VisualTraceCheck/plotClustCheck.m at main · NERD-CO/VisualTraceCheck · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
function [] = plotClustCheck( waveForms , nidx , stage)
%UNTITLED4 Summary of this function goes here
%   Detailed explanation goes here

tempWaves = transpose(waveForms);

if stage == 1

    cols = 'rgbkm';
    % nums = [1,2,5];
    nidxes = unique(nidx);
    for ti = 1:numel(nidxes)

        figure;

        nuM = nidxes(ti);
        plot(tempWaves(nidx == nuM,:)','k')
        hold on
        plot(mean(tempWaves(nidx == nuM,:)),'Color',cols(ti));
        ylim([-5000 5000])
        title(num2str(size(tempWaves(nidx == nuM,:),1)));


    end

    figure;
    for ti = 1:numel(nidxes)


        nuM = nidxes(ti);

        plot(mean(tempWaves(nidx == nuM,:)),'Color',cols(ti));
        hold on
        ylim([-5000 5000])


    end

elseif stage == 2

    colors = 'rbgck';

    allCvals = unique(nidx);
    cluVals = allCvals(allCvals ~= 0);

    if any(isnan(cluVals))
        disp('Only one cluster')
        return
    end


    for ni = 1:numel(cluVals)

        nuM = cluVals(ni);

        corInd = nidx == nuM;

        figure;
        plot(tempWaves(corInd,:)',colors(ni))
        ylim([min(tempWaves(:)) max(tempWaves(:))]);

        numWaves = sum(corInd);

        titleStr = ['Clust ', num2str(ni) , ' = '  , num2str(numWaves)];
        title(titleStr);

        figure;
        plot(mean(tempWaves(corInd,:))',colors(ni),'LineWidth',2)
        ylim([min(tempWaves(:)) max(tempWaves(:))]);

    end


    figure;
    for ni = 1:numel(cluVals)

        nuM = cluVals(ni);

        corInd = nidx == nuM;
        plot(mean(tempWaves(corInd,:))',colors(ni),'LineWidth',2)
        hold on

    end


end


end