conservatory
/
picrin
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add some user functions to manipulate ports

This commit is contained in:
Yuichi Nishiwaki 2014-02-11 13:45:41 +09:00
parent b404f30a50
commit fc7b59100d
2 changed files with 48 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/picrin/port.h
View File

@ -37,6 +37,11 @@ struct pic_port *pic_stdin(pic_state *);
struct pic_port *pic_stdout(pic_state *); struct pic_port *pic_stdout(pic_state *);
struct pic_port *pic_stderr(pic_state *); struct pic_port *pic_stderr(pic_state *);
struct pic_port *pic_open_output_string(pic_state *);
struct pic_string *pic_get_output_string(pic_state *, struct pic_port *);
void pic_close_port(pic_state *pic, struct pic_port *);
#if defined(__cplusplus) #if defined(__cplusplus)
} }
#endif #endif

63
src/port.c
View File

@ -54,6 +54,46 @@ port_new_stdport(pic_state *pic, XFILE *file, short dir)
return pic_obj_value(port); return pic_obj_value(port);
} }
struct pic_port *
pic_open_output_string(pic_state *pic)
{
struct pic_port *port;
port = (struct pic_port *)pic_obj_alloc(pic, sizeof(struct pic_port *), PIC_TT_PORT);
port->file = xmopen();
port->flags = PIC_PORT_OUT | PIC_PORT_TEXT;
port->status = PIC_PORT_OPEN;
return port;
}
struct pic_string *
pic_get_output_string(pic_state *pic, struct pic_port *port)
{
long endpos;
char *buf;
/* get endpos */
xfflush(port->file);
endpos = xftell(port->file);
xrewind(port->file);
/* copy to buf */
buf = (char *)pic_alloc(pic, endpos);
xfread(buf, 1, endpos, port->file);
return pic_str_new(pic, buf, endpos);
}
void
pic_close_port(pic_state *pic, struct pic_port *port)
{
if (xfclose(port->file) == EOF) {
pic_error(pic, "close-port: failure");
}
port->status = PIC_PORT_CLOSE;
}
static pic_value static pic_value
pic_port_input_port_p(pic_state *pic) pic_port_input_port_p(pic_state *pic)
{ {
@ -188,10 +228,7 @@ pic_port_close_port(pic_state *pic)
pic_get_args(pic, "p", &port); pic_get_args(pic, "p", &port);
if (xfclose(port->file) == EOF) { pic_close_port(pic, port);
pic_error(pic, "close-port: failure");
}
port->status = PIC_PORT_CLOSE;
return pic_none_value(); return pic_none_value();
} }
@ -251,10 +288,7 @@ pic_port_open_output_string(pic_state *pic)
pic_get_args(pic, ""); pic_get_args(pic, "");
port = (struct pic_port *)pic_obj_alloc(pic, sizeof(struct pic_port *), PIC_TT_PORT); port = pic_open_output_string(pic);
port->file = xmopen();
port->flags = PIC_PORT_OUT | PIC_PORT_TEXT;
port->status = PIC_PORT_OPEN;
return pic_obj_value(port); return pic_obj_value(port);
} }
@ -263,23 +297,12 @@ static pic_value
pic_port_get_output_string(pic_state *pic) pic_port_get_output_string(pic_state *pic)
{ {
struct pic_port *port = pic_stdout(pic);; struct pic_port *port = pic_stdout(pic);;
long endpos;
char *buf;
pic_get_args(pic, "|p", &port); pic_get_args(pic, "|p", &port);
assert_port_profile(port, PIC_PORT_OUT | PIC_PORT_TEXT, PIC_PORT_OPEN, "get-output-string"); assert_port_profile(port, PIC_PORT_OUT | PIC_PORT_TEXT, PIC_PORT_OPEN, "get-output-string");
/* get endpos */ return pic_obj_value(pic_get_output_string(pic, port));
xfflush(port->file);
endpos = xftell(port->file);
xrewind(port->file);
/* copy to buf */
buf = (char *)pic_alloc(pic, endpos);
xfread(buf, 1, endpos, port->file);
return pic_obj_value(pic_str_new(pic, buf, endpos));
} }
static pic_value static pic_value