ref: fe1eb39db7ae6904924f3ab1f6f9b34416f2eb1b
dir: /sys/src/cmd/reform/pm.c/
#include <u.h>
#include <libc.h>
#include <fcall.h>
#include <thread.h>
#include <9p.h>
#include <draw.h>
#include <memdraw.h>
enum
{
Mhz = 1000*1000,
Pwmsrcclk = 25*Mhz,
Kbdlightmax = 8,
Scharge = 0,
Sovervolted,
Scooldown,
Sundervolted,
Smissing,
Sfullycharged,
Spowersave,
Psomoff = 1,
Light = 1,
Temp,
Battery,
Kbdoled,
Pmctl,
KbdoledW = 126,
KbdoledH = 32,
Lcd = 0,
Kbd,
Tb,
Nlights = 5, /* trackball has 5 leds */
PWMSAR = 0x0c/4,
PWMPR = 0x10/4,
TMUTMR = 0x00/4,
TMR_ME = 1<<31,
TMR_ALPF_SHIFT = 26,
TMR_MSITE_SHIFT = 13,
TMUTSR = 0x04/4,
TSR_MIE = 1<<30,
TSR_ORL = 1<<29,
TSR_ORH = 1<<28,
TMUTMTMIR = 0x08/4,
TMUTIER = 0x20/4,
TMUTIDR = 0x24/4,
TIDR_MASK = 0xe0000000,
TMUTISCR = 0x28/4,
TMUTICSCR = 0x2c/4,
TMUTTCFGR = 0x80/4,
TMUTSCFGR = 0x84/4,
TMUTRITSR0 = 0x100/4,
TMUTRITSR1 = 0x110/4,
TMUTRITSR2 = 0x120/4,
TMUTTR0CR = 0xf10/4,
TMUTTR1CR = 0xf14/4,
TMUTTR2CR = 0xf18/4,
TMUTTR3CR = 0xf1c/4,
CR_CAL_PTR_SHIFT = 16,
SPIx_RXDATA = 0x00/4,
SPIx_TXDATA = 0x04/4,
SPIx_CONREG = 0x08/4,
CON_BURST_LENGTH = 1<<20,
CON_PRE_DIVIDER = 1<<12,
CON_POST_DIVIDER = 1<<8,
CON_CHAN_MASTER = 1<<4,
CON_XCH = 1<<2,
CON_EN = 1<<0,
SPIx_CONFIGREG = 0x0c/4,
CONFIG_SS_CTL_NCSS = 1<<8,
CONFIG_SCLK_PHA_1 = 1<<0,
SPIx_STATREG = 0x18/4,
STAT_RR = 1<<3,
};
static char *uid = "pm";
static Reqqueue *lpcreq;
static u32int *pwm2, *tmu, *spi2;
static int kbdlight = 0, tbleds[5] = {0};
static int hidkb = -1, hidtb = -1;
static Memimage *kbdoled, *image;
static u8int kbdoledraw[4+KbdoledW*KbdoledH/8] = {'W', 'B', 'I', 'T', 0};
static char udidkb[] = "Reform Keyboard";
static char udidtb[] = "Reform Trackball";
static void
wr(u32int *base, int reg, u32int v)
{
//fprint(2, "[0%x] ← 0x%ux\n", reg*4, v);
if(base != nil)
base[reg] = v;
}
static u32int
rd(u32int *base, int reg)
{
return base != nil ? base[reg] : -1;
}
static char *
readall(int f)
{
int bufsz, sz, n;
char *s;
bufsz = 2047;
s = nil;
for(sz = 0;; sz += n){
if(bufsz-sz < 2048){
bufsz *= 2;
s = realloc(s, bufsz);
}
if((n = readn(f, s+sz, bufsz-sz-1)) < 1)
break;
}
if(n < 0 || sz < 1){
if(n == 0)
werrstr("empty");
free(s);
return nil;
}
s[sz] = 0;
return s;
}
static int
openhidctl(int *fd, char *match)
{
char path[32], *s, *k, *e;
int f;
if(*fd < 0 && (f = open("/dev/usb/ctl", OREAD)) >= 0){
if((s = readall(f)) != nil &&
(k = strstr(s, match)) != nil &&
(k = strstr(k+strlen(match), "' ")) != nil &&
(e = strchr(k+2, ' ')) != nil){
*e = 0;
snprint(path, sizeof(path), "/dev/hidU%sctl", k+2);
if((*fd = open(path, OWRITE)) >= 0 && write(*fd, "rawon", 5) != 5){
close(*fd);
*fd = -1;
}
}
free(s);
close(f);
}
return *fd < 0;
}
static int
loadkbdoled(char *data, int offset, int size)
{
int x, y, i, k, v, bpl, used;
static Rectangle r;
char hdr[5*12+1];
u8int *p, q;
ulong chan;
if(openhidctl(&hidkb, udidkb) != 0)
return -1;
if(size == 0)
return write(hidkb, "WCLR", 4);
used = 0;
if(offset == 0){
if(size < 60){
werrstr("invalid header");
return -1;
}
memmove(hdr, data, 60);
hdr[11] = 0;
hdr[60] = 0;
if((chan = strtochan(data)) == 0){
werrstr("bad channel string %s", (char*)data);
return -1;
}
r.min.x = atoi(data+1*12);
r.min.y = atoi(data+2*12);
r.max.x = atoi(data+3*12);
r.max.y = atoi(data+4*12);
if(badrect(r)){
werrstr("bad rect");
return -1;
}
data += 60;
size -= 60;
used += 60;
if(image == nil || chan != image->chan || !eqrect(r, image->r)){
freememimage(image);
if((image = allocmemimage(r, chan)) == nil)
return -1;
}
r.max.y = r.min.y;
}
if(image == nil){
werrstr("no header");
return -1;
}
bpl = bytesperline(image->r, image->depth);
i = size / bpl;
if(i < 1)
return used;
v = loadmemimage(image, Rect(r.min.x, r.max.y, r.max.x, r.max.y+i), (uchar*)data, size);
if(v <= 0){
werrstr("loadmemimage: failed");
return -1;
}
r.max.y += i;
used += v;
if(r.max.y < image->r.max.y)
return used;
memimagedraw(kbdoled, kbdoled->r, image, image->r.min, nil, ZP, S);
bpl = bytesperline(kbdoled->r, kbdoled->depth);
for(y = 0, i = 4; y < KbdoledH; y += 8){
for(x = v = 0; x < KbdoledW; x++, v = (v+1)&7){
SET(p);
if(v == 0)
p = byteaddr(kbdoled, Pt(x,y));
for(k = q = 0; k < 8; k++)
q |= ((p[bpl*k] >> (7-v)) & 1) << k;
kbdoledraw[i++] = q;
}
}
if(write(hidkb, kbdoledraw, sizeof(kbdoledraw)) != sizeof(kbdoledraw))
return -1;
return used;
}
static int
setlight(int k, int *p)
{
u32int v, i;
for(i = 0; i < Nlights; i++){
if(p[i] < 0)
p[i] = 0;
else if(k == Tb)
p[i] = p[i] > 0;
else if(p[i] > 100)
p[i] = 100;
}
if(k == Lcd){
v = Pwmsrcclk / rd(pwm2, PWMSAR);
wr(pwm2, PWMPR, (Pwmsrcclk/(v*p[0]/100))-2);
return 0;
}else if(k == Kbd && openhidctl(&hidkb, udidkb) == 0){
v = Kbdlightmax*p[0]/100;
if(fprint(hidkb, "LITE%d", v) > 0){
kbdlight = p[0];
return 0;
}
close(hidkb);
hidkb = -1;
}else if(k == Tb && openhidctl(&hidtb, udidtb) == 0){
if(fprint(hidtb, "LEDS%d%d%d%d%d", p[0], p[1], p[2], p[3], p[4]) > 0){
memmove(tbleds, p, sizeof(tbleds));
return 0;
}
close(hidtb);
hidtb = -1;
}
return -1;
}
static void
getlight(int k, int *v)
{
u32int m;
int i;
SET(m);
if(k == Lcd){
m = Pwmsrcclk / rd(pwm2, PWMSAR);
v[0] = Pwmsrcclk / (rd(pwm2, PWMPR)+2);
}else if(k == Kbd){
m = 100;
v[0] = kbdlight;
}else if(k == Tb){
m = 100;
memmove(v, tbleds, sizeof(tbleds));
}
for(i = 0; i < Nlights; i++)
v[i] = v[i]*100/m;
}
static int
getcputemp(void)
{
u32int s;
int i, c;
/* enable: all sites, ALPF 11=0.125 */
wr(tmu, TMUTMR, TMR_ME | 3<<TMR_ALPF_SHIFT | 0<<TMR_MSITE_SHIFT);
sleep(50);
s = rd(tmu, TMUTSR);
if(s & TSR_MIE){
werrstr("monitoring interval exceeded");
return -1;
}
if(s & (TSR_ORL|TSR_ORH)){
werrstr("out of range");
return -1;
}
for(i = 0; (c = rd(tmu, TMUTRITSR0)) >= 0 && i < 10; i++)
sleep(10);
wr(tmu, TMUTMR, 0);
return c & 0xff;
}
static void
tmuinit(void)
{
/* without proper calibration data sensing is useless */
static u8int cfg[4][12] = {
{0x23, 0x29, 0x2f, 0x35, 0x3d, 0x43, 0x4b, 0x51, 0x57, 0x5f, 0x67, 0x6f},
{0x1b, 0x23, 0x2b, 0x33, 0x3b, 0x43, 0x4b, 0x55, 0x5d, 0x67, 0x70, 0},
{0x17, 0x23, 0x2d, 0x37, 0x41, 0x4b, 0x57, 0x63, 0x6f, 0},
{0x15, 0x21, 0x2d, 0x39, 0x45, 0x53, 0x5f, 0x71, 0},
};
int i, j;
wr(tmu, TMUTMR, 0); /* disable */
wr(tmu, TMUTIER, 0); /* disable all interrupts */
wr(tmu, TMUTMTMIR, 0xf); /* no monitoring interval */
/* configure default ranges */
wr(tmu, TMUTTR0CR, 11<<CR_CAL_PTR_SHIFT | 0);
wr(tmu, TMUTTR1CR, 10<<CR_CAL_PTR_SHIFT | 38);
wr(tmu, TMUTTR2CR, 8<<CR_CAL_PTR_SHIFT | 72);
wr(tmu, TMUTTR3CR, 7<<CR_CAL_PTR_SHIFT | 97);
/* calibration data */
for(i = 0; i < 4; i++){
for(j = 0; j < 12 && cfg[i][j] != 0; j++){
wr(tmu, TMUTTCFGR, i<<16|j);
wr(tmu, TMUTSCFGR, cfg[i][j]);
}
}
}
static void
lpccall(char cmd, u8int arg, void *ret)
{
u32int con;
int i;
con =
/* 8 bits burst */
CON_BURST_LENGTH*(8-1) |
/* clk=25Mhz, pre=1, post=2⁶ → 25Mhz/1/2⁶ ≲ 400kHz */
CON_PRE_DIVIDER*(1-1) | CON_POST_DIVIDER*6 |
/* master mode on channel 0 */
CON_CHAN_MASTER<<0 |
/* enable */
CON_EN;
wr(spi2, SPIx_CONREG, con);
wr(spi2, SPIx_CONFIGREG,
/* tx shift - rising edge; rx latch - falling edge */
CONFIG_SCLK_PHA_1 |
CONFIG_SS_CTL_NCSS);
wr(spi2, SPIx_TXDATA, 0xb5);
wr(spi2, SPIx_TXDATA, cmd);
wr(spi2, SPIx_TXDATA, arg);
wr(spi2, SPIx_CONREG, con | CON_XCH);
/*
* LPC buffers 3 bytes without responding, but spends some time
* to prepare the response. 50ms should be safe, add a bit more
* to be sure LPC is blocked waiting for the chip select to go
* active again.
*/
sleep(80);
while(rd(spi2, SPIx_STATREG) & STAT_RR)
rd(spi2, SPIx_RXDATA);
/* expecting 8 bytes, start the exchange */
for(i = 0; i < 8; i++)
wr(spi2, SPIx_TXDATA, 0);
wr(spi2, SPIx_CONREG, con | CON_XCH);
sleep(80);
for(i = 0; i < 8; i++)
((u8int*)ret)[i] = rd(spi2, SPIx_RXDATA);
wr(spi2, SPIx_CONREG, con & ~CON_EN);
}
static void
readbattery(Req *r)
{
int hh, mm, ss, full, remain, warn, safe;
u8int st[8], ch[8];
s16int current;
char msg[256];
char *state;
lpccall('c', 0, ch);
lpccall('q', 0, st);
current = (s16int)(st[2] | st[3]<<8);
remain = ch[0]|ch[1]<<8;
warn = ch[2]|ch[3]<<8;
full = ch[4]|ch[5]<<8;
safe = remain - warn;
hh = mm = ss = 0;
state = "unknown";
switch(st[5]){
case Sfullycharged: state = "full"; break;
case Sovervolted: state = "balancing"; break;
case Scooldown: state = "cooldown"; break;
case Spowersave: state = "powersave"; break;
case Scharge:
if(current < 0)
state = "charging";
if(current > 0)
state = "discharging";
if(current != 0){
ss = (current < 0 ? full - safe : safe) * 3600 / abs(current);
hh = ss/3600;
ss -= 3600*(ss/3600);
mm = ss/60;
ss -= 60*(ss/60);
}
break;
case Smissing:
state = "missing";
break;
}
snprint(msg, sizeof(msg), "%d mA %d %d %d %d ? mV %d ? %02d:%02d:%02d %s\n",
st[4],
remain, full, full, warn,
st[0]|st[1]<<8,
hh, mm, ss,
state
);
readstr(r, msg);
respond(r, nil);
}
static void
readpmctl(Req *r)
{
char msg[256], *s, *e;
static char lpcfw[9*3+1];
u8int v[16], q[8];
int i;
if(lpcfw[0] == 0){
lpccall('f', 0, msg+0);
lpccall('f', 1, msg+8);
lpccall('f', 2, msg+16);
snprint(lpcfw, sizeof(lpcfw), "%.*s %.*s %.*s", 8, msg+0, 8, msg+8, 8, msg+16);
}
lpccall('v', 0, v+0);
lpccall('v', 1, v+8);
lpccall('q', 0, q);
s = msg;
e = s+sizeof(msg);
s = seprint(s, e, "version %s\n", lpcfw);
s = seprint(s, e, "voltage(mV) %d\n", (s16int)(q[0] | q[1]<<8));
s = seprint(s, e, "current(mA) %d\n", (s16int)(q[2] | q[3]<<8));
s = seprint(s, e, "cells(mV)");
for(i = 0; i < 16; i += 2)
s = seprint(s, e, " %d", v[i] | v[i+1]<<8);
s = seprint(s, e, "\n");
USED(s);
readstr(r, msg);
respond(r, nil);
}
static void
fsread(Req *r)
{
int c, v[2+Nlights];
char msg[256];
void *aux;
msg[0] = 0;
if(r->ifcall.offset == 0){
aux = r->fid->file->aux;
if(aux == (void*)Light){
getlight(Lcd, v);
getlight(Kbd, v+1);
getlight(Tb, v+2);
snprint(msg, sizeof(msg),
"lcd %d\nkbd %d\ntb %d %d %d %d %d\n",
v[0], v[1], v[2], v[3], v[4], v[5], v[6]
);
}else if(aux == (void*)Temp){
if((c = getcputemp()) < 0){
responderror(r);
return;
}
snprint(msg, sizeof(msg), "%d.0\n", c);
}else if(aux == (void*)Battery){
reqqueuepush(lpcreq, r, readbattery);
return;
}else if(aux == (void*)Pmctl){
reqqueuepush(lpcreq, r, readpmctl);
return;
}
}
readstr(r, msg);
respond(r, nil);
}
static void
fswrite(Req *r)
{
int nf, v[Nlights], p, k, i;
char msg[256], *f[1+Nlights];
void *aux;
aux = r->fid->file->aux;
if(aux == (void*)Kbdoled){
if((k = loadkbdoled(r->ifcall.data, r->ifcall.offset, r->ifcall.count)) < 0){
Err:
responderror(r);
return;
}
r->ofcall.count = k;
respond(r, nil);
return;
}
snprint(msg, sizeof(msg), "%.*s", utfnlen(r->ifcall.data, r->ifcall.count), r->ifcall.data);
nf = tokenize(msg, f, nelem(f));
if(aux == (void*)Light){
if(nf < 2){
Bad:
respond(r, "invalid ctl message");
return;
}
if(strcmp(f[0], "lcd") == 0)
k = Lcd;
else if(strcmp(f[0], "kbd") == 0)
k = Kbd;
else if(strcmp(f[0], "tb") == 0)
k = Tb;
else
goto Bad;
getlight(k, v);
for(i = 0; i < nf-1; i++){
if(*f[i+1] != '+' && *f[i+1] != '-')
v[i] = 0;
v[i] += atoi(f[i+1]);
}
if(setlight(k, v) != 0)
goto Err;
}else if(aux == (void*)Pmctl){
p = -1;
if(nf >= 2 && strcmp(f[0], "power") == 0){
if(nf == 2 && strcmp(f[1], "off") == 0){
/*
* LPC firmware might not be up to date so try
* shutting down through the keyboard first
*/
if(openhidctl(&hidkb, udidkb) == 0){
write(hidkb, "PWR0", 4);
sleep(2000); /* give it a chance */
}
p = 1; /* fall back to lpc shutdown */
}
}
if(p < 0)
goto Bad;
lpccall('p', p, msg);
}
r->ofcall.count = r->ifcall.count;
respond(r, nil);
}
static void
fsflush(Req *r)
{
void *aux;
Req *o;
o = r->oldreq;
aux = o->fid->file->aux;
if(o->ifcall.type == Tread && (aux == (void*)Battery || aux == (void*)Pmctl))
reqqueueflush(lpcreq, o);
respond(r, nil);
}
static Srv fs = {
.read = fsread,
.write = fswrite,
.flush = fsflush,
};
static void
usage(void)
{
fprint(2, "usage: %s [-D] [-m mountpoint] [-s service]\n", argv0);
exits("usage");
}
void
threadmain(int argc, char **argv)
{
char *mtpt, *srv;
mtpt = "/dev";
srv = nil;
ARGBEGIN{
case 'D':
chatty9p = 1;
break;
case 'm':
mtpt = EARGF(usage());
break;
case 's':
srv = EARGF(usage());
break;
default:
usage();
}ARGEND
if((tmu = segattach(0, "tmu", 0, 0xf20)) == (void*)-1)
sysfatal("no tmu");
if((pwm2 = segattach(0, "pwm2", 0, 0x18)) == (void*)-1)
sysfatal("no pwm2");
if((spi2 = segattach(0, "ecspi2", 0, 0x20)) == (void*)-1)
sysfatal("no spi2");
tmuinit();
if(memimageinit() != 0)
sysfatal("%r");
if((kbdoled = allocmemimage(Rect(0, 0, KbdoledW, KbdoledH), GREY1)) == nil)
sysfatal("%r");
lpcreq = reqqueuecreate();
fs.tree = alloctree(uid, uid, DMDIR|0555, nil);
createfile(fs.tree->root, "battery", uid, 0444, (void*)Battery);
createfile(fs.tree->root, "cputemp", uid, 0444, (void*)Temp);
createfile(fs.tree->root, "light", uid, 0666, (void*)Light);
createfile(fs.tree->root, "kbdoled", uid, 0222, (void*)Kbdoled);
createfile(fs.tree->root, "pmctl", uid, 0666, (void*)Pmctl);
threadpostmountsrv(&fs, srv, mtpt, MAFTER);
threadexits(nil);
}