Ejemplo n.º 1
0
        public static string InsertSample(string value, string name)
        {
            LLex l = new LLex(new StringLoadInfo(value), name);

            l.InsertString(0, LOCAL_PROFILER + "BeginMikuSample(\"" + "[lua]:require " + name + " &line:1" + "\") ");
            int lastPos = 0;
            int nextPos = l.pos;

            l.Next();
            int tokenType = l.Token.TokenType;

            lastPos = nextPos;
            nextPos = l.pos;

            InsertSample(l, ref lastPos, ref nextPos, tokenType, false);

            return(l.code);
        }
Ejemplo n.º 2
0
        static void InsertSample(LLex l, ref int lastPos, ref int nextPos, int tokenType, bool onlyFun)
        {
            Stack <int>  tokens  = new Stack <int>();
            List <Token> history = new List <Token>(16);

            bool needLastSample = true;
            bool hasReturn      = false;
            int  lastStackToken = -1;

            while (tokenType != (int)TK.EOS)
            {
                switch (tokenType)
                {
                case (int)TK.FUNCTION:
                    hasReturn = false;
                    tokens.Push(tokenType);
                    lastStackToken = tokenType;
                    string funName = "";
                    bool   isLeft  = false;

                    bool isForward = false;
                    int  index     = history.Count - 2;
                    if (index >= 0)
                    {
                        var hisToken = history[index];
                        while (hisToken is JumpToken)
                        {
                            index--;
                            if (index < 0)
                            {
                                break;
                            }
                            hisToken = history[index];
                        }
                        isForward = hisToken.TokenType == (int)'=';
                        index--;
                        if (isForward && index >= 0)
                        {
                            hisToken = history[index];
                            while (hisToken is JumpToken)
                            {
                                index--;
                                if (index < 0)
                                {
                                    break;
                                }
                                hisToken = history[index];
                            }
                            while (!(hisToken is JumpToken))
                            {
                                if (hisToken is NameToken)
                                {
                                    funName = ((NameToken)hisToken).SemInfo + funName;
                                }
                                else if ((hisToken.TokenType == (int)':'))
                                {
                                    funName = ':' + funName;
                                }
                                else if ((hisToken.TokenType == (int)'.') ||
                                         (hisToken.TokenType == (int)'['))
                                {
                                    funName = '.' + funName;
                                }
                                else if (hisToken is StringToken)
                                {
                                    funName = ((StringToken)hisToken).SemInfo + funName;
                                }
                                else
                                {
                                    break;
                                }
                                index--;
                                if (index < 0)
                                {
                                    break;
                                }
                                hisToken = history[index];
                            }
                        }
                    }

                    while (tokenType != (int)TK.EOS)
                    {
                        l.Next();
                        tokenType = l.Token.TokenType;

                        lastPos = nextPos;
                        nextPos = l.pos;


                        if (!isLeft && !isForward)
                        {
                            if (l.Token is NameToken)
                            {
                                funName += ((NameToken)l.Token).SemInfo;
                            }
                            else if ((l.Token.TokenType == (int)':'))
                            {
                                funName += ':';
                            }
                            else if ((l.Token.TokenType == (int)'.'))
                            {
                                funName += '.';
                            }
                        }

                        if (tokenType == (int)'(')
                        {
                            isLeft = true;
                        }

                        if (tokenType == (int)')')
                        {
                            if (string.IsNullOrEmpty(funName))
                            {
                                funName = "";
                            }
                            string profilerStr = string.Format(" BeginMikuSample(\"[lua]:{0} {1}&line:{2}\") ", funName, l.Source, l.LineNumber);
                            l.InsertString(nextPos - 1, profilerStr);
                            nextPos = l.pos;
                            break;
                        }
                    }
                    break;

                case (int)TK.IF:
                case (int)TK.FOR:
                case (int)TK.WHILE:
                    if (tokens.Count > 0)
                    {
                        tokens.Push(tokenType);
                        lastStackToken = tokenType;
                    }
                    break;

                case (int)TK.RETURN:
                    int insertPos = lastPos - 1;

                    if (tokens.Count == 0)
                    {
                        needLastSample = false;
                    }
                    while (tokenType != (int)TK.EOS)
                    {
                        l.Next();

                        tokenType = l.Token.TokenType;
                        if (!(l.Token is JumpToken))
                        {
                            lastPos = nextPos;
                            nextPos = l.pos;
                        }

                        if (tokenType == (int)TK.FUNCTION)
                        {
                            InsertSample(l, ref lastPos, ref nextPos, tokenType, true);
                            tokenType = l.Token.TokenType;
                        }

                        if (tokenType == (int)TK.END ||
                            tokenType == (int)TK.ELSEIF ||
                            tokenType == (int)TK.ELSE ||
                            tokenType == (int)TK.EOS)
                        {
                            lastPos = lastPos - 1;

                            string returnStr = l.ReadString(insertPos, lastPos - 1);
                            returnStr = " return miku_unpack_return_value(" + returnStr.Substring(6, returnStr.Length - 6) + ") ";
                            l.Replace(insertPos, lastPos - 1, returnStr);

                            nextPos = l.pos;
                            if (tokenType == (int)TK.END)
                            {
                                if (tokens.Count > 0)
                                {
                                    tokens.Pop();
                                }
                                if (onlyFun && tokens.Count <= 0)
                                {
                                    l.Next();
                                    lastPos = nextPos;
                                    nextPos = l.pos;
                                    return;
                                }
                            }
                            break;
                        }
                    }

                    if (lastStackToken != (int)TK.IF)
                    {
                        hasReturn = true;
                    }
                    break;

                case (int)TK.END:
                    if (tokens.Count > 0)
                    {
                        int token = tokens.Pop();
                        if (token == (int)TK.FUNCTION)
                        {
                            if (!hasReturn)
                            {
                                l.InsertString(lastPos - 1, " EndMikuSample() ");
                                lastPos = nextPos;
                                nextPos = l.pos;
                            }
                            if (onlyFun && tokens.Count <= 0)
                            {
                                l.Next();
                                lastPos = nextPos;
                                nextPos = l.pos;
                                return;
                            }
                        }
                        if (tokens.Count > 0)
                        {
                            var tA = tokens.ToArray();
                            lastStackToken = tA[tA.Length - 1];
                        }
                        hasReturn = false;
                    }
                    break;
                }
                l.Next();
                history.Add(l.Token);
                if (history.Count >= 16)
                {
                    history.RemoveAt(0);
                }

                tokenType = l.Token.TokenType;
                lastPos   = nextPos;
                nextPos   = l.pos;
            }

            if (needLastSample)
            {
                l.InsertString(nextPos, "\n EndMikuSample()");
            }
        }