コード例 #1
0
ファイル: TransformVisitor.cpp プロジェクト: fengpsky/freeze
void
FreezeScript::TransformVisitor::visitObject(const ObjectRefPtr& dest)
{
    //
    // Support struct->class and class->class transforms.
    //
    Slice::TypePtr type = dest->getType();
    ObjectRefPtr src = ObjectRefPtr::dynamicCast(_src);
    StructDataPtr s = StructDataPtr::dynamicCast(_src);
    if(!src && !s)
    {
        typeMismatchError(type, _src->getType());
    }
    else if(_info->doDefaultTransform(type))
    {
        if(src)
        {
            //
            // class->class transform
            //
            ObjectDataPtr srcValue = src->getValue();
            Slice::TypePtr srcType = src->getType();
            if(!srcValue)
            {
                //
                // Allow a nil value from type Object.
                //
                if(Slice::BuiltinPtr::dynamicCast(srcType) || isCompatible(type, srcType))
                {
                    dest->setValue(0);
                }
                else
                {
                    typeMismatchError(type, srcType);
                }
            }
            else
            {
                Slice::TypePtr srcValueType = srcValue->getType();
                if(isCompatible(type, srcValueType))
                {
                    //
                    // If the types are in the same Slice unit, then we can simply
                    // copy the reference. Otherwise, we check the object map to
                    // see if an equivalent object has already been created, and
                    // if not, then we have to create one.
                    //
                    if(type->unit().get() == srcValueType->unit().get())
                    {
                        dest->setValue(srcValue);
                    }
                    else
                    {
                        ObjectDataMap& objectDataMap = _info->getObjectDataMap();
                        ObjectDataMap::iterator p = objectDataMap.find(srcValue.get());
                        if(p != objectDataMap.end() && p->second)
                        {
                            dest->setValue(p->second);
                        }
                        else
                        {
                            //
                            // If the type has been renamed, we need to get its equivalent
                            // in the new Slice definitions.
                            //
                            Slice::TypePtr newType = _info->getRenamedType(srcValueType);
                            if(!newType)
                            {
                                string name = typeToString(srcValueType);
                                Slice::TypeList l = type->unit()->lookupType(name, false);
                                if(l.empty())
                                {
                                    throw ClassNotFoundException(name);
                                }
                                newType = l.front();
                            }

                            //
                            // Use createObject() so that an initializer is invoked if necessary.
                            //
                            DataPtr newObj = _info->getDataFactory()->createObject(newType, false);
                            ObjectRefPtr newRef = ObjectRefPtr::dynamicCast(newObj);
                            assert(newRef);

                            ObjectDataPtr newValue = newRef->getValue();
                            try
                            {
                                transformObject(newValue, srcValue);
                            }
                            catch(...)
                            {
                                newObj->destroy();
                                throw;
                            }

                            dest->setValue(newValue);
                            newObj->destroy();
                        }
                    }
                }
                else
                {
                    typeMismatchError(type, srcValueType);
                }
            }
        }
        else
        {
            //
            // struct->class transform
            //
            Slice::TypePtr srcType = _src->getType();
            if(isCompatible(type, srcType))
            {
                //
                // If the type has been renamed, we need to get its equivalent
                // in the new Slice definitions.
                //
                Slice::TypePtr newType = _info->getRenamedType(srcType);
                if(!newType)
                {
                    string name = typeToString(srcType);
                    Slice::TypeList l = type->unit()->lookupType(name, false);
                    if(l.empty())
                    {
                        throw ClassNotFoundException(name);
                    }
                    newType = l.front();
                }

                //
                // Use createObject() so that an initializer is invoked if necessary.
                //
                DataPtr newObj = _info->getDataFactory()->createObject(newType, false);
                ObjectRefPtr newRef = ObjectRefPtr::dynamicCast(newObj);
                assert(newRef);

                ObjectDataPtr newValue = newRef->getValue();
                try
                {
                    //
                    // Transform members with the same name.
                    //
                    DataMemberMap srcMap = s->getMembers();
                    DataMemberMap destMap = newValue->getMembers();
                    string typeName = typeToString(type);
                    for(DataMemberMap::iterator p = destMap.begin(); p != destMap.end(); ++p)
                    {
                        DataMemberMap::iterator q = srcMap.find(p->first);
                        if(q != srcMap.end())
                        {
                            string context = typeName + " member " + p->first + " value";
                            TransformVisitor v(q->second, _info, context);
                            p->second->visit(v);
                        }
                    }
                }
                catch(...)
                {
                    newObj->destroy();
                    throw;
                }

                dest->setValue(newValue);
                newObj->destroy();
            }
            else
            {
                typeMismatchError(type, srcType);
            }
        }
    }
    _info->executeCustomTransform(dest, _src);
}