DataloggerExynos4412/INA220Source.c at master · ESLab/DataloggerExynos4412 · 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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#include "DataSource.h"
#include "INA220.h"
#include "Evaluator.h"

#include <stdlib.h>
#include <string.h>

#define CurrentFactor 5
#define ShuntResistance 0.02

static DataSource *AllocateDataSource(const char *arguments);
static void FreeDataSource(DataSource *source);
static bool IsDataAvailable();
static void SampleData(DataSource *source);
static double GetData(DataSource *source,int channel);
static double GetDefaultCalibration(DataSource *source,int channel);
static int ChannelNumberForName(DataSource *source,const char *name);

DataSourceDefinition INA220SourceDefinition=
{
	.AllocateDataSource=AllocateDataSource,
	.Name="INA220",
	.Arguments="I2C address",
	.ChannelNamesAndUnits=(const char *[]) {
		"Voltage","V",
		"Current","A",
		"Power","W",
		NULL,
	},
};

typedef struct INA220Source
{
	DataSource source;
	int address;
	bool ready;
	int voltage,current,power;
} INA220Source;

static DataSource *AllocateDataSource(const char *arguments)
{
	int address=EvaluateExpression(arguments,strlen(arguments));
	if(!InitializeINA220(address,0xfff*CurrentFactor)) return NULL; // Should this be 0x1000?

	INA220Source *source=calloc(sizeof(INA220Source),1);
	if(!source) exit(2);

	source->source.FreeDataSource=FreeDataSource;
	source->source.IsDataAvailable=IsDataAvailable;
	source->source.SampleData=SampleData;
	source->source.GetData=GetData;
	source->source.GetDefaultCalibration=GetDefaultCalibration;
	source->source.ChannelNumberForName=ChannelNumberForName;
	source->source.IsImportant=true;

	source->address=address;
	source->ready=false;

	return &source->source;
}

static void FreeDataSource(DataSource *source)
{
	free(source);
}

static bool IsDataAvailable(DataSource *sourcearg)
{
	INA220Source *source=(INA220Source *)sourcearg;

	if(source->ready) return true;
	int voltageandflags=ReadINA220VoltageAndFlags(source->address);
	source->voltage=voltageandflags>>3;
	if(voltageandflags&0x0002) source->ready=true;
	return source->ready;
}

static void SampleData(DataSource *sourcearg)
{
	INA220Source *source=(INA220Source *)sourcearg;

	if(source->ready) source->voltage=ReadINA220VoltageAndFlags(source->address)>>3;
	source->current=ReadINA220Current(source->address);
	source->power=ReadINA220Power(source->address);
	source->ready=false;
}

static double GetData(DataSource *sourcearg,int channel)
{
	INA220Source *source=(INA220Source *)sourcearg;

	switch(channel)
	{
		case 0: return source->voltage;
		case 1: return source->current;
		case 2: return source->power;
		default: return 0;
	}
}

static double GetDefaultCalibration(DataSource *source,int channel)
{
	switch(channel)
	{
		case 0: return 0.004;
		case 1: return 0.00001/CurrentFactor/ShuntResistance;
		case 2: return 0.004*0.00001/CurrentFactor/ShuntResistance*5000;
		default: return 0;
	}
}

static int ChannelNumberForName(DataSource *source,const char *name)
{
	if(strcasecmp(name,"Voltage")==0) return 0;
	else if(strcasecmp(name,"Current")==0) return 1;
	else if(strcasecmp(name,"Power")==0) return 2;
	else return -1;
}