зеркало из https://github.com/github/ruby.git
738 строки
22 KiB
C
738 строки
22 KiB
C
#include "win32ole.h"
|
|
|
|
struct olevariantdata {
|
|
VARIANT realvar;
|
|
VARIANT var;
|
|
};
|
|
|
|
static void olevariant_free(void *ptr);
|
|
static size_t olevariant_size(const void *ptr);
|
|
static void ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar);
|
|
static void ole_val2variant_err(VALUE val, VARIANT *var);
|
|
static void ole_set_byref(VARIANT *realvar, VARIANT *var, VARTYPE vt);
|
|
static VALUE folevariant_s_allocate(VALUE klass);
|
|
static VALUE folevariant_s_array(VALUE klass, VALUE dims, VALUE vvt);
|
|
static void check_type_val2variant(VALUE val);
|
|
static VALUE folevariant_initialize(VALUE self, VALUE args);
|
|
static LONG *ary2safe_array_index(int ary_size, VALUE *ary, SAFEARRAY *psa);
|
|
static void unlock_safe_array(SAFEARRAY *psa);
|
|
static SAFEARRAY *get_locked_safe_array(VALUE val);
|
|
static VALUE folevariant_ary_aref(int argc, VALUE *argv, VALUE self);
|
|
static VALUE folevariant_ary_aset(int argc, VALUE *argv, VALUE self);
|
|
static VALUE folevariant_value(VALUE self);
|
|
static VALUE folevariant_vartype(VALUE self);
|
|
static VALUE folevariant_set_value(VALUE self, VALUE val);
|
|
|
|
static const rb_data_type_t olevariant_datatype = {
|
|
"win32ole_variant",
|
|
{NULL, olevariant_free, olevariant_size,},
|
|
0, 0, RUBY_TYPED_FREE_IMMEDIATELY
|
|
};
|
|
|
|
static void
|
|
olevariant_free(void *ptr)
|
|
{
|
|
struct olevariantdata *pvar = ptr;
|
|
VariantClear(&(pvar->realvar));
|
|
VariantClear(&(pvar->var));
|
|
free(pvar);
|
|
}
|
|
|
|
static size_t
|
|
olevariant_size(const void *ptr)
|
|
{
|
|
return ptr ? sizeof(struct olevariantdata) : 0;
|
|
}
|
|
|
|
static void
|
|
ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (((vt & ~VT_BYREF) == (VT_ARRAY | VT_UI1)) && RB_TYPE_P(val, T_STRING)) {
|
|
long len = RSTRING_LEN(val);
|
|
void *pdest = NULL;
|
|
SAFEARRAY *p = NULL;
|
|
SAFEARRAY *psa = SafeArrayCreateVector(VT_UI1, 0, len);
|
|
if (!psa) {
|
|
rb_raise(rb_eRuntimeError, "fail to SafeArrayCreateVector");
|
|
}
|
|
hr = SafeArrayAccessData(psa, &pdest);
|
|
if (SUCCEEDED(hr)) {
|
|
memcpy(pdest, RSTRING_PTR(val), len);
|
|
SafeArrayUnaccessData(psa);
|
|
V_VT(&(pvar->realvar)) = (vt & ~VT_BYREF);
|
|
p = V_ARRAY(&(pvar->realvar));
|
|
if (p != NULL) {
|
|
SafeArrayDestroy(p);
|
|
}
|
|
V_ARRAY(&(pvar->realvar)) = psa;
|
|
if (vt & VT_BYREF) {
|
|
V_VT(&(pvar->var)) = vt;
|
|
V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
|
|
} else {
|
|
hr = VariantCopy(&(pvar->var), &(pvar->realvar));
|
|
}
|
|
} else {
|
|
if (psa)
|
|
SafeArrayDestroy(psa);
|
|
}
|
|
} else if (vt & VT_ARRAY) {
|
|
if (val == Qnil) {
|
|
V_VT(&(pvar->var)) = vt;
|
|
if (vt & VT_BYREF) {
|
|
V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
|
|
}
|
|
} else {
|
|
hr = ole_val_ary2variant_ary(val, &(pvar->realvar), (VARTYPE)(vt & ~VT_BYREF));
|
|
if (SUCCEEDED(hr)) {
|
|
if (vt & VT_BYREF) {
|
|
V_VT(&(pvar->var)) = vt;
|
|
V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
|
|
} else {
|
|
hr = VariantCopy(&(pvar->var), &(pvar->realvar));
|
|
}
|
|
}
|
|
}
|
|
#if (defined(_MSC_VER) && (_MSC_VER >= 1300)) || defined(__CYGWIN__) || defined(__MINGW32__)
|
|
} else if ( (vt & ~VT_BYREF) == VT_I8 || (vt & ~VT_BYREF) == VT_UI8) {
|
|
ole_val2variant_ex(val, &(pvar->realvar), (vt & ~VT_BYREF));
|
|
ole_val2variant_ex(val, &(pvar->var), (vt & ~VT_BYREF));
|
|
V_VT(&(pvar->var)) = vt;
|
|
if (vt & VT_BYREF) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
}
|
|
#endif
|
|
} else if ( (vt & ~VT_BYREF) == VT_ERROR) {
|
|
ole_val2variant_err(val, &(pvar->realvar));
|
|
if (vt & VT_BYREF) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
} else {
|
|
hr = VariantCopy(&(pvar->var), &(pvar->realvar));
|
|
}
|
|
} else {
|
|
if (val == Qnil) {
|
|
V_VT(&(pvar->var)) = vt;
|
|
if (vt == (VT_BYREF | VT_VARIANT)) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
} else {
|
|
V_VT(&(pvar->realvar)) = vt & ~VT_BYREF;
|
|
if (vt & VT_BYREF) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
}
|
|
}
|
|
} else {
|
|
ole_val2variant_ex(val, &(pvar->realvar), (VARTYPE)(vt & ~VT_BYREF));
|
|
if (vt == (VT_BYREF | VT_VARIANT)) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
} else if (vt & VT_BYREF) {
|
|
if ( (vt & ~VT_BYREF) != V_VT(&(pvar->realvar))) {
|
|
hr = VariantChangeTypeEx(&(pvar->realvar), &(pvar->realvar),
|
|
cWIN32OLE_lcid, 0, (VARTYPE)(vt & ~VT_BYREF));
|
|
}
|
|
if (SUCCEEDED(hr)) {
|
|
ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
|
|
}
|
|
} else {
|
|
if (vt == V_VT(&(pvar->realvar))) {
|
|
hr = VariantCopy(&(pvar->var), &(pvar->realvar));
|
|
} else {
|
|
hr = VariantChangeTypeEx(&(pvar->var), &(pvar->realvar),
|
|
cWIN32OLE_lcid, 0, vt);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (FAILED(hr)) {
|
|
ole_raise(hr, eWIN32OLERuntimeError, "failed to change type");
|
|
}
|
|
}
|
|
|
|
static void
|
|
ole_val2variant_err(VALUE val, VARIANT *var)
|
|
{
|
|
VALUE v = val;
|
|
if (rb_obj_is_kind_of(v, cWIN32OLE_VARIANT)) {
|
|
v = folevariant_value(v);
|
|
}
|
|
if (!(FIXNUM_P(v) || RB_TYPE_P(v, T_BIGNUM) || v == Qnil)) {
|
|
rb_raise(eWIN32OLERuntimeError, "failed to convert VT_ERROR VARIANT:`%"PRIsVALUE"'", rb_inspect(v));
|
|
}
|
|
V_VT(var) = VT_ERROR;
|
|
if (v != Qnil) {
|
|
V_ERROR(var) = RB_NUM2LONG(val);
|
|
} else {
|
|
V_ERROR(var) = 0;
|
|
}
|
|
}
|
|
|
|
static void
|
|
ole_set_byref(VARIANT *realvar, VARIANT *var, VARTYPE vt)
|
|
{
|
|
V_VT(var) = vt;
|
|
if (vt == (VT_VARIANT|VT_BYREF)) {
|
|
V_VARIANTREF(var) = realvar;
|
|
} else {
|
|
if (V_VT(realvar) != (vt & ~VT_BYREF)) {
|
|
rb_raise(eWIN32OLERuntimeError, "variant type mismatch");
|
|
}
|
|
switch(vt & ~VT_BYREF) {
|
|
case VT_I1:
|
|
V_I1REF(var) = &V_I1(realvar);
|
|
break;
|
|
case VT_UI1:
|
|
V_UI1REF(var) = &V_UI1(realvar);
|
|
break;
|
|
case VT_I2:
|
|
V_I2REF(var) = &V_I2(realvar);
|
|
break;
|
|
case VT_UI2:
|
|
V_UI2REF(var) = &V_UI2(realvar);
|
|
break;
|
|
case VT_I4:
|
|
V_I4REF(var) = &V_I4(realvar);
|
|
break;
|
|
case VT_UI4:
|
|
V_UI4REF(var) = &V_UI4(realvar);
|
|
break;
|
|
case VT_R4:
|
|
V_R4REF(var) = &V_R4(realvar);
|
|
break;
|
|
case VT_R8:
|
|
V_R8REF(var) = &V_R8(realvar);
|
|
break;
|
|
|
|
#if (defined(_MSC_VER) && (_MSC_VER >= 1300)) || defined(__CYGWIN__) || defined(__MINGW32__)
|
|
#ifdef V_I8REF
|
|
case VT_I8:
|
|
V_I8REF(var) = &V_I8(realvar);
|
|
break;
|
|
#endif
|
|
#ifdef V_UI8REF
|
|
case VT_UI8:
|
|
V_UI8REF(var) = &V_UI8(realvar);
|
|
break;
|
|
#endif
|
|
#endif
|
|
case VT_INT:
|
|
V_INTREF(var) = &V_INT(realvar);
|
|
break;
|
|
|
|
case VT_UINT:
|
|
V_UINTREF(var) = &V_UINT(realvar);
|
|
break;
|
|
|
|
case VT_CY:
|
|
V_CYREF(var) = &V_CY(realvar);
|
|
break;
|
|
case VT_DATE:
|
|
V_DATEREF(var) = &V_DATE(realvar);
|
|
break;
|
|
case VT_BSTR:
|
|
V_BSTRREF(var) = &V_BSTR(realvar);
|
|
break;
|
|
case VT_DISPATCH:
|
|
V_DISPATCHREF(var) = &V_DISPATCH(realvar);
|
|
break;
|
|
case VT_ERROR:
|
|
V_ERRORREF(var) = &V_ERROR(realvar);
|
|
break;
|
|
case VT_BOOL:
|
|
V_BOOLREF(var) = &V_BOOL(realvar);
|
|
break;
|
|
case VT_UNKNOWN:
|
|
V_UNKNOWNREF(var) = &V_UNKNOWN(realvar);
|
|
break;
|
|
case VT_ARRAY:
|
|
V_ARRAYREF(var) = &V_ARRAY(realvar);
|
|
break;
|
|
default:
|
|
rb_raise(eWIN32OLERuntimeError, "unknown type specified(setting BYREF):%d", vt);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static VALUE
|
|
folevariant_s_allocate(VALUE klass)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
VALUE obj;
|
|
ole_initialize();
|
|
obj = TypedData_Make_Struct(klass, struct olevariantdata, &olevariant_datatype, pvar);
|
|
VariantInit(&(pvar->var));
|
|
VariantInit(&(pvar->realvar));
|
|
return obj;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* array(ary, vt)
|
|
*
|
|
* Returns Ruby object wrapping OLE variant whose variant type is VT_ARRAY.
|
|
* The first argument should be Array object which specifies dimensions
|
|
* and each size of dimensions of OLE array.
|
|
* The second argument specifies variant type of the element of OLE array.
|
|
*
|
|
* The following create 2 dimensions OLE array. The first dimensions size
|
|
* is 3, and the second is 4.
|
|
*
|
|
* ole_ary = WIN32OLE::Variant.array([3,4], VT_I4)
|
|
* ruby_ary = ole_ary.value # => [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_s_array(VALUE klass, VALUE elems, VALUE vvt)
|
|
{
|
|
VALUE obj = Qnil;
|
|
VARTYPE vt;
|
|
struct olevariantdata *pvar;
|
|
SAFEARRAYBOUND *psab = NULL;
|
|
SAFEARRAY *psa = NULL;
|
|
UINT dim = 0;
|
|
UINT i = 0;
|
|
|
|
ole_initialize();
|
|
|
|
vt = RB_NUM2UINT(vvt);
|
|
vt = (vt | VT_ARRAY);
|
|
Check_Type(elems, T_ARRAY);
|
|
obj = folevariant_s_allocate(klass);
|
|
|
|
TypedData_Get_Struct(obj, struct olevariantdata, &olevariant_datatype, pvar);
|
|
dim = RARRAY_LEN(elems);
|
|
|
|
psab = ALLOC_N(SAFEARRAYBOUND, dim);
|
|
|
|
if(!psab) {
|
|
rb_raise(rb_eRuntimeError, "memory allocation error");
|
|
}
|
|
|
|
for (i = 0; i < dim; i++) {
|
|
psab[i].cElements = RB_FIX2INT(rb_ary_entry(elems, i));
|
|
psab[i].lLbound = 0;
|
|
}
|
|
|
|
psa = SafeArrayCreate((VARTYPE)(vt & VT_TYPEMASK), dim, psab);
|
|
if (psa == NULL) {
|
|
if (psab) free(psab);
|
|
rb_raise(rb_eRuntimeError, "memory allocation error(SafeArrayCreate)");
|
|
}
|
|
|
|
V_VT(&(pvar->var)) = vt;
|
|
if (vt & VT_BYREF) {
|
|
V_VT(&(pvar->realvar)) = (vt & ~VT_BYREF);
|
|
V_ARRAY(&(pvar->realvar)) = psa;
|
|
V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
|
|
} else {
|
|
V_ARRAY(&(pvar->var)) = psa;
|
|
}
|
|
if (psab) free(psab);
|
|
return obj;
|
|
}
|
|
|
|
static void
|
|
check_type_val2variant(VALUE val)
|
|
{
|
|
VALUE elem;
|
|
int len = 0;
|
|
int i = 0;
|
|
if(!rb_obj_is_kind_of(val, cWIN32OLE) &&
|
|
!rb_obj_is_kind_of(val, cWIN32OLE_VARIANT) &&
|
|
!rb_obj_is_kind_of(val, rb_cTime)) {
|
|
switch (TYPE(val)) {
|
|
case T_ARRAY:
|
|
len = RARRAY_LEN(val);
|
|
for(i = 0; i < len; i++) {
|
|
elem = rb_ary_entry(val, i);
|
|
check_type_val2variant(elem);
|
|
}
|
|
break;
|
|
case T_STRING:
|
|
case T_FIXNUM:
|
|
case T_BIGNUM:
|
|
case T_FLOAT:
|
|
case T_TRUE:
|
|
case T_FALSE:
|
|
case T_NIL:
|
|
break;
|
|
default:
|
|
rb_raise(rb_eTypeError, "can not convert WIN32OLE::Variant from type %s",
|
|
rb_obj_classname(val));
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Document-class: WIN32OLE::Variant
|
|
*
|
|
* +WIN32OLE::Variant+ objects represents OLE variant.
|
|
*
|
|
* Win32OLE converts Ruby object into OLE variant automatically when
|
|
* invoking OLE methods. If OLE method requires the argument which is
|
|
* different from the variant by automatic conversion of Win32OLE, you
|
|
* can convert the specified variant type by using WIN32OLE::Variant class.
|
|
*
|
|
* param = WIN32OLE::Variant.new(10, WIN32OLE::VARIANT::VT_R4)
|
|
* oleobj.method(param)
|
|
*
|
|
* WIN32OLE::Variant does not support VT_RECORD variant. Use WIN32OLE::Record
|
|
* class instead of WIN32OLE::Variant if the VT_RECORD variant is needed.
|
|
*/
|
|
|
|
/*
|
|
* call-seq:
|
|
* new(val, vartype) #=> WIN32OLE::Variant object.
|
|
*
|
|
* Returns Ruby object wrapping OLE variant.
|
|
* The first argument specifies Ruby object to convert OLE variant variable.
|
|
* The second argument specifies VARIANT type.
|
|
* In some situation, you need the WIN32OLE::Variant object to pass OLE method
|
|
*
|
|
* shell = WIN32OLE.new("Shell.Application")
|
|
* folder = shell.NameSpace("C:\\Windows")
|
|
* item = folder.ParseName("tmp.txt")
|
|
* # You can't use Ruby String object to call FolderItem.InvokeVerb.
|
|
* # Instead, you have to use WIN32OLE::Variant object to call the method.
|
|
* shortcut = WIN32OLE::Variant.new("Create Shortcut(\&S)")
|
|
* item.invokeVerb(shortcut)
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_initialize(VALUE self, VALUE args)
|
|
{
|
|
int len = 0;
|
|
VARIANT var;
|
|
VALUE val;
|
|
VALUE vvt;
|
|
VARTYPE vt;
|
|
struct olevariantdata *pvar;
|
|
|
|
len = RARRAY_LEN(args);
|
|
rb_check_arity(len, 1, 3);
|
|
VariantInit(&var);
|
|
val = rb_ary_entry(args, 0);
|
|
|
|
check_type_val2variant(val);
|
|
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
if (len == 1) {
|
|
ole_val2variant(val, &(pvar->var));
|
|
} else {
|
|
vvt = rb_ary_entry(args, 1);
|
|
vt = RB_NUM2INT(vvt);
|
|
if ((vt & VT_TYPEMASK) == VT_RECORD) {
|
|
rb_raise(rb_eArgError, "not supported VT_RECORD WIN32OLE::Variant object");
|
|
}
|
|
ole_val2olevariantdata(val, vt, pvar);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
static SAFEARRAY *
|
|
get_locked_safe_array(VALUE val)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
SAFEARRAY *psa = NULL;
|
|
HRESULT hr;
|
|
TypedData_Get_Struct(val, struct olevariantdata, &olevariant_datatype, pvar);
|
|
if (!(V_VT(&(pvar->var)) & VT_ARRAY)) {
|
|
rb_raise(rb_eTypeError, "variant type is not VT_ARRAY.");
|
|
}
|
|
psa = V_ISBYREF(&(pvar->var)) ? *V_ARRAYREF(&(pvar->var)) : V_ARRAY(&(pvar->var));
|
|
if (psa == NULL) {
|
|
return psa;
|
|
}
|
|
hr = SafeArrayLock(psa);
|
|
if (FAILED(hr)) {
|
|
ole_raise(hr, rb_eRuntimeError, "failed to SafeArrayLock");
|
|
}
|
|
return psa;
|
|
}
|
|
|
|
static LONG *
|
|
ary2safe_array_index(int ary_size, VALUE *ary, SAFEARRAY *psa)
|
|
{
|
|
long dim;
|
|
LONG *pid;
|
|
long i;
|
|
dim = SafeArrayGetDim(psa);
|
|
if (dim != ary_size) {
|
|
rb_raise(rb_eArgError, "unmatch number of indices");
|
|
}
|
|
pid = ALLOC_N(LONG, dim);
|
|
if (pid == NULL) {
|
|
rb_raise(rb_eRuntimeError, "failed to allocate memory for indices");
|
|
}
|
|
for (i = 0; i < dim; i++) {
|
|
pid[i] = RB_NUM2INT(ary[i]);
|
|
}
|
|
return pid;
|
|
}
|
|
|
|
static void
|
|
unlock_safe_array(SAFEARRAY *psa)
|
|
{
|
|
HRESULT hr;
|
|
hr = SafeArrayUnlock(psa);
|
|
if (FAILED(hr)) {
|
|
ole_raise(hr, rb_eRuntimeError, "failed to SafeArrayUnlock");
|
|
}
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* variant[i,j,...] #=> element of OLE array.
|
|
*
|
|
* Returns the element of WIN32OLE::Variant object(OLE array).
|
|
* This method is available only when the variant type of
|
|
* WIN32OLE::Variant object is VT_ARRAY.
|
|
*
|
|
* REMARK:
|
|
* The all indices should be 0 or natural number and
|
|
* lower than or equal to max indices.
|
|
* (This point is different with Ruby Array indices.)
|
|
*
|
|
* obj = WIN32OLE::Variant.new([[1,2,3],[4,5,6]])
|
|
* p obj[0,0] # => 1
|
|
* p obj[1,0] # => 4
|
|
* p obj[2,0] # => WIN32OLE::RuntimeError
|
|
* p obj[0, -1] # => WIN32OLE::RuntimeError
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_ary_aref(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
SAFEARRAY *psa;
|
|
VALUE val = Qnil;
|
|
VARIANT variant;
|
|
LONG *pid;
|
|
HRESULT hr;
|
|
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
if (!V_ISARRAY(&(pvar->var))) {
|
|
rb_raise(eWIN32OLERuntimeError,
|
|
"`[]' is not available for this variant type object");
|
|
}
|
|
psa = get_locked_safe_array(self);
|
|
if (psa == NULL) {
|
|
return val;
|
|
}
|
|
|
|
pid = ary2safe_array_index(argc, argv, psa);
|
|
|
|
VariantInit(&variant);
|
|
V_VT(&variant) = (V_VT(&(pvar->var)) & ~VT_ARRAY) | VT_BYREF;
|
|
hr = SafeArrayPtrOfIndex(psa, pid, &V_BYREF(&variant));
|
|
if (FAILED(hr)) {
|
|
ole_raise(hr, eWIN32OLERuntimeError, "failed to SafeArrayPtrOfIndex");
|
|
}
|
|
val = ole_variant2val(&variant);
|
|
|
|
unlock_safe_array(psa);
|
|
if (pid) free(pid);
|
|
return val;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* variant[i,j,...] = val #=> set the element of OLE array
|
|
*
|
|
* Set the element of WIN32OLE::Variant object(OLE array) to val.
|
|
* This method is available only when the variant type of
|
|
* WIN32OLE::Variant object is VT_ARRAY.
|
|
*
|
|
* REMARK:
|
|
* The all indices should be 0 or natural number and
|
|
* lower than or equal to max indices.
|
|
* (This point is different with Ruby Array indices.)
|
|
*
|
|
* obj = WIN32OLE::Variant.new([[1,2,3],[4,5,6]])
|
|
* obj[0,0] = 7
|
|
* obj[1,0] = 8
|
|
* p obj.value # => [[7,2,3], [8,5,6]]
|
|
* obj[2,0] = 9 # => WIN32OLE::RuntimeError
|
|
* obj[0, -1] = 9 # => WIN32OLE::RuntimeError
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_ary_aset(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
SAFEARRAY *psa;
|
|
VARIANT var;
|
|
VARTYPE vt;
|
|
LONG *pid;
|
|
HRESULT hr;
|
|
VOID *p = NULL;
|
|
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
if (!V_ISARRAY(&(pvar->var))) {
|
|
rb_raise(eWIN32OLERuntimeError,
|
|
"`[]' is not available for this variant type object");
|
|
}
|
|
psa = get_locked_safe_array(self);
|
|
if (psa == NULL) {
|
|
rb_raise(rb_eRuntimeError, "failed to get SafeArray pointer");
|
|
}
|
|
|
|
pid = ary2safe_array_index(argc-1, argv, psa);
|
|
|
|
VariantInit(&var);
|
|
vt = (V_VT(&(pvar->var)) & ~VT_ARRAY);
|
|
p = val2variant_ptr(argv[argc-1], &var, vt);
|
|
if ((V_VT(&var) == VT_DISPATCH && V_DISPATCH(&var) == NULL) ||
|
|
(V_VT(&var) == VT_UNKNOWN && V_UNKNOWN(&var) == NULL)) {
|
|
rb_raise(eWIN32OLERuntimeError, "argument does not have IDispatch or IUnknown Interface");
|
|
}
|
|
hr = SafeArrayPutElement(psa, pid, p);
|
|
if (FAILED(hr)) {
|
|
ole_raise(hr, eWIN32OLERuntimeError, "failed to SafeArrayPutElement");
|
|
}
|
|
|
|
unlock_safe_array(psa);
|
|
if (pid) free(pid);
|
|
return argv[argc-1];
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* value #=> Ruby object.
|
|
*
|
|
* Returns Ruby object value from OLE variant.
|
|
* obj = WIN32OLE::Variant.new(1, WIN32OLE::VARIANT::VT_BSTR)
|
|
* obj.value # => "1" (not Integer object, but String object "1")
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_value(VALUE self)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
VALUE val = Qnil;
|
|
VARTYPE vt;
|
|
int dim;
|
|
SAFEARRAY *psa;
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
|
|
val = ole_variant2val(&(pvar->var));
|
|
vt = V_VT(&(pvar->var));
|
|
|
|
if ((vt & ~VT_BYREF) == (VT_UI1|VT_ARRAY)) {
|
|
if (vt & VT_BYREF) {
|
|
psa = *V_ARRAYREF(&(pvar->var));
|
|
} else {
|
|
psa = V_ARRAY(&(pvar->var));
|
|
}
|
|
if (!psa) {
|
|
return val;
|
|
}
|
|
dim = SafeArrayGetDim(psa);
|
|
if (dim == 1) {
|
|
val = rb_funcall(val, rb_intern("pack"), 1, rb_str_new2("C*"));
|
|
}
|
|
}
|
|
return val;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* vartype #=> OLE variant type.
|
|
*
|
|
* Returns OLE variant type.
|
|
* obj = WIN32OLE::Variant.new("string")
|
|
* obj.vartype # => WIN32OLE::VARIANT::VT_BSTR
|
|
*
|
|
*/
|
|
static VALUE
|
|
folevariant_vartype(VALUE self)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
return RB_INT2FIX(V_VT(&pvar->var));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* variant.value = val #=> set WIN32OLE::Variant value to val.
|
|
*
|
|
* Sets variant value to val. If the val type does not match variant value
|
|
* type(vartype), then val is changed to match variant value type(vartype)
|
|
* before setting val.
|
|
* This method is not available when vartype is VT_ARRAY(except VT_UI1|VT_ARRAY).
|
|
* If the vartype is VT_UI1|VT_ARRAY, the val should be String object.
|
|
*
|
|
* obj = WIN32OLE::Variant.new(1) # obj.vartype is WIN32OLE::VARIANT::VT_I4
|
|
* obj.value = 3.2 # 3.2 is changed to 3 when setting value.
|
|
* p obj.value # => 3
|
|
*/
|
|
static VALUE
|
|
folevariant_set_value(VALUE self, VALUE val)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
VARTYPE vt;
|
|
TypedData_Get_Struct(self, struct olevariantdata, &olevariant_datatype, pvar);
|
|
vt = V_VT(&(pvar->var));
|
|
if (V_ISARRAY(&(pvar->var)) && ((vt & ~VT_BYREF) != (VT_UI1|VT_ARRAY) || !RB_TYPE_P(val, T_STRING))) {
|
|
rb_raise(eWIN32OLERuntimeError,
|
|
"`value=' is not available for this variant type object");
|
|
}
|
|
ole_val2olevariantdata(val, vt, pvar);
|
|
return Qnil;
|
|
}
|
|
|
|
void
|
|
ole_variant2variant(VALUE val, VARIANT *var)
|
|
{
|
|
struct olevariantdata *pvar;
|
|
TypedData_Get_Struct(val, struct olevariantdata, &olevariant_datatype, pvar);
|
|
VariantCopy(var, &(pvar->var));
|
|
}
|
|
|
|
VALUE cWIN32OLE_VARIANT;
|
|
|
|
void
|
|
Init_win32ole_variant(void)
|
|
{
|
|
#undef rb_intern
|
|
cWIN32OLE_VARIANT = rb_define_class_under(cWIN32OLE, "Variant", rb_cObject);
|
|
/* Alias of WIN32OLE::Variant, for the backward compatibility */
|
|
rb_define_const(rb_cObject, "WIN32OLE_VARIANT", cWIN32OLE_VARIANT);
|
|
rb_define_alloc_func(cWIN32OLE_VARIANT, folevariant_s_allocate);
|
|
rb_define_singleton_method(cWIN32OLE_VARIANT, "array", folevariant_s_array, 2);
|
|
rb_define_method(cWIN32OLE_VARIANT, "initialize", folevariant_initialize, -2);
|
|
rb_define_method(cWIN32OLE_VARIANT, "value", folevariant_value, 0);
|
|
rb_define_method(cWIN32OLE_VARIANT, "value=", folevariant_set_value, 1);
|
|
rb_define_method(cWIN32OLE_VARIANT, "vartype", folevariant_vartype, 0);
|
|
rb_define_method(cWIN32OLE_VARIANT, "[]", folevariant_ary_aref, -1);
|
|
rb_define_method(cWIN32OLE_VARIANT, "[]=", folevariant_ary_aset, -1);
|
|
|
|
/*
|
|
* represents VT_EMPTY OLE object.
|
|
*/
|
|
rb_define_const(cWIN32OLE_VARIANT, "Empty",
|
|
rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, RB_INT2FIX(VT_EMPTY)));
|
|
|
|
/*
|
|
* represents VT_NULL OLE object.
|
|
*/
|
|
rb_define_const(cWIN32OLE_VARIANT, "Null",
|
|
rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, RB_INT2FIX(VT_NULL)));
|
|
|
|
/*
|
|
* represents Nothing of VB.NET or VB.
|
|
*/
|
|
rb_define_const(cWIN32OLE_VARIANT, "Nothing",
|
|
rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, RB_INT2FIX(VT_DISPATCH)));
|
|
|
|
/*
|
|
* represents VT_ERROR variant with DISP_E_PARAMNOTFOUND.
|
|
* This constants is used for not specified parameter.
|
|
*
|
|
* fso = WIN32OLE.new("Scripting.FileSystemObject")
|
|
* fso.openTextFile(filename, WIN32OLE::Variant::NoParam, false)
|
|
*/
|
|
rb_define_const(cWIN32OLE_VARIANT, "NoParam",
|
|
rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, INT2NUM(DISP_E_PARAMNOTFOUND), RB_INT2FIX(VT_ERROR)));
|
|
}
|