Home | History | Annotate | Download | only in transform
      1 /*
      2  * Copyright 2003 The Apache Software Foundation
      3  *
      4  *  Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  *  Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 package org.mockito.cglib.transform;
     17 
     18 import org.mockito.asm.ClassVisitor;
     19 import org.mockito.cglib.core.ClassGenerator;
     20 import org.mockito.cglib.core.Transformer;
     21 
     22 public class TransformingClassGenerator implements ClassGenerator {
     23     private ClassGenerator gen;
     24     private ClassTransformer t;
     25 
     26     public TransformingClassGenerator(ClassGenerator gen, ClassTransformer t) {
     27         this.gen = gen;
     28         this.t = t;
     29     }
     30 
     31     public void generateClass(ClassVisitor v) throws Exception {
     32         t.setTarget(v);
     33         gen.generateClass(t);
     34     }
     35 }
     36